3aa6cc4a94866f086ae17d2fea601ed208152930
[util-vserver.git] / src / vtag.c
1 // $Id$    --*- c -*--
2
3 // Copyright (C) 2007 Daniel Hokka Zakrisson
4 //  
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.
8 //  
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.
13 //  
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.
17
18
19 #ifdef HAVE_CONFIG_H
20 #  include <config.h>
21 #endif
22
23 #include "util.h"
24
25 #include <vserver.h>
26 #include <getopt.h>
27 #include <errno.h>
28 #include <sys/types.h>
29
30
31 #define ENSC_WRAPPERS_PREFIX    "vtag: "
32 #define ENSC_WRAPPERS_VSERVER   1
33 #include <wrappers.h>
34
35 #define CMD_HELP                0x1000
36 #define CMD_VERSION             0x1001
37 #define CMD_TAG                 0x4000
38 #define CMD_CREATE              0x4001
39 #define CMD_MIGRATE             0x4002
40 #define CMD_SILENTEXIST         0x4003
41
42
43 struct option const
44 CMDLINE_OPTIONS[] = {
45   { "help",         no_argument,       0, CMD_HELP },
46   { "version",      no_argument,       0, CMD_VERSION },
47   { "tag",          required_argument, 0, CMD_TAG },
48   { "create",       no_argument,       0, CMD_CREATE },
49   { "migrate",      no_argument,       0, CMD_MIGRATE },
50   { "silentexist",  no_argument,       0, CMD_SILENTEXIST },
51   { 0,0,0,0 },
52 };
53
54 struct Arguments {
55     bool                do_create;
56     bool                do_migrate;
57     bool                is_silentexist;
58     int                 verbosity;
59     tag_t               tag;
60 };
61
62 int             wrapper_exit_code = 255;
63
64 static void
65 showHelp(int fd, char const *cmd, int res)
66 {
67   WRITE_MSG(fd, "Usage:\n    ");
68   WRITE_STR(fd, cmd);
69   WRITE_MSG(fd,
70             " (--create|--migrate) --tag <tag> <opts>* [--] <program> <args>*\n    "
71             "\n"
72             "<opts> can be:\n"
73             "    --silentexist   ...  be silent when context exists already; useful\n"
74             "                         for '--create' only\n"
75             "\n"
76             "'vtag --create' exits with code 254 iff the context exists already.\n"
77             "\n"
78             "Please report bugs to " PACKAGE_BUGREPORT "\n");
79
80   exit(res);
81 }
82
83 static void
84 showVersion()
85 {
86   WRITE_MSG(1,
87             "vtag " VERSION " -- sets the process's filesystem tag\n"
88             "This program is part of " PACKAGE_STRING "\n\n"
89             "Copyright (C) 2007 Daniel Hokka Zakrisson\n"
90             VERSION_COPYRIGHT_DISCLAIMER);
91   exit(0);
92 }
93
94 static inline ALWAYSINLINE int
95 doit(struct Arguments const *args, char *argv[])
96 {
97   tag_t                 tag;
98
99   if (!vc_isSupported(vcFEATURE_PPTAG))
100     goto exec;
101
102   if (args->do_create) {
103     tag = vc_tag_create(args->tag);
104     if (tag==VC_NOCTX) {
105       switch (errno) {
106         case EEXIST     :
107           if (!args->is_silentexist)
108             perror(ENSC_WRAPPERS_PREFIX "vc_tag_create()");
109           return 254;
110         default         :
111           perror(ENSC_WRAPPERS_PREFIX "vc_tag_create()");
112           return wrapper_exit_code;
113       }
114     }
115   }
116   else
117     tag = args->tag;
118
119   if (args->do_migrate)
120     Evc_tag_migrate(tag);
121
122 exec:
123   execvp(argv[optind], argv+optind);
124
125   PERROR_Q(ENSC_WRAPPERS_PREFIX "execvp", argv[optind]);
126   return wrapper_exit_code;
127 }
128
129 int main (int argc, char *argv[])
130 {
131   struct Arguments              args = {
132     .tag               = VC_NOCTX,
133     .do_create         = false,
134     .do_migrate        = false,
135     .is_silentexist    = false,
136     .verbosity         = 1,
137   };
138
139   while (1) {
140     int         c = getopt_long(argc, argv, "+", CMDLINE_OPTIONS, 0);
141     if (c==-1) break;
142     
143     switch (c) {
144       case CMD_HELP             :  showHelp(1, argv[0], 0);
145       case CMD_VERSION          :  showVersion();
146       case CMD_CREATE           :  args.do_create      = true;   break;
147       case CMD_MIGRATE          :  args.do_migrate     = true;   break;
148       case CMD_SILENTEXIST      :  args.is_silentexist = true;   break;
149       case CMD_TAG              :  args.tag = Evc_tagopt2tag(optarg,true); break;
150
151       default           :
152         WRITE_MSG(2, "Try '");
153         WRITE_STR(2, argv[0]);
154         WRITE_MSG(2, " --help' for more information.\n");
155         return wrapper_exit_code;
156         break;
157     }
158   }
159
160   if (!args.do_create && !args.do_migrate)
161     WRITE_MSG(2, "Neither '--create' nor '--migrate' specified; try '--help' for more information\n");
162   else if (args.do_create && args.do_migrate)
163     WRITE_MSG(2, "Can not specify '--create' and '--migrate' at the same time; try '--help' for more information\n");
164   else if (args.tag==VC_NOCTX)
165     WRITE_MSG(2, "You must specify the tag with '--tag'; try '--help' for more information\n");
166   else if (optind>=argc)
167     WRITE_MSG(2, "No command given; use '--help' for more information.\n");
168   else
169     return doit(&args, argv);
170
171   return wrapper_exit_code;
172 }