3 // Copyright (C) 2004 Enrico Scholz <enrico.scholz@informatik.tu-chemnitz.de>
5 // This program is free software; you can redistribute it and/or modify
6 // it under the terms of the GNU General Public License as published by
7 // the Free Software Foundation; version 2 of the License.
9 // This program is distributed in the hope that it will be useful,
10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
12 // GNU General Public License for more details.
14 // You should have received a copy of the GNU General Public License
15 // along with this program; if not, write to the Free Software
16 // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
24 #include "attribute-util.h"
25 #include <lib/vserver.h>
31 #define ENSC_WRAPPERS_PREFIX "vattribute: "
32 #define ENSC_WRAPPERS_VSERVER 1
33 #define ENSC_WRAPPERS_UNISTD 1
36 #define CMD_HELP 0x1000
37 #define CMD_VERSION 0x1001
38 #define CMD_XID 0x2000
39 #define CMD_SET 0x2001
40 #define CMD_CCAP 0x2002
41 #define CMD_FLAG 0x2003
42 #define CMD_SECURE 0x2004
43 #define CMD_BCAP 0x2005
44 #define CMD_GET 0x2006
45 #define CMD_UMASK 0x2007
47 int wrapper_exit_code = 1;
51 { "help", no_argument, 0, CMD_HELP },
52 { "version", no_argument, 0, CMD_VERSION },
53 { "xid", required_argument, 0, CMD_XID },
54 { "set", no_argument, 0, CMD_SET },
55 { "get", no_argument, 0, CMD_GET },
56 { "ccap", required_argument, 0, CMD_CCAP },
57 { "bcap", required_argument, 0, CMD_BCAP },
58 { "flag", required_argument, 0, CMD_FLAG },
59 { "secure", no_argument, 0, CMD_SECURE },
60 { "umask", required_argument, 0, CMD_UMASK },
66 struct vc_ctx_flags flags;
67 struct vc_ctx_caps caps;
68 struct vc_umask umask;
73 showHelp(int fd, char const *cmd, int res)
75 WRITE_MSG(fd, "Usage:\n ");
78 " [--xid <xid>] {--get|--set [--bcap [~!]<cap>] [--ccap [~!]<cap>]\n"
79 " [--flag [~!]<flag>] [--secure]} -- [<program> <args>*]\n"
81 " --bcap <cap> ... system capability to be set\n"
82 " --ccap <cap> ... context capability to be set\n"
83 " --flag <flag> ... context flag to be set\n"
84 " --umask <mask> ... unshare mask to be set\n"
86 "Please report bugs to " PACKAGE_BUGREPORT "\n");
95 "vattribute " VERSION " -- sets/gets attributes of vservers\n"
96 "This program is part of " PACKAGE_STRING "\n\n"
97 "Copyright (C) 2004 Enrico Scholz\n"
98 VERSION_COPYRIGHT_DISCLAIMER);
103 parseFlags(char const *str, struct vc_ctx_flags *flags)
105 struct vc_err_listparser err;
108 rc = vc_list2cflag(str,0, &err, flags);
111 WRITE_MSG(2, "Unknown flag '");
112 Vwrite(2, err.ptr, err.len);
114 exit(wrapper_exit_code);
119 parseBCaps(char const *str, struct vc_ctx_caps *caps)
121 struct vc_err_listparser err;
124 rc = vc_list2bcap(str,0, &err, caps);
127 WRITE_MSG(2, "Unknown bcap '");
128 Vwrite(2, err.ptr, err.len);
130 exit(wrapper_exit_code);
135 parseCCaps(char const *str, struct vc_ctx_caps *caps)
137 struct vc_err_listparser err;
140 rc = vc_list2ccap(str,0, &err, caps);
143 WRITE_MSG(2, "Unknown ccap '");
144 Vwrite(2, err.ptr, err.len);
146 exit(wrapper_exit_code);
151 parseUMask(char const *str, struct vc_umask *umask)
153 struct vc_err_listparser err;
156 rc = vc_list2umask(str, 0, &err, umask);
159 WRITE_MSG(2, "Unknown namespace '");
160 Vwrite(2, err.ptr, err.len);
162 exit(wrapper_exit_code);
167 parseSecure(struct vc_ctx_flags UNUSED * flags,
168 struct vc_ctx_caps UNUSED * caps)
170 caps->ccaps = ~vc_get_insecureccaps();
172 caps->bcaps = ~vc_get_insecurebcaps();
175 // TODO: generalize this
176 flags->flagword = VC_VXF_HIDE_NETIF;
177 flags->mask = VC_VXF_HIDE_NETIF;
181 printAttrs(struct Arguments *args)
183 struct vc_ctx_flags flags;
184 struct vc_ctx_caps caps;
185 struct vc_umask umask = { .mask = ~0, .umask = 0x20200 };
187 Evc_get_cflags(args->xid, &flags);
188 Evc_get_ccaps(args->xid, &caps);
190 print_bitfield(1, bcap, "bcapabilities", &caps.bcaps);
191 print_bitfield(1, ccap, "ccapabilities", &caps.ccaps);
192 print_bitfield(1, cflag, "flags", &flags.flagword);
193 if (vc_get_umask(args->xid, &umask) == 0) {
194 print_bitfield(1, umask, "umask", &umask.umask);
200 int main(int argc, char *argv[])
202 struct Arguments args = {
204 .flags = { .flagword = 0, .mask = 0 },
205 .caps = { .bcaps = 0, .bmask = 0,.ccaps = 0, .cmask = 0 },
206 .umask = { .umask = 0, .mask = 0 },
211 int c = getopt_long(argc, argv, "+", CMDLINE_OPTIONS, 0);
215 case CMD_HELP : showHelp(1, argv[0], 0);
216 case CMD_VERSION : showVersion();
217 case CMD_SET : args.mode = CMD_SET; break;
218 case CMD_GET : args.mode = CMD_GET; break;
219 case CMD_XID : args.xid = Evc_xidopt2xid(optarg,true); break;
220 case CMD_FLAG : parseFlags(optarg, &args.flags); break;
221 case CMD_CCAP : parseCCaps(optarg, &args.caps); break;
222 case CMD_BCAP : parseBCaps(optarg, &args.caps); break;
223 case CMD_SECURE : parseSecure(&args.flags, &args.caps); break;
224 case CMD_UMASK : parseUMask(optarg, &args.umask); break;
226 WRITE_MSG(2, "Try '");
227 WRITE_STR(2, argv[0]);
228 WRITE_MSG(2, " --help' for more information.\n");
234 if (args.xid==VC_NOCTX) args.xid = Evc_get_task_xid(0);
236 if (args.mode == CMD_SET) {
237 if ((args.caps.cmask || args.caps.bmask) &&
238 vc_set_ccaps(args.xid, &args.caps)==-1)
239 perror(ENSC_WRAPPERS_PREFIX "vc_set_ccaps()");
240 else if (args.flags.mask &&
241 vc_set_cflags(args.xid, &args.flags)==-1)
242 perror(ENSC_WRAPPERS_PREFIX "vc_set_flags()");
243 else if (args.umask.mask &&
244 vc_set_umask(args.xid, &args.umask)==-1)
245 perror(ENSC_WRAPPERS_PREFIX "vc_set_umask()");
246 else if (optind<argc)
247 EexecvpD(argv[optind], argv+optind);
251 else if (args.mode == CMD_GET) {
254 EexecvpD(argv[optind], argv+optind);