1 /* ----------------------------------------------------------------------- * 2 * 3 * Copyright 2010 Intel Corp. - All Rights Reserved 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, Inc., 53 Temple Place Ste 330, 8 * Boston MA 02111-1307, USA; either version 2 of the License, or 9 * (at your option) any later version; incorporated herein by reference. 10 * 11 * ----------------------------------------------------------------------- */ 12 13 /* 14 * syslxopt.c 15 * 16 * parse cmdline for extlinux and syslinux installer 17 * 18 */ 19 #include <stdio.h> 20 #include <stdlib.h> 21 #include <stddef.h> 22 #include <stdint.h> 23 #include <string.h> 24 #include <getopt.h> 25 #include <sysexits.h> 26 #include "version.h" 27 #include "syslxcom.h" 28 #include "syslxfs.h" 29 #include "syslxopt.h" 30 31 /* These are the options we can set their values */ 32 struct sys_options opt = { 33 .sectors = 0, 34 .heads = 0, 35 .raid_mode = 0, 36 .stupid_mode = 0, 37 .reset_adv = 0, 38 .set_once = NULL, 39 .update_only = -1, 40 .directory = NULL, 41 .device = NULL, 42 .offset = 0, 43 .menu_save = NULL, 44 .install_mbr = 0, 45 .activate_partition = 0, 46 .force = 0, 47 .bootsecfile = NULL, 48 }; 49 50 const struct option long_options[] = { 51 {"force", 0, NULL, 'f'}, /* DOS/Win32/mtools only */ 52 {"install", 0, NULL, 'i'}, 53 {"directory", 1, NULL, 'd'}, 54 {"offset", 1, NULL, 't'}, 55 {"update", 0, NULL, 'U'}, 56 {"zipdrive", 0, NULL, 'z'}, 57 {"sectors", 1, NULL, 'S'}, 58 {"stupid", 0, NULL, 's'}, 59 {"heads", 1, NULL, 'H'}, 60 {"raid-mode", 0, NULL, 'r'}, 61 {"version", 0, NULL, 'v'}, 62 {"help", 0, NULL, 'h'}, 63 {"once", 1, NULL, OPT_ONCE}, 64 {"clear-once", 0, NULL, 'O'}, 65 {"reset-adv", 0, NULL, OPT_RESET_ADV}, 66 {"menu-save", 1, NULL, 'M'}, 67 {"mbr", 0, NULL, 'm'}, /* DOS/Win32 only */ 68 {"active", 0, NULL, 'a'}, /* DOS/Win32 only */ 69 {"device", 1, NULL, OPT_DEVICE}, 70 {NULL, 0, NULL, 0} 71 }; 72 73 const char short_options[] = "t:fid:UuzsS:H:rvho:OM:ma"; 74 75 void __attribute__ ((noreturn)) usage(int rv, enum syslinux_mode mode) 76 { 77 switch (mode) { 78 case MODE_SYSLINUX: 79 /* For unmounted fs installation (syslinux) */ 80 fprintf(stderr, 81 "Usage: %s [options] device\n" 82 " --offset -t Offset of the file system on the device \n" 83 " --directory -d Directory for installation target\n", 84 program); 85 break; 86 87 case MODE_EXTLINUX: 88 /* Mounted fs installation (extlinux) */ 89 /* Actually extlinux can also use -d to provide a directory too... */ 90 fprintf(stderr, 91 "Usage: %s [options] directory\n" 92 " --device Force use of a specific block device (experts only)\n", 93 program); 94 break; 95 96 case MODE_SYSLINUX_DOSWIN: 97 /* For fs installation under Windows (syslinux.exe) */ 98 fprintf(stderr, 99 "Usage: %s [options] <drive>: [bootsecfile]\n" 100 " --directory -d Directory for installation target\n", 101 program); 102 break; 103 } 104 105 fprintf(stderr, 106 " --install -i Install over the current bootsector\n" 107 " --update -U Update a previous installation\n" 108 " --zip -z Force zipdrive geometry (-H 64 -S 32)\n" 109 " --sectors=# -S Force the number of sectors per track\n" 110 " --heads=# -H Force number of heads\n" 111 " --stupid -s Slow, safe and stupid mode\n" 112 " --raid -r Fall back to the next device on boot failure\n" 113 " --once=... %s Execute a command once upon boot\n" 114 " --clear-once -O Clear the boot-once command\n" 115 " --reset-adv Reset auxilliary data\n", 116 mode == MODE_SYSLINUX ? " " : "-o"); 117 /* 118 * Have to chop this roughly in half for the DOS installer due 119 * to limited output buffer size 120 */ 121 fprintf(stderr, 122 " --menu-save= -M Set the label to select as default on the next boot\n"); 123 if (mode == MODE_SYSLINUX_DOSWIN) 124 fprintf(stderr, 125 " --mbr -m Install an MBR\n" 126 " --active -a Mark partition as active\n"); 127 128 if (mode == MODE_SYSLINUX_DOSWIN || mode == MODE_SYSLINUX) 129 fprintf(stderr, 130 " --force -f Ignore precautions\n"); 131 132 exit(rv); 133 } 134 135 void parse_options(int argc, char *argv[], enum syslinux_mode mode) 136 { 137 int o; 138 139 program = argv[0]; 140 while ((o = getopt_long(argc, argv, short_options, 141 long_options, NULL)) != EOF) { 142 switch (o) { 143 case 'f': 144 opt.force = 1; 145 break; 146 case 'z': 147 opt.heads = 64; 148 opt.sectors = 32; 149 break; 150 case 'S': 151 opt.sectors = strtoul(optarg, NULL, 0); 152 if (opt.sectors < 1 || opt.sectors > 63) { 153 fprintf(stderr, 154 "%s: invalid number of sectors: %u (must be 1-63)\n", 155 program, opt.sectors); 156 exit(EX_USAGE); 157 } 158 break; 159 case 'H': 160 opt.heads = strtoul(optarg, NULL, 0); 161 if (opt.heads < 1 || opt.heads > 256) { 162 fprintf(stderr, 163 "%s: invalid number of heads: %u (must be 1-256)\n", 164 program, opt.heads); 165 exit(EX_USAGE); 166 } 167 break; 168 case 'r': 169 opt.raid_mode = 1; 170 break; 171 case 's': 172 opt.stupid_mode = 1; 173 break; 174 case 'i': 175 opt.update_only = 0; 176 break; 177 case 'u': 178 case 'U': 179 opt.update_only = 1; 180 break; 181 case 'h': 182 usage(0, mode); 183 break; 184 case 'o': 185 if (mode == MODE_SYSLINUX) { 186 fprintf(stderr, "%s: -o will change meaning in a future version, use -t or --offset\n", program); 187 goto opt_offset; 188 } 189 /* else fall through */ 190 case OPT_ONCE: 191 opt.set_once = optarg; 192 break; 193 case 't': 194 opt_offset: 195 opt.offset = strtoul(optarg, NULL, 0); 196 break; 197 case 'O': 198 opt.set_once = ""; 199 break; 200 case 'd': 201 opt.directory = optarg; 202 break; 203 case OPT_RESET_ADV: 204 opt.reset_adv = 1; 205 break; 206 case 'M': 207 opt.menu_save = optarg; 208 break; 209 case 'm': 210 opt.install_mbr = 1; 211 break; 212 case 'a': 213 opt.activate_partition = 1; 214 break; 215 case OPT_DEVICE: 216 if (mode != MODE_EXTLINUX) 217 usage(EX_USAGE, mode); 218 opt.device = optarg; 219 break; 220 case 'v': 221 fprintf(stderr, 222 "%s " VERSION_STR " Copyright 1994-" YEAR_STR 223 " H. Peter Anvin et al\n", program); 224 exit(0); 225 default: 226 fprintf(stderr, "%s: Unknown option: -%c\n", program, optopt); 227 usage(EX_USAGE, mode); 228 } 229 } 230 231 switch (mode) { 232 case MODE_SYSLINUX: 233 case MODE_SYSLINUX_DOSWIN: 234 opt.device = argv[optind++]; 235 break; 236 case MODE_EXTLINUX: 237 if (!opt.directory) 238 opt.directory = argv[optind++]; 239 break; 240 } 241 242 if (argv[optind] && (mode == MODE_SYSLINUX_DOSWIN)) 243 /* Allow for the boot-sector argument */ 244 opt.bootsecfile = argv[optind++]; 245 if (argv[optind]) 246 usage(EX_USAGE, mode); /* Excess arguments */ 247 } 248 249 /* 250 * Make any user-specified ADV modifications in memory 251 */ 252 int modify_adv(void) 253 { 254 int rv = 0; 255 256 if (opt.reset_adv) 257 syslinux_reset_adv(syslinux_adv); 258 259 if (opt.set_once) { 260 if (syslinux_setadv(ADV_BOOTONCE, strlen(opt.set_once), opt.set_once)) { 261 fprintf(stderr, "%s: not enough space for boot-once command\n", 262 program); 263 rv = -1; 264 } 265 } 266 if (opt.menu_save) { 267 if (syslinux_setadv(ADV_MENUSAVE, strlen(opt.menu_save), opt.menu_save)) { 268 fprintf(stderr, "%s: not enough space for menu-save label\n", 269 program); 270 rv = -1; 271 } 272 } 273 274 return rv; 275 } 276
