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 | } |
---|