vdevmap.c (5245B)
1 // $Id$ --*- c -*-- 2 3 // Copyright (C) 2006 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 #include <lib/internal.h> 25 26 #include <vserver.h> 27 28 #include <getopt.h> 29 #include <errno.h> 30 31 #define ENSC_WRAPPERS_PREFIX "vdevmap: " 32 #define ENSC_WRAPPERS_UNISTD 1 33 #define ENSC_WRAPPERS_VSERVER 1 34 #include <wrappers.h> 35 36 #define CMD_HELP 0x1000 37 #define CMD_VERSION 0x1001 38 39 int wrapper_exit_code = 1; 40 41 struct option const 42 CMDLINE_OPTIONS[] = { 43 { "help", no_argument, 0, CMD_HELP }, 44 { "version", no_argument, 0, CMD_VERSION }, 45 { "xid", required_argument, 0, 'x' }, 46 { "set", no_argument, 0, 's' }, 47 { "unset", no_argument, 0, 'u' }, 48 { "open", no_argument, 0, 'o' }, 49 { "create", no_argument, 0, 'c' }, 50 { "remap", no_argument, 0, 'r' }, 51 { "flags", required_argument, 0, 'f' }, 52 { "device", required_argument, 0, 'd' }, 53 { "target", required_argument, 0, 't' }, 54 {0,0,0,0} 55 }; 56 57 static void 58 showHelp(int fd, char const *cmd) 59 { 60 WRITE_MSG(fd, "Usage: "); 61 WRITE_STR(fd, cmd); 62 WRITE_MSG(fd, 63 " --xid <xid> {--set|--unset} [--flags <flags>] [--open] [--create]\n" 64 " [--device <dev>] [--remap --target <dev>]\n" 65 "\n" 66 " --flags <flags> Set the specified flags\n" 67 " --open Allow opening of the device\n" 68 " --create If CAP_MKNOD is given, allow mknod(2)\n" 69 " --device <dev> Device to apply the command to\n" 70 " --remap Remap the device to the target\n" 71 " --target <dev> Target for --remap\n" 72 "\n" 73 "EXAMPLES\n" 74 " Remap /dev/hda1 to /dev/vroot1 for xid 42\n" 75 " vdevmap --xid 42 --set --open --device /dev/hda1 --target /dev/vroot1 --remap\n" 76 " Let xid 42 create all device nodes\n" 77 " vdevmap --xid 42 --set --open --create --target /dev/null\n" 78 " vdevmap --xid 42 --set --open --create --target /dev/root\n" 79 " Let xid 43 create just /dev/null\n" 80 " vdevmap --xid 43 --set --open --create --device /dev/null\n" 81 "\n" 82 "Please report bugs to " PACKAGE_BUGREPORT "\n"); 83 84 exit(0); 85 } 86 87 static void 88 showVersion() 89 { 90 WRITE_MSG(1, 91 "vdevmap " VERSION " -- manages device mappings\n" 92 "This program is part of " PACKAGE_STRING "\n\n" 93 "Copyright (C) 2006 Daniel Hokka Zakrisson\n" 94 VERSION_COPYRIGHT_DISCLAIMER); 95 exit(0); 96 } 97 98 int main(int argc, char *argv[]) 99 { 100 xid_t xid = VC_NOCTX; 101 bool allow_open = false; 102 bool allow_create = false; 103 bool do_remap = false; 104 uint32_t flags = 0; 105 char *device = NULL; 106 char *target = NULL; 107 bool set = true; 108 unsigned long tmp = 0; 109 110 while (1) { 111 int c = getopt_long(argc, argv, "+x:suocrf:d:t:", CMDLINE_OPTIONS, 0); 112 if (c==-1) break; 113 114 switch (c) { 115 case CMD_HELP : showHelp(1, argv[0]); 116 case CMD_VERSION : showVersion(); 117 case 'x' : xid = Evc_xidopt2xid(optarg, true); break; 118 case 'o' : allow_open = true; break; 119 case 'c' : allow_create = true; break; 120 case 'r' : do_remap = true; break; 121 case 'd' : device = optarg; break; 122 case 't' : target = optarg; break; 123 case 's' : set = 1; break; 124 case 'u' : set = 0; break; 125 case 'f' : 126 if (!isNumberUnsigned(optarg, &tmp, false)) { 127 WRITE_MSG(2, "Invalid flags argument: '"); 128 WRITE_STR(2, optarg); 129 WRITE_MSG(2, "'; try '--help' for more information\n"); 130 return EXIT_FAILURE; 131 } 132 flags |= (uint32_t) tmp; 133 break; 134 135 default : 136 WRITE_MSG(2, "Try '"); 137 WRITE_STR(2, argv[0]); 138 WRITE_MSG(2, " --help' for more information.\n"); 139 return EXIT_FAILURE; 140 break; 141 } 142 } 143 144 if (allow_open) flags |= VC_DATTR_OPEN; 145 if (allow_create) flags |= VC_DATTR_CREATE; 146 if (do_remap) flags |= VC_DATTR_REMAP; 147 148 if (!target && do_remap) 149 WRITE_MSG(2, "Remapping specified without a target; try '--help' for more information\n"); 150 else if (xid==VC_NOCTX) 151 WRITE_MSG(2, "No xid specified; try '--help' for more information\n"); 152 else if (optind!=argc) 153 WRITE_MSG(2, "Unused argument(s); try '--help' for more information\n"); 154 else if (!device && !target) 155 WRITE_MSG(2, "Device and target are missing; try '--help' for more information\n"); 156 else if (set && vc_set_mapping(xid, device, target, flags)==-1) 157 perror("vc_set_mapping()"); 158 else if (!set && vc_unset_mapping(xid, device, target, flags)==-1) 159 perror("vc_unset_mapping()"); 160 else 161 return EXIT_SUCCESS; 162 163 return EXIT_FAILURE; 164 }