36 |
#include <signal.h> |
#include <signal.h> |
37 |
#include <blkid/blkid.h> |
#include <blkid/blkid.h> |
38 |
|
|
39 |
|
#include "log.h" |
40 |
|
|
41 |
#ifndef DEBUG |
#ifndef DEBUG |
42 |
#define DEBUG 0 |
#define DEBUG 0 |
43 |
#endif |
#endif |
48 |
#define dbgPrintf(format, args...) |
#define dbgPrintf(format, args...) |
49 |
#endif |
#endif |
50 |
|
|
51 |
int debug = 0; /* Currently just for template debugging */ |
int debug = 0; /* Currently just for template debugging */ |
52 |
|
|
53 |
#define _(A) (A) |
#define _(A) (A) |
54 |
|
|
58 |
#define NOOP_OPCODE 0x90 |
#define NOOP_OPCODE 0x90 |
59 |
#define JMP_SHORT_OPCODE 0xeb |
#define JMP_SHORT_OPCODE 0xeb |
60 |
|
|
61 |
|
int isEfi = 0; |
62 |
|
|
63 |
|
#if defined(__aarch64__) |
64 |
|
#define isEfiOnly 1 |
65 |
|
#else |
66 |
|
#define isEfiOnly 0 |
67 |
|
#endif |
68 |
|
|
69 |
|
char *saved_command_line = NULL; |
70 |
|
|
71 |
/* comments get lumped in with indention */ |
/* comments get lumped in with indention */ |
72 |
struct lineElement { |
struct lineElement { |
73 |
char * item; |
char *item; |
74 |
char * indent; |
char *indent; |
75 |
}; |
}; |
76 |
|
|
77 |
enum lineType_e { |
enum lineType_e { |
78 |
LT_WHITESPACE = 1 << 0, |
LT_UNIDENTIFIED = 0, |
79 |
LT_TITLE = 1 << 1, |
LT_WHITESPACE = 1 << 0, |
80 |
LT_KERNEL = 1 << 2, |
LT_TITLE = 1 << 1, |
81 |
LT_INITRD = 1 << 3, |
LT_KERNEL = 1 << 2, |
82 |
LT_HYPER = 1 << 4, |
LT_INITRD = 1 << 3, |
83 |
LT_DEFAULT = 1 << 5, |
LT_HYPER = 1 << 4, |
84 |
LT_MBMODULE = 1 << 6, |
LT_DEFAULT = 1 << 5, |
85 |
LT_ROOT = 1 << 7, |
LT_MBMODULE = 1 << 6, |
86 |
LT_FALLBACK = 1 << 8, |
LT_ROOT = 1 << 7, |
87 |
LT_KERNELARGS = 1 << 9, |
LT_FALLBACK = 1 << 8, |
88 |
LT_BOOT = 1 << 10, |
LT_KERNELARGS = 1 << 9, |
89 |
LT_BOOTROOT = 1 << 11, |
LT_BOOT = 1 << 10, |
90 |
LT_LBA = 1 << 12, |
LT_BOOTROOT = 1 << 11, |
91 |
LT_OTHER = 1 << 13, |
LT_LBA = 1 << 12, |
92 |
LT_GENERIC = 1 << 14, |
LT_OTHER = 1 << 13, |
93 |
LT_ECHO = 1 << 16, |
LT_GENERIC = 1 << 14, |
94 |
LT_MENUENTRY = 1 << 17, |
LT_ECHO = 1 << 16, |
95 |
LT_ENTRY_END = 1 << 18, |
LT_MENUENTRY = 1 << 17, |
96 |
LT_SET_VARIABLE = 1 << 19, |
LT_ENTRY_END = 1 << 18, |
97 |
LT_UNKNOWN = 1 << 20, |
LT_SET_VARIABLE = 1 << 19, |
98 |
|
LT_KERNEL_EFI = 1 << 20, |
99 |
|
LT_INITRD_EFI = 1 << 21, |
100 |
|
LT_KERNEL_16 = 1 << 22, |
101 |
|
LT_INITRD_16 = 1 << 23, |
102 |
|
LT_DEVTREE = 1 << 24, |
103 |
|
LT_UNKNOWN = 1 << 25, |
104 |
}; |
}; |
105 |
|
|
106 |
struct singleLine { |
struct singleLine { |
107 |
char * indent; |
char *indent; |
108 |
int numElements; |
int numElements; |
109 |
struct lineElement * elements; |
struct lineElement *elements; |
110 |
struct singleLine * next; |
struct singleLine *next; |
111 |
enum lineType_e type; |
enum lineType_e type; |
112 |
}; |
}; |
113 |
|
|
114 |
struct singleEntry { |
struct singleEntry { |
115 |
struct singleLine * lines; |
struct singleLine *lines; |
116 |
int skip; |
int skip; |
117 |
int multiboot; |
int multiboot; |
118 |
struct singleEntry * next; |
struct singleEntry *next; |
119 |
}; |
}; |
120 |
|
|
121 |
#define GRUBBY_BADIMAGE_OKAY (1 << 0) |
#define GRUBBY_BADIMAGE_OKAY (1 << 0) |
129 |
#define NEED_ARGS (1 << 3) |
#define NEED_ARGS (1 << 3) |
130 |
#define NEED_MB (1 << 4) |
#define NEED_MB (1 << 4) |
131 |
#define NEED_END (1 << 5) |
#define NEED_END (1 << 5) |
132 |
|
#define NEED_DEVTREE (1 << 6) |
133 |
|
|
134 |
#define MAIN_DEFAULT (1 << 0) |
#define MAIN_DEFAULT (1 << 0) |
135 |
|
#define FIRST_ENTRY_INDEX 0 /* boot entry index value begin and increment |
136 |
|
from this initial value */ |
137 |
|
#define NO_DEFAULT_ENTRY -1 /* indicates that no specific default boot |
138 |
|
entry was set or currently exists */ |
139 |
#define DEFAULT_SAVED -2 |
#define DEFAULT_SAVED -2 |
140 |
#define DEFAULT_SAVED_GRUB2 -3 |
#define DEFAULT_SAVED_GRUB2 -3 |
141 |
|
|
142 |
struct keywordTypes { |
struct keywordTypes { |
143 |
char * key; |
char *key; |
144 |
enum lineType_e type; |
enum lineType_e type; |
145 |
char nextChar; |
char nextChar; |
146 |
char separatorChar; |
char separatorChar; |
147 |
}; |
}; |
148 |
|
|
149 |
struct configFileInfo; |
struct configFileInfo; |
150 |
|
|
151 |
typedef const char *(*findConfigFunc)(struct configFileInfo *); |
typedef const char *(*findConfigFunc) (struct configFileInfo *); |
152 |
typedef const int (*writeLineFunc)(struct configFileInfo *, |
typedef const int (*writeLineFunc) (struct configFileInfo *, |
153 |
struct singleLine *line); |
struct singleLine * line); |
154 |
|
typedef char *(*getEnvFunc) (struct configFileInfo *, char *name); |
155 |
|
typedef int (*setEnvFunc) (struct configFileInfo *, char *name, char *value); |
156 |
|
|
157 |
struct configFileInfo { |
struct configFileInfo { |
158 |
char * defaultConfig; |
char *defaultConfig; |
159 |
findConfigFunc findConfig; |
findConfigFunc findConfig; |
160 |
writeLineFunc writeLine; |
writeLineFunc writeLine; |
161 |
struct keywordTypes * keywords; |
getEnvFunc getEnv; |
162 |
int defaultIsIndex; |
setEnvFunc setEnv; |
163 |
int defaultIsVariable; |
struct keywordTypes *keywords; |
164 |
int defaultSupportSaved; |
int caseInsensitive; |
165 |
enum lineType_e entryStart; |
int defaultIsIndex; |
166 |
enum lineType_e entryEnd; |
int defaultIsVariable; |
167 |
int needsBootPrefix; |
int defaultSupportSaved; |
168 |
int argsInQuotes; |
int defaultIsSaved; |
169 |
int maxTitleLength; |
int defaultIsUnquoted; |
170 |
int titleBracketed; |
enum lineType_e entryStart; |
171 |
int titlePosition; |
enum lineType_e entryEnd; |
172 |
int mbHyperFirst; |
int needsBootPrefix; |
173 |
int mbInitRdIsModule; |
int argsInQuotes; |
174 |
int mbConcatArgs; |
int maxTitleLength; |
175 |
int mbAllowExtraInitRds; |
int titleBracketed; |
176 |
|
int titlePosition; |
177 |
|
int mbHyperFirst; |
178 |
|
int mbInitRdIsModule; |
179 |
|
int mbConcatArgs; |
180 |
|
int mbAllowExtraInitRds; |
181 |
|
char *envFile; |
182 |
}; |
}; |
183 |
|
|
184 |
struct keywordTypes grubKeywords[] = { |
struct keywordTypes grubKeywords[] = { |
185 |
{ "title", LT_TITLE, ' ' }, |
{"title", LT_TITLE, ' '}, |
186 |
{ "root", LT_BOOTROOT, ' ' }, |
{"root", LT_BOOTROOT, ' '}, |
187 |
{ "default", LT_DEFAULT, ' ' }, |
{"default", LT_DEFAULT, ' '}, |
188 |
{ "fallback", LT_FALLBACK, ' ' }, |
{"fallback", LT_FALLBACK, ' '}, |
189 |
{ "kernel", LT_KERNEL, ' ' }, |
{"kernel", LT_KERNEL, ' '}, |
190 |
{ "initrd", LT_INITRD, ' ', ' ' }, |
{"initrd", LT_INITRD, ' ', ' '}, |
191 |
{ "module", LT_MBMODULE, ' ' }, |
{"module", LT_MBMODULE, ' '}, |
192 |
{ "kernel", LT_HYPER, ' ' }, |
{"kernel", LT_HYPER, ' '}, |
193 |
{ NULL, 0, 0 }, |
{NULL, 0, 0}, |
194 |
}; |
}; |
195 |
|
|
196 |
const char *grubFindConfig(struct configFileInfo *cfi) { |
const char *grubFindConfig(struct configFileInfo *cfi) |
197 |
static const char *configFiles[] = { |
{ |
198 |
"/etc/grub.conf", |
static const char *configFiles[] = { |
199 |
"/boot/grub/grub.conf", |
"/boot/grub/grub.conf", |
200 |
"/boot/grub/menu.lst", |
"/boot/grub/menu.lst", |
201 |
NULL |
"/etc/grub.conf", |
202 |
}; |
NULL |
203 |
static int i = -1; |
}; |
204 |
|
static int i = -1; |
205 |
if (i == -1) { |
|
206 |
for (i = 0; configFiles[i] != NULL; i++) { |
if (i == -1) { |
207 |
dbgPrintf("Checking \"%s\": ", configFiles[i]); |
for (i = 0; configFiles[i] != NULL; i++) { |
208 |
if (!access(configFiles[i], R_OK)) { |
dbgPrintf("Checking \"%s\": ", configFiles[i]); |
209 |
dbgPrintf("found\n"); |
if (!access(configFiles[i], R_OK)) { |
210 |
return configFiles[i]; |
dbgPrintf("found\n"); |
211 |
} |
return configFiles[i]; |
212 |
dbgPrintf("not found\n"); |
} |
213 |
|
dbgPrintf("not found\n"); |
214 |
|
} |
215 |
} |
} |
216 |
} |
return configFiles[i]; |
|
return configFiles[i]; |
|
217 |
} |
} |
218 |
|
|
219 |
struct configFileInfo grubConfigType = { |
struct configFileInfo grubConfigType = { |
220 |
.findConfig = grubFindConfig, |
.findConfig = grubFindConfig, |
221 |
.keywords = grubKeywords, |
.keywords = grubKeywords, |
222 |
.defaultIsIndex = 1, |
.defaultIsIndex = 1, |
223 |
.defaultSupportSaved = 1, |
.defaultSupportSaved = 1, |
224 |
.entryStart = LT_TITLE, |
.entryStart = LT_TITLE, |
225 |
.needsBootPrefix = 1, |
.needsBootPrefix = 1, |
226 |
.mbHyperFirst = 1, |
.mbHyperFirst = 1, |
227 |
.mbInitRdIsModule = 1, |
.mbInitRdIsModule = 1, |
228 |
.mbAllowExtraInitRds = 1, |
.mbAllowExtraInitRds = 1, |
229 |
|
.titlePosition = 1, |
230 |
}; |
}; |
231 |
|
|
232 |
struct keywordTypes grub2Keywords[] = { |
struct keywordTypes grub2Keywords[] = { |
233 |
{ "menuentry", LT_MENUENTRY, ' ' }, |
{"menuentry", LT_MENUENTRY, ' '}, |
234 |
{ "}", LT_ENTRY_END, ' ' }, |
{"}", LT_ENTRY_END, ' '}, |
235 |
{ "echo", LT_ECHO, ' ' }, |
{"echo", LT_ECHO, ' '}, |
236 |
{ "set", LT_SET_VARIABLE,' ', '=' }, |
{"set", LT_SET_VARIABLE, ' ', '='}, |
237 |
{ "root", LT_BOOTROOT, ' ' }, |
{"root", LT_BOOTROOT, ' '}, |
238 |
{ "default", LT_DEFAULT, ' ' }, |
{"default", LT_DEFAULT, ' '}, |
239 |
{ "fallback", LT_FALLBACK, ' ' }, |
{"fallback", LT_FALLBACK, ' '}, |
240 |
{ "linux", LT_KERNEL, ' ' }, |
{"linux", LT_KERNEL, ' '}, |
241 |
{ "initrd", LT_INITRD, ' ', ' ' }, |
{"linuxefi", LT_KERNEL_EFI, ' '}, |
242 |
{ "module", LT_MBMODULE, ' ' }, |
{"linux16", LT_KERNEL_16, ' '}, |
243 |
{ "kernel", LT_HYPER, ' ' }, |
{"initrd", LT_INITRD, ' ', ' '}, |
244 |
{ NULL, 0, 0 }, |
{"initrdefi", LT_INITRD_EFI, ' ', ' '}, |
245 |
}; |
{"initrd16", LT_INITRD_16, ' ', ' '}, |
246 |
|
{"module", LT_MBMODULE, ' '}, |
247 |
const char *grub2FindConfig(struct configFileInfo *cfi) { |
{"kernel", LT_HYPER, ' '}, |
248 |
static const char *configFiles[] = { |
{"devicetree", LT_DEVTREE, ' '}, |
249 |
"/boot/grub/grub-efi.cfg", |
{NULL, 0, 0}, |
250 |
"/boot/grub/grub.cfg", |
}; |
251 |
NULL |
|
252 |
}; |
const char *grub2FindConfig(struct configFileInfo *cfi) |
253 |
static int i = -1; |
{ |
254 |
static const char *grub_cfg = "/boot/grub/grub.cfg"; |
static const char *configFiles[] = { |
255 |
|
"/boot/grub/grub-efi.cfg", |
256 |
if (i == -1) { |
"/boot/grub/grub.cfg", |
257 |
for (i = 0; configFiles[i] != NULL; i++) { |
"/etc/grub2-efi.cfg", |
258 |
dbgPrintf("Checking \"%s\": ", configFiles[i]); |
"/etc/grub2.cfg", |
259 |
if (!access(configFiles[i], R_OK)) { |
"/boot/grub2/grub.cfg", |
260 |
|
"/boot/grub2-efi/grub.cfg", |
261 |
|
NULL |
262 |
|
}; |
263 |
|
static int i = -1; |
264 |
|
static const char *grub_cfg = "/boot/grub/grub.cfg"; |
265 |
|
int rc = -1; |
266 |
|
|
267 |
|
if (i == -1) { |
268 |
|
for (i = 0; configFiles[i] != NULL; i++) { |
269 |
|
dbgPrintf("Checking \"%s\": ", configFiles[i]); |
270 |
|
if ((rc = access(configFiles[i], R_OK))) { |
271 |
|
if (errno == EACCES) { |
272 |
|
printf |
273 |
|
("Unable to access bootloader configuration file " |
274 |
|
"\"%s\": %m\n", configFiles[i]); |
275 |
|
exit(1); |
276 |
|
} |
277 |
|
continue; |
278 |
|
} else { |
279 |
|
dbgPrintf("found\n"); |
280 |
|
return configFiles[i]; |
281 |
|
} |
282 |
|
} |
283 |
|
} |
284 |
|
|
285 |
|
/* Ubuntu renames grub2 to grub, so check for the grub.d directory |
286 |
|
* that isn't in grub1, and if it exists, return the config file path |
287 |
|
* that they use. */ |
288 |
|
if (configFiles[i] == NULL && !access("/etc/grub.d/", R_OK)) { |
289 |
dbgPrintf("found\n"); |
dbgPrintf("found\n"); |
290 |
return configFiles[i]; |
return grub_cfg; |
|
} |
|
291 |
} |
} |
|
} |
|
292 |
|
|
293 |
/* Ubuntu renames grub2 to grub, so check for the grub.d directory |
dbgPrintf("not found\n"); |
294 |
* that isn't in grub1, and if it exists, return the config file path |
return configFiles[i]; |
295 |
* that they use. */ |
} |
296 |
if (configFiles[i] == NULL && !access("/etc/grub.d/", R_OK)) { |
|
297 |
dbgPrintf("found\n"); |
/* kind of hacky. It'll give the first 1024 bytes, ish. */ |
298 |
return grub_cfg; |
static char *grub2GetEnv(struct configFileInfo *info, char *name) |
299 |
} |
{ |
300 |
|
static char buf[1025]; |
301 |
|
char *s = NULL; |
302 |
|
char *ret = NULL; |
303 |
|
char *envFile = info->envFile ? info->envFile : "/boot/grub/grubenv"; |
304 |
|
int rc = |
305 |
|
asprintf(&s, "grub-editenv %s list | grep '^%s='", envFile, name); |
306 |
|
|
307 |
|
if (rc < 0) |
308 |
|
return NULL; |
309 |
|
|
310 |
|
FILE *f = popen(s, "r"); |
311 |
|
if (!f) |
312 |
|
goto out; |
313 |
|
|
314 |
|
memset(buf, '\0', sizeof(buf)); |
315 |
|
ret = fgets(buf, 1024, f); |
316 |
|
pclose(f); |
317 |
|
|
318 |
|
if (ret) { |
319 |
|
ret += strlen(name) + 1; |
320 |
|
ret[strlen(ret) - 1] = '\0'; |
321 |
|
} |
322 |
|
dbgPrintf("grub2GetEnv(%s): %s\n", name, ret); |
323 |
|
out: |
324 |
|
free(s); |
325 |
|
return ret; |
326 |
|
} |
327 |
|
|
328 |
|
static int sPopCount(const char *s, const char *c) |
329 |
|
{ |
330 |
|
int ret = 0; |
331 |
|
if (!s) |
332 |
|
return -1; |
333 |
|
for (int i = 0; s[i] != '\0'; i++) |
334 |
|
for (int j = 0; c[j] != '\0'; j++) |
335 |
|
if (s[i] == c[j]) |
336 |
|
ret++; |
337 |
|
return ret; |
338 |
|
} |
339 |
|
|
340 |
|
static char *shellEscape(const char *s) |
341 |
|
{ |
342 |
|
int l = strlen(s) + sPopCount(s, "'") * 2; |
343 |
|
|
344 |
dbgPrintf("not found\n"); |
char *ret = calloc(l + 1, sizeof(*ret)); |
345 |
return configFiles[i]; |
if (!ret) |
346 |
|
return NULL; |
347 |
|
for (int i = 0, j = 0; s[i] != '\0'; i++, j++) { |
348 |
|
if (s[i] == '\'') |
349 |
|
ret[j++] = '\\'; |
350 |
|
ret[j] = s[i]; |
351 |
|
} |
352 |
|
return ret; |
353 |
|
} |
354 |
|
|
355 |
|
static void unquote(char *s) |
356 |
|
{ |
357 |
|
int l = strlen(s); |
358 |
|
|
359 |
|
if ((s[l - 1] == '\'' && s[0] == '\'') |
360 |
|
|| (s[l - 1] == '"' && s[0] == '"')) { |
361 |
|
memmove(s, s + 1, l - 2); |
362 |
|
s[l - 2] = '\0'; |
363 |
|
} |
364 |
|
} |
365 |
|
|
366 |
|
static int grub2SetEnv(struct configFileInfo *info, char *name, char *value) |
367 |
|
{ |
368 |
|
char *s = NULL; |
369 |
|
int rc = 0; |
370 |
|
char *envFile = info->envFile ? info->envFile : "/boot/grub/grubenv"; |
371 |
|
|
372 |
|
unquote(value); |
373 |
|
value = shellEscape(value); |
374 |
|
if (!value) |
375 |
|
return -1; |
376 |
|
|
377 |
|
rc = asprintf(&s, "grub-editenv %s set '%s=%s'", envFile, name, value); |
378 |
|
free(value); |
379 |
|
if (rc < 0) |
380 |
|
return -1; |
381 |
|
|
382 |
|
dbgPrintf("grub2SetEnv(%s): %s\n", name, s); |
383 |
|
rc = system(s); |
384 |
|
free(s); |
385 |
|
return rc; |
386 |
|
} |
387 |
|
|
388 |
|
/* this is a gigantic hack to avoid clobbering grub2 variables... */ |
389 |
|
static int is_special_grub2_variable(const char *name) |
390 |
|
{ |
391 |
|
if (!strcmp(name, "\"${next_entry}\"")) |
392 |
|
return 1; |
393 |
|
if (!strcmp(name, "\"${prev_saved_entry}\"")) |
394 |
|
return 1; |
395 |
|
return 0; |
396 |
} |
} |
397 |
|
|
398 |
int sizeOfSingleLine(struct singleLine * line) { |
int sizeOfSingleLine(struct singleLine *line) |
399 |
int i; |
{ |
400 |
int count = 0; |
int count = 0; |
401 |
|
|
402 |
for (i = 0; i < line->numElements; i++) { |
for (int i = 0; i < line->numElements; i++) { |
403 |
int indentSize = 0; |
int indentSize = 0; |
404 |
|
|
405 |
count = count + strlen(line->elements[i].item); |
count = count + strlen(line->elements[i].item); |
406 |
|
|
407 |
indentSize = strlen(line->elements[i].indent); |
indentSize = strlen(line->elements[i].indent); |
408 |
if (indentSize > 0) |
if (indentSize > 0) |
409 |
count = count + indentSize; |
count = count + indentSize; |
410 |
else |
else |
411 |
/* be extra safe and add room for whitespaces */ |
/* be extra safe and add room for whitespaces */ |
412 |
count = count + 1; |
count = count + 1; |
413 |
} |
} |
414 |
|
|
415 |
/* room for trailing terminator */ |
/* room for trailing terminator */ |
416 |
count = count + 1; |
count = count + 1; |
417 |
|
|
418 |
return count; |
return count; |
419 |
} |
} |
420 |
|
|
421 |
static int isquote(char q) |
static int isquote(char q) |
422 |
{ |
{ |
423 |
if (q == '\'' || q == '\"') |
if (q == '\'' || q == '\"') |
424 |
return 1; |
return 1; |
425 |
return 0; |
return 0; |
426 |
} |
} |
427 |
|
|
428 |
char *grub2ExtractTitle(struct singleLine * line) { |
static int iskernel(enum lineType_e type) |
429 |
char * current; |
{ |
430 |
char * current_indent; |
return (type == LT_KERNEL || type == LT_KERNEL_EFI |
431 |
int current_len; |
|| type == LT_KERNEL_16); |
432 |
int current_indent_len; |
} |
433 |
int i; |
|
434 |
|
static int isinitrd(enum lineType_e type) |
435 |
/* bail out if line does not start with menuentry */ |
{ |
436 |
if (strcmp(line->elements[0].item, "menuentry")) |
return (type == LT_INITRD || type == LT_INITRD_EFI |
437 |
return NULL; |
|| type == LT_INITRD_16); |
438 |
|
} |
|
i = 1; |
|
|
current = line->elements[i].item; |
|
|
current_len = strlen(current); |
|
|
|
|
|
/* if second word is quoted, strip the quotes and return single word */ |
|
|
if (isquote(*current) && isquote(current[current_len - 1])) { |
|
|
char *tmp; |
|
|
|
|
|
tmp = strdup(current); |
|
|
*(tmp + current_len - 1) = '\0'; |
|
|
return ++tmp; |
|
|
} |
|
439 |
|
|
440 |
/* if no quotes, return second word verbatim */ |
char *grub2ExtractTitle(struct singleLine *line) |
441 |
if (!isquote(*current)) |
{ |
442 |
return current; |
char *current; |
443 |
|
char *current_indent; |
444 |
/* second element start with a quote, so we have to find the element |
int current_len; |
445 |
* whose last character is also quote (assuming it's the closing one) */ |
int current_indent_len; |
446 |
int resultMaxSize; |
int i; |
447 |
char * result; |
|
448 |
|
/* bail out if line does not start with menuentry */ |
449 |
resultMaxSize = sizeOfSingleLine(line); |
if (strcmp(line->elements[0].item, "menuentry")) |
450 |
result = malloc(resultMaxSize); |
return NULL; |
451 |
snprintf(result, resultMaxSize, "%s", ++current); |
|
452 |
|
i = 1; |
|
i++; |
|
|
for (; i < line->numElements; ++i) { |
|
453 |
current = line->elements[i].item; |
current = line->elements[i].item; |
454 |
current_len = strlen(current); |
current_len = strlen(current); |
|
current_indent = line->elements[i].indent; |
|
|
current_indent_len = strlen(current_indent); |
|
455 |
|
|
456 |
strncat(result, current_indent, current_indent_len); |
/* if second word is quoted, strip the quotes and return single word */ |
457 |
if (!isquote(current[current_len-1])) { |
if (isquote(*current) && isquote(current[current_len - 1])) { |
458 |
strncat(result, current, current_len); |
char *tmp; |
459 |
} else { |
|
460 |
strncat(result, current, current_len - 1); |
tmp = strdup(current + 1); |
461 |
break; |
if (!tmp) |
462 |
|
return NULL; |
463 |
|
tmp[strlen(tmp) - 1] = '\0'; |
464 |
|
return tmp; |
465 |
|
} |
466 |
|
|
467 |
|
/* if no quotes, return second word verbatim */ |
468 |
|
if (!isquote(*current)) |
469 |
|
return current; |
470 |
|
|
471 |
|
/* second element start with a quote, so we have to find the element |
472 |
|
* whose last character is also quote (assuming it's the closing one) */ |
473 |
|
int resultMaxSize; |
474 |
|
char *result; |
475 |
|
/* need to ensure that ' does not match " as we search */ |
476 |
|
char quote_char = *current; |
477 |
|
|
478 |
|
resultMaxSize = sizeOfSingleLine(line); |
479 |
|
result = malloc(resultMaxSize); |
480 |
|
snprintf(result, resultMaxSize, "%s", ++current); |
481 |
|
|
482 |
|
i++; |
483 |
|
for (; i < line->numElements; ++i) { |
484 |
|
current = line->elements[i].item; |
485 |
|
current_len = strlen(current); |
486 |
|
current_indent = line->elements[i].indent; |
487 |
|
current_indent_len = strlen(current_indent); |
488 |
|
|
489 |
|
strncat(result, current_indent, current_indent_len); |
490 |
|
if (current[current_len - 1] != quote_char) { |
491 |
|
strncat(result, current, current_len); |
492 |
|
} else { |
493 |
|
strncat(result, current, current_len - 1); |
494 |
|
break; |
495 |
|
} |
496 |
} |
} |
497 |
} |
return result; |
|
return result; |
|
498 |
} |
} |
499 |
|
|
500 |
struct configFileInfo grub2ConfigType = { |
struct configFileInfo grub2ConfigType = { |
501 |
.findConfig = grub2FindConfig, |
.findConfig = grub2FindConfig, |
502 |
.keywords = grub2Keywords, |
.getEnv = grub2GetEnv, |
503 |
.defaultIsIndex = 1, |
.setEnv = grub2SetEnv, |
504 |
.defaultSupportSaved = 1, |
.keywords = grub2Keywords, |
505 |
.defaultIsVariable = 1, |
.defaultIsIndex = 1, |
506 |
.entryStart = LT_MENUENTRY, |
.defaultSupportSaved = 1, |
507 |
.entryEnd = LT_ENTRY_END, |
.defaultIsVariable = 1, |
508 |
.titlePosition = 1, |
.entryStart = LT_MENUENTRY, |
509 |
.needsBootPrefix = 1, |
.entryEnd = LT_ENTRY_END, |
510 |
.mbHyperFirst = 1, |
.titlePosition = 1, |
511 |
.mbInitRdIsModule = 1, |
.needsBootPrefix = 1, |
512 |
.mbAllowExtraInitRds = 1, |
.mbHyperFirst = 1, |
513 |
|
.mbInitRdIsModule = 1, |
514 |
|
.mbAllowExtraInitRds = 1, |
515 |
}; |
}; |
516 |
|
|
517 |
struct keywordTypes yabootKeywords[] = { |
struct keywordTypes yabootKeywords[] = { |
518 |
{ "label", LT_TITLE, '=' }, |
{"label", LT_TITLE, '='}, |
519 |
{ "root", LT_ROOT, '=' }, |
{"root", LT_ROOT, '='}, |
520 |
{ "default", LT_DEFAULT, '=' }, |
{"default", LT_DEFAULT, '='}, |
521 |
{ "image", LT_KERNEL, '=' }, |
{"image", LT_KERNEL, '='}, |
522 |
{ "bsd", LT_GENERIC, '=' }, |
{"bsd", LT_GENERIC, '='}, |
523 |
{ "macos", LT_GENERIC, '=' }, |
{"macos", LT_GENERIC, '='}, |
524 |
{ "macosx", LT_GENERIC, '=' }, |
{"macosx", LT_GENERIC, '='}, |
525 |
{ "magicboot", LT_GENERIC, '=' }, |
{"magicboot", LT_GENERIC, '='}, |
526 |
{ "darwin", LT_GENERIC, '=' }, |
{"darwin", LT_GENERIC, '='}, |
527 |
{ "timeout", LT_GENERIC, '=' }, |
{"timeout", LT_GENERIC, '='}, |
528 |
{ "install", LT_GENERIC, '=' }, |
{"install", LT_GENERIC, '='}, |
529 |
{ "fstype", LT_GENERIC, '=' }, |
{"fstype", LT_GENERIC, '='}, |
530 |
{ "hfstype", LT_GENERIC, '=' }, |
{"hfstype", LT_GENERIC, '='}, |
531 |
{ "delay", LT_GENERIC, '=' }, |
{"delay", LT_GENERIC, '='}, |
532 |
{ "defaultos", LT_GENERIC, '=' }, |
{"defaultos", LT_GENERIC, '='}, |
533 |
{ "init-message", LT_GENERIC, '=' }, |
{"init-message", LT_GENERIC, '='}, |
534 |
{ "enablecdboot", LT_GENERIC, ' ' }, |
{"enablecdboot", LT_GENERIC, ' '}, |
535 |
{ "enableofboot", LT_GENERIC, ' ' }, |
{"enableofboot", LT_GENERIC, ' '}, |
536 |
{ "enablenetboot", LT_GENERIC, ' ' }, |
{"enablenetboot", LT_GENERIC, ' '}, |
537 |
{ "nonvram", LT_GENERIC, ' ' }, |
{"nonvram", LT_GENERIC, ' '}, |
538 |
{ "hide", LT_GENERIC, ' ' }, |
{"hide", LT_GENERIC, ' '}, |
539 |
{ "protect", LT_GENERIC, ' ' }, |
{"protect", LT_GENERIC, ' '}, |
540 |
{ "nobless", LT_GENERIC, ' ' }, |
{"nobless", LT_GENERIC, ' '}, |
541 |
{ "nonvram", LT_GENERIC, ' ' }, |
{"nonvram", LT_GENERIC, ' '}, |
542 |
{ "brokenosx", LT_GENERIC, ' ' }, |
{"brokenosx", LT_GENERIC, ' '}, |
543 |
{ "usemount", LT_GENERIC, ' ' }, |
{"usemount", LT_GENERIC, ' '}, |
544 |
{ "mntpoint", LT_GENERIC, '=' }, |
{"mntpoint", LT_GENERIC, '='}, |
545 |
{ "partition", LT_GENERIC, '=' }, |
{"partition", LT_GENERIC, '='}, |
546 |
{ "device", LT_GENERIC, '=' }, |
{"device", LT_GENERIC, '='}, |
547 |
{ "fstype", LT_GENERIC, '=' }, |
{"fstype", LT_GENERIC, '='}, |
548 |
{ "initrd", LT_INITRD, '=', ';' }, |
{"initrd", LT_INITRD, '=', ';'}, |
549 |
{ "append", LT_KERNELARGS, '=' }, |
{"append", LT_KERNELARGS, '='}, |
550 |
{ "boot", LT_BOOT, '=' }, |
{"boot", LT_BOOT, '='}, |
551 |
{ "lba", LT_LBA, ' ' }, |
{"lba", LT_LBA, ' '}, |
552 |
{ NULL, 0, 0 }, |
{NULL, 0, 0}, |
553 |
}; |
}; |
554 |
|
|
555 |
struct keywordTypes liloKeywords[] = { |
struct keywordTypes liloKeywords[] = { |
556 |
{ "label", LT_TITLE, '=' }, |
{"label", LT_TITLE, '='}, |
557 |
{ "root", LT_ROOT, '=' }, |
{"root", LT_ROOT, '='}, |
558 |
{ "default", LT_DEFAULT, '=' }, |
{"default", LT_DEFAULT, '='}, |
559 |
{ "image", LT_KERNEL, '=' }, |
{"image", LT_KERNEL, '='}, |
560 |
{ "other", LT_OTHER, '=' }, |
{"other", LT_OTHER, '='}, |
561 |
{ "initrd", LT_INITRD, '=' }, |
{"initrd", LT_INITRD, '='}, |
562 |
{ "append", LT_KERNELARGS, '=' }, |
{"append", LT_KERNELARGS, '='}, |
563 |
{ "boot", LT_BOOT, '=' }, |
{"boot", LT_BOOT, '='}, |
564 |
{ "lba", LT_LBA, ' ' }, |
{"lba", LT_LBA, ' '}, |
565 |
{ NULL, 0, 0 }, |
{NULL, 0, 0}, |
566 |
}; |
}; |
567 |
|
|
568 |
struct keywordTypes eliloKeywords[] = { |
struct keywordTypes eliloKeywords[] = { |
569 |
{ "label", LT_TITLE, '=' }, |
{"label", LT_TITLE, '='}, |
570 |
{ "root", LT_ROOT, '=' }, |
{"root", LT_ROOT, '='}, |
571 |
{ "default", LT_DEFAULT, '=' }, |
{"default", LT_DEFAULT, '='}, |
572 |
{ "image", LT_KERNEL, '=' }, |
{"image", LT_KERNEL, '='}, |
573 |
{ "initrd", LT_INITRD, '=' }, |
{"initrd", LT_INITRD, '='}, |
574 |
{ "append", LT_KERNELARGS, '=' }, |
{"append", LT_KERNELARGS, '='}, |
575 |
{ "vmm", LT_HYPER, '=' }, |
{"vmm", LT_HYPER, '='}, |
576 |
{ NULL, 0, 0 }, |
{NULL, 0, 0}, |
577 |
}; |
}; |
578 |
|
|
579 |
struct keywordTypes siloKeywords[] = { |
struct keywordTypes siloKeywords[] = { |
580 |
{ "label", LT_TITLE, '=' }, |
{"label", LT_TITLE, '='}, |
581 |
{ "root", LT_ROOT, '=' }, |
{"root", LT_ROOT, '='}, |
582 |
{ "default", LT_DEFAULT, '=' }, |
{"default", LT_DEFAULT, '='}, |
583 |
{ "image", LT_KERNEL, '=' }, |
{"image", LT_KERNEL, '='}, |
584 |
{ "other", LT_OTHER, '=' }, |
{"other", LT_OTHER, '='}, |
585 |
{ "initrd", LT_INITRD, '=' }, |
{"initrd", LT_INITRD, '='}, |
586 |
{ "append", LT_KERNELARGS, '=' }, |
{"append", LT_KERNELARGS, '='}, |
587 |
{ "boot", LT_BOOT, '=' }, |
{"boot", LT_BOOT, '='}, |
588 |
{ NULL, 0, 0 }, |
{NULL, 0, 0}, |
589 |
}; |
}; |
590 |
|
|
591 |
struct keywordTypes ziplKeywords[] = { |
struct keywordTypes ziplKeywords[] = { |
592 |
{ "target", LT_BOOTROOT, '=' }, |
{"target", LT_BOOTROOT, '='}, |
593 |
{ "image", LT_KERNEL, '=' }, |
{"image", LT_KERNEL, '='}, |
594 |
{ "ramdisk", LT_INITRD, '=' }, |
{"ramdisk", LT_INITRD, '='}, |
595 |
{ "parameters", LT_KERNELARGS, '=' }, |
{"parameters", LT_KERNELARGS, '='}, |
596 |
{ "default", LT_DEFAULT, '=' }, |
{"default", LT_DEFAULT, '='}, |
597 |
{ NULL, 0, 0 }, |
{NULL, 0, 0}, |
598 |
}; |
}; |
599 |
|
|
600 |
struct keywordTypes extlinuxKeywords[] = { |
struct keywordTypes extlinuxKeywords[] = { |
601 |
{ "label", LT_TITLE, ' ' }, |
{"label", LT_TITLE, ' '}, |
602 |
{ "root", LT_ROOT, ' ' }, |
{"root", LT_ROOT, ' '}, |
603 |
{ "default", LT_DEFAULT, ' ' }, |
{"default", LT_DEFAULT, ' '}, |
604 |
{ "kernel", LT_KERNEL, ' ' }, |
{"kernel", LT_KERNEL, ' '}, |
605 |
{ "initrd", LT_INITRD, ' ', ',' }, |
{"initrd", LT_INITRD, ' ', ','}, |
606 |
{ "append", LT_KERNELARGS, ' ' }, |
{"append", LT_KERNELARGS, ' '}, |
607 |
{ "prompt", LT_UNKNOWN, ' ' }, |
{"prompt", LT_UNKNOWN, ' '}, |
608 |
{ NULL, 0, 0 }, |
{"fdt", LT_DEVTREE, ' '}, |
609 |
|
{"fdtdir", LT_DEVTREE, ' '}, |
610 |
|
{NULL, 0, 0}, |
611 |
}; |
}; |
612 |
|
|
613 |
int useextlinuxmenu; |
int useextlinuxmenu; |
614 |
struct configFileInfo eliloConfigType = { |
struct configFileInfo eliloConfigType = { |
615 |
.defaultConfig = "/boot/efi/EFI/redhat/elilo.conf", |
.defaultConfig = "/boot/efi/EFI/redhat/elilo.conf", |
616 |
.keywords = eliloKeywords, |
.keywords = eliloKeywords, |
617 |
.entryStart = LT_KERNEL, |
.entryStart = LT_KERNEL, |
618 |
.needsBootPrefix = 1, |
.needsBootPrefix = 1, |
619 |
.argsInQuotes = 1, |
.argsInQuotes = 1, |
620 |
.mbConcatArgs = 1, |
.mbConcatArgs = 1, |
621 |
|
.titlePosition = 1, |
622 |
}; |
}; |
623 |
|
|
624 |
struct configFileInfo liloConfigType = { |
struct configFileInfo liloConfigType = { |
625 |
.defaultConfig = "/etc/lilo.conf", |
.defaultConfig = "/etc/lilo.conf", |
626 |
.keywords = liloKeywords, |
.keywords = liloKeywords, |
627 |
.entryStart = LT_KERNEL, |
.entryStart = LT_KERNEL, |
628 |
.argsInQuotes = 1, |
.argsInQuotes = 1, |
629 |
.maxTitleLength = 15, |
.maxTitleLength = 15, |
630 |
|
.titlePosition = 1, |
631 |
}; |
}; |
632 |
|
|
633 |
struct configFileInfo yabootConfigType = { |
struct configFileInfo yabootConfigType = { |
634 |
.defaultConfig = "/etc/yaboot.conf", |
.defaultConfig = "/etc/yaboot.conf", |
635 |
.keywords = yabootKeywords, |
.keywords = yabootKeywords, |
636 |
.entryStart = LT_KERNEL, |
.entryStart = LT_KERNEL, |
637 |
.needsBootPrefix = 1, |
.needsBootPrefix = 1, |
638 |
.argsInQuotes = 1, |
.argsInQuotes = 1, |
639 |
.maxTitleLength = 15, |
.maxTitleLength = 15, |
640 |
.mbAllowExtraInitRds = 1, |
.mbAllowExtraInitRds = 1, |
641 |
|
.titlePosition = 1, |
642 |
}; |
}; |
643 |
|
|
644 |
struct configFileInfo siloConfigType = { |
struct configFileInfo siloConfigType = { |
645 |
.defaultConfig = "/etc/silo.conf", |
.defaultConfig = "/etc/silo.conf", |
646 |
.keywords = siloKeywords, |
.keywords = siloKeywords, |
647 |
.entryStart = LT_KERNEL, |
.entryStart = LT_KERNEL, |
648 |
.needsBootPrefix = 1, |
.needsBootPrefix = 1, |
649 |
.argsInQuotes = 1, |
.argsInQuotes = 1, |
650 |
.maxTitleLength = 15, |
.maxTitleLength = 15, |
651 |
|
.titlePosition = 1, |
652 |
}; |
}; |
653 |
|
|
654 |
struct configFileInfo ziplConfigType = { |
struct configFileInfo ziplConfigType = { |
655 |
.defaultConfig = "/etc/zipl.conf", |
.defaultConfig = "/etc/zipl.conf", |
656 |
.keywords = ziplKeywords, |
.keywords = ziplKeywords, |
657 |
.entryStart = LT_TITLE, |
.entryStart = LT_TITLE, |
658 |
.argsInQuotes = 1, |
.argsInQuotes = 1, |
659 |
.titleBracketed = 1, |
.titleBracketed = 1, |
660 |
}; |
}; |
661 |
|
|
662 |
struct configFileInfo extlinuxConfigType = { |
struct configFileInfo extlinuxConfigType = { |
663 |
.defaultConfig = "/boot/extlinux/extlinux.conf", |
.defaultConfig = "/boot/extlinux/extlinux.conf", |
664 |
.keywords = extlinuxKeywords, |
.keywords = extlinuxKeywords, |
665 |
.entryStart = LT_TITLE, |
.caseInsensitive = 1, |
666 |
.needsBootPrefix = 1, |
.entryStart = LT_TITLE, |
667 |
.maxTitleLength = 255, |
.needsBootPrefix = 1, |
668 |
.mbAllowExtraInitRds = 1, |
.maxTitleLength = 255, |
669 |
|
.mbAllowExtraInitRds = 1, |
670 |
|
.defaultIsUnquoted = 1, |
671 |
|
.titlePosition = 1, |
672 |
}; |
}; |
673 |
|
|
674 |
struct grubConfig { |
struct grubConfig { |
675 |
struct singleLine * theLines; |
struct singleLine *theLines; |
676 |
struct singleEntry * entries; |
struct singleEntry *entries; |
677 |
char * primaryIndent; |
char *primaryIndent; |
678 |
char * secondaryIndent; |
char *secondaryIndent; |
679 |
int defaultImage; /* -1 if none specified -- this value is |
int defaultImage; /* -1 if none specified -- this value is |
680 |
* written out, overriding original */ |
* written out, overriding original */ |
681 |
int fallbackImage; /* just like defaultImage */ |
int fallbackImage; /* just like defaultImage */ |
682 |
int flags; |
int flags; |
683 |
struct configFileInfo * cfi; |
struct configFileInfo *cfi; |
684 |
|
int isModified; /* assumes only one entry added |
685 |
|
per invocation of grubby */ |
686 |
}; |
}; |
687 |
|
|
688 |
blkid_cache blkid; |
blkid_cache blkid; |
689 |
|
|
690 |
struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index); |
struct singleEntry *findEntryByIndex(struct grubConfig *cfg, int index); |
691 |
struct singleEntry * findEntryByPath(struct grubConfig * cfg, |
struct singleEntry *findEntryByPath(struct grubConfig *cfg, |
692 |
const char * path, const char * prefix, |
const char *path, const char *prefix, |
693 |
int * index); |
int *index); |
694 |
static int readFile(int fd, char ** bufPtr); |
struct singleEntry *findEntryByTitle(struct grubConfig *cfg, char *title, |
695 |
static void lineInit(struct singleLine * line); |
int *index); |
696 |
struct singleLine * lineDup(struct singleLine * line); |
static int readFile(int fd, char **bufPtr); |
697 |
static void lineFree(struct singleLine * line); |
static void lineInit(struct singleLine *line); |
698 |
static int lineWrite(FILE * out, struct singleLine * line, |
struct singleLine *lineDup(struct singleLine *line); |
699 |
struct configFileInfo * cfi); |
static void lineFree(struct singleLine *line); |
700 |
static int getNextLine(char ** bufPtr, struct singleLine * line, |
static int lineWrite(FILE * out, struct singleLine *line, |
701 |
struct configFileInfo * cfi); |
struct configFileInfo *cfi); |
702 |
static char * getRootSpecifier(char * str); |
static int getNextLine(char **bufPtr, struct singleLine *line, |
703 |
static void requote(struct singleLine *line, struct configFileInfo * cfi); |
struct configFileInfo *cfi); |
704 |
static void insertElement(struct singleLine * line, |
static char *getRootSpecifier(char *str); |
705 |
const char * item, int insertHere, |
static void requote(struct singleLine *line, struct configFileInfo *cfi); |
706 |
struct configFileInfo * cfi); |
static void insertElement(struct singleLine *line, |
707 |
static void removeElement(struct singleLine * line, int removeHere); |
const char *item, int insertHere, |
708 |
static struct keywordTypes * getKeywordByType(enum lineType_e type, |
struct configFileInfo *cfi); |
709 |
struct configFileInfo * cfi); |
static void removeElement(struct singleLine *line, int removeHere); |
710 |
static enum lineType_e getTypeByKeyword(char * keyword, |
static struct keywordTypes *getKeywordByType(enum lineType_e type, |
711 |
struct configFileInfo * cfi); |
struct configFileInfo *cfi); |
712 |
static struct singleLine * getLineByType(enum lineType_e type, |
static enum lineType_e getTypeByKeyword(char *keyword, |
713 |
struct singleLine * line); |
struct configFileInfo *cfi); |
714 |
static int checkForExtLinux(struct grubConfig * config); |
static struct singleLine *getLineByType(enum lineType_e type, |
715 |
struct singleLine * addLineTmpl(struct singleEntry * entry, |
struct singleLine *line); |
716 |
struct singleLine * tmplLine, |
static int checkForExtLinux(struct grubConfig *config); |
717 |
struct singleLine * prevLine, |
struct singleLine *addLineTmpl(struct singleEntry *entry, |
718 |
const char * val, |
struct singleLine *tmplLine, |
719 |
struct configFileInfo * cfi); |
struct singleLine *prevLine, |
720 |
struct singleLine * addLine(struct singleEntry * entry, |
const char *val, struct configFileInfo *cfi); |
721 |
struct configFileInfo * cfi, |
struct singleLine *addLine(struct singleEntry *entry, |
722 |
enum lineType_e type, char * defaultIndent, |
struct configFileInfo *cfi, |
723 |
const char * val); |
enum lineType_e type, char *defaultIndent, |
724 |
|
const char *val); |
725 |
|
|
726 |
static char * sdupprintf(const char *format, ...) |
static char *sdupprintf(const char *format, ...) |
727 |
#ifdef __GNUC__ |
#ifdef __GNUC__ |
728 |
__attribute__ ((format (printf, 1, 2))); |
__attribute__ ((format(printf, 1, 2))); |
729 |
#else |
#else |
730 |
; |
; |
731 |
#endif |
#endif |
732 |
|
|
733 |
static char * sdupprintf(const char *format, ...) { |
static char *sdupprintf(const char *format, ...) |
734 |
char *buf = NULL; |
{ |
735 |
char c; |
char *buf = NULL; |
736 |
va_list args; |
char c; |
737 |
size_t size = 0; |
va_list args; |
738 |
va_start(args, format); |
size_t size = 0; |
739 |
|
va_start(args, format); |
740 |
/* XXX requires C99 vsnprintf behavior */ |
|
741 |
size = vsnprintf(&c, 1, format, args) + 1; |
/* XXX requires C99 vsnprintf behavior */ |
742 |
if (size == -1) { |
size = vsnprintf(&c, 1, format, args) + 1; |
743 |
printf("ERROR: vsnprintf behavior is not C99\n"); |
if (size == -1) { |
744 |
abort(); |
printf("ERROR: vsnprintf behavior is not C99\n"); |
745 |
} |
abort(); |
746 |
|
} |
747 |
|
|
748 |
va_end(args); |
va_end(args); |
749 |
va_start(args, format); |
va_start(args, format); |
750 |
|
|
751 |
buf = malloc(size); |
buf = malloc(size); |
752 |
if (buf == NULL) |
if (buf == NULL) |
753 |
return NULL; |
return NULL; |
754 |
vsnprintf(buf, size, format, args); |
vsnprintf(buf, size, format, args); |
755 |
va_end (args); |
va_end(args); |
756 |
|
|
757 |
return buf; |
return buf; |
758 |
} |
} |
759 |
|
|
760 |
static struct keywordTypes * getKeywordByType(enum lineType_e type, |
static inline int |
761 |
struct configFileInfo * cfi) { |
kwcmp(struct keywordTypes *kw, const char * label, int case_insensitive) |
762 |
struct keywordTypes * kw; |
{ |
763 |
for (kw = cfi->keywords; kw->key; kw++) { |
int kwl = strlen(kw->key); |
764 |
if (kw->type == type) |
int ll = strlen(label); |
765 |
return kw; |
int rc; |
766 |
|
int (*snc)(const char *s1, const char *s2, size_t n) = |
767 |
|
case_insensitive ? strncasecmp : strncmp; |
768 |
|
int (*sc)(const char *s1, const char *s2) = |
769 |
|
case_insensitive ? strcasecmp : strcmp; |
770 |
|
|
771 |
|
rc = snc(kw->key, label, kwl); |
772 |
|
if (rc) |
773 |
|
return rc; |
774 |
|
|
775 |
|
for (int i = kwl; i < ll; i++) { |
776 |
|
if (isspace(label[i])) |
777 |
|
return 0; |
778 |
|
if (kw->separatorChar && label[i] == kw->separatorChar) |
779 |
|
return 0; |
780 |
|
else if (kw->nextChar && label[i] == kw->nextChar) |
781 |
|
return 0; |
782 |
|
return sc(kw->key+kwl, label+kwl); |
783 |
} |
} |
784 |
return NULL; |
return 0; |
785 |
} |
} |
786 |
|
|
787 |
static char *getKeyByType(enum lineType_e type, struct configFileInfo * cfi) { |
static enum lineType_e preferredLineType(enum lineType_e type, |
788 |
struct keywordTypes *kt = getKeywordByType(type, cfi); |
struct configFileInfo *cfi) |
789 |
if (kt) |
{ |
790 |
return kt->key; |
if (isEfi && cfi == &grub2ConfigType) { |
791 |
return "unknown"; |
switch (type) { |
792 |
|
case LT_KERNEL: |
793 |
|
return isEfiOnly ? LT_KERNEL : LT_KERNEL_EFI; |
794 |
|
case LT_INITRD: |
795 |
|
return isEfiOnly ? LT_INITRD : LT_INITRD_EFI; |
796 |
|
default: |
797 |
|
return type; |
798 |
|
} |
799 |
|
#if defined(__i386__) || defined(__x86_64__) |
800 |
|
} else if (cfi == &grub2ConfigType) { |
801 |
|
switch (type) { |
802 |
|
case LT_KERNEL: |
803 |
|
return LT_KERNEL_16; |
804 |
|
case LT_INITRD: |
805 |
|
return LT_INITRD_16; |
806 |
|
default: |
807 |
|
return type; |
808 |
|
} |
809 |
|
#endif |
810 |
|
} |
811 |
|
return type; |
812 |
} |
} |
813 |
|
|
814 |
static char * getpathbyspec(char *device) { |
static struct keywordTypes *getKeywordByType(enum lineType_e type, |
815 |
if (!blkid) |
struct configFileInfo *cfi) |
816 |
blkid_get_cache(&blkid, NULL); |
{ |
817 |
|
for (struct keywordTypes * kw = cfi->keywords; kw->key; kw++) { |
818 |
|
if (kw->type == type) |
819 |
|
return kw; |
820 |
|
} |
821 |
|
return NULL; |
822 |
|
} |
823 |
|
|
824 |
return blkid_get_devname(blkid, device, NULL); |
static char *getKeyByType(enum lineType_e type, struct configFileInfo *cfi) |
825 |
|
{ |
826 |
|
struct keywordTypes *kt = getKeywordByType(type, cfi); |
827 |
|
if (kt) |
828 |
|
return kt->key; |
829 |
|
return "unknown"; |
830 |
} |
} |
831 |
|
|
832 |
static char * getuuidbydev(char *device) { |
static char *getpathbyspec(char *device) |
833 |
if (!blkid) |
{ |
834 |
blkid_get_cache(&blkid, NULL); |
if (!blkid) |
835 |
|
blkid_get_cache(&blkid, NULL); |
836 |
|
|
837 |
return blkid_get_tag_value(blkid, "UUID", device); |
return blkid_get_devname(blkid, device, NULL); |
838 |
} |
} |
839 |
|
|
840 |
static enum lineType_e getTypeByKeyword(char * keyword, |
static char *getuuidbydev(char *device) |
841 |
struct configFileInfo * cfi) { |
{ |
842 |
struct keywordTypes * kw; |
if (!blkid) |
843 |
for (kw = cfi->keywords; kw->key; kw++) { |
blkid_get_cache(&blkid, NULL); |
|
if (!strcmp(keyword, kw->key)) |
|
|
return kw->type; |
|
|
} |
|
|
return LT_UNKNOWN; |
|
|
} |
|
844 |
|
|
845 |
static struct singleLine * getLineByType(enum lineType_e type, |
return blkid_get_tag_value(blkid, "UUID", device); |
|
struct singleLine * line) { |
|
|
dbgPrintf("getLineByType(%d): ", type); |
|
|
for (; line; line = line->next) { |
|
|
dbgPrintf("%d:%s ", line->type, |
|
|
line->numElements ? line->elements[0].item : "(empty)"); |
|
|
if (line->type & type) break; |
|
|
} |
|
|
dbgPrintf(line ? "\n" : " (failed)\n"); |
|
|
return line; |
|
846 |
} |
} |
847 |
|
|
848 |
static int isBracketedTitle(struct singleLine * line) { |
static enum lineType_e getTypeByKeyword(char *keyword, |
849 |
if (line->numElements == 1 && *line->elements[0].item == '[') { |
struct configFileInfo *cfi) |
850 |
int len = strlen(line->elements[0].item); |
{ |
851 |
if (*(line->elements[0].item + len - 1) == ']') { |
for (struct keywordTypes * kw = cfi->keywords; kw->key; kw++) { |
852 |
/* FIXME: this is a hack... */ |
if (!kwcmp(kw, keyword, cfi->caseInsensitive)) |
853 |
if (strcmp(line->elements[0].item, "[defaultboot]")) { |
return kw->type; |
854 |
return 1; |
} |
855 |
} |
return LT_UNKNOWN; |
|
} |
|
|
} |
|
|
return 0; |
|
856 |
} |
} |
857 |
|
|
858 |
static int isEntryStart(struct singleLine * line, |
static struct singleLine *getLineByType(enum lineType_e type, |
859 |
struct configFileInfo * cfi) { |
struct singleLine *line) |
860 |
return line->type == cfi->entryStart || line->type == LT_OTHER || |
{ |
861 |
(cfi->titleBracketed && isBracketedTitle(line)); |
dbgPrintf("getLineByType(%d): ", type); |
862 |
|
for (; line; line = line->next) { |
863 |
|
dbgPrintf("%d:%s ", line->type, |
864 |
|
line->numElements ? line->elements[0]. |
865 |
|
item : "(empty)"); |
866 |
|
if (line->type & type) |
867 |
|
break; |
868 |
|
} |
869 |
|
dbgPrintf(line ? "\n" : " (failed)\n"); |
870 |
|
return line; |
871 |
} |
} |
872 |
|
|
873 |
/* extract the title from within brackets (for zipl) */ |
static int isBracketedTitle(struct singleLine *line) |
874 |
static char * extractTitle(struct singleLine * line) { |
{ |
875 |
/* bracketed title... let's extract it (leaks a byte) */ |
if (line->numElements == 1 && *line->elements[0].item == '[') { |
876 |
char * title; |
int len = strlen(line->elements[0].item); |
877 |
title = strdup(line->elements[0].item); |
if (*(line->elements[0].item + len - 1) == ']') { |
878 |
title++; |
/* FIXME: this is a hack... */ |
879 |
*(title + strlen(title) - 1) = '\0'; |
if (strcmp(line->elements[0].item, "[defaultboot]")) { |
880 |
return title; |
return 1; |
881 |
} |
} |
882 |
|
} |
883 |
static int readFile(int fd, char ** bufPtr) { |
} |
884 |
int alloced = 0, size = 0, i = 0; |
return 0; |
885 |
char * buf = NULL; |
} |
|
|
|
|
do { |
|
|
size += i; |
|
|
if ((size + 1024) > alloced) { |
|
|
alloced += 4096; |
|
|
buf = realloc(buf, alloced + 1); |
|
|
} |
|
|
} while ((i = read(fd, buf + size, 1024)) > 0); |
|
|
|
|
|
if (i < 0) { |
|
|
fprintf(stderr, _("error reading input: %s\n"), strerror(errno)); |
|
|
free(buf); |
|
|
return 1; |
|
|
} |
|
|
|
|
|
buf = realloc(buf, size + 2); |
|
|
if (size == 0) |
|
|
buf[size++] = '\n'; |
|
|
else |
|
|
if (buf[size - 1] != '\n') |
|
|
buf[size++] = '\n'; |
|
|
buf[size] = '\0'; |
|
886 |
|
|
887 |
*bufPtr = buf; |
static int isEntryStart(struct singleLine *line, struct configFileInfo *cfi) |
888 |
|
{ |
889 |
|
return line->type == cfi->entryStart || line->type == LT_OTHER || |
890 |
|
(cfi->titleBracketed && isBracketedTitle(line)); |
891 |
|
} |
892 |
|
|
893 |
return 0; |
/* extract the title from within brackets (for zipl) */ |
894 |
|
static char *extractTitle(struct grubConfig *cfg, struct singleLine *line) |
895 |
|
{ |
896 |
|
/* bracketed title... let's extract it */ |
897 |
|
char *title = NULL; |
898 |
|
if (line->type == LT_TITLE) { |
899 |
|
char *tmp = line->elements[cfg->cfi->titlePosition].item; |
900 |
|
if (cfg->cfi->titleBracketed) { |
901 |
|
tmp++; |
902 |
|
title = strdup(tmp); |
903 |
|
*(title + strlen(title) - 1) = '\0'; |
904 |
|
} else { |
905 |
|
title = strdup(tmp); |
906 |
|
} |
907 |
|
} else if (line->type == LT_MENUENTRY) |
908 |
|
title = strdup(line->elements[1].item); |
909 |
|
else |
910 |
|
return NULL; |
911 |
|
return title; |
912 |
} |
} |
913 |
|
|
914 |
static void lineInit(struct singleLine * line) { |
static int readFile(int fd, char **bufPtr) |
915 |
line->indent = NULL; |
{ |
916 |
line->elements = NULL; |
int alloced = 0, size = 0, i = 0; |
917 |
line->numElements = 0; |
char *buf = NULL; |
|
line->next = NULL; |
|
|
} |
|
|
|
|
|
struct singleLine * lineDup(struct singleLine * line) { |
|
|
int i; |
|
|
struct singleLine * newLine = malloc(sizeof(*newLine)); |
|
|
|
|
|
newLine->indent = strdup(line->indent); |
|
|
newLine->next = NULL; |
|
|
newLine->type = line->type; |
|
|
newLine->numElements = line->numElements; |
|
|
newLine->elements = malloc(sizeof(*newLine->elements) * |
|
|
newLine->numElements); |
|
|
|
|
|
for (i = 0; i < newLine->numElements; i++) { |
|
|
newLine->elements[i].indent = strdup(line->elements[i].indent); |
|
|
newLine->elements[i].item = strdup(line->elements[i].item); |
|
|
} |
|
918 |
|
|
919 |
return newLine; |
do { |
920 |
} |
size += i; |
921 |
|
if ((size + 1024) > alloced) { |
922 |
|
alloced += 4096; |
923 |
|
buf = realloc(buf, alloced + 1); |
924 |
|
} |
925 |
|
} while ((i = read(fd, buf + size, 1024)) > 0); |
926 |
|
|
927 |
static void lineFree(struct singleLine * line) { |
if (i < 0) { |
928 |
int i; |
fprintf(stderr, _("error reading input: %s\n"), |
929 |
|
strerror(errno)); |
930 |
|
free(buf); |
931 |
|
return 1; |
932 |
|
} |
933 |
|
|
934 |
if (line->indent) free(line->indent); |
buf = realloc(buf, size + 2); |
935 |
|
if (size == 0) |
936 |
|
buf[size++] = '\n'; |
937 |
|
else if (buf[size - 1] != '\n') |
938 |
|
buf[size++] = '\n'; |
939 |
|
buf[size] = '\0'; |
940 |
|
|
941 |
for (i = 0; i < line->numElements; i++) { |
*bufPtr = buf; |
|
free(line->elements[i].item); |
|
|
free(line->elements[i].indent); |
|
|
} |
|
942 |
|
|
943 |
if (line->elements) free(line->elements); |
return 0; |
|
lineInit(line); |
|
944 |
} |
} |
945 |
|
|
946 |
static int lineWrite(FILE * out, struct singleLine * line, |
static void lineInit(struct singleLine *line) |
947 |
struct configFileInfo * cfi) { |
{ |
948 |
int i; |
line->type = LT_UNIDENTIFIED; |
949 |
|
line->indent = NULL; |
950 |
|
line->elements = NULL; |
951 |
|
line->numElements = 0; |
952 |
|
line->next = NULL; |
953 |
|
} |
954 |
|
|
955 |
if (fprintf(out, "%s", line->indent) == -1) return -1; |
struct singleLine *lineDup(struct singleLine *line) |
956 |
|
{ |
957 |
|
struct singleLine *newLine = malloc(sizeof(*newLine)); |
958 |
|
|
959 |
for (i = 0; i < line->numElements; i++) { |
newLine->indent = strdup(line->indent); |
960 |
/* Need to handle this, because we strip the quotes from |
newLine->next = NULL; |
961 |
* menuentry when read it. */ |
newLine->type = line->type; |
962 |
if (line->type == LT_MENUENTRY && i == 1) { |
newLine->numElements = line->numElements; |
963 |
if(!isquote(*line->elements[i].item)) |
newLine->elements = malloc(sizeof(*newLine->elements) * |
964 |
fprintf(out, "\'%s\'", line->elements[i].item); |
newLine->numElements); |
|
else |
|
|
fprintf(out, "%s", line->elements[i].item); |
|
|
fprintf(out, "%s", line->elements[i].indent); |
|
965 |
|
|
966 |
continue; |
for (int i = 0; i < newLine->numElements; i++) { |
967 |
|
newLine->elements[i].indent = strdup(line->elements[i].indent); |
968 |
|
newLine->elements[i].item = strdup(line->elements[i].item); |
969 |
} |
} |
970 |
|
|
971 |
if (i == 1 && line->type == LT_KERNELARGS && cfi->argsInQuotes) |
return newLine; |
972 |
if (fputc('"', out) == EOF) return -1; |
} |
|
|
|
|
if (fprintf(out, "%s", line->elements[i].item) == -1) return -1; |
|
|
if (i < line->numElements - 1) |
|
|
if (fprintf(out, "%s", line->elements[i].indent) == -1) return -1; |
|
|
} |
|
973 |
|
|
974 |
if (line->type == LT_KERNELARGS && cfi->argsInQuotes) |
static void lineFree(struct singleLine *line) |
975 |
if (fputc('"', out) == EOF) return -1; |
{ |
976 |
|
if (line->indent) |
977 |
|
free(line->indent); |
978 |
|
|
979 |
if (fprintf(out, "\n") == -1) return -1; |
for (int i = 0; i < line->numElements; i++) { |
980 |
|
free(line->elements[i].item); |
981 |
|
free(line->elements[i].indent); |
982 |
|
} |
983 |
|
|
984 |
return 0; |
if (line->elements) |
985 |
|
free(line->elements); |
986 |
|
lineInit(line); |
987 |
} |
} |
988 |
|
|
989 |
/* we've guaranteed that the buffer ends w/ \n\0 */ |
static int lineWrite(FILE * out, struct singleLine *line, |
990 |
static int getNextLine(char ** bufPtr, struct singleLine * line, |
struct configFileInfo *cfi) |
991 |
struct configFileInfo * cfi) { |
{ |
992 |
char * end; |
if (fprintf(out, "%s", line->indent) == -1) |
993 |
char * start = *bufPtr; |
return -1; |
|
char * chptr; |
|
|
int elementsAlloced = 0; |
|
|
struct lineElement * element; |
|
|
int first = 1; |
|
994 |
|
|
995 |
lineFree(line); |
for (int i = 0; i < line->numElements; i++) { |
996 |
|
/* Need to handle this, because we strip the quotes from |
997 |
|
* menuentry when read it. */ |
998 |
|
if (line->type == LT_MENUENTRY && i == 1) { |
999 |
|
if (!isquote(*line->elements[i].item)) { |
1000 |
|
int substring = 0; |
1001 |
|
/* If the line contains nested quotes, we did |
1002 |
|
* not strip the "interna" quotes and we must |
1003 |
|
* use the right quotes again when writing |
1004 |
|
* the updated file. */ |
1005 |
|
for (int j = i; j < line->numElements; j++) { |
1006 |
|
if (strchr(line->elements[i].item, '\'') |
1007 |
|
!= NULL) { |
1008 |
|
substring = 1; |
1009 |
|
fprintf(out, "\"%s\"", |
1010 |
|
line->elements[i].item); |
1011 |
|
break; |
1012 |
|
} |
1013 |
|
} |
1014 |
|
if (!substring) |
1015 |
|
fprintf(out, "\'%s\'", |
1016 |
|
line->elements[i].item); |
1017 |
|
} else { |
1018 |
|
fprintf(out, "%s", line->elements[i].item); |
1019 |
|
} |
1020 |
|
fprintf(out, "%s", line->elements[i].indent); |
1021 |
|
|
1022 |
end = strchr(start, '\n'); |
continue; |
1023 |
*end = '\0'; |
} |
|
*bufPtr = end + 1; |
|
1024 |
|
|
1025 |
for (chptr = start; *chptr && isspace(*chptr); chptr++) ; |
if (i == 1 && line->type == LT_KERNELARGS && cfi->argsInQuotes) |
1026 |
|
if (fputc('"', out) == EOF) |
1027 |
|
return -1; |
1028 |
|
|
1029 |
|
if (fprintf(out, "%s", line->elements[i].item) == -1) |
1030 |
|
return -1; |
1031 |
|
if (i < line->numElements - 1 || line->type == LT_SET_VARIABLE) |
1032 |
|
if (fprintf(out, "%s", line->elements[i].indent) == -1) |
1033 |
|
return -1; |
1034 |
|
} |
1035 |
|
|
1036 |
|
if (line->type == LT_KERNELARGS && cfi->argsInQuotes) |
1037 |
|
if (fputc('"', out) == EOF) |
1038 |
|
return -1; |
1039 |
|
|
1040 |
line->indent = strndup(start, chptr - start); |
if (fprintf(out, "\n") == -1) |
1041 |
start = chptr; |
return -1; |
1042 |
|
|
1043 |
while (start < end) { |
return 0; |
1044 |
/* we know !isspace(*start) */ |
} |
1045 |
|
|
1046 |
if (elementsAlloced == line->numElements) { |
/* we've guaranteed that the buffer ends w/ \n\0 */ |
1047 |
elementsAlloced += 5; |
static int getNextLine(char **bufPtr, struct singleLine *line, |
1048 |
line->elements = realloc(line->elements, |
struct configFileInfo *cfi) |
1049 |
sizeof(*line->elements) * elementsAlloced); |
{ |
1050 |
} |
char *end; |
1051 |
|
char *start = *bufPtr; |
1052 |
|
char *chptr; |
1053 |
|
int elementsAlloced = 0; |
1054 |
|
struct lineElement *element; |
1055 |
|
int first = 1; |
1056 |
|
|
1057 |
element = line->elements + line->numElements; |
lineFree(line); |
1058 |
|
|
1059 |
chptr = start; |
end = strchr(start, '\n'); |
1060 |
while (*chptr && !isspace(*chptr)) { |
*end = '\0'; |
1061 |
if (first && *chptr == '=') break; |
*bufPtr = end + 1; |
|
chptr++; |
|
|
} |
|
|
element->item = strndup(start, chptr - start); |
|
|
start = chptr; |
|
1062 |
|
|
1063 |
/* lilo actually accepts the pathological case of append = " foo " */ |
for (chptr = start; *chptr && isspace(*chptr); chptr++) ; |
|
if (*start == '=') |
|
|
chptr = start + 1; |
|
|
else |
|
|
chptr = start; |
|
|
|
|
|
do { |
|
|
for (; *chptr && isspace(*chptr); chptr++); |
|
|
if (*chptr == '=') |
|
|
chptr = chptr + 1; |
|
|
} while (isspace(*chptr)); |
|
1064 |
|
|
1065 |
element->indent = strndup(start, chptr - start); |
line->indent = strndup(start, chptr - start); |
1066 |
start = chptr; |
start = chptr; |
1067 |
|
|
1068 |
line->numElements++; |
while (start < end) { |
1069 |
first = 0; |
/* we know !isspace(*start) */ |
|
} |
|
1070 |
|
|
1071 |
if (!line->numElements) |
if (elementsAlloced == line->numElements) { |
1072 |
line->type = LT_WHITESPACE; |
elementsAlloced += 5; |
1073 |
else { |
line->elements = realloc(line->elements, |
1074 |
line->type = getTypeByKeyword(line->elements[0].item, cfi); |
sizeof(*line->elements) * |
1075 |
if (line->type == LT_UNKNOWN) { |
elementsAlloced); |
1076 |
/* zipl does [title] instead of something reasonable like all |
} |
|
* the other boot loaders. kind of ugly */ |
|
|
if (cfi->titleBracketed && isBracketedTitle(line)) { |
|
|
line->type = LT_TITLE; |
|
|
} |
|
|
|
|
|
/* this is awkward, but we need to be able to handle keywords |
|
|
that begin with a # (specifically for #boot in grub.conf), |
|
|
but still make comments lines with no elements (everything |
|
|
stored in the indent */ |
|
|
if (*line->elements[0].item == '#') { |
|
|
char * fullLine; |
|
|
int len; |
|
|
int i; |
|
|
|
|
|
len = strlen(line->indent); |
|
|
for (i = 0; i < line->numElements; i++) |
|
|
len += strlen(line->elements[i].item) + |
|
|
strlen(line->elements[i].indent); |
|
1077 |
|
|
1078 |
fullLine = malloc(len + 1); |
element = line->elements + line->numElements; |
|
strcpy(fullLine, line->indent); |
|
|
free(line->indent); |
|
|
line->indent = fullLine; |
|
1079 |
|
|
1080 |
for (i = 0; i < line->numElements; i++) { |
chptr = start; |
1081 |
strcat(fullLine, line->elements[i].item); |
while (*chptr && !isspace(*chptr)) { |
1082 |
strcat(fullLine, line->elements[i].indent); |
if (first && *chptr == '=') |
1083 |
free(line->elements[i].item); |
break; |
1084 |
free(line->elements[i].indent); |
chptr++; |
1085 |
} |
} |
1086 |
|
if (line->type == LT_UNIDENTIFIED) |
1087 |
|
line->type = getTypeByKeyword(start, cfi); |
1088 |
|
element->item = strndup(start, chptr - start); |
1089 |
|
start = chptr; |
1090 |
|
|
1091 |
|
/* lilo actually accepts the pathological case of |
1092 |
|
* append = " foo " */ |
1093 |
|
if (*start == '=') |
1094 |
|
chptr = start + 1; |
1095 |
|
else |
1096 |
|
chptr = start; |
1097 |
|
|
1098 |
|
do { |
1099 |
|
for (; *chptr && isspace(*chptr); chptr++) ; |
1100 |
|
if (*chptr == '=') |
1101 |
|
chptr = chptr + 1; |
1102 |
|
} while (isspace(*chptr)); |
1103 |
|
|
1104 |
|
element->indent = strndup(start, chptr - start); |
1105 |
|
start = chptr; |
1106 |
|
|
1107 |
|
line->numElements++; |
1108 |
|
first = 0; |
1109 |
|
} |
1110 |
|
|
1111 |
|
if (!line->numElements) |
1112 |
line->type = LT_WHITESPACE; |
line->type = LT_WHITESPACE; |
1113 |
line->numElements = 0; |
else { |
1114 |
} |
line->type = getTypeByKeyword(line->elements[0].item, cfi); |
1115 |
} else { |
if (line->type == LT_UNKNOWN) { |
1116 |
struct keywordTypes *kw; |
/* zipl does [title] instead of something reasonable |
1117 |
|
* like all the other boot loaders. kind of ugly */ |
1118 |
|
if (cfi->titleBracketed && isBracketedTitle(line)) { |
1119 |
|
line->type = LT_TITLE; |
1120 |
|
} |
1121 |
|
|
1122 |
kw = getKeywordByType(line->type, cfi); |
/* this is awkward, but we need to be able to handle |
1123 |
|
* keywords that begin with a # (specifically for |
1124 |
|
* #boot in grub.conf), but still make comments lines |
1125 |
|
* with no elements (everything stored in the indent |
1126 |
|
*/ |
1127 |
|
if (*line->elements[0].item == '#') { |
1128 |
|
char *fullLine; |
1129 |
|
int len; |
1130 |
|
|
1131 |
|
len = strlen(line->indent); |
1132 |
|
for (int i = 0; i < line->numElements; i++) |
1133 |
|
len += strlen(line->elements[i].item) + |
1134 |
|
strlen(line->elements[i].indent); |
1135 |
|
|
1136 |
|
fullLine = malloc(len + 1); |
1137 |
|
strcpy(fullLine, line->indent); |
1138 |
|
free(line->indent); |
1139 |
|
line->indent = fullLine; |
1140 |
|
|
1141 |
|
for (int i = 0; i < line->numElements; i++) { |
1142 |
|
strcat(fullLine, |
1143 |
|
line->elements[i].item); |
1144 |
|
strcat(fullLine, |
1145 |
|
line->elements[i].indent); |
1146 |
|
free(line->elements[i].item); |
1147 |
|
free(line->elements[i].indent); |
1148 |
|
} |
1149 |
|
|
1150 |
/* space isn't the only separator, we need to split |
line->type = LT_WHITESPACE; |
1151 |
* elements up more |
line->numElements = 0; |
|
*/ |
|
|
if (!isspace(kw->separatorChar)) { |
|
|
int i; |
|
|
char indent[2] = ""; |
|
|
indent[0] = kw->separatorChar; |
|
|
for (i = 1; i < line->numElements; i++) { |
|
|
char *p; |
|
|
int j; |
|
|
int numNewElements; |
|
|
|
|
|
numNewElements = 0; |
|
|
p = line->elements[i].item; |
|
|
while (*p != '\0') { |
|
|
if (*p == kw->separatorChar) |
|
|
numNewElements++; |
|
|
p++; |
|
|
} |
|
|
if (line->numElements + numNewElements >= elementsAlloced) { |
|
|
elementsAlloced += numNewElements + 5; |
|
|
line->elements = realloc(line->elements, |
|
|
sizeof(*line->elements) * elementsAlloced); |
|
|
} |
|
|
|
|
|
for (j = line->numElements; j > i; j--) { |
|
|
line->elements[j + numNewElements] = line->elements[j]; |
|
1152 |
} |
} |
1153 |
line->numElements += numNewElements; |
} else if (line->type == LT_INITRD) { |
1154 |
|
struct keywordTypes *kw; |
1155 |
|
|
1156 |
p = line->elements[i].item; |
kw = getKeywordByType(line->type, cfi); |
|
while (*p != '\0') { |
|
1157 |
|
|
1158 |
while (*p != kw->separatorChar && *p != '\0') p++; |
/* space isn't the only separator, we need to split |
1159 |
if (*p == '\0') { |
* elements up more |
1160 |
break; |
*/ |
1161 |
} |
if (!isspace(kw->separatorChar)) { |
1162 |
|
char indent[2] = ""; |
1163 |
|
indent[0] = kw->separatorChar; |
1164 |
|
for (int i = 1; i < line->numElements; i++) { |
1165 |
|
char *p; |
1166 |
|
int numNewElements; |
1167 |
|
|
1168 |
|
numNewElements = 0; |
1169 |
|
p = line->elements[i].item; |
1170 |
|
while (*p != '\0') { |
1171 |
|
if (*p == kw->separatorChar) |
1172 |
|
numNewElements++; |
1173 |
|
p++; |
1174 |
|
} |
1175 |
|
if (line->numElements + |
1176 |
|
numNewElements >= elementsAlloced) { |
1177 |
|
elementsAlloced += |
1178 |
|
numNewElements + 5; |
1179 |
|
line->elements = |
1180 |
|
realloc(line->elements, |
1181 |
|
sizeof(*line-> |
1182 |
|
elements) * |
1183 |
|
elementsAlloced); |
1184 |
|
} |
1185 |
|
|
1186 |
line->elements[i + 1].indent = line->elements[i].indent; |
for (int j = line->numElements; j > i; |
1187 |
line->elements[i].indent = strdup(indent); |
j--) { |
1188 |
*p++ = '\0'; |
line->elements[j + |
1189 |
i++; |
numNewElements] = |
1190 |
line->elements[i].item = strdup(p); |
line->elements[j]; |
1191 |
|
} |
1192 |
|
line->numElements += numNewElements; |
1193 |
|
|
1194 |
|
p = line->elements[i].item; |
1195 |
|
while (*p != '\0') { |
1196 |
|
|
1197 |
|
while (*p != kw->separatorChar |
1198 |
|
&& *p != '\0') |
1199 |
|
p++; |
1200 |
|
if (*p == '\0') { |
1201 |
|
break; |
1202 |
|
} |
1203 |
|
|
1204 |
|
line->elements[i + 1].indent = |
1205 |
|
line->elements[i].indent; |
1206 |
|
line->elements[i].indent = |
1207 |
|
strdup(indent); |
1208 |
|
*p++ = '\0'; |
1209 |
|
i++; |
1210 |
|
line->elements[i].item = |
1211 |
|
strdup(p); |
1212 |
|
} |
1213 |
|
} |
1214 |
|
} |
1215 |
|
} else if (line->type == LT_SET_VARIABLE) { |
1216 |
|
/* and if it's a "set blah=" we need to split it |
1217 |
|
* yet a third way to avoid rhbz# XXX FIXME :/ |
1218 |
|
*/ |
1219 |
|
char *eq; |
1220 |
|
int l; |
1221 |
|
int numElements = line->numElements; |
1222 |
|
struct lineElement *newElements; |
1223 |
|
eq = strchr(line->elements[1].item, '='); |
1224 |
|
if (!eq) |
1225 |
|
return 0; |
1226 |
|
l = eq - line->elements[1].item; |
1227 |
|
if (eq[1] != 0) |
1228 |
|
numElements++; |
1229 |
|
newElements = calloc(numElements,sizeof (*newElements)); |
1230 |
|
memcpy(&newElements[0], &line->elements[0], |
1231 |
|
sizeof (newElements[0])); |
1232 |
|
newElements[1].item = |
1233 |
|
strndup(line->elements[1].item, l); |
1234 |
|
newElements[1].indent = "="; |
1235 |
|
*(eq++) = '\0'; |
1236 |
|
newElements[2].item = strdup(eq); |
1237 |
|
free(line->elements[1].item); |
1238 |
|
if (line->elements[1].indent) |
1239 |
|
newElements[2].indent = line->elements[1].indent; |
1240 |
|
for (int i = 2; i < line->numElements; i++) { |
1241 |
|
newElements[i+1].item = line->elements[i].item; |
1242 |
|
newElements[i+1].indent = |
1243 |
|
line->elements[i].indent; |
1244 |
} |
} |
1245 |
} |
free(line->elements); |
1246 |
|
line->elements = newElements; |
1247 |
|
line->numElements = numElements; |
1248 |
} |
} |
1249 |
} |
} |
|
} |
|
1250 |
|
|
1251 |
return 0; |
return 0; |
1252 |
} |
} |
1253 |
|
|
1254 |
static struct grubConfig * readConfig(const char * inName, |
static int isnumber(const char *s) |
1255 |
struct configFileInfo * cfi) { |
{ |
1256 |
int in; |
int i; |
1257 |
char * incoming = NULL, * head; |
for (i = 0; s[i] != '\0'; i++) |
1258 |
int rc; |
if (s[i] < '0' || s[i] > '9') |
1259 |
int sawEntry = 0; |
return 0; |
1260 |
int movedLine = 0; |
return i; |
1261 |
struct grubConfig * cfg; |
} |
|
struct singleLine * last = NULL, * line, * defaultLine = NULL; |
|
|
char * end; |
|
|
struct singleEntry * entry = NULL; |
|
|
int i, len; |
|
|
char * buf; |
|
|
|
|
|
if (!strcmp(inName, "-")) { |
|
|
in = 0; |
|
|
} else { |
|
|
if ((in = open(inName, O_RDONLY)) < 0) { |
|
|
fprintf(stderr, _("error opening %s for read: %s\n"), |
|
|
inName, strerror(errno)); |
|
|
return NULL; |
|
|
} |
|
|
} |
|
|
|
|
|
rc = readFile(in, &incoming); |
|
|
close(in); |
|
|
if (rc) return NULL; |
|
|
|
|
|
head = incoming; |
|
|
cfg = malloc(sizeof(*cfg)); |
|
|
cfg->primaryIndent = strdup(""); |
|
|
cfg->secondaryIndent = strdup("\t"); |
|
|
cfg->flags = GRUB_CONFIG_NO_DEFAULT; |
|
|
cfg->cfi = cfi; |
|
|
cfg->theLines = NULL; |
|
|
cfg->entries = NULL; |
|
|
cfg->fallbackImage = 0; |
|
|
|
|
|
/* copy everything we have */ |
|
|
while (*head) { |
|
|
line = malloc(sizeof(*line)); |
|
|
lineInit(line); |
|
1262 |
|
|
1263 |
if (getNextLine(&head, line, cfi)) { |
static struct grubConfig *readConfig(const char *inName, |
1264 |
free(line); |
struct configFileInfo *cfi) |
1265 |
/* XXX memory leak of everything in cfg */ |
{ |
1266 |
return NULL; |
int in; |
1267 |
} |
char *incoming = NULL, *head; |
1268 |
|
int rc; |
1269 |
if (!sawEntry && line->numElements) { |
int sawEntry = 0; |
1270 |
free(cfg->primaryIndent); |
int movedLine = 0; |
1271 |
cfg->primaryIndent = strdup(line->indent); |
struct grubConfig *cfg; |
1272 |
} else if (line->numElements) { |
struct singleLine *last = NULL, *line, *defaultLine = NULL; |
1273 |
free(cfg->secondaryIndent); |
char *end; |
1274 |
cfg->secondaryIndent = strdup(line->indent); |
struct singleEntry *entry = NULL; |
1275 |
} |
int len; |
1276 |
|
char *buf; |
1277 |
if (isEntryStart(line, cfi) || (cfg->entries && !sawEntry)) { |
|
1278 |
sawEntry = 1; |
if (inName == NULL) { |
1279 |
if (!entry) { |
printf("Could not find bootloader configuration\n"); |
1280 |
cfg->entries = malloc(sizeof(*entry)); |
exit(1); |
1281 |
entry = cfg->entries; |
} else if (!strcmp(inName, "-")) { |
1282 |
} else { |
in = 0; |
1283 |
entry->next = malloc(sizeof(*entry)); |
} else { |
1284 |
entry = entry->next; |
if ((in = open(inName, O_RDONLY)) < 0) { |
1285 |
} |
fprintf(stderr, _("error opening %s for read: %s\n"), |
1286 |
|
inName, strerror(errno)); |
1287 |
|
return NULL; |
1288 |
|
} |
1289 |
|
} |
1290 |
|
|
1291 |
entry->skip = 0; |
rc = readFile(in, &incoming); |
1292 |
entry->multiboot = 0; |
close(in); |
1293 |
entry->lines = NULL; |
if (rc) |
1294 |
entry->next = NULL; |
return NULL; |
1295 |
} |
|
1296 |
|
head = incoming; |
1297 |
if (line->type == LT_SET_VARIABLE) { |
cfg = malloc(sizeof(*cfg)); |
1298 |
int i; |
cfg->primaryIndent = strdup(""); |
1299 |
dbgPrintf("found 'set' command (%d elements): ", line->numElements); |
cfg->secondaryIndent = strdup("\t"); |
1300 |
dbgPrintf("%s", line->indent); |
cfg->flags = GRUB_CONFIG_NO_DEFAULT; |
1301 |
for (i = 0; i < line->numElements; i++) |
cfg->cfi = cfi; |
1302 |
dbgPrintf("\"%s\"%s", line->elements[i].item, line->elements[i].indent); |
cfg->theLines = NULL; |
1303 |
dbgPrintf("\n"); |
cfg->entries = NULL; |
1304 |
struct keywordTypes *kwType = getKeywordByType(LT_DEFAULT, cfi); |
cfg->fallbackImage = 0; |
1305 |
if (kwType && line->numElements == 3 && |
cfg->isModified = 0; |
1306 |
!strcmp(line->elements[1].item, kwType->key)) { |
|
1307 |
dbgPrintf("Line sets default config\n"); |
/* copy everything we have */ |
1308 |
cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT; |
while (*head) { |
1309 |
defaultLine = line; |
line = malloc(sizeof(*line)); |
1310 |
} |
lineInit(line); |
1311 |
} else if (line->type == LT_DEFAULT && line->numElements == 2) { |
|
1312 |
cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT; |
if (getNextLine(&head, line, cfi)) { |
1313 |
defaultLine = line; |
free(line); |
1314 |
|
/* XXX memory leak of everything in cfg */ |
1315 |
} else if (line->type == LT_KERNEL) { |
return NULL; |
1316 |
/* if by some freak chance this is multiboot and the "module" |
} |
|
* lines came earlier in the template, make sure to use LT_HYPER |
|
|
* instead of LT_KERNEL now |
|
|
*/ |
|
|
if (entry->multiboot) |
|
|
line->type = LT_HYPER; |
|
|
|
|
|
} else if (line->type == LT_MBMODULE) { |
|
|
/* go back and fix the LT_KERNEL line to indicate LT_HYPER |
|
|
* instead, now that we know this is a multiboot entry. |
|
|
* This only applies to grub, but that's the only place we |
|
|
* should find LT_MBMODULE lines anyway. |
|
|
*/ |
|
|
struct singleLine * l; |
|
|
for (l = entry->lines; l; l = l->next) { |
|
|
if (l->type == LT_HYPER) |
|
|
break; |
|
|
else if (l->type == LT_KERNEL) { |
|
|
l->type = LT_HYPER; |
|
|
break; |
|
|
} |
|
|
} |
|
|
entry->multiboot = 1; |
|
|
|
|
|
} else if (line->type == LT_HYPER) { |
|
|
entry->multiboot = 1; |
|
|
|
|
|
} else if (line->type == LT_FALLBACK && line->numElements == 2) { |
|
|
cfg->fallbackImage = strtol(line->elements[1].item, &end, 10); |
|
|
if (*end) cfg->fallbackImage = -1; |
|
|
|
|
|
} else if (line->type == LT_TITLE && line->numElements > 1) { |
|
|
/* make the title a single argument (undoing our parsing) */ |
|
|
len = 0; |
|
|
for (i = 1; i < line->numElements; i++) { |
|
|
len += strlen(line->elements[i].item); |
|
|
len += strlen(line->elements[i].indent); |
|
|
} |
|
|
buf = malloc(len + 1); |
|
|
*buf = '\0'; |
|
1317 |
|
|
1318 |
for (i = 1; i < line->numElements; i++) { |
if (!sawEntry && line->numElements) { |
1319 |
strcat(buf, line->elements[i].item); |
free(cfg->primaryIndent); |
1320 |
free(line->elements[i].item); |
cfg->primaryIndent = strdup(line->indent); |
1321 |
|
} else if (line->numElements) { |
1322 |
|
free(cfg->secondaryIndent); |
1323 |
|
cfg->secondaryIndent = strdup(line->indent); |
1324 |
|
} |
1325 |
|
|
1326 |
if ((i + 1) != line->numElements) { |
if (isEntryStart(line, cfi) || (cfg->entries && !sawEntry)) { |
1327 |
strcat(buf, line->elements[i].indent); |
sawEntry = 1; |
1328 |
free(line->elements[i].indent); |
if (!entry) { |
1329 |
} |
cfg->entries = malloc(sizeof(*entry)); |
1330 |
} |
entry = cfg->entries; |
1331 |
|
} else { |
1332 |
line->elements[1].indent = |
entry->next = malloc(sizeof(*entry)); |
1333 |
line->elements[line->numElements - 1].indent; |
entry = entry->next; |
1334 |
line->elements[1].item = buf; |
} |
|
line->numElements = 2; |
|
|
} else if (line->type == LT_MENUENTRY && line->numElements > 3) { |
|
|
/* let --remove-kernel="TITLE=what" work */ |
|
|
len = 0; |
|
|
char *extras; |
|
|
char *title; |
|
|
|
|
|
for (i = 1; i < line->numElements; i++) { |
|
|
len += strlen(line->elements[i].item); |
|
|
len += strlen(line->elements[i].indent); |
|
|
} |
|
|
buf = malloc(len + 1); |
|
|
*buf = '\0'; |
|
|
|
|
|
/* allocate mem for extra flags. */ |
|
|
extras = malloc(len + 1); |
|
|
*extras = '\0'; |
|
|
|
|
|
/* get title. */ |
|
|
for (i = 0; i < line->numElements; i++) { |
|
|
if (!strcmp(line->elements[i].item, "menuentry")) |
|
|
continue; |
|
|
if (isquote(*line->elements[i].item)) |
|
|
title = line->elements[i].item + 1; |
|
|
else |
|
|
title = line->elements[i].item; |
|
1335 |
|
|
1336 |
len = strlen(title); |
entry->skip = 0; |
1337 |
if (isquote(title[len-1])) { |
entry->multiboot = 0; |
1338 |
strncat(buf, title,len-1); |
entry->lines = NULL; |
1339 |
break; |
entry->next = NULL; |
|
} else { |
|
|
strcat(buf, title); |
|
|
strcat(buf, line->elements[i].indent); |
|
1340 |
} |
} |
|
} |
|
1341 |
|
|
1342 |
/* get extras */ |
if (line->type == LT_SET_VARIABLE) { |
1343 |
int count = 0; |
dbgPrintf("found 'set' command (%d elements): ", |
1344 |
for (i = 0; i < line->numElements; i++) { |
line->numElements); |
1345 |
if (count >= 2) { |
dbgPrintf("%s", line->indent); |
1346 |
strcat(extras, line->elements[i].item); |
for (int i = 0; i < line->numElements; i++) |
1347 |
strcat(extras, line->elements[i].indent); |
dbgPrintf("\"%s\"%s", line->elements[i].item, |
1348 |
} |
line->elements[i].indent); |
1349 |
|
dbgPrintf("\n"); |
1350 |
if (!strcmp(line->elements[i].item, "menuentry")) |
struct keywordTypes *kwType = |
1351 |
continue; |
getKeywordByType(LT_DEFAULT, cfi); |
1352 |
|
if (kwType && line->numElements == 3 |
1353 |
/* count ' or ", there should be two in menuentry line. */ |
&& !strcmp(line->elements[1].item, kwType->key) |
1354 |
if (isquote(*line->elements[i].item)) |
&& !is_special_grub2_variable( |
1355 |
count++; |
line->elements[2].item)) { |
1356 |
|
dbgPrintf("Line sets default config\n"); |
1357 |
len = strlen(line->elements[i].item); |
cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT; |
1358 |
|
defaultLine = line; |
1359 |
if (isquote(line->elements[i].item[len -1])) |
} |
1360 |
count++; |
} else if (iskernel(line->type)) { |
1361 |
|
/* if by some freak chance this is multiboot and the |
1362 |
/* ok, we get the final ' or ", others are extras. */ |
* "module" lines came earlier in the template, make |
1363 |
} |
* sure to use LT_HYPER instead of LT_KERNEL now |
1364 |
line->elements[1].indent = |
*/ |
1365 |
line->elements[line->numElements - 2].indent; |
if (entry && entry->multiboot) |
1366 |
line->elements[1].item = buf; |
line->type = LT_HYPER; |
|
line->elements[2].indent = |
|
|
line->elements[line->numElements - 2].indent; |
|
|
line->elements[2].item = extras; |
|
|
line->numElements = 3; |
|
|
} else if (line->type == LT_KERNELARGS && cfi->argsInQuotes) { |
|
|
/* Strip off any " which may be present; they'll be put back |
|
|
on write. This is one of the few (the only?) places that grubby |
|
|
canonicalizes the output */ |
|
|
|
|
|
if (line->numElements >= 2) { |
|
|
int last, len; |
|
|
|
|
|
if (isquote(*line->elements[1].item)) |
|
|
memmove(line->elements[1].item, line->elements[1].item + 1, |
|
|
strlen(line->elements[1].item + 1) + 1); |
|
|
|
|
|
last = line->numElements - 1; |
|
|
len = strlen(line->elements[last].item) - 1; |
|
|
if (isquote(line->elements[last].item[len])) |
|
|
line->elements[last].item[len] = '\0'; |
|
|
} |
|
|
} |
|
|
|
|
|
/* If we find a generic config option which should live at the |
|
|
top of the file, move it there. Old versions of grubby were |
|
|
probably responsible for putting new images in the wrong |
|
|
place in front of it anyway. */ |
|
|
if (sawEntry && line->type == LT_GENERIC) { |
|
|
struct singleLine **l = &cfg->theLines; |
|
|
struct singleLine **last_nonws = &cfg->theLines; |
|
|
while (*l) { |
|
|
if ((*l)->type != LT_WHITESPACE) |
|
|
last_nonws = &((*l)->next); |
|
|
l = &((*l)->next); |
|
|
} |
|
|
line->next = *last_nonws; |
|
|
*last_nonws = line; |
|
|
movedLine = 1; |
|
|
continue; /* without setting 'last' */ |
|
|
} |
|
|
|
|
|
/* If a second line of whitespace happens after a generic option |
|
|
which was moved, drop it. */ |
|
|
if (movedLine && line->type == LT_WHITESPACE && last->type == LT_WHITESPACE) { |
|
|
lineFree(line); |
|
|
free(line); |
|
|
movedLine = 0; |
|
|
continue; |
|
|
} |
|
|
movedLine = 0; |
|
1367 |
|
|
1368 |
if (sawEntry) { |
} else if (line->type == LT_MBMODULE) { |
1369 |
if (!entry->lines) |
/* go back and fix the LT_KERNEL line to indicate |
1370 |
entry->lines = line; |
* LT_HYPER instead, now that we know this is a |
1371 |
else |
* multiboot entry. This only applies to grub, but |
1372 |
last->next = line; |
* that's the only place we should find LT_MBMODULE |
1373 |
dbgPrintf("readConfig added %s to %p\n", getKeyByType(line->type, cfi), entry); |
* lines anyway. |
1374 |
|
*/ |
1375 |
/* we could have seen this outside of an entry... if so, we |
for (struct singleLine * l = entry->lines; l; |
1376 |
* ignore it like any other line we don't grok */ |
l = l->next) { |
1377 |
if (line->type == LT_ENTRY_END && sawEntry) |
if (l->type == LT_HYPER) |
1378 |
sawEntry = 0; |
break; |
1379 |
} else { |
else if (iskernel(l->type)) { |
1380 |
if (!cfg->theLines) |
l->type = LT_HYPER; |
1381 |
cfg->theLines = line; |
break; |
1382 |
else |
} |
1383 |
last->next = line; |
} |
1384 |
dbgPrintf("readConfig added %s to cfg\n", getKeyByType(line->type, cfi)); |
entry->multiboot = 1; |
|
} |
|
1385 |
|
|
1386 |
last = line; |
} else if (line->type == LT_HYPER) { |
1387 |
} |
entry->multiboot = 1; |
1388 |
|
|
1389 |
free(incoming); |
} else if (line->type == LT_FALLBACK && line->numElements == 2) { |
1390 |
|
cfg->fallbackImage = |
1391 |
|
strtol(line->elements[1].item, &end, 10); |
1392 |
|
if (*end) |
1393 |
|
cfg->fallbackImage = -1; |
1394 |
|
|
1395 |
|
} else if ((line->type == LT_DEFAULT && cfi->defaultIsUnquoted) |
1396 |
|
|| (line->type == LT_TITLE |
1397 |
|
&& line->numElements > 1)) { |
1398 |
|
/* make the title/default a single argument (undoing |
1399 |
|
* our parsing) */ |
1400 |
|
len = 0; |
1401 |
|
for (int i = 1; i < line->numElements; i++) { |
1402 |
|
len += strlen(line->elements[i].item); |
1403 |
|
len += strlen(line->elements[i].indent); |
1404 |
|
} |
1405 |
|
buf = malloc(len + 1); |
1406 |
|
*buf = '\0'; |
1407 |
|
|
1408 |
dbgPrintf("defaultLine is %s\n", defaultLine ? "set" : "unset"); |
for (int i = 1; i < line->numElements; i++) { |
1409 |
if (defaultLine) { |
strcat(buf, line->elements[i].item); |
1410 |
if (defaultLine->numElements > 2 && |
free(line->elements[i].item); |
1411 |
cfi->defaultSupportSaved && |
|
1412 |
!strncmp(defaultLine->elements[2].item,"\"${saved_entry}\"", 16)) { |
if ((i + 1) != line->numElements) { |
1413 |
cfg->defaultImage = DEFAULT_SAVED_GRUB2; |
strcat(buf, line->elements[i].indent); |
1414 |
} else if (cfi->defaultIsVariable) { |
free(line->elements[i].indent); |
1415 |
char *value = defaultLine->elements[2].item; |
} |
1416 |
while (*value && (*value == '"' || *value == '\'' || |
} |
|
*value == ' ' || *value == '\t')) |
|
|
value++; |
|
|
cfg->defaultImage = strtol(value, &end, 10); |
|
|
while (*end && (*end == '"' || *end == '\'' || |
|
|
*end == ' ' || *end == '\t')) |
|
|
end++; |
|
|
if (*end) cfg->defaultImage = -1; |
|
|
} else if (cfi->defaultSupportSaved && |
|
|
!strncmp(defaultLine->elements[1].item, "saved", 5)) { |
|
|
cfg->defaultImage = DEFAULT_SAVED; |
|
|
} else if (cfi->defaultIsIndex) { |
|
|
cfg->defaultImage = strtol(defaultLine->elements[1].item, &end, 10); |
|
|
if (*end) cfg->defaultImage = -1; |
|
|
} else if (defaultLine->numElements >= 2) { |
|
|
i = 0; |
|
|
while ((entry = findEntryByIndex(cfg, i))) { |
|
|
for (line = entry->lines; line; line = line->next) |
|
|
if (line->type == LT_TITLE) break; |
|
|
|
|
|
if (!cfi->titleBracketed) { |
|
|
if (line && (line->numElements >= 2) && |
|
|
!strcmp(defaultLine->elements[1].item, |
|
|
line->elements[1].item)) break; |
|
|
} else if (line) { |
|
|
if (!strcmp(defaultLine->elements[1].item, |
|
|
extractTitle(line))) break; |
|
|
} |
|
|
i++; |
|
|
entry = NULL; |
|
|
} |
|
1417 |
|
|
1418 |
if (entry){ |
line->elements[1].indent = |
1419 |
cfg->defaultImage = i; |
line->elements[line->numElements - 1].indent; |
1420 |
}else{ |
line->elements[1].item = buf; |
1421 |
cfg->defaultImage = -1; |
line->numElements = 2; |
1422 |
} |
} else if (line->type == LT_MENUENTRY && line->numElements > 3) { |
1423 |
} |
/* let --remove-kernel="TITLE=what" work */ |
1424 |
} else { |
len = 0; |
1425 |
cfg->defaultImage = 0; |
char *extras; |
1426 |
} |
char *title; |
1427 |
|
/* initially unseen value */ |
1428 |
|
char quote_char = '\0'; |
1429 |
|
|
1430 |
|
for (int i = 1; i < line->numElements; i++) { |
1431 |
|
len += strlen(line->elements[i].item); |
1432 |
|
len += strlen(line->elements[i].indent); |
1433 |
|
} |
1434 |
|
buf = malloc(len + 1); |
1435 |
|
*buf = '\0'; |
1436 |
|
|
1437 |
return cfg; |
/* allocate mem for extra flags. */ |
1438 |
} |
extras = malloc(len + 1); |
1439 |
|
*extras = '\0'; |
1440 |
|
|
1441 |
|
/* get title. */ |
1442 |
|
for (int i = 0; i < line->numElements; i++) { |
1443 |
|
if (!strcmp |
1444 |
|
(line->elements[i].item, "menuentry")) |
1445 |
|
continue; |
1446 |
|
if (isquote(*line->elements[i].item) |
1447 |
|
&& quote_char == '\0') { |
1448 |
|
/* ensure we properly pair off quotes */ |
1449 |
|
quote_char = *line->elements[i].item; |
1450 |
|
title = line->elements[i].item + 1; |
1451 |
|
} else { |
1452 |
|
title = line->elements[i].item; |
1453 |
|
} |
1454 |
|
|
1455 |
static void writeDefault(FILE * out, char * indent, |
len = strlen(title); |
1456 |
char * separator, struct grubConfig * cfg) { |
if (title[len - 1] == quote_char) { |
1457 |
struct singleEntry * entry; |
strncat(buf, title, len - 1); |
1458 |
struct singleLine * line; |
break; |
1459 |
int i; |
} else { |
1460 |
|
strcat(buf, title); |
1461 |
if (!cfg->defaultImage && cfg->flags == GRUB_CONFIG_NO_DEFAULT) return; |
strcat(buf, line->elements[i].indent); |
1462 |
|
} |
1463 |
if (cfg->defaultImage == DEFAULT_SAVED) |
} |
|
fprintf(out, "%sdefault%ssaved\n", indent, separator); |
|
|
else if (cfg->defaultImage == DEFAULT_SAVED_GRUB2) |
|
|
fprintf(out, "%sset default=\"${saved_entry}\"\n", indent); |
|
|
else if (cfg->defaultImage > -1) { |
|
|
if (cfg->cfi->defaultIsIndex) { |
|
|
if (cfg->cfi->defaultIsVariable) { |
|
|
fprintf(out, "%sset default=\"%d\"\n", indent, |
|
|
cfg->defaultImage); |
|
|
} else { |
|
|
fprintf(out, "%sdefault%s%d\n", indent, separator, |
|
|
cfg->defaultImage); |
|
|
} |
|
|
} else { |
|
|
int image = cfg->defaultImage; |
|
1464 |
|
|
1465 |
entry = cfg->entries; |
/* get extras */ |
1466 |
while (entry && entry->skip) entry = entry->next; |
int count = 0; |
1467 |
|
quote_char = '\0'; |
1468 |
|
for (int i = 0; i < line->numElements; i++) { |
1469 |
|
if (count >= 2) { |
1470 |
|
strcat(extras, line->elements[i].item); |
1471 |
|
strcat(extras, |
1472 |
|
line->elements[i].indent); |
1473 |
|
} |
1474 |
|
|
1475 |
i = 0; |
if (!strcmp |
1476 |
while (entry && i < image) { |
(line->elements[i].item, "menuentry")) |
1477 |
entry = entry->next; |
continue; |
1478 |
|
|
1479 |
|
/* count ' or ", there should be two in menuentry line. */ |
1480 |
|
if (isquote(*line->elements[i].item) |
1481 |
|
&& quote_char == '\0') { |
1482 |
|
/* ensure we properly pair off quotes */ |
1483 |
|
quote_char = *line->elements[i].item; |
1484 |
|
count++; |
1485 |
|
} |
1486 |
|
|
1487 |
while (entry && entry->skip) entry = entry->next; |
len = strlen(line->elements[i].item); |
|
i++; |
|
|
} |
|
1488 |
|
|
1489 |
if (!entry) return; |
if (line->elements[i].item[len - 1] == |
1490 |
|
quote_char) |
1491 |
|
count++; |
1492 |
|
|
1493 |
line = getLineByType(LT_TITLE, entry->lines); |
/* ok, we get the final ' or ", others are extras. */ |
1494 |
|
} |
1495 |
|
line->elements[1].indent = |
1496 |
|
line->elements[line->numElements - 2].indent; |
1497 |
|
line->elements[1].item = buf; |
1498 |
|
line->elements[2].indent = |
1499 |
|
line->elements[line->numElements - 2].indent; |
1500 |
|
line->elements[2].item = extras; |
1501 |
|
line->numElements = 3; |
1502 |
|
} else if (line->type == LT_KERNELARGS && cfi->argsInQuotes) { |
1503 |
|
/* Strip off any " which may be present; they'll be |
1504 |
|
* put back on write. This is one of the few (the |
1505 |
|
* only?) places that grubby canonicalizes the output |
1506 |
|
*/ |
1507 |
|
if (line->numElements >= 2) { |
1508 |
|
int last, len; |
1509 |
|
|
1510 |
if (line && line->numElements >= 2) |
if (isquote(*line->elements[1].item)) |
1511 |
fprintf(out, "%sdefault%s%s\n", indent, separator, |
memmove(line->elements[1].item, |
1512 |
line->elements[1].item); |
line->elements[1].item + 1, |
1513 |
else if (line && (line->numElements == 1) && |
strlen(line->elements[1].item + |
1514 |
cfg->cfi->titleBracketed) { |
1) + 1); |
1515 |
fprintf(out, "%sdefault%s%s\n", indent, separator, |
|
1516 |
extractTitle(line)); |
last = line->numElements - 1; |
1517 |
} |
len = strlen(line->elements[last].item) - 1; |
1518 |
} |
if (isquote(line->elements[last].item[len])) |
1519 |
} |
line->elements[last].item[len] = '\0'; |
1520 |
} |
} |
1521 |
|
} |
1522 |
|
|
1523 |
static int writeConfig(struct grubConfig * cfg, char * outName, |
if (line->type == LT_DEFAULT && line->numElements == 2) { |
1524 |
const char * prefix) { |
cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT; |
1525 |
FILE * out; |
defaultLine = line; |
1526 |
struct singleLine * line; |
} |
|
struct singleEntry * entry; |
|
|
char * tmpOutName; |
|
|
int needs = MAIN_DEFAULT; |
|
|
struct stat sb; |
|
|
int i; |
|
|
|
|
|
if (!strcmp(outName, "-")) { |
|
|
out = stdout; |
|
|
tmpOutName = NULL; |
|
|
} else { |
|
|
if (!lstat(outName, &sb) && S_ISLNK(sb.st_mode)) { |
|
|
char * buf; |
|
|
int len = 256; |
|
|
int rc; |
|
|
|
|
|
/* most likely the symlink is relative, so change our |
|
|
directory to the dir of the symlink */ |
|
|
rc = chdir(dirname(strdupa(outName))); |
|
|
do { |
|
|
buf = alloca(len + 1); |
|
|
rc = readlink(basename(outName), buf, len); |
|
|
if (rc == len) len += 256; |
|
|
} while (rc == len); |
|
|
|
|
|
if (rc < 0) { |
|
|
fprintf(stderr, _("grubby: error readlink link %s: %s\n"), |
|
|
outName, strerror(errno)); |
|
|
return 1; |
|
|
} |
|
|
|
|
|
outName = buf; |
|
|
outName[rc] = '\0'; |
|
|
} |
|
|
|
|
|
tmpOutName = alloca(strlen(outName) + 2); |
|
|
sprintf(tmpOutName, "%s-", outName); |
|
|
out = fopen(tmpOutName, "w"); |
|
|
if (!out) { |
|
|
fprintf(stderr, _("grubby: error creating %s: %s\n"), tmpOutName, |
|
|
strerror(errno)); |
|
|
return 1; |
|
|
} |
|
|
|
|
|
if (!stat(outName, &sb)) { |
|
|
if (chmod(tmpOutName, sb.st_mode & ~(S_IFMT))) { |
|
|
fprintf(stderr, _("grubby: error setting perms on %s: %s\n"), |
|
|
tmpOutName, strerror(errno)); |
|
|
fclose(out); |
|
|
unlink(tmpOutName); |
|
|
return 1; |
|
|
} |
|
|
} |
|
|
} |
|
1527 |
|
|
1528 |
line = cfg->theLines; |
/* If we find a generic config option which should live at the |
1529 |
struct keywordTypes *defaultKw = getKeywordByType(LT_DEFAULT, cfg->cfi); |
top of the file, move it there. Old versions of grubby were |
1530 |
while (line) { |
probably responsible for putting new images in the wrong |
1531 |
if (line->type == LT_SET_VARIABLE && defaultKw && |
place in front of it anyway. */ |
1532 |
line->numElements == 3 && |
if (sawEntry && line->type == LT_GENERIC) { |
1533 |
!strcmp(line->elements[1].item, defaultKw->key)) { |
struct singleLine **l = &cfg->theLines; |
1534 |
writeDefault(out, line->indent, line->elements[0].indent, cfg); |
struct singleLine **last_nonws = &cfg->theLines; |
1535 |
needs &= ~MAIN_DEFAULT; |
while (*l) { |
1536 |
} else if (line->type == LT_DEFAULT) { |
if ((*l)->type != LT_WHITESPACE) |
1537 |
writeDefault(out, line->indent, line->elements[0].indent, cfg); |
last_nonws = &((*l)->next); |
1538 |
needs &= ~MAIN_DEFAULT; |
l = &((*l)->next); |
1539 |
} else if (line->type == LT_FALLBACK) { |
} |
1540 |
if (cfg->fallbackImage > -1) |
line->next = *last_nonws; |
1541 |
fprintf(out, "%s%s%s%d\n", line->indent, |
*last_nonws = line; |
1542 |
line->elements[0].item, line->elements[0].indent, |
movedLine = 1; |
1543 |
cfg->fallbackImage); |
continue; /* without setting 'last' */ |
1544 |
} else { |
} |
|
if (lineWrite(out, line, cfg->cfi) == -1) { |
|
|
fprintf(stderr, _("grubby: error writing %s: %s\n"), |
|
|
tmpOutName, strerror(errno)); |
|
|
fclose(out); |
|
|
unlink(tmpOutName); |
|
|
return 1; |
|
|
} |
|
|
} |
|
1545 |
|
|
1546 |
line = line->next; |
/* If a second line of whitespace happens after a generic |
1547 |
} |
* option which was moved, drop it. */ |
1548 |
|
if (movedLine && line->type == LT_WHITESPACE |
1549 |
|
&& last->type == LT_WHITESPACE) { |
1550 |
|
lineFree(line); |
1551 |
|
free(line); |
1552 |
|
movedLine = 0; |
1553 |
|
continue; |
1554 |
|
} |
1555 |
|
movedLine = 0; |
1556 |
|
|
1557 |
if (needs & MAIN_DEFAULT) { |
if (sawEntry) { |
1558 |
writeDefault(out, cfg->primaryIndent, "=", cfg); |
if (!entry->lines) |
1559 |
needs &= ~MAIN_DEFAULT; |
entry->lines = line; |
1560 |
} |
else |
1561 |
|
last->next = line; |
1562 |
|
dbgPrintf("readConfig added %s to %p\n", |
1563 |
|
getKeyByType(line->type, cfi), entry); |
1564 |
|
|
1565 |
i = 0; |
/* we could have seen this outside of an entry... if |
1566 |
while ((entry = findEntryByIndex(cfg, i++))) { |
* so, we ignore it like any other line we don't grok |
1567 |
if (entry->skip) continue; |
*/ |
1568 |
|
if (line->type == LT_ENTRY_END && sawEntry) |
1569 |
|
sawEntry = 0; |
1570 |
|
} else { |
1571 |
|
if (!cfg->theLines) |
1572 |
|
cfg->theLines = line; |
1573 |
|
else |
1574 |
|
last->next = line; |
1575 |
|
dbgPrintf("readConfig added %s to cfg\n", |
1576 |
|
getKeyByType(line->type, cfi)); |
1577 |
|
} |
1578 |
|
|
1579 |
line = entry->lines; |
last = line; |
|
while (line) { |
|
|
if (lineWrite(out, line, cfg->cfi) == -1) { |
|
|
fprintf(stderr, _("grubby: error writing %s: %s\n"), |
|
|
tmpOutName, strerror(errno)); |
|
|
fclose(out); |
|
|
unlink(tmpOutName); |
|
|
return 1; |
|
|
} |
|
|
line = line->next; |
|
1580 |
} |
} |
|
} |
|
1581 |
|
|
1582 |
if (tmpOutName) { |
free(incoming); |
|
if (rename(tmpOutName, outName)) { |
|
|
fprintf(stderr, _("grubby: error moving %s to %s: %s\n"), |
|
|
tmpOutName, outName, strerror(errno)); |
|
|
unlink(outName); |
|
|
return 1; |
|
|
} |
|
|
} |
|
1583 |
|
|
1584 |
return 0; |
dbgPrintf("defaultLine is %s\n", defaultLine ? "set" : "unset"); |
1585 |
|
if (defaultLine) { |
1586 |
|
if (defaultLine->numElements > 2 && |
1587 |
|
cfi->defaultSupportSaved && |
1588 |
|
!strncmp(defaultLine->elements[2].item, |
1589 |
|
"\"${saved_entry}\"", 16)) { |
1590 |
|
cfg->cfi->defaultIsSaved = 1; |
1591 |
|
cfg->defaultImage = DEFAULT_SAVED_GRUB2; |
1592 |
|
if (cfg->cfi->getEnv) { |
1593 |
|
char *defTitle = |
1594 |
|
cfi->getEnv(cfg->cfi, "saved_entry"); |
1595 |
|
if (defTitle) { |
1596 |
|
int index = 0; |
1597 |
|
if (isnumber(defTitle)) { |
1598 |
|
index = atoi(defTitle); |
1599 |
|
entry = |
1600 |
|
findEntryByIndex(cfg, |
1601 |
|
index); |
1602 |
|
} else { |
1603 |
|
entry = |
1604 |
|
findEntryByTitle(cfg, |
1605 |
|
defTitle, |
1606 |
|
&index); |
1607 |
|
} |
1608 |
|
if (entry) |
1609 |
|
cfg->defaultImage = index; |
1610 |
|
} |
1611 |
|
} |
1612 |
|
} else if (cfi->defaultIsVariable) { |
1613 |
|
if (defaultLine->numElements == 2) { |
1614 |
|
char *value = defaultLine->elements[1].item + 8; |
1615 |
|
while (*value && (*value == '"' || |
1616 |
|
*value == '\'' || |
1617 |
|
*value == ' ' || |
1618 |
|
*value == '\t')) |
1619 |
|
value++; |
1620 |
|
cfg->defaultImage = strtol(value, &end, 10); |
1621 |
|
while (*end && (*end == '"' || *end == '\'' || |
1622 |
|
*end == ' ' || *end == '\t')) |
1623 |
|
end++; |
1624 |
|
if (*end) |
1625 |
|
cfg->defaultImage = NO_DEFAULT_ENTRY; |
1626 |
|
} else if (defaultLine->numElements == 3) { |
1627 |
|
char *value = defaultLine->elements[2].item; |
1628 |
|
while (*value && (*value == '"' || |
1629 |
|
*value == '\'' || |
1630 |
|
*value == ' ' || |
1631 |
|
*value == '\t')) |
1632 |
|
value++; |
1633 |
|
cfg->defaultImage = strtol(value, &end, 10); |
1634 |
|
while (*end && (*end == '"' || *end == '\'' || |
1635 |
|
*end == ' ' || *end == '\t')) |
1636 |
|
end++; |
1637 |
|
if (*end) |
1638 |
|
cfg->defaultImage = NO_DEFAULT_ENTRY; |
1639 |
|
} |
1640 |
|
} else if (cfi->defaultSupportSaved && |
1641 |
|
!strncmp(defaultLine->elements[1].item, "saved", |
1642 |
|
5)) { |
1643 |
|
cfg->defaultImage = DEFAULT_SAVED; |
1644 |
|
} else if (cfi->defaultIsIndex) { |
1645 |
|
cfg->defaultImage = |
1646 |
|
strtol(defaultLine->elements[1].item, &end, 10); |
1647 |
|
if (*end) |
1648 |
|
cfg->defaultImage = NO_DEFAULT_ENTRY; |
1649 |
|
} else if (defaultLine->numElements >= 2) { |
1650 |
|
int i = 0; |
1651 |
|
while ((entry = findEntryByIndex(cfg, i))) { |
1652 |
|
for (line = entry->lines; line; |
1653 |
|
line = line->next) |
1654 |
|
if (line->type == LT_TITLE) |
1655 |
|
break; |
1656 |
|
|
1657 |
|
if (!cfi->titleBracketed) { |
1658 |
|
if (line && (line->numElements >= 2) && |
1659 |
|
!strcmp(defaultLine->elements[1]. |
1660 |
|
item, |
1661 |
|
line->elements[1].item)) |
1662 |
|
break; |
1663 |
|
} else if (line) { |
1664 |
|
if (!strcmp |
1665 |
|
(defaultLine->elements[1].item, |
1666 |
|
extractTitle(cfg, line))) |
1667 |
|
break; |
1668 |
|
} |
1669 |
|
i++; |
1670 |
|
entry = NULL; |
1671 |
|
} |
1672 |
|
|
1673 |
|
if (entry) { |
1674 |
|
cfg->defaultImage = i; |
1675 |
|
} else { |
1676 |
|
cfg->defaultImage = NO_DEFAULT_ENTRY; |
1677 |
|
} |
1678 |
|
} |
1679 |
|
} else if (cfg->cfi->defaultIsSaved && cfg->cfi->getEnv) { |
1680 |
|
char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry"); |
1681 |
|
if (defTitle) { |
1682 |
|
int index = 0; |
1683 |
|
if (isnumber(defTitle)) { |
1684 |
|
index = atoi(defTitle); |
1685 |
|
entry = findEntryByIndex(cfg, index); |
1686 |
|
} else { |
1687 |
|
entry = findEntryByTitle(cfg, defTitle, &index); |
1688 |
|
} |
1689 |
|
if (entry) |
1690 |
|
cfg->defaultImage = index; |
1691 |
|
} |
1692 |
|
} else { |
1693 |
|
cfg->defaultImage = FIRST_ENTRY_INDEX; |
1694 |
|
} |
1695 |
|
|
1696 |
|
return cfg; |
1697 |
} |
} |
1698 |
|
|
1699 |
static int numEntries(struct grubConfig *cfg) { |
static void writeDefault(FILE * out, char *indent, |
1700 |
int i = 0; |
char *separator, struct grubConfig *cfg) |
1701 |
struct singleEntry * entry; |
{ |
1702 |
|
struct singleEntry *entry; |
1703 |
entry = cfg->entries; |
struct singleLine *line; |
1704 |
while (entry) { |
int i; |
1705 |
if (!entry->skip) |
|
1706 |
i++; |
if (!cfg->defaultImage && cfg->flags == GRUB_CONFIG_NO_DEFAULT) |
1707 |
entry = entry->next; |
return; |
1708 |
} |
|
1709 |
return i; |
if (cfg->defaultImage == DEFAULT_SAVED) |
1710 |
|
fprintf(out, "%sdefault%ssaved\n", indent, separator); |
1711 |
|
else if (cfg->cfi->defaultIsSaved) { |
1712 |
|
fprintf(out, "%sset default=\"${saved_entry}\"\n", indent); |
1713 |
|
if (cfg->defaultImage >= FIRST_ENTRY_INDEX && cfg->cfi->setEnv) { |
1714 |
|
char *title; |
1715 |
|
int trueIndex, currentIndex; |
1716 |
|
|
1717 |
|
trueIndex = 0; |
1718 |
|
currentIndex = 0; |
1719 |
|
|
1720 |
|
while ((entry = findEntryByIndex(cfg, currentIndex))) { |
1721 |
|
if (!entry->skip) { |
1722 |
|
if (trueIndex == cfg->defaultImage) { |
1723 |
|
break; |
1724 |
|
} |
1725 |
|
trueIndex++; |
1726 |
|
} |
1727 |
|
currentIndex++; |
1728 |
|
} |
1729 |
|
line = getLineByType(LT_MENUENTRY, entry->lines); |
1730 |
|
if (!line) |
1731 |
|
line = getLineByType(LT_TITLE, entry->lines); |
1732 |
|
if (line) { |
1733 |
|
title = extractTitle(cfg, line); |
1734 |
|
if (title) |
1735 |
|
cfg->cfi->setEnv(cfg->cfi, |
1736 |
|
"saved_entry", title); |
1737 |
|
} |
1738 |
|
} |
1739 |
|
} else if (cfg->defaultImage >= FIRST_ENTRY_INDEX) { |
1740 |
|
if (cfg->cfi->defaultIsIndex) { |
1741 |
|
if (cfg->cfi->defaultIsVariable) { |
1742 |
|
fprintf(out, "%sset default=\"%d\"\n", indent, |
1743 |
|
cfg->defaultImage); |
1744 |
|
} else { |
1745 |
|
fprintf(out, "%sdefault%s%d\n", indent, |
1746 |
|
separator, cfg->defaultImage); |
1747 |
|
} |
1748 |
|
} else { |
1749 |
|
int image = cfg->defaultImage; |
1750 |
|
|
1751 |
|
entry = cfg->entries; |
1752 |
|
while (entry && entry->skip) |
1753 |
|
entry = entry->next; |
1754 |
|
|
1755 |
|
i = 0; |
1756 |
|
while (entry && i < image) { |
1757 |
|
entry = entry->next; |
1758 |
|
|
1759 |
|
while (entry && entry->skip) |
1760 |
|
entry = entry->next; |
1761 |
|
i++; |
1762 |
|
} |
1763 |
|
|
1764 |
|
if (!entry) |
1765 |
|
return; |
1766 |
|
|
1767 |
|
line = getLineByType(LT_TITLE, entry->lines); |
1768 |
|
|
1769 |
|
if (line && line->numElements >= 2) |
1770 |
|
fprintf(out, "%sdefault%s%s\n", indent, |
1771 |
|
separator, line->elements[1].item); |
1772 |
|
else if (line && (line->numElements == 1) |
1773 |
|
&& cfg->cfi->titleBracketed) { |
1774 |
|
char *title = extractTitle(cfg, line); |
1775 |
|
if (title) { |
1776 |
|
fprintf(out, "%sdefault%s%s\n", indent, |
1777 |
|
separator, title); |
1778 |
|
free(title); |
1779 |
|
} |
1780 |
|
} |
1781 |
|
} |
1782 |
|
} |
1783 |
|
} |
1784 |
|
|
1785 |
|
static int writeConfig(struct grubConfig *cfg, char *outName, |
1786 |
|
const char *prefix) |
1787 |
|
{ |
1788 |
|
FILE *out; |
1789 |
|
struct singleLine *line; |
1790 |
|
struct singleEntry *entry; |
1791 |
|
char *tmpOutName; |
1792 |
|
int needs = MAIN_DEFAULT; |
1793 |
|
struct stat sb; |
1794 |
|
int i; |
1795 |
|
int rc = 0; |
1796 |
|
|
1797 |
|
if (!strcmp(outName, "-")) { |
1798 |
|
out = stdout; |
1799 |
|
tmpOutName = NULL; |
1800 |
|
} else { |
1801 |
|
if (!lstat(outName, &sb) && S_ISLNK(sb.st_mode)) { |
1802 |
|
char *buf; |
1803 |
|
int len = 256; |
1804 |
|
int rc; |
1805 |
|
|
1806 |
|
/* most likely the symlink is relative, so change our |
1807 |
|
directory to the dir of the symlink */ |
1808 |
|
char *dir = strdupa(outName); |
1809 |
|
rc = chdir(dirname(dir)); |
1810 |
|
do { |
1811 |
|
buf = alloca(len + 1); |
1812 |
|
rc = readlink(basename(outName), buf, len); |
1813 |
|
if (rc == len) |
1814 |
|
len += 256; |
1815 |
|
} while (rc == len); |
1816 |
|
|
1817 |
|
if (rc < 0) { |
1818 |
|
fprintf(stderr, |
1819 |
|
_ |
1820 |
|
("grubby: error readlink link %s: %s\n"), |
1821 |
|
outName, strerror(errno)); |
1822 |
|
return 1; |
1823 |
|
} |
1824 |
|
|
1825 |
|
outName = buf; |
1826 |
|
outName[rc] = '\0'; |
1827 |
|
} |
1828 |
|
|
1829 |
|
tmpOutName = alloca(strlen(outName) + 2); |
1830 |
|
sprintf(tmpOutName, "%s-", outName); |
1831 |
|
out = fopen(tmpOutName, "w"); |
1832 |
|
if (!out) { |
1833 |
|
fprintf(stderr, _("grubby: error creating %s: %s\n"), |
1834 |
|
tmpOutName, strerror(errno)); |
1835 |
|
return 1; |
1836 |
|
} |
1837 |
|
|
1838 |
|
if (!stat(outName, &sb)) { |
1839 |
|
if (chmod(tmpOutName, sb.st_mode & ~(S_IFMT))) { |
1840 |
|
fprintf(stderr, |
1841 |
|
_ |
1842 |
|
("grubby: error setting perms on %s: %s\n"), |
1843 |
|
tmpOutName, strerror(errno)); |
1844 |
|
fclose(out); |
1845 |
|
unlink(tmpOutName); |
1846 |
|
return 1; |
1847 |
|
} |
1848 |
|
} |
1849 |
|
} |
1850 |
|
|
1851 |
|
line = cfg->theLines; |
1852 |
|
struct keywordTypes *defaultKw = getKeywordByType(LT_DEFAULT, cfg->cfi); |
1853 |
|
while (line) { |
1854 |
|
if (line->type == LT_SET_VARIABLE && defaultKw && |
1855 |
|
line->numElements == 3 && |
1856 |
|
!strcmp(line->elements[1].item, defaultKw->key) && |
1857 |
|
!is_special_grub2_variable(line->elements[2].item)) { |
1858 |
|
writeDefault(out, line->indent, |
1859 |
|
line->elements[0].indent, cfg); |
1860 |
|
needs &= ~MAIN_DEFAULT; |
1861 |
|
} else if (line->type == LT_DEFAULT) { |
1862 |
|
writeDefault(out, line->indent, |
1863 |
|
line->elements[0].indent, cfg); |
1864 |
|
needs &= ~MAIN_DEFAULT; |
1865 |
|
} else if (line->type == LT_FALLBACK) { |
1866 |
|
if (cfg->fallbackImage > -1) |
1867 |
|
fprintf(out, "%s%s%s%d\n", line->indent, |
1868 |
|
line->elements[0].item, |
1869 |
|
line->elements[0].indent, |
1870 |
|
cfg->fallbackImage); |
1871 |
|
} else { |
1872 |
|
if (lineWrite(out, line, cfg->cfi) == -1) { |
1873 |
|
fprintf(stderr, |
1874 |
|
_("grubby: error writing %s: %s\n"), |
1875 |
|
tmpOutName, strerror(errno)); |
1876 |
|
fclose(out); |
1877 |
|
unlink(tmpOutName); |
1878 |
|
return 1; |
1879 |
|
} |
1880 |
|
} |
1881 |
|
|
1882 |
|
line = line->next; |
1883 |
|
} |
1884 |
|
|
1885 |
|
if (needs & MAIN_DEFAULT) { |
1886 |
|
writeDefault(out, cfg->primaryIndent, "=", cfg); |
1887 |
|
needs &= ~MAIN_DEFAULT; |
1888 |
|
} |
1889 |
|
|
1890 |
|
i = 0; |
1891 |
|
while ((entry = findEntryByIndex(cfg, i++))) { |
1892 |
|
if (entry->skip) |
1893 |
|
continue; |
1894 |
|
|
1895 |
|
line = entry->lines; |
1896 |
|
while (line) { |
1897 |
|
if (lineWrite(out, line, cfg->cfi) == -1) { |
1898 |
|
fprintf(stderr, |
1899 |
|
_("grubby: error writing %s: %s\n"), |
1900 |
|
tmpOutName, strerror(errno)); |
1901 |
|
fclose(out); |
1902 |
|
unlink(tmpOutName); |
1903 |
|
return 1; |
1904 |
|
} |
1905 |
|
line = line->next; |
1906 |
|
} |
1907 |
|
} |
1908 |
|
|
1909 |
|
if (tmpOutName) { |
1910 |
|
/* write userspace buffers */ |
1911 |
|
if (fflush(out)) |
1912 |
|
rc = 1; |
1913 |
|
|
1914 |
|
/* purge the write-back cache with fsync() */ |
1915 |
|
if (fsync(fileno(out))) |
1916 |
|
rc = 1; |
1917 |
|
|
1918 |
|
if (fclose(out)) |
1919 |
|
rc = 1; |
1920 |
|
|
1921 |
|
if (rc == 0 && rename(tmpOutName, outName)) { |
1922 |
|
unlink(tmpOutName); |
1923 |
|
rc = 1; |
1924 |
|
} |
1925 |
|
|
1926 |
|
/* fsync() the destination directory after rename */ |
1927 |
|
if (rc == 0) { |
1928 |
|
int dirfd; |
1929 |
|
|
1930 |
|
dirfd = open(dirname(strdupa(outName)), O_RDONLY); |
1931 |
|
if (dirfd < 0) |
1932 |
|
rc = 1; |
1933 |
|
else if (fsync(dirfd)) |
1934 |
|
rc = 1; |
1935 |
|
|
1936 |
|
if (dirfd >= 0) |
1937 |
|
close(dirfd); |
1938 |
|
} |
1939 |
|
|
1940 |
|
if (rc == 1) |
1941 |
|
fprintf(stderr, |
1942 |
|
_("grubby: error flushing data: %m\n")); |
1943 |
|
} |
1944 |
|
|
1945 |
|
return rc; |
1946 |
|
} |
1947 |
|
|
1948 |
|
static int numEntries(struct grubConfig *cfg) |
1949 |
|
{ |
1950 |
|
int i = 0; |
1951 |
|
struct singleEntry *entry; |
1952 |
|
|
1953 |
|
entry = cfg->entries; |
1954 |
|
while (entry) { |
1955 |
|
if (!entry->skip) |
1956 |
|
i++; |
1957 |
|
entry = entry->next; |
1958 |
|
} |
1959 |
|
return i; |
1960 |
} |
} |
1961 |
|
|
1962 |
static char *findDiskForRoot() |
static char *findDiskForRoot() |
1963 |
{ |
{ |
1964 |
int fd; |
int fd; |
1965 |
char buf[65536]; |
char buf[65536]; |
1966 |
char *devname; |
char *devname; |
1967 |
char *chptr; |
char *chptr; |
1968 |
int rc; |
int rc; |
1969 |
|
|
1970 |
|
if ((fd = open(_PATH_MOUNTED, O_RDONLY)) < 0) { |
1971 |
|
fprintf(stderr, "grubby: failed to open %s: %s\n", |
1972 |
|
_PATH_MOUNTED, strerror(errno)); |
1973 |
|
return NULL; |
1974 |
|
} |
1975 |
|
|
1976 |
|
rc = read(fd, buf, sizeof(buf) - 1); |
1977 |
|
if (rc <= 0) { |
1978 |
|
fprintf(stderr, "grubby: failed to read %s: %s\n", |
1979 |
|
_PATH_MOUNTED, strerror(errno)); |
1980 |
|
close(fd); |
1981 |
|
return NULL; |
1982 |
|
} |
1983 |
|
close(fd); |
1984 |
|
buf[rc] = '\0'; |
1985 |
|
chptr = buf; |
1986 |
|
|
1987 |
if ((fd = open(_PATH_MOUNTED, O_RDONLY)) < 0) { |
char *foundanswer = NULL; |
|
fprintf(stderr, "grubby: failed to open %s: %s\n", |
|
|
_PATH_MOUNTED, strerror(errno)); |
|
|
return NULL; |
|
|
} |
|
1988 |
|
|
1989 |
rc = read(fd, buf, sizeof(buf) - 1); |
while (chptr && chptr != buf + rc) { |
1990 |
if (rc <= 0) { |
devname = chptr; |
|
fprintf(stderr, "grubby: failed to read %s: %s\n", |
|
|
_PATH_MOUNTED, strerror(errno)); |
|
|
close(fd); |
|
|
return NULL; |
|
|
} |
|
|
close(fd); |
|
|
buf[rc] = '\0'; |
|
|
chptr = buf; |
|
|
|
|
|
char *foundanswer = NULL; |
|
|
|
|
|
while (chptr && chptr != buf+rc) { |
|
|
devname = chptr; |
|
|
|
|
|
/* |
|
|
* The first column of a mtab entry is the device, but if the entry is a |
|
|
* special device it won't start with /, so move on to the next line. |
|
|
*/ |
|
|
if (*devname != '/') { |
|
|
chptr = strchr(chptr, '\n'); |
|
|
if (chptr) |
|
|
chptr++; |
|
|
continue; |
|
|
} |
|
|
|
|
|
/* Seek to the next space */ |
|
|
chptr = strchr(chptr, ' '); |
|
|
if (!chptr) { |
|
|
fprintf(stderr, "grubby: error parsing %s: %s\n", |
|
|
_PATH_MOUNTED, strerror(errno)); |
|
|
return NULL; |
|
|
} |
|
|
|
|
|
/* |
|
|
* The second column of a mtab entry is the mount point, we are looking |
|
|
* for '/' obviously. |
|
|
*/ |
|
|
if (*(++chptr) == '/' && *(++chptr) == ' ') { |
|
|
/* remember the last / entry in mtab */ |
|
|
foundanswer = devname; |
|
|
} |
|
|
|
|
|
/* Next line */ |
|
|
chptr = strchr(chptr, '\n'); |
|
|
if (chptr) |
|
|
chptr++; |
|
|
} |
|
1991 |
|
|
1992 |
/* Return the last / entry found */ |
/* |
1993 |
if (foundanswer) { |
* The first column of a mtab entry is the device, but if the |
1994 |
chptr = strchr(foundanswer, ' '); |
* entry is a special device it won't start with /, so move |
1995 |
*chptr = '\0'; |
* on to the next line. |
1996 |
return strdup(foundanswer); |
*/ |
1997 |
} |
if (*devname != '/') { |
1998 |
|
chptr = strchr(chptr, '\n'); |
1999 |
|
if (chptr) |
2000 |
|
chptr++; |
2001 |
|
continue; |
2002 |
|
} |
2003 |
|
|
2004 |
|
/* Seek to the next space */ |
2005 |
|
chptr = strchr(chptr, ' '); |
2006 |
|
if (!chptr) { |
2007 |
|
fprintf(stderr, "grubby: error parsing %s: %s\n", |
2008 |
|
_PATH_MOUNTED, strerror(errno)); |
2009 |
|
return NULL; |
2010 |
|
} |
2011 |
|
|
2012 |
|
/* |
2013 |
|
* The second column of a mtab entry is the mount point, we |
2014 |
|
* are looking for '/' obviously. |
2015 |
|
*/ |
2016 |
|
if (*(++chptr) == '/' && *(++chptr) == ' ') { |
2017 |
|
/* remember the last / entry in mtab */ |
2018 |
|
foundanswer = devname; |
2019 |
|
} |
2020 |
|
|
2021 |
|
/* Next line */ |
2022 |
|
chptr = strchr(chptr, '\n'); |
2023 |
|
if (chptr) |
2024 |
|
chptr++; |
2025 |
|
} |
2026 |
|
|
2027 |
return NULL; |
/* Return the last / entry found */ |
2028 |
|
if (foundanswer) { |
2029 |
|
chptr = strchr(foundanswer, ' '); |
2030 |
|
*chptr = '\0'; |
2031 |
|
return strdup(foundanswer); |
2032 |
|
} |
2033 |
|
|
2034 |
|
return NULL; |
2035 |
} |
} |
2036 |
|
|
2037 |
void printEntry(struct singleEntry * entry) { |
void printEntry(struct singleEntry *entry, FILE * f) |
2038 |
int i; |
{ |
2039 |
struct singleLine * line; |
int i; |
2040 |
|
struct singleLine *line; |
2041 |
|
|
2042 |
|
for (line = entry->lines; line; line = line->next) { |
2043 |
|
log_message(f, "DBG: %s", line->indent); |
2044 |
|
for (i = 0; i < line->numElements; i++) { |
2045 |
|
/* Need to handle this, because we strip the quotes from |
2046 |
|
* menuentry when read it. */ |
2047 |
|
if (line->type == LT_MENUENTRY && i == 1) { |
2048 |
|
if (!isquote(*line->elements[i].item)) |
2049 |
|
log_message(f, "\'%s\'", |
2050 |
|
line->elements[i].item); |
2051 |
|
else |
2052 |
|
log_message(f, "%s", |
2053 |
|
line->elements[i].item); |
2054 |
|
log_message(f, "%s", line->elements[i].indent); |
2055 |
|
|
2056 |
for (line = entry->lines; line; line = line->next) { |
continue; |
2057 |
fprintf(stderr, "DBG: %s", line->indent); |
} |
2058 |
for (i = 0; i < line->numElements; i++) { |
|
2059 |
/* Need to handle this, because we strip the quotes from |
log_message(f, "%s%s", |
2060 |
* menuentry when read it. */ |
line->elements[i].item, |
2061 |
if (line->type == LT_MENUENTRY && i == 1) { |
line->elements[i].indent); |
2062 |
if(!isquote(*line->elements[i].item)) |
} |
2063 |
fprintf(stderr, "\'%s\'", line->elements[i].item); |
log_message(f, "\n"); |
|
else |
|
|
fprintf(stderr, "%s", line->elements[i].item); |
|
|
fprintf(stderr, "%s", line->elements[i].indent); |
|
|
|
|
|
continue; |
|
|
} |
|
|
|
|
|
fprintf(stderr, "%s%s", |
|
|
line->elements[i].item, line->elements[i].indent); |
|
2064 |
} |
} |
|
fprintf(stderr, "\n"); |
|
|
} |
|
2065 |
} |
} |
2066 |
|
|
2067 |
void notSuitablePrintf(struct singleEntry * entry, const char *fmt, ...) |
void notSuitablePrintf(struct singleEntry *entry, int okay, const char *fmt, |
2068 |
|
...) |
2069 |
{ |
{ |
2070 |
va_list argp; |
static int once; |
2071 |
|
va_list argp, argq; |
2072 |
|
|
2073 |
if (!debug) |
va_start(argp, fmt); |
|
return; |
|
2074 |
|
|
2075 |
va_start(argp, fmt); |
va_copy(argq, argp); |
2076 |
fprintf(stderr, "DBG: Image entry failed: "); |
if (!once) { |
2077 |
vfprintf(stderr, fmt, argp); |
log_time(NULL); |
2078 |
printEntry(entry); |
log_message(NULL, "command line: %s\n", saved_command_line); |
2079 |
va_end(argp); |
} |
2080 |
|
log_message(NULL, "DBG: Image entry %s: ", |
2081 |
|
okay ? "succeeded" : "failed"); |
2082 |
|
log_vmessage(NULL, fmt, argq); |
2083 |
|
|
2084 |
|
printEntry(entry, NULL); |
2085 |
|
va_end(argq); |
2086 |
|
|
2087 |
|
if (!debug) { |
2088 |
|
once = 1; |
2089 |
|
va_end(argp); |
2090 |
|
return; |
2091 |
|
} |
2092 |
|
|
2093 |
|
if (okay) { |
2094 |
|
va_end(argp); |
2095 |
|
return; |
2096 |
|
} |
2097 |
|
|
2098 |
|
if (!once) |
2099 |
|
log_message(stderr, "DBG: command line: %s\n", |
2100 |
|
saved_command_line); |
2101 |
|
once = 1; |
2102 |
|
fprintf(stderr, "DBG: Image entry failed: "); |
2103 |
|
vfprintf(stderr, fmt, argp); |
2104 |
|
printEntry(entry, stderr); |
2105 |
|
va_end(argp); |
2106 |
} |
} |
2107 |
|
|
2108 |
#define beginswith(s, c) ((s) && (s)[0] == (c)) |
#define beginswith(s, c) ((s) && (s)[0] == (c)) |
2118 |
return s[slen] == c; |
return s[slen] == c; |
2119 |
} |
} |
2120 |
|
|
2121 |
int suitableImage(struct singleEntry * entry, const char * bootPrefix, |
int suitableImage(struct singleEntry *entry, const char *bootPrefix, |
2122 |
int skipRemoved, int flags) { |
int skipRemoved, int flags) |
2123 |
struct singleLine * line; |
{ |
2124 |
char * fullName; |
struct singleLine *line; |
2125 |
int i; |
char *fullName; |
2126 |
char * dev; |
int i; |
2127 |
char * rootspec; |
char *dev; |
2128 |
char * rootdev; |
char *rootspec; |
2129 |
|
char *rootdev; |
|
if (skipRemoved && entry->skip) { |
|
|
notSuitablePrintf(entry, "marked to skip\n"); |
|
|
return 0; |
|
|
} |
|
2130 |
|
|
2131 |
line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines); |
if (skipRemoved && entry->skip) { |
2132 |
if (!line) { |
notSuitablePrintf(entry, 0, "marked to skip\n"); |
2133 |
notSuitablePrintf(entry, "no line found\n"); |
return 0; |
2134 |
return 0; |
} |
|
} |
|
|
if (line->numElements < 2) { |
|
|
notSuitablePrintf(entry, "line has only %d elements\n", |
|
|
line->numElements); |
|
|
return 0; |
|
|
} |
|
2135 |
|
|
2136 |
if (flags & GRUBBY_BADIMAGE_OKAY) return 1; |
line = |
2137 |
|
getLineByType(LT_KERNEL | LT_HYPER | LT_KERNEL_EFI | LT_KERNEL_16, |
2138 |
|
entry->lines); |
2139 |
|
if (!line) { |
2140 |
|
notSuitablePrintf(entry, 0, "no line found\n"); |
2141 |
|
return 0; |
2142 |
|
} |
2143 |
|
if (line->numElements < 2) { |
2144 |
|
notSuitablePrintf(entry, 0, "line has only %d elements\n", |
2145 |
|
line->numElements); |
2146 |
|
return 0; |
2147 |
|
} |
2148 |
|
|
2149 |
fullName = alloca(strlen(bootPrefix) + |
if (flags & GRUBBY_BADIMAGE_OKAY) { |
2150 |
strlen(line->elements[1].item) + 1); |
notSuitablePrintf(entry, 1, "\n"); |
2151 |
rootspec = getRootSpecifier(line->elements[1].item); |
return 1; |
2152 |
int rootspec_offset = rootspec ? strlen(rootspec) : 0; |
} |
|
int hasslash = endswith(bootPrefix, '/') || |
|
|
beginswith(line->elements[1].item + rootspec_offset, '/'); |
|
|
sprintf(fullName, "%s%s%s", bootPrefix, hasslash ? "" : "/", |
|
|
line->elements[1].item + rootspec_offset); |
|
|
if (access(fullName, R_OK)) { |
|
|
notSuitablePrintf(entry, "access to %s failed\n", fullName); |
|
|
return 0; |
|
|
} |
|
|
for (i = 2; i < line->numElements; i++) |
|
|
if (!strncasecmp(line->elements[i].item, "root=", 5)) break; |
|
|
if (i < line->numElements) { |
|
|
dev = line->elements[i].item + 5; |
|
|
} else { |
|
|
/* look for a lilo style LT_ROOT line */ |
|
|
line = getLineByType(LT_ROOT, entry->lines); |
|
2153 |
|
|
2154 |
if (line && line->numElements >= 2) { |
fullName = alloca(strlen(bootPrefix) + |
2155 |
dev = line->elements[1].item; |
strlen(line->elements[1].item) + 1); |
2156 |
|
rootspec = getRootSpecifier(line->elements[1].item); |
2157 |
|
int rootspec_offset = rootspec ? strlen(rootspec) : 0; |
2158 |
|
int hasslash = endswith(bootPrefix, '/') || |
2159 |
|
beginswith(line->elements[1].item + rootspec_offset, '/'); |
2160 |
|
sprintf(fullName, "%s%s%s", bootPrefix, hasslash ? "" : "/", |
2161 |
|
line->elements[1].item + rootspec_offset); |
2162 |
|
if (access(fullName, R_OK)) { |
2163 |
|
notSuitablePrintf(entry, 0, "access to %s failed\n", fullName); |
2164 |
|
return 0; |
2165 |
|
} |
2166 |
|
for (i = 2; i < line->numElements; i++) |
2167 |
|
if (!strncasecmp(line->elements[i].item, "root=", 5)) |
2168 |
|
break; |
2169 |
|
if (i < line->numElements) { |
2170 |
|
dev = line->elements[i].item + 5; |
2171 |
} else { |
} else { |
2172 |
/* didn't succeed in finding a LT_ROOT, let's try LT_KERNELARGS. |
/* look for a lilo style LT_ROOT line */ |
2173 |
* grub+multiboot uses LT_MBMODULE for the args, so check that too. |
line = getLineByType(LT_ROOT, entry->lines); |
2174 |
*/ |
|
2175 |
line = getLineByType(LT_KERNELARGS|LT_MBMODULE, entry->lines); |
if (line && line->numElements >= 2) { |
2176 |
|
dev = line->elements[1].item; |
2177 |
/* failed to find one */ |
} else { |
2178 |
if (!line) { |
/* didn't succeed in finding a LT_ROOT, let's try |
2179 |
notSuitablePrintf(entry, "no line found\n"); |
* LT_KERNELARGS. grub+multiboot uses LT_MBMODULE |
2180 |
return 0; |
* for the args, so check that too. |
2181 |
} |
*/ |
2182 |
|
line = |
2183 |
for (i = 1; i < line->numElements; i++) |
getLineByType(LT_KERNELARGS | LT_MBMODULE, |
2184 |
if (!strncasecmp(line->elements[i].item, "root=", 5)) break; |
entry->lines); |
2185 |
if (i < line->numElements) |
|
2186 |
dev = line->elements[i].item + 5; |
/* failed to find one */ |
2187 |
else { |
if (!line) { |
2188 |
notSuitablePrintf(entry, "no root= entry found\n"); |
notSuitablePrintf(entry, 0, "no line found\n"); |
2189 |
/* it failed too... can't find root= */ |
return 0; |
2190 |
return 0; |
} |
2191 |
} |
|
2192 |
|
for (i = 1; i < line->numElements; i++) |
2193 |
|
if (!strncasecmp |
2194 |
|
(line->elements[i].item, "root=", 5)) |
2195 |
|
break; |
2196 |
|
if (i < line->numElements) |
2197 |
|
dev = line->elements[i].item + 5; |
2198 |
|
else { |
2199 |
|
notSuitablePrintf(entry, 0, |
2200 |
|
"no root= entry found\n"); |
2201 |
|
/* it failed too... can't find root= */ |
2202 |
|
return 0; |
2203 |
|
} |
2204 |
|
} |
2205 |
} |
} |
|
} |
|
2206 |
|
|
|
dev = getpathbyspec(dev); |
|
|
if (!getpathbyspec(dev)) { |
|
|
notSuitablePrintf(entry, "can't find blkid entry for %s\n", dev); |
|
|
return 0; |
|
|
} else |
|
2207 |
dev = getpathbyspec(dev); |
dev = getpathbyspec(dev); |
2208 |
|
if (!getpathbyspec(dev)) { |
2209 |
|
notSuitablePrintf(entry, 0, "can't find blkid entry for %s\n", |
2210 |
|
dev); |
2211 |
|
return 0; |
2212 |
|
} else |
2213 |
|
dev = getpathbyspec(dev); |
2214 |
|
|
2215 |
rootdev = findDiskForRoot(); |
rootdev = findDiskForRoot(); |
2216 |
if (!rootdev) { |
if (!rootdev) { |
2217 |
notSuitablePrintf(entry, "can't find root device\n"); |
notSuitablePrintf(entry, 0, "can't find root device\n"); |
2218 |
return 0; |
return 0; |
2219 |
} |
} |
2220 |
|
|
2221 |
if (!getuuidbydev(rootdev) || !getuuidbydev(dev)) { |
if (!getuuidbydev(rootdev) || !getuuidbydev(dev)) { |
2222 |
notSuitablePrintf(entry, "uuid missing: rootdev %s, dev %s\n", |
notSuitablePrintf(entry, 0, |
2223 |
getuuidbydev(rootdev), getuuidbydev(dev)); |
"uuid missing: rootdev %s, dev %s\n", |
2224 |
free(rootdev); |
getuuidbydev(rootdev), getuuidbydev(dev)); |
2225 |
return 0; |
free(rootdev); |
2226 |
} |
return 0; |
2227 |
|
} |
2228 |
|
|
2229 |
if (strcmp(getuuidbydev(rootdev), getuuidbydev(dev))) { |
if (strcmp(getuuidbydev(rootdev), getuuidbydev(dev))) { |
2230 |
notSuitablePrintf(entry, "uuid mismatch: rootdev %s, dev %s\n", |
notSuitablePrintf(entry, 0, |
2231 |
getuuidbydev(rootdev), getuuidbydev(dev)); |
"uuid mismatch: rootdev %s, dev %s\n", |
2232 |
free(rootdev); |
getuuidbydev(rootdev), getuuidbydev(dev)); |
2233 |
return 0; |
free(rootdev); |
2234 |
} |
return 0; |
2235 |
|
} |
2236 |
|
|
2237 |
free(rootdev); |
free(rootdev); |
2238 |
|
notSuitablePrintf(entry, 1, "\n"); |
2239 |
|
|
2240 |
return 1; |
return 1; |
2241 |
} |
} |
2242 |
|
|
2243 |
/* returns the first match on or after the one pointed to by index (if index |
/* returns the first match on or after the one pointed to by index (if index |
2244 |
is not NULL) which is not marked as skip */ |
is not NULL) which is not marked as skip */ |
2245 |
struct singleEntry * findEntryByPath(struct grubConfig * config, |
struct singleEntry *findEntryByPath(struct grubConfig *config, |
2246 |
const char * kernel, const char * prefix, |
const char *kernel, const char *prefix, |
2247 |
int * index) { |
int *index) |
2248 |
struct singleEntry * entry = NULL; |
{ |
2249 |
struct singleLine * line; |
struct singleEntry *entry = NULL; |
2250 |
int i; |
struct singleLine *line; |
2251 |
char * chptr; |
int i; |
2252 |
char * rootspec = NULL; |
char *chptr; |
2253 |
enum lineType_e checkType = LT_KERNEL; |
char *rootspec = NULL; |
2254 |
|
enum lineType_e checkType = LT_KERNEL; |
2255 |
|
|
2256 |
|
if (isdigit(*kernel)) { |
2257 |
|
int *indexVars = alloca(sizeof(*indexVars) * strlen(kernel)); |
2258 |
|
|
2259 |
|
i = 0; |
2260 |
|
indexVars[i] = strtol(kernel, &chptr, 10); |
2261 |
|
while (*chptr == ',') { |
2262 |
|
i++; |
2263 |
|
kernel = chptr + 1; |
2264 |
|
indexVars[i] = strtol(kernel, &chptr, 10); |
2265 |
|
} |
2266 |
|
|
2267 |
if (isdigit(*kernel)) { |
if (*chptr) { |
2268 |
int * indexVars = alloca(sizeof(*indexVars) * strlen(kernel)); |
/* can't parse it, bail */ |
2269 |
|
return NULL; |
2270 |
|
} |
2271 |
|
|
2272 |
i = 0; |
indexVars[i + 1] = -1; |
|
indexVars[i] = strtol(kernel, &chptr, 10); |
|
|
while (*chptr == ',') { |
|
|
i++; |
|
|
kernel = chptr + 1; |
|
|
indexVars[i] = strtol(kernel, &chptr, 10); |
|
|
} |
|
2273 |
|
|
2274 |
if (*chptr) { |
i = 0; |
2275 |
/* can't parse it, bail */ |
if (index) { |
2276 |
return NULL; |
while (i < *index) { |
2277 |
} |
i++; |
2278 |
|
if (indexVars[i] == -1) |
2279 |
|
return NULL; |
2280 |
|
} |
2281 |
|
} |
2282 |
|
|
2283 |
indexVars[i + 1] = -1; |
entry = findEntryByIndex(config, indexVars[i]); |
2284 |
|
if (!entry) |
2285 |
i = 0; |
return NULL; |
2286 |
if (index) { |
|
2287 |
while (i < *index) i++; |
line = |
2288 |
if (indexVars[i] == -1) return NULL; |
getLineByType(LT_KERNEL | LT_HYPER | LT_KERNEL_EFI | |
2289 |
} |
LT_KERNEL_16, entry->lines); |
2290 |
|
if (!line) |
2291 |
|
return NULL; |
2292 |
|
|
2293 |
|
if (index) |
2294 |
|
*index = indexVars[i]; |
2295 |
|
return entry; |
2296 |
|
} |
2297 |
|
|
2298 |
|
if (!strcmp(kernel, "DEFAULT")) { |
2299 |
|
if (index && *index > config->defaultImage) { |
2300 |
|
entry = NULL; |
2301 |
|
} else { |
2302 |
|
entry = findEntryByIndex(config, config->defaultImage); |
2303 |
|
if (entry && entry->skip) |
2304 |
|
entry = NULL; |
2305 |
|
else if (index) |
2306 |
|
*index = config->defaultImage; |
2307 |
|
} |
2308 |
|
} else if (!strcmp(kernel, "ALL")) { |
2309 |
|
if (index) |
2310 |
|
i = *index; |
2311 |
|
else |
2312 |
|
i = 0; |
2313 |
|
|
2314 |
entry = findEntryByIndex(config, indexVars[i]); |
while ((entry = findEntryByIndex(config, i))) { |
2315 |
if (!entry) return NULL; |
if (!entry->skip) |
2316 |
|
break; |
2317 |
|
i++; |
2318 |
|
} |
2319 |
|
|
2320 |
line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines); |
if (entry && index) |
2321 |
if (!line) return NULL; |
*index = i; |
2322 |
|
} else { |
2323 |
|
if (index) |
2324 |
|
i = *index; |
2325 |
|
else |
2326 |
|
i = 0; |
2327 |
|
|
2328 |
if (index) *index = indexVars[i]; |
if (!strncmp(kernel, "TITLE=", 6)) { |
2329 |
return entry; |
prefix = ""; |
2330 |
} |
checkType = LT_TITLE | LT_MENUENTRY; |
2331 |
|
kernel += 6; |
2332 |
if (!strcmp(kernel, "DEFAULT")) { |
} |
2333 |
if (index && *index > config->defaultImage) { |
|
2334 |
entry = NULL; |
for (entry = findEntryByIndex(config, i); entry; |
2335 |
} else { |
entry = entry->next, i++) { |
2336 |
entry = findEntryByIndex(config, config->defaultImage); |
if (entry->skip) |
2337 |
if (entry && entry->skip) |
continue; |
2338 |
entry = NULL; |
|
2339 |
else if (index) |
dbgPrintf("findEntryByPath looking for %d %s in %p\n", |
2340 |
*index = config->defaultImage; |
checkType, kernel, entry); |
2341 |
} |
|
2342 |
} else if (!strcmp(kernel, "ALL")) { |
/* check all the lines matching checkType */ |
2343 |
if (index) |
for (line = entry->lines; line; line = line->next) { |
2344 |
i = *index; |
enum lineType_e ct = checkType; |
2345 |
else |
if (entry->multiboot && checkType == LT_KERNEL) |
2346 |
i = 0; |
ct = LT_KERNEL | LT_KERNEL_EFI | |
2347 |
|
LT_MBMODULE | LT_HYPER | |
2348 |
|
LT_KERNEL_16; |
2349 |
|
else if (checkType & LT_KERNEL) |
2350 |
|
ct = checkType | LT_KERNEL_EFI | |
2351 |
|
LT_KERNEL_16; |
2352 |
|
line = getLineByType(ct, line); |
2353 |
|
if (!line) |
2354 |
|
break; /* not found in this entry */ |
2355 |
|
|
2356 |
|
if (line && line->type != LT_MENUENTRY && |
2357 |
|
line->numElements >= 2) { |
2358 |
|
rootspec = |
2359 |
|
getRootSpecifier(line->elements[1]. |
2360 |
|
item); |
2361 |
|
if (!strcmp |
2362 |
|
(line->elements[1].item + |
2363 |
|
((rootspec != |
2364 |
|
NULL) ? strlen(rootspec) : 0), |
2365 |
|
kernel + strlen(prefix))) |
2366 |
|
break; |
2367 |
|
} |
2368 |
|
if (line->type == LT_MENUENTRY && |
2369 |
|
!strcmp(line->elements[1].item, kernel)) |
2370 |
|
break; |
2371 |
|
} |
2372 |
|
|
2373 |
while ((entry = findEntryByIndex(config, i))) { |
/* make sure this entry has a kernel identifier; this skips |
2374 |
if (!entry->skip) break; |
* non-Linux boot entries (could find netbsd etc, though, which is |
2375 |
i++; |
* unfortunate) |
2376 |
|
*/ |
2377 |
|
if (line |
2378 |
|
&& getLineByType(LT_KERNEL | LT_HYPER | |
2379 |
|
LT_KERNEL_EFI | LT_KERNEL_16, |
2380 |
|
entry->lines)) |
2381 |
|
break; /* found 'im! */ |
2382 |
|
} |
2383 |
|
|
2384 |
|
if (index) |
2385 |
|
*index = i; |
2386 |
} |
} |
2387 |
|
|
2388 |
if (entry && index) |
return entry; |
2389 |
*index = i; |
} |
|
} else { |
|
|
if (index) |
|
|
i = *index; |
|
|
else |
|
|
i = 0; |
|
2390 |
|
|
2391 |
if (!strncmp(kernel, "TITLE=", 6)) { |
struct singleEntry *findEntryByTitle(struct grubConfig *cfg, char *title, |
2392 |
prefix = ""; |
int *index) |
2393 |
checkType = LT_TITLE|LT_MENUENTRY; |
{ |
2394 |
kernel += 6; |
struct singleEntry *entry; |
2395 |
} |
struct singleLine *line; |
2396 |
|
int i; |
2397 |
for (entry = findEntryByIndex(config, i); entry; entry = entry->next, i++) { |
char *newtitle; |
2398 |
if (entry->skip) continue; |
|
2399 |
|
for (i = 0, entry = cfg->entries; entry; entry = entry->next, i++) { |
2400 |
dbgPrintf("findEntryByPath looking for %d %s in %p\n", checkType, kernel, entry); |
if (index && i < *index) |
2401 |
|
continue; |
2402 |
/* check all the lines matching checkType */ |
line = getLineByType(LT_TITLE, entry->lines); |
2403 |
for (line = entry->lines; line; line = line->next) { |
if (!line) |
2404 |
line = getLineByType(entry->multiboot && checkType == LT_KERNEL ? |
line = getLineByType(LT_MENUENTRY, entry->lines); |
2405 |
LT_KERNEL|LT_MBMODULE|LT_HYPER : |
if (!line) |
2406 |
checkType, line); |
continue; |
2407 |
if (!line) break; /* not found in this entry */ |
newtitle = grub2ExtractTitle(line); |
2408 |
|
if (!newtitle) |
2409 |
if (line && line->type != LT_MENUENTRY && |
continue; |
2410 |
line->numElements >= 2) { |
if (!strcmp(title, newtitle)) |
|
rootspec = getRootSpecifier(line->elements[1].item); |
|
|
if (!strcmp(line->elements[1].item + |
|
|
((rootspec != NULL) ? strlen(rootspec) : 0), |
|
|
kernel + strlen(prefix))) |
|
2411 |
break; |
break; |
|
} |
|
|
if(line->type == LT_MENUENTRY && |
|
|
!strcmp(line->elements[1].item, kernel)) |
|
|
break; |
|
|
} |
|
|
|
|
|
/* make sure this entry has a kernel identifier; this skips |
|
|
* non-Linux boot entries (could find netbsd etc, though, which is |
|
|
* unfortunate) |
|
|
*/ |
|
|
if (line && getLineByType(LT_KERNEL|LT_HYPER, entry->lines)) |
|
|
break; /* found 'im! */ |
|
2412 |
} |
} |
2413 |
|
|
2414 |
if (index) *index = i; |
if (!entry) |
2415 |
} |
return NULL; |
2416 |
|
|
2417 |
return entry; |
if (index) |
2418 |
|
*index = i; |
2419 |
|
return entry; |
2420 |
} |
} |
2421 |
|
|
2422 |
struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index) { |
struct singleEntry *findEntryByIndex(struct grubConfig *cfg, int index) |
2423 |
struct singleEntry * entry; |
{ |
2424 |
|
struct singleEntry *entry; |
2425 |
|
|
2426 |
entry = cfg->entries; |
entry = cfg->entries; |
2427 |
while (index && entry) { |
while (index && entry) { |
2428 |
entry = entry->next; |
entry = entry->next; |
2429 |
index--; |
index--; |
2430 |
} |
} |
2431 |
|
|
2432 |
return entry; |
return entry; |
2433 |
} |
} |
2434 |
|
|
2435 |
/* Find a good template to use for the new kernel. An entry is |
/* Find a good template to use for the new kernel. An entry is |
2437 |
* is going to be removed). Try and use the default entry, but |
* is going to be removed). Try and use the default entry, but |
2438 |
* if that doesn't work just take the first. If we can't find one, |
* if that doesn't work just take the first. If we can't find one, |
2439 |
* bail. */ |
* bail. */ |
2440 |
struct singleEntry * findTemplate(struct grubConfig * cfg, const char * prefix, |
struct singleEntry *findTemplate(struct grubConfig *cfg, const char *prefix, |
2441 |
int * indexPtr, int skipRemoved, int flags) { |
int *indexPtr, int skipRemoved, int flags) |
2442 |
struct singleEntry * entry, * entry2; |
{ |
2443 |
int index; |
struct singleEntry *entry, *entry2; |
2444 |
|
int index; |
2445 |
if (cfg->defaultImage > -1) { |
|
2446 |
entry = findEntryByIndex(cfg, cfg->defaultImage); |
if (cfg->cfi->defaultIsSaved) { |
2447 |
if (entry && suitableImage(entry, prefix, skipRemoved, flags)) { |
if (cfg->cfi->getEnv) { |
2448 |
if (indexPtr) *indexPtr = cfg->defaultImage; |
char *defTitle = |
2449 |
return entry; |
cfg->cfi->getEnv(cfg->cfi, "saved_entry"); |
2450 |
|
if (defTitle) { |
2451 |
|
int index = 0; |
2452 |
|
if (isnumber(defTitle)) { |
2453 |
|
index = atoi(defTitle); |
2454 |
|
entry = findEntryByIndex(cfg, index); |
2455 |
|
} else { |
2456 |
|
entry = |
2457 |
|
findEntryByTitle(cfg, defTitle, |
2458 |
|
&index); |
2459 |
|
} |
2460 |
|
if (entry |
2461 |
|
&& suitableImage(entry, prefix, skipRemoved, |
2462 |
|
flags)) { |
2463 |
|
cfg->defaultImage = index; |
2464 |
|
if (indexPtr) |
2465 |
|
*indexPtr = index; |
2466 |
|
return entry; |
2467 |
|
} |
2468 |
|
} |
2469 |
|
} |
2470 |
|
} else if (cfg->defaultImage >= FIRST_ENTRY_INDEX) { |
2471 |
|
entry = findEntryByIndex(cfg, cfg->defaultImage); |
2472 |
|
if (entry && suitableImage(entry, prefix, skipRemoved, flags)) { |
2473 |
|
if (indexPtr) |
2474 |
|
*indexPtr = cfg->defaultImage; |
2475 |
|
return entry; |
2476 |
|
} |
2477 |
} |
} |
|
} |
|
2478 |
|
|
2479 |
index = 0; |
index = 0; |
2480 |
while ((entry = findEntryByIndex(cfg, index))) { |
while ((entry = findEntryByIndex(cfg, index))) { |
2481 |
if (suitableImage(entry, prefix, skipRemoved, flags)) { |
if (suitableImage(entry, prefix, skipRemoved, flags)) { |
2482 |
int j; |
int j, unmodifiedIndex; |
2483 |
for (j = 0; j < index; j++) { |
|
2484 |
entry2 = findEntryByIndex(cfg, j); |
unmodifiedIndex = index; |
2485 |
if (entry2->skip) index--; |
|
2486 |
} |
for (j = 0; j < unmodifiedIndex; j++) { |
2487 |
if (indexPtr) *indexPtr = index; |
entry2 = findEntryByIndex(cfg, j); |
2488 |
|
if (entry2->skip) |
2489 |
|
index--; |
2490 |
|
} |
2491 |
|
if (indexPtr) |
2492 |
|
*indexPtr = index; |
2493 |
|
|
2494 |
return entry; |
return entry; |
2495 |
} |
} |
2496 |
|
|
2497 |
index++; |
index++; |
2498 |
} |
} |
2499 |
|
|
2500 |
fprintf(stderr, _("grubby fatal error: unable to find a suitable template\n")); |
fprintf(stderr, |
2501 |
|
_("grubby fatal error: unable to find a suitable template\n")); |
2502 |
|
|
2503 |
return NULL; |
return NULL; |
2504 |
} |
} |
2505 |
|
|
2506 |
char * findBootPrefix(void) { |
char *findBootPrefix(void) |
2507 |
struct stat sb, sb2; |
{ |
2508 |
|
struct stat sb, sb2; |
2509 |
|
|
2510 |
stat("/", &sb); |
stat("/", &sb); |
2511 |
#ifdef __ia64__ |
#ifdef __ia64__ |
2512 |
stat("/boot/efi/EFI/redhat/", &sb2); |
stat("/boot/efi/EFI/redhat/", &sb2); |
2513 |
#else |
#else |
2514 |
stat("/boot", &sb2); |
stat("/boot", &sb2); |
2515 |
#endif |
#endif |
2516 |
|
|
2517 |
if (sb.st_dev == sb2.st_dev) |
if (sb.st_dev == sb2.st_dev) |
2518 |
return strdup(""); |
return strdup(""); |
2519 |
|
|
2520 |
#ifdef __ia64__ |
#ifdef __ia64__ |
2521 |
return strdup("/boot/efi/EFI/redhat/"); |
return strdup("/boot/efi/EFI/redhat/"); |
2522 |
#else |
#else |
2523 |
return strdup("/boot"); |
return strdup("/boot"); |
2524 |
#endif |
#endif |
2525 |
} |
} |
2526 |
|
|
2527 |
void markRemovedImage(struct grubConfig * cfg, const char * image, |
void markRemovedImage(struct grubConfig *cfg, const char *image, |
2528 |
const char * prefix) { |
const char *prefix) |
2529 |
struct singleEntry * entry; |
{ |
2530 |
|
struct singleEntry *entry; |
|
if (!image) |
|
|
return; |
|
|
|
|
|
/* check and see if we're removing the default image */ |
|
|
if (isdigit(*image)) { |
|
|
entry = findEntryByPath(cfg, image, prefix, NULL); |
|
|
if(entry) |
|
|
entry->skip = 1; |
|
|
return; |
|
|
} |
|
2531 |
|
|
2532 |
while ((entry = findEntryByPath(cfg, image, prefix, NULL))) |
if (!image) |
2533 |
entry->skip = 1; |
return; |
|
} |
|
2534 |
|
|
2535 |
void setDefaultImage(struct grubConfig * config, int hasNew, |
/* check and see if we're removing the default image */ |
2536 |
const char * defaultKernelPath, int newIsDefault, |
if (isdigit(*image)) { |
2537 |
const char * prefix, int flags) { |
entry = findEntryByPath(cfg, image, prefix, NULL); |
2538 |
struct singleEntry * entry, * entry2, * newDefault; |
if (entry) |
2539 |
int i, j; |
entry->skip = 1; |
2540 |
|
return; |
|
if (newIsDefault) { |
|
|
config->defaultImage = 0; |
|
|
return; |
|
|
} else if (defaultKernelPath) { |
|
|
i = 0; |
|
|
if (findEntryByPath(config, defaultKernelPath, prefix, &i)) { |
|
|
config->defaultImage = i; |
|
|
} else { |
|
|
config->defaultImage = -1; |
|
|
return; |
|
2541 |
} |
} |
|
} |
|
|
|
|
|
/* defaultImage now points to what we'd like to use, but before any order |
|
|
changes */ |
|
|
if ((config->defaultImage == DEFAULT_SAVED) || |
|
|
(config->defaultImage == DEFAULT_SAVED_GRUB2)) |
|
|
/* default is set to saved, we don't want to change it */ |
|
|
return; |
|
|
|
|
|
if (config->defaultImage > -1) |
|
|
entry = findEntryByIndex(config, config->defaultImage); |
|
|
else |
|
|
entry = NULL; |
|
|
|
|
|
if (entry && !entry->skip) { |
|
|
/* we can preserve the default */ |
|
|
if (hasNew) |
|
|
config->defaultImage++; |
|
|
|
|
|
/* count the number of entries erased before this one */ |
|
|
for (j = 0; j < config->defaultImage; j++) { |
|
|
entry2 = findEntryByIndex(config, j); |
|
|
if (entry2->skip) config->defaultImage--; |
|
|
} |
|
|
} else if (hasNew) { |
|
|
config->defaultImage = 0; |
|
|
} else { |
|
|
/* Either we just erased the default (or the default line was bad |
|
|
* to begin with) and didn't put a new one in. We'll use the first |
|
|
* valid image. */ |
|
|
newDefault = findTemplate(config, prefix, &config->defaultImage, 1, |
|
|
flags); |
|
|
if (!newDefault) |
|
|
config->defaultImage = -1; |
|
|
} |
|
|
} |
|
|
|
|
|
void setFallbackImage(struct grubConfig * config, int hasNew) { |
|
|
struct singleEntry * entry, * entry2; |
|
|
int j; |
|
|
|
|
|
if (config->fallbackImage == -1) return; |
|
|
|
|
|
entry = findEntryByIndex(config, config->fallbackImage); |
|
|
if (!entry || entry->skip) { |
|
|
config->fallbackImage = -1; |
|
|
return; |
|
|
} |
|
2542 |
|
|
2543 |
if (hasNew) |
while ((entry = findEntryByPath(cfg, image, prefix, NULL))) |
2544 |
config->fallbackImage++; |
entry->skip = 1; |
|
|
|
|
/* count the number of entries erased before this one */ |
|
|
for (j = 0; j < config->fallbackImage; j++) { |
|
|
entry2 = findEntryByIndex(config, j); |
|
|
if (entry2->skip) config->fallbackImage--; |
|
|
} |
|
2545 |
} |
} |
2546 |
|
|
2547 |
void displayEntry(struct singleEntry * entry, const char * prefix, int index) { |
void setDefaultImage(struct grubConfig *config, int isAddingBootEntry, |
2548 |
struct singleLine * line; |
const char *defaultKernelPath, int newBootEntryIsDefault, |
2549 |
char * root = NULL; |
const char *prefix, int flags, |
2550 |
int i; |
int newDefaultBootEntryIndex, int newBootEntryIndex) |
2551 |
|
{ |
2552 |
printf("index=%d\n", index); |
struct singleEntry *bootEntry, *newDefault; |
2553 |
|
int indexToVerify, firstKernelEntryIndex, currentLookupIndex; |
|
line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines); |
|
|
if (!line) { |
|
|
printf("non linux entry\n"); |
|
|
return; |
|
|
} |
|
2554 |
|
|
2555 |
printf("kernel=%s%s\n", prefix, line->elements[1].item); |
/* initialize */ |
2556 |
|
currentLookupIndex = FIRST_ENTRY_INDEX; |
2557 |
|
|
2558 |
if (line->numElements >= 3) { |
/* handle the two cases where the user explictly picks the default |
2559 |
printf("args=\""); |
* boot entry index as it would exist post-modification */ |
|
i = 2; |
|
|
while (i < line->numElements) { |
|
|
if (!strncmp(line->elements[i].item, "root=", 5)) { |
|
|
root = line->elements[i].item + 5; |
|
|
} else { |
|
|
printf("%s%s", line->elements[i].item, |
|
|
line->elements[i].indent); |
|
|
} |
|
2560 |
|
|
2561 |
i++; |
/* Case 1: user chose to make the latest boot entry the default */ |
2562 |
|
if (newBootEntryIsDefault) { |
2563 |
|
config->defaultImage = newBootEntryIndex; |
2564 |
|
return; |
2565 |
} |
} |
|
printf("\"\n"); |
|
|
} else { |
|
|
line = getLineByType(LT_KERNELARGS, entry->lines); |
|
|
if (line) { |
|
|
char * s; |
|
2566 |
|
|
2567 |
printf("args=\""); |
/* Case 2: user picked an arbitrary index as the default boot entry */ |
2568 |
i = 1; |
if (newDefaultBootEntryIndex >= FIRST_ENTRY_INDEX) { |
2569 |
while (i < line->numElements) { |
indexToVerify = newDefaultBootEntryIndex; |
|
if (!strncmp(line->elements[i].item, "root=", 5)) { |
|
|
root = line->elements[i].item + 5; |
|
|
} else { |
|
|
s = line->elements[i].item; |
|
2570 |
|
|
2571 |
printf("%s%s", s, line->elements[i].indent); |
/* user chose to make latest boot entry the default */ |
2572 |
|
if (newDefaultBootEntryIndex == newBootEntryIndex) { |
2573 |
|
config->defaultImage = newBootEntryIndex; |
2574 |
|
return; |
2575 |
} |
} |
2576 |
|
|
2577 |
i++; |
/* the user picks the default index based on the |
2578 |
} |
* order of the bootloader configuration after |
2579 |
|
* modification; ensure we are checking for the |
2580 |
|
* existence of the correct entry */ |
2581 |
|
if (newBootEntryIndex < newDefaultBootEntryIndex) { |
2582 |
|
if (!config->isModified) |
2583 |
|
indexToVerify--; |
2584 |
|
} |
2585 |
|
|
2586 |
s = line->elements[i - 1].indent; |
/* verify the user selected index will exist */ |
2587 |
printf("\"\n"); |
if (findEntryByIndex(config, indexToVerify)) { |
2588 |
} |
config->defaultImage = newDefaultBootEntryIndex; |
2589 |
} |
} else { |
2590 |
|
config->defaultImage = NO_DEFAULT_ENTRY; |
2591 |
|
} |
2592 |
|
|
2593 |
if (!root) { |
return; |
2594 |
line = getLineByType(LT_ROOT, entry->lines); |
} |
|
if (line && line->numElements >= 2) |
|
|
root=line->elements[1].item; |
|
|
} |
|
2595 |
|
|
2596 |
if (root) { |
/* handle cases where the index value may shift */ |
|
char * s = alloca(strlen(root) + 1); |
|
|
|
|
|
strcpy(s, root); |
|
|
if (s[strlen(s) - 1] == '"') |
|
|
s[strlen(s) - 1] = '\0'; |
|
|
/* make sure the root doesn't have a trailing " */ |
|
|
printf("root=%s\n", s); |
|
|
} |
|
2597 |
|
|
2598 |
line = getLineByType(LT_INITRD, entry->lines); |
/* check validity of existing default or first-entry-found |
2599 |
|
selection */ |
2600 |
|
if (defaultKernelPath) { |
2601 |
|
/* we must initialize this */ |
2602 |
|
firstKernelEntryIndex = 0; |
2603 |
|
/* user requested first-entry-found */ |
2604 |
|
if (!findEntryByPath(config, defaultKernelPath, |
2605 |
|
prefix, &firstKernelEntryIndex)) { |
2606 |
|
/* don't change default if can't find match */ |
2607 |
|
config->defaultImage = NO_DEFAULT_ENTRY; |
2608 |
|
return; |
2609 |
|
} |
2610 |
|
|
2611 |
if (line && line->numElements >= 2) { |
config->defaultImage = firstKernelEntryIndex; |
|
printf("initrd=%s", prefix); |
|
|
for (i = 1; i < line->numElements; i++) |
|
|
printf("%s%s", line->elements[i].item, line->elements[i].indent); |
|
|
printf("\n"); |
|
|
} |
|
2612 |
|
|
2613 |
line = getLineByType(LT_TITLE, entry->lines); |
/* this is where we start looking for decrement later */ |
2614 |
if (line) { |
currentLookupIndex = config->defaultImage; |
|
printf("title=%s\n", line->elements[1].item); |
|
|
} else { |
|
|
char * title; |
|
|
line = getLineByType(LT_MENUENTRY, entry->lines); |
|
|
title = grub2ExtractTitle(line); |
|
|
if (title) |
|
|
printf("title=%s\n", title); |
|
|
} |
|
|
} |
|
2615 |
|
|
2616 |
int parseSysconfigGrub(int * lbaPtr, char ** bootPtr) { |
if (isAddingBootEntry && !config->isModified && |
2617 |
FILE * in; |
(newBootEntryIndex < config->defaultImage)) { |
2618 |
char buf[1024]; |
/* increment because new entry added before default */ |
2619 |
char * chptr; |
config->defaultImage++; |
2620 |
char * start; |
} |
2621 |
char * param; |
} else { |
2622 |
|
/* check to see if the default is stored in the environment */ |
2623 |
in = fopen("/etc/conf.d/grub", "r"); |
if (config->defaultImage < FIRST_ENTRY_INDEX) { |
2624 |
if (!in) return 1; |
if (config->defaultImage == DEFAULT_SAVED || config->defaultImage == DEFAULT_SAVED_GRUB2) |
2625 |
|
{ |
2626 |
if (lbaPtr) *lbaPtr = 0; |
if (config->cfi->defaultIsSaved) { |
2627 |
if (bootPtr) *bootPtr = NULL; |
if (config->cfi->getEnv) { |
2628 |
|
char *defaultTitle = config->cfi->getEnv(config->cfi, "saved_entry"); |
2629 |
while (fgets(buf, sizeof(buf), in)) { |
|
2630 |
start = buf; |
if (defaultTitle) { |
2631 |
while (isspace(*start)) start++; |
if (isnumber(defaultTitle)) { |
2632 |
if (*start == '#') continue; |
currentLookupIndex = atoi(defaultTitle); |
2633 |
|
} else { |
2634 |
chptr = strchr(start, '='); |
findEntryByTitle(config, defaultTitle, ¤tLookupIndex); |
2635 |
if (!chptr) continue; |
} |
2636 |
chptr--; |
/* set the default Image to an actual index */ |
2637 |
while (*chptr && isspace(*chptr)) chptr--; |
config->defaultImage = currentLookupIndex; |
2638 |
chptr++; |
} |
2639 |
*chptr = '\0'; |
} |
2640 |
|
} |
2641 |
param = chptr + 1; |
} |
2642 |
while (*param && isspace(*param)) param++; |
} else { |
2643 |
if (*param == '=') { |
/* use pre-existing default entry from the file*/ |
2644 |
param++; |
currentLookupIndex = config->defaultImage; |
2645 |
while (*param && isspace(*param)) param++; |
} |
|
} |
|
|
|
|
|
chptr = param; |
|
|
while (*chptr && !isspace(*chptr)) chptr++; |
|
|
*chptr = '\0'; |
|
|
|
|
|
if (!strcmp(start, "forcelba") && !strcmp(param, "1") && lbaPtr) |
|
|
*lbaPtr = 1; |
|
|
else if (!strcmp(start, "boot") && bootPtr) |
|
|
*bootPtr = strdup(param); |
|
|
} |
|
2646 |
|
|
2647 |
fclose(in); |
if (isAddingBootEntry |
2648 |
|
&& (newBootEntryIndex <= config->defaultImage)) { |
2649 |
|
config->defaultImage++; |
2650 |
|
|
2651 |
return 0; |
if (config->isModified) { |
2652 |
} |
currentLookupIndex++; |
2653 |
|
} |
2654 |
|
} |
2655 |
|
} |
2656 |
|
|
2657 |
void dumpSysconfigGrub(void) { |
/* sanity check - is this entry index valid? */ |
2658 |
char * boot; |
bootEntry = findEntryByIndex(config, currentLookupIndex); |
|
int lba; |
|
|
|
|
|
if (!parseSysconfigGrub(&lba, &boot)) { |
|
|
if (lba) printf("lba\n"); |
|
|
if (boot) printf("boot=%s\n", boot); |
|
|
} |
|
|
} |
|
2659 |
|
|
2660 |
int displayInfo(struct grubConfig * config, char * kernel, |
if ((bootEntry && bootEntry->skip) || !bootEntry) { |
2661 |
const char * prefix) { |
/* entry is to be skipped or is invalid */ |
2662 |
int i = 0; |
if (isAddingBootEntry) { |
2663 |
struct singleEntry * entry; |
config->defaultImage = newBootEntryIndex; |
2664 |
struct singleLine * line; |
return; |
2665 |
|
} |
2666 |
entry = findEntryByPath(config, kernel, prefix, &i); |
newDefault = |
2667 |
if (!entry) { |
findTemplate(config, prefix, &config->defaultImage, 1, |
2668 |
fprintf(stderr, _("grubby: kernel not found\n")); |
flags); |
2669 |
return 1; |
if (!newDefault) { |
2670 |
} |
config->defaultImage = NO_DEFAULT_ENTRY; |
2671 |
|
} |
2672 |
|
|
2673 |
/* this is a horrible hack to support /etc/conf.d/grub; there must |
return; |
|
be a better way */ |
|
|
if (config->cfi == &grubConfigType) { |
|
|
dumpSysconfigGrub(); |
|
|
} else { |
|
|
line = getLineByType(LT_BOOT, config->theLines); |
|
|
if (line && line->numElements >= 1) { |
|
|
printf("boot=%s\n", line->elements[1].item); |
|
2674 |
} |
} |
2675 |
|
|
2676 |
line = getLineByType(LT_LBA, config->theLines); |
currentLookupIndex--; |
|
if (line) printf("lba\n"); |
|
|
} |
|
2677 |
|
|
2678 |
displayEntry(entry, prefix, i); |
/* decrement index by the total number of entries deleted */ |
2679 |
|
|
2680 |
i++; |
for (indexToVerify = currentLookupIndex; |
2681 |
while ((entry = findEntryByPath(config, kernel, prefix, &i))) { |
indexToVerify >= FIRST_ENTRY_INDEX; indexToVerify--) { |
|
displayEntry(entry, prefix, i); |
|
|
i++; |
|
|
} |
|
2682 |
|
|
2683 |
return 0; |
bootEntry = findEntryByIndex(config, indexToVerify); |
|
} |
|
2684 |
|
|
2685 |
struct singleLine * addLineTmpl(struct singleEntry * entry, |
if (bootEntry && bootEntry->skip) { |
2686 |
struct singleLine * tmplLine, |
config->defaultImage--; |
2687 |
struct singleLine * prevLine, |
} |
|
const char * val, |
|
|
struct configFileInfo * cfi) |
|
|
{ |
|
|
struct singleLine * newLine = lineDup(tmplLine); |
|
|
|
|
|
if (val) { |
|
|
/* override the inherited value with our own. |
|
|
* This is a little weak because it only applies to elements[1] |
|
|
*/ |
|
|
if (newLine->numElements > 1) |
|
|
removeElement(newLine, 1); |
|
|
insertElement(newLine, val, 1, cfi); |
|
|
|
|
|
/* but try to keep the rootspec from the template... sigh */ |
|
|
if (tmplLine->type & (LT_HYPER|LT_KERNEL|LT_MBMODULE|LT_INITRD)) { |
|
|
char * rootspec = getRootSpecifier(tmplLine->elements[1].item); |
|
|
if (rootspec != NULL) { |
|
|
free(newLine->elements[1].item); |
|
|
newLine->elements[1].item = |
|
|
sdupprintf("%s%s", rootspec, val); |
|
|
} |
|
2688 |
} |
} |
|
} |
|
|
|
|
|
dbgPrintf("addLineTmpl(%s)\n", newLine->numElements ? |
|
|
newLine->elements[0].item : ""); |
|
|
|
|
|
if (!entry->lines) { |
|
|
/* first one on the list */ |
|
|
entry->lines = newLine; |
|
|
} else if (prevLine) { |
|
|
/* add after prevLine */ |
|
|
newLine->next = prevLine->next; |
|
|
prevLine->next = newLine; |
|
|
} |
|
|
|
|
|
return newLine; |
|
2689 |
} |
} |
2690 |
|
|
2691 |
/* val may be NULL */ |
void setFallbackImage(struct grubConfig *config, int hasNew) |
2692 |
struct singleLine * addLine(struct singleEntry * entry, |
{ |
2693 |
struct configFileInfo * cfi, |
struct singleEntry *entry, *entry2; |
2694 |
enum lineType_e type, char * defaultIndent, |
int j; |
|
const char * val) { |
|
|
struct singleLine * line, * prev; |
|
|
struct keywordTypes * kw; |
|
|
struct singleLine tmpl; |
|
|
|
|
|
/* NB: This function shouldn't allocate items on the heap, rather on the |
|
|
* stack since it calls addLineTmpl which will make copies. |
|
|
*/ |
|
|
|
|
|
if (type == LT_TITLE && cfi->titleBracketed) { |
|
|
/* we're doing a bracketed title (zipl) */ |
|
|
tmpl.type = type; |
|
|
tmpl.numElements = 1; |
|
|
tmpl.elements = alloca(sizeof(*tmpl.elements)); |
|
|
tmpl.elements[0].item = alloca(strlen(val)+3); |
|
|
sprintf(tmpl.elements[0].item, "[%s]", val); |
|
|
tmpl.elements[0].indent = ""; |
|
|
val = NULL; |
|
|
} else if (type == LT_MENUENTRY) { |
|
|
char *lineend = "--class gnu-linux --class gnu --class os {"; |
|
|
if (!val) { |
|
|
fprintf(stderr, "Line type LT_MENUENTRY requires a value\n"); |
|
|
abort(); |
|
|
} |
|
|
kw = getKeywordByType(type, cfi); |
|
|
if (!kw) { |
|
|
fprintf(stderr, "Looking up keyword for unknown type %d\n", type); |
|
|
abort(); |
|
|
} |
|
|
tmpl.indent = ""; |
|
|
tmpl.type = type; |
|
|
tmpl.numElements = 3; |
|
|
tmpl.elements = alloca(sizeof(*tmpl.elements) * tmpl.numElements); |
|
|
tmpl.elements[0].item = kw->key; |
|
|
tmpl.elements[0].indent = alloca(2); |
|
|
sprintf(tmpl.elements[0].indent, "%c", kw->nextChar); |
|
|
tmpl.elements[1].item = (char *)val; |
|
|
tmpl.elements[1].indent = alloca(2); |
|
|
sprintf(tmpl.elements[1].indent, "%c", kw->nextChar); |
|
|
tmpl.elements[2].item = alloca(strlen(lineend)+1); |
|
|
strcpy(tmpl.elements[2].item, lineend); |
|
|
tmpl.elements[2].indent = ""; |
|
|
} else { |
|
|
kw = getKeywordByType(type, cfi); |
|
|
if (!kw) { |
|
|
fprintf(stderr, "Looking up keyword for unknown type %d\n", type); |
|
|
abort(); |
|
|
} |
|
|
tmpl.type = type; |
|
|
tmpl.numElements = val ? 2 : 1; |
|
|
tmpl.elements = alloca(sizeof(*tmpl.elements) * tmpl.numElements); |
|
|
tmpl.elements[0].item = kw->key; |
|
|
tmpl.elements[0].indent = alloca(2); |
|
|
sprintf(tmpl.elements[0].indent, "%c", kw->nextChar); |
|
|
if (val) { |
|
|
tmpl.elements[1].item = (char *)val; |
|
|
tmpl.elements[1].indent = ""; |
|
|
} |
|
|
} |
|
|
|
|
|
/* The last non-empty line gives us the indention to us and the line |
|
|
to insert after. Note that comments are considered empty lines, which |
|
|
may not be ideal? If there are no lines or we are looking at the |
|
|
first line, we use defaultIndent (the first line is normally indented |
|
|
differently from the rest) */ |
|
|
for (line = entry->lines, prev = NULL; line; line = line->next) { |
|
|
if (line->numElements) prev = line; |
|
|
/* fall back on the last line if prev isn't otherwise set */ |
|
|
if (!line->next && !prev) prev = line; |
|
|
} |
|
|
|
|
|
struct singleLine *menuEntry; |
|
|
menuEntry = getLineByType(LT_MENUENTRY, entry->lines); |
|
|
if (tmpl.type == LT_ENTRY_END) { |
|
|
if (menuEntry) |
|
|
tmpl.indent = menuEntry->indent; |
|
|
else |
|
|
tmpl.indent = defaultIndent ?: ""; |
|
|
} else if (tmpl.type != LT_MENUENTRY) { |
|
|
if (menuEntry) |
|
|
tmpl.indent = "\t"; |
|
|
else if (prev == entry->lines) |
|
|
tmpl.indent = defaultIndent ?: ""; |
|
|
else |
|
|
tmpl.indent = prev->indent; |
|
|
} |
|
|
|
|
|
return addLineTmpl(entry, &tmpl, prev, val, cfi); |
|
|
} |
|
2695 |
|
|
2696 |
void removeLine(struct singleEntry * entry, struct singleLine * line) { |
if (config->fallbackImage == -1) |
2697 |
struct singleLine * prev; |
return; |
|
int i; |
|
2698 |
|
|
2699 |
for (i = 0; i < line->numElements; i++) { |
entry = findEntryByIndex(config, config->fallbackImage); |
2700 |
free(line->elements[i].item); |
if (!entry || entry->skip) { |
2701 |
free(line->elements[i].indent); |
config->fallbackImage = -1; |
2702 |
} |
return; |
2703 |
free(line->elements); |
} |
|
free(line->indent); |
|
2704 |
|
|
2705 |
if (line == entry->lines) { |
if (hasNew) |
2706 |
entry->lines = line->next; |
config->fallbackImage++; |
|
} else { |
|
|
prev = entry->lines; |
|
|
while (prev->next != line) prev = prev->next; |
|
|
prev->next = line->next; |
|
|
} |
|
2707 |
|
|
2708 |
free(line); |
/* count the number of entries erased before this one */ |
2709 |
|
for (j = 0; j < config->fallbackImage; j++) { |
2710 |
|
entry2 = findEntryByIndex(config, j); |
2711 |
|
if (entry2->skip) |
2712 |
|
config->fallbackImage--; |
2713 |
|
} |
2714 |
} |
} |
2715 |
|
|
2716 |
static void requote(struct singleLine *tmplLine, struct configFileInfo * cfi) |
void displayEntry(struct grubConfig *config, struct singleEntry *entry, const char *prefix, int index) |
2717 |
{ |
{ |
2718 |
struct singleLine newLine = { |
struct singleLine *line; |
2719 |
.indent = tmplLine->indent, |
char *root = NULL; |
2720 |
.type = tmplLine->type, |
int i; |
2721 |
.next = tmplLine->next, |
int j; |
|
}; |
|
|
int firstQuotedItem = -1; |
|
|
int quoteLen = 0; |
|
|
int j; |
|
|
int element = 0; |
|
|
char *c; |
|
|
|
|
|
c = malloc(strlen(tmplLine->elements[0].item) + 1); |
|
|
strcpy(c, tmplLine->elements[0].item); |
|
|
insertElement(&newLine, c, element++, cfi); |
|
|
free(c); |
|
|
c = NULL; |
|
|
|
|
|
for (j = 1; j < tmplLine->numElements; j++) { |
|
|
if (firstQuotedItem == -1) { |
|
|
quoteLen += strlen(tmplLine->elements[j].item); |
|
|
|
|
|
if (isquote(tmplLine->elements[j].item[0])) { |
|
|
firstQuotedItem = j; |
|
|
quoteLen += strlen(tmplLine->elements[j].indent); |
|
|
} else { |
|
|
c = malloc(quoteLen + 1); |
|
|
strcpy(c, tmplLine->elements[j].item); |
|
|
insertElement(&newLine, c, element++, cfi); |
|
|
free(c); |
|
|
quoteLen = 0; |
|
|
} |
|
|
} else { |
|
|
int itemlen = strlen(tmplLine->elements[j].item); |
|
|
quoteLen += itemlen; |
|
|
quoteLen += strlen(tmplLine->elements[j].indent); |
|
|
|
|
|
if (isquote(tmplLine->elements[j].item[itemlen - 1])) { |
|
|
c = malloc(quoteLen + 1); |
|
|
c[0] = '\0'; |
|
|
for (int i = firstQuotedItem; i < j+1; i++) { |
|
|
strcat(c, tmplLine->elements[i].item); |
|
|
strcat(c, tmplLine->elements[i].indent); |
|
|
} |
|
|
insertElement(&newLine, c, element++, cfi); |
|
|
free(c); |
|
|
|
|
|
firstQuotedItem = -1; |
|
|
quoteLen = 0; |
|
|
} |
|
|
} |
|
|
} |
|
|
while (tmplLine->numElements) |
|
|
removeElement(tmplLine, 0); |
|
|
if (tmplLine->elements) |
|
|
free(tmplLine->elements); |
|
|
|
|
|
tmplLine->numElements = newLine.numElements; |
|
|
tmplLine->elements = newLine.elements; |
|
|
} |
|
|
|
|
|
static void insertElement(struct singleLine * line, |
|
|
const char * item, int insertHere, |
|
|
struct configFileInfo * cfi) |
|
|
{ |
|
|
struct keywordTypes * kw; |
|
|
char indent[2] = ""; |
|
|
|
|
|
/* sanity check */ |
|
|
if (insertHere > line->numElements) { |
|
|
dbgPrintf("insertElement() adjusting insertHere from %d to %d\n", |
|
|
insertHere, line->numElements); |
|
|
insertHere = line->numElements; |
|
|
} |
|
|
|
|
|
line->elements = realloc(line->elements, (line->numElements + 1) * |
|
|
sizeof(*line->elements)); |
|
|
memmove(&line->elements[insertHere+1], |
|
|
&line->elements[insertHere], |
|
|
(line->numElements - insertHere) * |
|
|
sizeof(*line->elements)); |
|
|
line->elements[insertHere].item = strdup(item); |
|
|
|
|
|
kw = getKeywordByType(line->type, cfi); |
|
|
|
|
|
if (line->numElements == 0) { |
|
|
indent[0] = '\0'; |
|
|
} else if (insertHere == 0) { |
|
|
indent[0] = kw->nextChar; |
|
|
} else if (kw->separatorChar != '\0') { |
|
|
indent[0] = kw->separatorChar; |
|
|
} else { |
|
|
indent[0] = ' '; |
|
|
} |
|
|
|
|
|
if (insertHere > 0 && line->elements[insertHere-1].indent[0] == '\0') { |
|
|
/* move the end-of-line forward */ |
|
|
line->elements[insertHere].indent = |
|
|
line->elements[insertHere-1].indent; |
|
|
line->elements[insertHere-1].indent = strdup(indent); |
|
|
} else { |
|
|
line->elements[insertHere].indent = strdup(indent); |
|
|
} |
|
2722 |
|
|
2723 |
line->numElements++; |
printf("index=%d\n", index); |
2724 |
|
|
2725 |
dbgPrintf("insertElement(%s, '%s%s', %d)\n", |
line = |
2726 |
line->elements[0].item, |
getLineByType(LT_KERNEL | LT_HYPER | LT_KERNEL_EFI | LT_KERNEL_16, |
2727 |
line->elements[insertHere].item, |
entry->lines); |
2728 |
line->elements[insertHere].indent, |
if (!line) { |
2729 |
insertHere); |
printf("non linux entry\n"); |
2730 |
} |
return; |
2731 |
|
} |
2732 |
|
|
2733 |
static void removeElement(struct singleLine * line, int removeHere) { |
if (!strncmp(prefix, line->elements[1].item, strlen(prefix))) |
2734 |
int i; |
printf("kernel=%s\n", line->elements[1].item); |
2735 |
|
else |
2736 |
|
printf("kernel=%s%s\n", prefix, line->elements[1].item); |
2737 |
|
|
2738 |
/* sanity check */ |
if (line->numElements >= 3) { |
2739 |
if (removeHere >= line->numElements) return; |
printf("args=\""); |
2740 |
|
i = 2; |
2741 |
|
while (i < line->numElements) { |
2742 |
|
if (!strncmp(line->elements[i].item, "root=", 5)) { |
2743 |
|
root = line->elements[i].item + 5; |
2744 |
|
} else { |
2745 |
|
printf("%s%s", line->elements[i].item, |
2746 |
|
line->elements[i].indent); |
2747 |
|
} |
2748 |
|
|
2749 |
dbgPrintf("removeElement(%s, %d:%s)\n", line->elements[0].item, |
i++; |
2750 |
removeHere, line->elements[removeHere].item); |
} |
2751 |
|
printf("\"\n"); |
2752 |
|
} else { |
2753 |
|
line = getLineByType(LT_KERNELARGS, entry->lines); |
2754 |
|
if (line) { |
2755 |
|
char *s; |
2756 |
|
|
2757 |
|
printf("args=\""); |
2758 |
|
i = 1; |
2759 |
|
while (i < line->numElements) { |
2760 |
|
if (!strncmp |
2761 |
|
(line->elements[i].item, "root=", 5)) { |
2762 |
|
root = line->elements[i].item + 5; |
2763 |
|
} else { |
2764 |
|
s = line->elements[i].item; |
2765 |
|
|
2766 |
free(line->elements[removeHere].item); |
printf("%s%s", s, |
2767 |
|
line->elements[i].indent); |
2768 |
|
} |
2769 |
|
|
2770 |
if (removeHere > 1) { |
i++; |
2771 |
/* previous argument gets this argument's post-indentation */ |
} |
|
free(line->elements[removeHere-1].indent); |
|
|
line->elements[removeHere-1].indent = |
|
|
line->elements[removeHere].indent; |
|
|
} else { |
|
|
free(line->elements[removeHere].indent); |
|
|
} |
|
2772 |
|
|
2773 |
/* now collapse the array, but don't bother to realloc smaller */ |
s = line->elements[i - 1].indent; |
2774 |
for (i = removeHere; i < line->numElements - 1; i++) |
printf("\"\n"); |
2775 |
line->elements[i] = line->elements[i + 1]; |
} |
|
|
|
|
line->numElements--; |
|
|
} |
|
|
|
|
|
int argMatch(const char * one, const char * two) { |
|
|
char * first, * second; |
|
|
char * chptr; |
|
|
|
|
|
first = strcpy(alloca(strlen(one) + 1), one); |
|
|
second = strcpy(alloca(strlen(two) + 1), two); |
|
|
|
|
|
chptr = strchr(first, '='); |
|
|
if (chptr) *chptr = '\0'; |
|
|
|
|
|
chptr = strchr(second, '='); |
|
|
if (chptr) *chptr = '\0'; |
|
|
|
|
|
return strcmp(first, second); |
|
|
} |
|
|
|
|
|
int updateActualImage(struct grubConfig * cfg, const char * image, |
|
|
const char * prefix, const char * addArgs, |
|
|
const char * removeArgs, int multibootArgs) { |
|
|
struct singleEntry * entry; |
|
|
struct singleLine * line, * rootLine; |
|
|
int index = 0; |
|
|
int i, k; |
|
|
const char ** newArgs, ** oldArgs; |
|
|
const char ** arg; |
|
|
int useKernelArgs, useRoot; |
|
|
int firstElement; |
|
|
int *usedElements; |
|
|
int doreplace; |
|
|
|
|
|
if (!image) return 0; |
|
|
|
|
|
if (!addArgs) { |
|
|
newArgs = malloc(sizeof(*newArgs)); |
|
|
*newArgs = NULL; |
|
|
} else { |
|
|
if (poptParseArgvString(addArgs, NULL, &newArgs)) { |
|
|
fprintf(stderr, |
|
|
_("grubby: error separating arguments '%s'\n"), addArgs); |
|
|
return 1; |
|
2776 |
} |
} |
|
} |
|
2777 |
|
|
2778 |
if (!removeArgs) { |
if (!root) { |
2779 |
oldArgs = malloc(sizeof(*oldArgs)); |
line = getLineByType(LT_ROOT, entry->lines); |
2780 |
*oldArgs = NULL; |
if (line && line->numElements >= 2) |
2781 |
} else { |
root = line->elements[1].item; |
|
if (poptParseArgvString(removeArgs, NULL, &oldArgs)) { |
|
|
fprintf(stderr, |
|
|
_("grubby: error separating arguments '%s'\n"), removeArgs); |
|
|
free(newArgs); |
|
|
return 1; |
|
2782 |
} |
} |
|
} |
|
2783 |
|
|
2784 |
|
if (root) { |
2785 |
|
char *s = alloca(strlen(root) + 1); |
2786 |
|
|
2787 |
useKernelArgs = (getKeywordByType(LT_KERNELARGS, cfg->cfi) |
strcpy(s, root); |
2788 |
&& (!multibootArgs || cfg->cfi->mbConcatArgs)); |
if (s[strlen(s) - 1] == '"') |
2789 |
|
s[strlen(s) - 1] = '\0'; |
2790 |
useRoot = (getKeywordByType(LT_ROOT, cfg->cfi) |
/* make sure the root doesn't have a trailing " */ |
2791 |
&& !multibootArgs); |
printf("root=%s\n", s); |
2792 |
|
} |
2793 |
|
|
2794 |
for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) { |
line = |
2795 |
|
getLineByType(LT_INITRD | LT_INITRD_EFI | LT_INITRD_16, |
2796 |
|
entry->lines); |
2797 |
|
|
2798 |
if (multibootArgs && !entry->multiboot) |
if (line && line->numElements >= 2) { |
2799 |
continue; |
if (!strncmp(prefix, line->elements[1].item, strlen(prefix))) |
2800 |
|
printf("initrd="); |
2801 |
|
else |
2802 |
|
printf("initrd=%s", prefix); |
2803 |
|
|
2804 |
/* Determine where to put the args. If this config supports |
for (i = 1; i < line->numElements; i++) |
2805 |
* LT_KERNELARGS, use that. Otherwise use |
printf("%s%s", line->elements[i].item, |
2806 |
* LT_HYPER/LT_KERNEL/LT_MBMODULE lines. |
line->elements[i].indent); |
2807 |
*/ |
printf("\n"); |
|
if (useKernelArgs) { |
|
|
line = getLineByType(LT_KERNELARGS, entry->lines); |
|
|
if (!line) { |
|
|
/* no LT_KERNELARGS, need to add it */ |
|
|
line = addLine(entry, cfg->cfi, LT_KERNELARGS, |
|
|
cfg->secondaryIndent, NULL); |
|
|
} |
|
|
firstElement = 1; |
|
|
|
|
|
} else if (multibootArgs) { |
|
|
line = getLineByType(LT_HYPER, entry->lines); |
|
|
if (!line) { |
|
|
/* a multiboot entry without LT_HYPER? */ |
|
|
continue; |
|
|
} |
|
|
firstElement = 2; |
|
|
|
|
|
} else { |
|
|
line = getLineByType(LT_KERNEL|LT_MBMODULE, entry->lines); |
|
|
if (!line) { |
|
|
/* no LT_KERNEL or LT_MBMODULE in this entry? */ |
|
|
continue; |
|
|
} |
|
|
firstElement = 2; |
|
2808 |
} |
} |
2809 |
|
|
2810 |
/* handle the elilo case which does: |
line = getLineByType(LT_TITLE, entry->lines); |
2811 |
* append="hypervisor args -- kernel args" |
if (line) { |
2812 |
*/ |
char *entryTitle; |
2813 |
if (entry->multiboot && cfg->cfi->mbConcatArgs) { |
/* if we can extractTitle, then it's a zipl config and |
2814 |
/* this is a multiboot entry, make sure there's |
* if not then we go ahead with what's existed prior */ |
2815 |
* -- on the args line |
entryTitle = extractTitle(config, line); |
2816 |
*/ |
if (!entryTitle) { |
2817 |
for (i = firstElement; i < line->numElements; i++) { |
entryTitle=line->elements[1].item; |
|
if (!strcmp(line->elements[i].item, "--")) |
|
|
break; |
|
|
} |
|
|
if (i == line->numElements) { |
|
|
/* assume all existing args are kernel args, |
|
|
* prepend -- to make it official |
|
|
*/ |
|
|
insertElement(line, "--", firstElement, cfg->cfi); |
|
|
i = firstElement; |
|
|
} |
|
|
if (!multibootArgs) { |
|
|
/* kernel args start after the -- */ |
|
|
firstElement = i + 1; |
|
|
} |
|
|
} else if (cfg->cfi->mbConcatArgs) { |
|
|
/* this is a non-multiboot entry, remove hyper args */ |
|
|
for (i = firstElement; i < line->numElements; i++) { |
|
|
if (!strcmp(line->elements[i].item, "--")) |
|
|
break; |
|
|
} |
|
|
if (i < line->numElements) { |
|
|
/* remove args up to -- */ |
|
|
while (strcmp(line->elements[firstElement].item, "--")) |
|
|
removeElement(line, firstElement); |
|
|
/* remove -- */ |
|
|
removeElement(line, firstElement); |
|
|
} |
|
|
} |
|
|
|
|
|
usedElements = calloc(line->numElements, sizeof(*usedElements)); |
|
|
|
|
|
for (k = 0, arg = newArgs; *arg; arg++, k++) { |
|
|
|
|
|
doreplace = 1; |
|
|
for (i = firstElement; i < line->numElements; i++) { |
|
|
if (multibootArgs && cfg->cfi->mbConcatArgs && |
|
|
!strcmp(line->elements[i].item, "--")) |
|
|
{ |
|
|
/* reached the end of hyper args, insert here */ |
|
|
doreplace = 0; |
|
|
break; |
|
|
} |
|
|
if (usedElements[i]) |
|
|
continue; |
|
|
if (!argMatch(line->elements[i].item, *arg)) { |
|
|
usedElements[i]=1; |
|
|
break; |
|
2818 |
} |
} |
2819 |
} |
printf("title=%s\n", entryTitle); |
2820 |
|
} else { |
2821 |
if (i < line->numElements && doreplace) { |
char *title; |
2822 |
/* direct replacement */ |
line = getLineByType(LT_MENUENTRY, entry->lines); |
2823 |
free(line->elements[i].item); |
if (line) { |
2824 |
line->elements[i].item = strdup(*arg); |
title = grub2ExtractTitle(line); |
2825 |
|
if (title) |
2826 |
} else if (useRoot && !strncmp(*arg, "root=/dev/", 10)) { |
printf("title=%s\n", title); |
|
/* root= replacement */ |
|
|
rootLine = getLineByType(LT_ROOT, entry->lines); |
|
|
if (rootLine) { |
|
|
free(rootLine->elements[1].item); |
|
|
rootLine->elements[1].item = strdup(*arg + 5); |
|
|
} else { |
|
|
rootLine = addLine(entry, cfg->cfi, LT_ROOT, |
|
|
cfg->secondaryIndent, *arg + 5); |
|
2827 |
} |
} |
2828 |
} |
} |
2829 |
|
|
2830 |
else { |
for (j = 0, line = entry->lines; line; line = line->next) { |
2831 |
/* insert/append */ |
if ((line->type & LT_MBMODULE) && line->numElements >= 2) { |
2832 |
insertElement(line, *arg, i, cfg->cfi); |
if (!strncmp |
2833 |
usedElements = realloc(usedElements, line->numElements * |
(prefix, line->elements[1].item, strlen(prefix))) |
2834 |
sizeof(*usedElements)); |
printf("mbmodule%d=", j); |
2835 |
memmove(&usedElements[i + 1], &usedElements[i], |
else |
2836 |
line->numElements - i - 1); |
printf("mbmodule%d=%s", j, prefix); |
2837 |
usedElements[i] = 1; |
|
2838 |
|
for (i = 1; i < line->numElements; i++) |
2839 |
/* if we updated a root= here even though there is a |
printf("%s%s", line->elements[i].item, |
2840 |
LT_ROOT available we need to remove the LT_ROOT entry |
line->elements[i].indent); |
2841 |
(this will happen if we switch from a device to a label) */ |
printf("\n"); |
2842 |
if (useRoot && !strncmp(*arg, "root=", 5)) { |
j++; |
2843 |
rootLine = getLineByType(LT_ROOT, entry->lines); |
} |
|
if (rootLine) |
|
|
removeLine(entry, rootLine); |
|
|
} |
|
|
} |
|
|
} |
|
|
|
|
|
free(usedElements); |
|
|
|
|
|
for (arg = oldArgs; *arg; arg++) { |
|
|
for (i = firstElement; i < line->numElements; i++) { |
|
|
if (multibootArgs && cfg->cfi->mbConcatArgs && |
|
|
!strcmp(line->elements[i].item, "--")) |
|
|
/* reached the end of hyper args, stop here */ |
|
|
break; |
|
|
if (!argMatch(line->elements[i].item, *arg)) { |
|
|
removeElement(line, i); |
|
|
break; |
|
|
} |
|
|
} |
|
|
/* handle removing LT_ROOT line too */ |
|
|
if (useRoot && !strncmp(*arg, "root=", 5)) { |
|
|
rootLine = getLineByType(LT_ROOT, entry->lines); |
|
|
if (rootLine) |
|
|
removeLine(entry, rootLine); |
|
|
} |
|
|
} |
|
|
|
|
|
if (line->numElements == 1) { |
|
|
/* don't need the line at all (note it has to be a |
|
|
LT_KERNELARGS for this to happen */ |
|
|
removeLine(entry, line); |
|
2844 |
} |
} |
2845 |
} |
} |
2846 |
|
|
2847 |
free(newArgs); |
int isSuseSystem(void) |
2848 |
free(oldArgs); |
{ |
2849 |
|
const char *path; |
2850 |
|
const static char default_path[] = "/etc/SuSE-release"; |
2851 |
|
|
2852 |
return 0; |
if ((path = getenv("GRUBBY_SUSE_RELEASE")) == NULL) |
2853 |
} |
path = default_path; |
2854 |
|
|
2855 |
int updateImage(struct grubConfig * cfg, const char * image, |
if (!access(path, R_OK)) |
|
const char * prefix, const char * addArgs, |
|
|
const char * removeArgs, |
|
|
const char * addMBArgs, const char * removeMBArgs) { |
|
|
int rc = 0; |
|
|
|
|
|
if (!image) return rc; |
|
|
|
|
|
/* update the main args first... */ |
|
|
if (addArgs || removeArgs) |
|
|
rc = updateActualImage(cfg, image, prefix, addArgs, removeArgs, 0); |
|
|
if (rc) return rc; |
|
|
|
|
|
/* and now any multiboot args */ |
|
|
if (addMBArgs || removeMBArgs) |
|
|
rc = updateActualImage(cfg, image, prefix, addMBArgs, removeMBArgs, 1); |
|
|
return rc; |
|
|
} |
|
|
|
|
|
int updateInitrd(struct grubConfig * cfg, const char * image, |
|
|
const char * prefix, const char * initrd) { |
|
|
struct singleEntry * entry; |
|
|
struct singleLine * line, * kernelLine, *endLine = NULL; |
|
|
int index = 0; |
|
|
|
|
|
if (!image) return 0; |
|
|
|
|
|
for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) { |
|
|
kernelLine = getLineByType(LT_KERNEL, entry->lines); |
|
|
if (!kernelLine) continue; |
|
|
|
|
|
line = getLineByType(LT_INITRD, entry->lines); |
|
|
if (line) |
|
|
removeLine(entry, line); |
|
|
if (prefix) { |
|
|
int prefixLen = strlen(prefix); |
|
|
if (!strncmp(initrd, prefix, prefixLen)) |
|
|
initrd += prefixLen; |
|
|
} |
|
|
endLine = getLineByType(LT_ENTRY_END, entry->lines); |
|
|
if (endLine) |
|
|
removeLine(entry, endLine); |
|
|
line = addLine(entry, cfg->cfi, LT_INITRD, kernelLine->indent, initrd); |
|
|
if (!line) |
|
|
return 1; |
|
|
if (endLine) { |
|
|
line = addLine(entry, cfg->cfi, LT_ENTRY_END, "", NULL); |
|
|
if (!line) |
|
2856 |
return 1; |
return 1; |
2857 |
|
return 0; |
2858 |
|
} |
2859 |
|
|
2860 |
|
int isSuseGrubConf(const char *path) |
2861 |
|
{ |
2862 |
|
FILE *grubConf; |
2863 |
|
char *line = NULL; |
2864 |
|
size_t len = 0, res = 0; |
2865 |
|
|
2866 |
|
grubConf = fopen(path, "r"); |
2867 |
|
if (!grubConf) { |
2868 |
|
dbgPrintf("Could not open SuSE configuration file '%s'\n", |
2869 |
|
path); |
2870 |
|
return 0; |
2871 |
} |
} |
2872 |
|
|
2873 |
break; |
while ((res = getline(&line, &len, grubConf)) != -1) { |
2874 |
} |
if (!strncmp(line, "setup", 5)) { |
2875 |
|
fclose(grubConf); |
2876 |
|
free(line); |
2877 |
|
return 1; |
2878 |
|
} |
2879 |
|
} |
2880 |
|
|
2881 |
return 0; |
dbgPrintf("SuSE configuration file '%s' does not appear to be valid\n", |
2882 |
|
path); |
2883 |
|
|
2884 |
|
fclose(grubConf); |
2885 |
|
free(line); |
2886 |
|
return 0; |
2887 |
} |
} |
2888 |
|
|
2889 |
int checkDeviceBootloader(const char * device, const unsigned char * boot) { |
int suseGrubConfGetLba(const char *path, int *lbaPtr) |
2890 |
int fd; |
{ |
2891 |
unsigned char bootSect[512]; |
FILE *grubConf; |
2892 |
int offset; |
char *line = NULL; |
2893 |
|
size_t res = 0, len = 0; |
|
fd = open(device, O_RDONLY); |
|
|
if (fd < 0) { |
|
|
fprintf(stderr, _("grubby: unable to open %s: %s\n"), |
|
|
device, strerror(errno)); |
|
|
return 1; |
|
|
} |
|
2894 |
|
|
2895 |
if (read(fd, bootSect, 512) != 512) { |
if (!path) |
2896 |
fprintf(stderr, _("grubby: unable to read %s: %s\n"), |
return 1; |
2897 |
device, strerror(errno)); |
if (!lbaPtr) |
2898 |
return 1; |
return 1; |
|
} |
|
|
close(fd); |
|
2899 |
|
|
2900 |
/* first three bytes should match, a jmp short should be in there */ |
grubConf = fopen(path, "r"); |
2901 |
if (memcmp(boot, bootSect, 3)) |
if (!grubConf) |
2902 |
return 0; |
return 1; |
2903 |
|
|
2904 |
if (boot[1] == JMP_SHORT_OPCODE) { |
while ((res = getline(&line, &len, grubConf)) != -1) { |
2905 |
offset = boot[2] + 2; |
if (line[res - 1] == '\n') |
2906 |
} else if (boot[1] == 0xe8 || boot[1] == 0xe9) { |
line[res - 1] = '\0'; |
2907 |
offset = (boot[3] << 8) + boot[2] + 2; |
else if (len > res) |
2908 |
} else if (boot[0] == JMP_SHORT_OPCODE) { |
line[res] = '\0'; |
2909 |
offset = boot[1] + 2; |
else { |
2910 |
/* |
line = realloc(line, res + 1); |
2911 |
* it looks like grub, when copying stage1 into the mbr, patches stage1 |
line[res] = '\0'; |
2912 |
* right after the JMP location, replacing other instructions such as |
} |
|
* JMPs for NOOPs. So, relax the check a little bit by skipping those |
|
|
* different bytes. |
|
|
*/ |
|
|
if ((bootSect[offset + 1] == NOOP_OPCODE) |
|
|
&& (bootSect[offset + 2] == NOOP_OPCODE)) { |
|
|
offset = offset + 3; |
|
|
} |
|
|
} else if (boot[0] == 0xe8 || boot[0] == 0xe9) { |
|
|
offset = (boot[2] << 8) + boot[1] + 2; |
|
|
} else { |
|
|
return 0; |
|
|
} |
|
2913 |
|
|
2914 |
if (memcmp(boot + offset, bootSect + offset, CODE_SEG_SIZE)) |
if (!strncmp(line, "setup", 5)) { |
2915 |
return 0; |
if (strstr(line, "--force-lba")) { |
2916 |
|
*lbaPtr = 1; |
2917 |
|
} else { |
2918 |
|
*lbaPtr = 0; |
2919 |
|
} |
2920 |
|
dbgPrintf("lba: %i\n", *lbaPtr); |
2921 |
|
break; |
2922 |
|
} |
2923 |
|
} |
2924 |
|
|
2925 |
return 2; |
free(line); |
2926 |
|
fclose(grubConf); |
2927 |
|
return 0; |
2928 |
} |
} |
2929 |
|
|
2930 |
int checkLiloOnRaid(char * mdDev, const unsigned char * boot) { |
int suseGrubConfGetInstallDevice(const char *path, char **devicePtr) |
2931 |
int fd; |
{ |
2932 |
char buf[65536]; |
FILE *grubConf; |
2933 |
char * end; |
char *line = NULL; |
2934 |
char * chptr; |
size_t res = 0, len = 0; |
2935 |
char * chptr2; |
char *lastParamPtr = NULL; |
2936 |
int rc; |
char *secLastParamPtr = NULL; |
2937 |
|
char installDeviceNumber = '\0'; |
2938 |
/* it's on raid; we need to parse /proc/mdstat and check all of the |
char *bounds = NULL; |
|
*raw* devices listed in there */ |
|
|
|
|
|
if (!strncmp(mdDev, "/dev/", 5)) |
|
|
mdDev += 5; |
|
2939 |
|
|
2940 |
if ((fd = open("/proc/mdstat", O_RDONLY)) < 0) { |
if (!path) |
2941 |
fprintf(stderr, _("grubby: failed to open /proc/mdstat: %s\n"), |
return 1; |
2942 |
strerror(errno)); |
if (!devicePtr) |
2943 |
return 2; |
return 1; |
|
} |
|
2944 |
|
|
2945 |
rc = read(fd, buf, sizeof(buf) - 1); |
grubConf = fopen(path, "r"); |
2946 |
if (rc < 0 || rc == (sizeof(buf) - 1)) { |
if (!grubConf) |
2947 |
fprintf(stderr, _("grubby: failed to read /proc/mdstat: %s\n"), |
return 1; |
|
strerror(errno)); |
|
|
close(fd); |
|
|
return 2; |
|
|
} |
|
|
close(fd); |
|
|
buf[rc] = '\0'; |
|
2948 |
|
|
2949 |
chptr = buf; |
while ((res = getline(&line, &len, grubConf)) != -1) { |
2950 |
while (*chptr) { |
if (strncmp(line, "setup", 5)) |
2951 |
end = strchr(chptr, '\n'); |
continue; |
2952 |
if (!end) break; |
|
2953 |
*end = '\0'; |
if (line[res - 1] == '\n') |
2954 |
|
line[res - 1] = '\0'; |
2955 |
|
else if (len > res) |
2956 |
|
line[res] = '\0'; |
2957 |
|
else { |
2958 |
|
line = realloc(line, res + 1); |
2959 |
|
line[res] = '\0'; |
2960 |
|
} |
2961 |
|
|
2962 |
if (!strncmp(chptr, mdDev, strlen(mdDev)) && |
lastParamPtr = bounds = line + res; |
|
chptr[strlen(mdDev)] == ' ') { |
|
2963 |
|
|
2964 |
/* found the device */ |
/* Last parameter in grub may be an optional IMAGE_DEVICE */ |
2965 |
while (*chptr && *chptr != ':') chptr++; |
while (!isspace(*lastParamPtr)) |
2966 |
chptr++; |
lastParamPtr--; |
2967 |
while (*chptr && isspace(*chptr)) chptr++; |
lastParamPtr++; |
2968 |
|
|
2969 |
/* skip the "active" bit */ |
secLastParamPtr = lastParamPtr - 2; |
2970 |
while (*chptr && !isspace(*chptr)) chptr++; |
dbgPrintf("lastParamPtr: %s\n", lastParamPtr); |
2971 |
while (*chptr && isspace(*chptr)) chptr++; |
|
2972 |
|
if (lastParamPtr + 3 > bounds) { |
2973 |
/* skip the raid level */ |
dbgPrintf("lastParamPtr going over boundary"); |
2974 |
while (*chptr && !isspace(*chptr)) chptr++; |
fclose(grubConf); |
2975 |
while (*chptr && isspace(*chptr)) chptr++; |
free(line); |
2976 |
|
return 1; |
2977 |
/* everything else is partition stuff */ |
} |
2978 |
while (*chptr) { |
if (!strncmp(lastParamPtr, "(hd", 3)) |
2979 |
chptr2 = chptr; |
lastParamPtr += 3; |
2980 |
while (*chptr2 && *chptr2 != '[') chptr2++; |
dbgPrintf("lastParamPtr: %c\n", *lastParamPtr); |
2981 |
if (!*chptr2) break; |
|
2982 |
|
/* |
2983 |
/* yank off the numbers at the end */ |
* Second last parameter will decide wether last parameter is |
2984 |
chptr2--; |
* an IMAGE_DEVICE or INSTALL_DEVICE |
2985 |
while (isdigit(*chptr2) && chptr2 > chptr) chptr2--; |
*/ |
2986 |
chptr2++; |
while (!isspace(*secLastParamPtr)) |
2987 |
*chptr2 = '\0'; |
secLastParamPtr--; |
2988 |
|
secLastParamPtr++; |
2989 |
/* Better, now we need the /dev/ back. We're done with |
|
2990 |
* everything before this point, so we can just put |
if (secLastParamPtr + 3 > bounds) { |
2991 |
* the /dev/ part there. There will always be room. */ |
dbgPrintf("secLastParamPtr going over boundary"); |
2992 |
memcpy(chptr - 5, "/dev/", 5); |
fclose(grubConf); |
2993 |
rc = checkDeviceBootloader(chptr - 5, boot); |
free(line); |
2994 |
if (rc != 2) { |
return 1; |
2995 |
return rc; |
} |
2996 |
} |
dbgPrintf("secLastParamPtr: %s\n", secLastParamPtr); |
2997 |
|
if (!strncmp(secLastParamPtr, "(hd", 3)) { |
2998 |
chptr = chptr2 + 1; |
secLastParamPtr += 3; |
2999 |
/* skip the [11] bit */ |
dbgPrintf("secLastParamPtr: %c\n", *secLastParamPtr); |
3000 |
while (*chptr && !isspace(*chptr)) chptr++; |
installDeviceNumber = *secLastParamPtr; |
3001 |
/* and move to the next one */ |
} else { |
3002 |
while (*chptr && isspace(*chptr)) chptr++; |
installDeviceNumber = *lastParamPtr; |
3003 |
} |
} |
3004 |
|
|
3005 |
/* we're good to go */ |
*devicePtr = malloc(6); |
3006 |
return 2; |
snprintf(*devicePtr, 6, "(hd%c)", installDeviceNumber); |
3007 |
|
dbgPrintf("installDeviceNumber: %c\n", installDeviceNumber); |
3008 |
|
fclose(grubConf); |
3009 |
|
free(line); |
3010 |
|
return 0; |
3011 |
} |
} |
3012 |
|
|
3013 |
chptr = end + 1; |
free(line); |
3014 |
} |
fclose(grubConf); |
3015 |
|
return 1; |
|
fprintf(stderr, |
|
|
_("grubby: raid device /dev/%s not found in /proc/mdstat\n"), |
|
|
mdDev); |
|
|
return 0; |
|
3016 |
} |
} |
3017 |
|
|
3018 |
int checkForLilo(struct grubConfig * config) { |
int grubGetBootFromDeviceMap(const char *device, char **bootPtr) |
3019 |
int fd; |
{ |
3020 |
unsigned char boot[512]; |
FILE *deviceMap; |
3021 |
struct singleLine * line; |
char *line = NULL; |
3022 |
|
size_t res = 0, len = 0; |
3023 |
for (line = config->theLines; line; line = line->next) |
char *devicePtr; |
3024 |
if (line->type == LT_BOOT) break; |
char *bounds = NULL; |
3025 |
|
const char *path; |
3026 |
if (!line) { |
const static char default_path[] = "/boot/grub/device.map"; |
|
fprintf(stderr, |
|
|
_("grubby: no boot line found in lilo configuration\n")); |
|
|
return 1; |
|
|
} |
|
3027 |
|
|
3028 |
if (line->numElements != 2) return 1; |
if (!device) |
3029 |
|
return 1; |
3030 |
|
if (!bootPtr) |
3031 |
|
return 1; |
3032 |
|
|
3033 |
fd = open("/boot/boot.b", O_RDONLY); |
if ((path = getenv("GRUBBY_GRUB_DEVICE_MAP")) == NULL) |
3034 |
if (fd < 0) { |
path = default_path; |
|
fprintf(stderr, _("grubby: unable to open %s: %s\n"), |
|
|
"/boot/boot.b", strerror(errno)); |
|
|
return 1; |
|
|
} |
|
3035 |
|
|
3036 |
if (read(fd, boot, 512) != 512) { |
dbgPrintf("opening grub device.map file from: %s\n", path); |
3037 |
fprintf(stderr, _("grubby: unable to read %s: %s\n"), |
deviceMap = fopen(path, "r"); |
3038 |
"/boot/boot.b", strerror(errno)); |
if (!deviceMap) |
3039 |
return 1; |
return 1; |
3040 |
} |
|
3041 |
close(fd); |
while ((res = getline(&line, &len, deviceMap)) != -1) { |
3042 |
|
if (!strncmp(line, "#", 1)) |
3043 |
|
continue; |
3044 |
|
|
3045 |
|
if (line[res - 1] == '\n') |
3046 |
|
line[res - 1] = '\0'; |
3047 |
|
else if (len > res) |
3048 |
|
line[res] = '\0'; |
3049 |
|
else { |
3050 |
|
line = realloc(line, res + 1); |
3051 |
|
line[res] = '\0'; |
3052 |
|
} |
3053 |
|
|
3054 |
if (!strncmp("/dev/md", line->elements[1].item, 7)) |
devicePtr = line; |
3055 |
return checkLiloOnRaid(line->elements[1].item, boot); |
bounds = line + res; |
3056 |
|
|
3057 |
return checkDeviceBootloader(line->elements[1].item, boot); |
while ((isspace(*line) && ((devicePtr + 1) <= bounds))) |
3058 |
} |
devicePtr++; |
3059 |
|
dbgPrintf("device: %s\n", devicePtr); |
3060 |
|
|
3061 |
|
if (!strncmp(devicePtr, device, strlen(device))) { |
3062 |
|
devicePtr += strlen(device); |
3063 |
|
while (isspace(*devicePtr) |
3064 |
|
&& ((devicePtr + 1) <= bounds)) |
3065 |
|
devicePtr++; |
3066 |
|
|
3067 |
int checkForGrub2(struct grubConfig * config) { |
*bootPtr = strdup(devicePtr); |
3068 |
if (!access("/etc/grub.d/", R_OK)) |
break; |
3069 |
return 2; |
} |
3070 |
|
} |
3071 |
|
|
3072 |
return 1; |
free(line); |
3073 |
|
fclose(deviceMap); |
3074 |
|
return 0; |
3075 |
} |
} |
3076 |
|
|
3077 |
int checkForGrub(struct grubConfig * config) { |
int suseGrubConfGetBoot(const char *path, char **bootPtr) |
3078 |
int fd; |
{ |
3079 |
unsigned char bootSect[512]; |
char *grubDevice; |
|
char * boot; |
|
3080 |
|
|
3081 |
if (parseSysconfigGrub(NULL, &boot)) |
if (suseGrubConfGetInstallDevice(path, &grubDevice)) |
3082 |
return 0; |
dbgPrintf("error looking for grub installation device\n"); |
3083 |
|
else |
3084 |
|
dbgPrintf("grubby installation device: %s\n", grubDevice); |
3085 |
|
|
3086 |
/* assume grub is not installed -- not an error condition */ |
if (grubGetBootFromDeviceMap(grubDevice, bootPtr)) |
3087 |
if (!boot) |
dbgPrintf("error looking for grub boot device\n"); |
3088 |
return 0; |
else |
3089 |
|
dbgPrintf("grubby boot device: %s\n", *bootPtr); |
3090 |
|
|
3091 |
fd = open("/boot/grub/stage1", O_RDONLY); |
free(grubDevice); |
|
if (fd < 0) |
|
|
/* this doesn't exist if grub hasn't been installed */ |
|
3092 |
return 0; |
return 0; |
3093 |
|
} |
3094 |
|
|
3095 |
if (read(fd, bootSect, 512) != 512) { |
int parseSuseGrubConf(int *lbaPtr, char **bootPtr) |
3096 |
fprintf(stderr, _("grubby: unable to read %s: %s\n"), |
{ |
3097 |
"/boot/grub/stage1", strerror(errno)); |
/* |
3098 |
close(fd); |
* This SuSE grub configuration file at this location is not your |
3099 |
return 1; |
* average grub configuration file, but instead the grub commands |
3100 |
} |
* used to setup grub on that system. |
3101 |
close(fd); |
*/ |
3102 |
|
const char *path; |
3103 |
|
const static char default_path[] = "/etc/grub.conf"; |
3104 |
|
|
3105 |
return checkDeviceBootloader(boot, bootSect); |
if ((path = getenv("GRUBBY_SUSE_GRUB_CONF")) == NULL) |
3106 |
} |
path = default_path; |
3107 |
|
|
3108 |
|
if (!isSuseGrubConf(path)) |
3109 |
|
return 1; |
3110 |
|
|
3111 |
int checkForExtLinux(struct grubConfig * config) { |
if (lbaPtr) { |
3112 |
int fd; |
*lbaPtr = 0; |
3113 |
unsigned char bootSect[512]; |
if (suseGrubConfGetLba(path, lbaPtr)) |
3114 |
char * boot; |
return 1; |
3115 |
char executable[] = "/boot/extlinux/extlinux"; |
} |
3116 |
|
|
3117 |
printf("entered: checkForExtLinux()\n"); |
if (bootPtr) { |
3118 |
|
*bootPtr = NULL; |
3119 |
|
suseGrubConfGetBoot(path, bootPtr); |
3120 |
|
} |
3121 |
|
|
|
if (parseSysconfigGrub(NULL, &boot)) |
|
3122 |
return 0; |
return 0; |
3123 |
|
} |
3124 |
|
|
3125 |
/* assume grub is not installed -- not an error condition */ |
int parseSysconfigGrub(int *lbaPtr, char **bootPtr) |
3126 |
if (!boot) |
{ |
3127 |
return 0; |
FILE *in; |
3128 |
|
char buf[1024]; |
3129 |
|
char *chptr; |
3130 |
|
char *start; |
3131 |
|
char *param; |
3132 |
|
|
3133 |
fd = open(executable, O_RDONLY); |
in = fopen("/etc/sysconfig/grub", "r"); |
3134 |
if (fd < 0) |
if (!in) |
3135 |
/* this doesn't exist if grub hasn't been installed */ |
return 1; |
|
return 0; |
|
3136 |
|
|
3137 |
if (read(fd, bootSect, 512) != 512) { |
if (lbaPtr) |
3138 |
fprintf(stderr, _("grubby: unable to read %s: %s\n"), |
*lbaPtr = 0; |
3139 |
executable, strerror(errno)); |
if (bootPtr) |
3140 |
return 1; |
*bootPtr = NULL; |
3141 |
} |
|
3142 |
close(fd); |
while (fgets(buf, sizeof(buf), in)) { |
3143 |
|
start = buf; |
3144 |
|
while (isspace(*start)) |
3145 |
|
start++; |
3146 |
|
if (*start == '#') |
3147 |
|
continue; |
3148 |
|
|
3149 |
|
chptr = strchr(start, '='); |
3150 |
|
if (!chptr) |
3151 |
|
continue; |
3152 |
|
chptr--; |
3153 |
|
while (*chptr && isspace(*chptr)) |
3154 |
|
chptr--; |
3155 |
|
chptr++; |
3156 |
|
*chptr = '\0'; |
3157 |
|
|
3158 |
|
param = chptr + 1; |
3159 |
|
while (*param && isspace(*param)) |
3160 |
|
param++; |
3161 |
|
if (*param == '=') { |
3162 |
|
param++; |
3163 |
|
while (*param && isspace(*param)) |
3164 |
|
param++; |
3165 |
|
} |
3166 |
|
|
3167 |
return checkDeviceBootloader(boot, bootSect); |
chptr = param; |
3168 |
} |
while (*chptr && !isspace(*chptr)) |
3169 |
|
chptr++; |
3170 |
|
*chptr = '\0'; |
3171 |
|
|
3172 |
|
if (!strcmp(start, "forcelba") && !strcmp(param, "1") && lbaPtr) |
3173 |
|
*lbaPtr = 1; |
3174 |
|
else if (!strcmp(start, "boot") && bootPtr) |
3175 |
|
*bootPtr = strdup(param); |
3176 |
|
} |
3177 |
|
|
3178 |
static char * getRootSpecifier(char * str) { |
fclose(in); |
|
char * idx, * rootspec = NULL; |
|
3179 |
|
|
3180 |
if (*str == '(') { |
return 0; |
|
idx = rootspec = strdup(str); |
|
|
while(*idx && (*idx != ')') && (!isspace(*idx))) idx++; |
|
|
*(++idx) = '\0'; |
|
|
} |
|
|
return rootspec; |
|
3181 |
} |
} |
3182 |
|
|
3183 |
static char * getInitrdVal(struct grubConfig * config, |
void dumpSysconfigGrub(void) |
3184 |
const char * prefix, struct singleLine *tmplLine, |
{ |
3185 |
const char * newKernelInitrd, |
char *boot = NULL; |
3186 |
char ** extraInitrds, int extraInitrdCount) |
int lba; |
|
{ |
|
|
char *initrdVal, *end; |
|
|
int i; |
|
|
size_t totalSize; |
|
|
size_t prefixLen; |
|
|
char separatorChar; |
|
|
|
|
|
prefixLen = strlen(prefix); |
|
|
totalSize = strlen(newKernelInitrd) - prefixLen + 1 /* \0 */; |
|
|
|
|
|
for (i = 0; i < extraInitrdCount; i++) { |
|
|
totalSize += sizeof(separatorChar); |
|
|
totalSize += strlen(extraInitrds[i]) - prefixLen; |
|
|
} |
|
3187 |
|
|
3188 |
initrdVal = end = malloc(totalSize); |
if (isSuseSystem()) { |
3189 |
|
if (parseSuseGrubConf(&lba, &boot)) { |
3190 |
|
free(boot); |
3191 |
|
return; |
3192 |
|
} |
3193 |
|
} else { |
3194 |
|
if (parseSysconfigGrub(&lba, &boot)) { |
3195 |
|
free(boot); |
3196 |
|
return; |
3197 |
|
} |
3198 |
|
} |
3199 |
|
|
3200 |
end = stpcpy (end, newKernelInitrd + prefixLen); |
if (lba) |
3201 |
|
printf("lba\n"); |
3202 |
|
if (boot) { |
3203 |
|
printf("boot=%s\n", boot); |
3204 |
|
free(boot); |
3205 |
|
} |
3206 |
|
} |
3207 |
|
|
3208 |
separatorChar = getKeywordByType(LT_INITRD, config->cfi)->separatorChar; |
int displayInfo(struct grubConfig *config, char *kernel, const char *prefix) |
3209 |
for (i = 0; i < extraInitrdCount; i++) { |
{ |
3210 |
const char *extraInitrd; |
int i = 0; |
3211 |
int j; |
struct singleEntry *entry; |
3212 |
|
struct singleLine *line; |
3213 |
|
|
3214 |
|
entry = findEntryByPath(config, kernel, prefix, &i); |
3215 |
|
if (!entry) { |
3216 |
|
fprintf(stderr, _("grubby: kernel not found\n")); |
3217 |
|
return 1; |
3218 |
|
} |
3219 |
|
|
3220 |
extraInitrd = extraInitrds[i] + prefixLen; |
/* this is a horrible hack to support /etc/sysconfig/grub; there must |
3221 |
/* Don't add entries that are already there */ |
be a better way */ |
3222 |
if (tmplLine != NULL) { |
if (config->cfi == &grubConfigType) { |
3223 |
for (j = 2; j < tmplLine->numElements; j++) |
dumpSysconfigGrub(); |
3224 |
if (strcmp(extraInitrd, tmplLine->elements[j].item) == 0) |
} else { |
3225 |
break; |
line = getLineByType(LT_BOOT, config->theLines); |
3226 |
|
if (line && line->numElements >= 1) { |
3227 |
|
printf("boot=%s\n", line->elements[1].item); |
3228 |
|
} |
3229 |
|
|
3230 |
if (j != tmplLine->numElements) |
line = getLineByType(LT_LBA, config->theLines); |
3231 |
continue; |
if (line) |
3232 |
|
printf("lba\n"); |
3233 |
} |
} |
3234 |
|
|
3235 |
*end++ = separatorChar; |
displayEntry(config, entry, prefix, i); |
3236 |
end = stpcpy(end, extraInitrd); |
|
3237 |
} |
i++; |
3238 |
|
while ((entry = findEntryByPath(config, kernel, prefix, &i))) { |
3239 |
|
displayEntry(config, entry, prefix, i); |
3240 |
|
i++; |
3241 |
|
} |
3242 |
|
|
3243 |
return initrdVal; |
return 0; |
3244 |
} |
} |
3245 |
|
|
3246 |
int addNewKernel(struct grubConfig * config, struct singleEntry * template, |
struct singleLine *addLineTmpl(struct singleEntry *entry, |
3247 |
const char * prefix, |
struct singleLine *tmplLine, |
3248 |
char * newKernelPath, char * newKernelTitle, |
struct singleLine *prevLine, |
3249 |
char * newKernelArgs, char * newKernelInitrd, |
const char *val, struct configFileInfo *cfi) |
3250 |
char ** extraInitrds, int extraInitrdCount, |
{ |
3251 |
char * newMBKernel, char * newMBKernelArgs) { |
struct singleLine *newLine = lineDup(tmplLine); |
|
struct singleEntry * new; |
|
|
struct singleLine * newLine = NULL, * tmplLine = NULL, * masterLine = NULL; |
|
|
int needs; |
|
|
char * chptr; |
|
|
|
|
|
if (!newKernelPath) return 0; |
|
|
|
|
|
/* if the newKernelTitle is too long silently munge it into something |
|
|
* we can live with. truncating is first check, then we'll just mess with |
|
|
* it until it looks better */ |
|
|
if (config->cfi->maxTitleLength && |
|
|
(strlen(newKernelTitle) > config->cfi->maxTitleLength)) { |
|
|
char * buf = alloca(config->cfi->maxTitleLength + 7); |
|
|
char * numBuf = alloca(config->cfi->maxTitleLength + 1); |
|
|
int i = 1; |
|
3252 |
|
|
3253 |
sprintf(buf, "TITLE=%.*s", config->cfi->maxTitleLength, newKernelTitle); |
if (isEfi && cfi == &grub2ConfigType) { |
3254 |
while (findEntryByPath(config, buf, NULL, NULL)) { |
enum lineType_e old = newLine->type; |
3255 |
sprintf(numBuf, "%d", i++); |
newLine->type = preferredLineType(newLine->type, cfi); |
3256 |
strcpy(buf + strlen(buf) - strlen(numBuf), numBuf); |
if (old != newLine->type) |
3257 |
|
newLine->elements[0].item = |
3258 |
|
getKeyByType(newLine->type, cfi); |
3259 |
} |
} |
3260 |
|
|
3261 |
newKernelTitle = buf + 6; |
if (val) { |
3262 |
} |
/* override the inherited value with our own. |
3263 |
|
* This is a little weak because it only applies to elements[1] |
3264 |
|
*/ |
3265 |
|
if (newLine->numElements > 1) |
3266 |
|
removeElement(newLine, 1); |
3267 |
|
insertElement(newLine, val, 1, cfi); |
3268 |
|
|
3269 |
|
/* but try to keep the rootspec from the template... sigh */ |
3270 |
|
if (tmplLine-> |
3271 |
|
type & (LT_HYPER | LT_KERNEL | LT_MBMODULE | LT_INITRD | |
3272 |
|
LT_KERNEL_EFI | LT_INITRD_EFI | LT_KERNEL_16 | |
3273 |
|
LT_INITRD_16)) { |
3274 |
|
char *rootspec = |
3275 |
|
getRootSpecifier(tmplLine->elements[1].item); |
3276 |
|
if (rootspec != NULL) { |
3277 |
|
free(newLine->elements[1].item); |
3278 |
|
newLine->elements[1].item = |
3279 |
|
sdupprintf("%s%s", rootspec, val); |
3280 |
|
} |
3281 |
|
} |
3282 |
|
} |
3283 |
|
|
3284 |
new = malloc(sizeof(*new)); |
dbgPrintf("addLineTmpl(%s)\n", newLine->numElements ? |
3285 |
new->skip = 0; |
newLine->elements[0].item : ""); |
|
new->multiboot = 0; |
|
|
new->next = config->entries; |
|
|
new->lines = NULL; |
|
|
config->entries = new; |
|
|
|
|
|
/* copy/update from the template */ |
|
|
needs = NEED_KERNEL | NEED_TITLE; |
|
|
if (newKernelInitrd) |
|
|
needs |= NEED_INITRD; |
|
|
if (newMBKernel) { |
|
|
needs |= NEED_MB; |
|
|
new->multiboot = 1; |
|
|
} |
|
3286 |
|
|
3287 |
if (template) { |
if (!entry->lines) { |
3288 |
for (masterLine = template->lines; |
/* first one on the list */ |
3289 |
masterLine && (tmplLine = lineDup(masterLine)); |
entry->lines = newLine; |
3290 |
lineFree(tmplLine), masterLine = masterLine->next) |
} else if (prevLine) { |
3291 |
{ |
/* add after prevLine */ |
3292 |
dbgPrintf("addNewKernel processing %d\n", tmplLine->type); |
newLine->next = prevLine->next; |
3293 |
|
prevLine->next = newLine; |
|
/* skip comments */ |
|
|
chptr = tmplLine->indent; |
|
|
while (*chptr && isspace(*chptr)) chptr++; |
|
|
if (*chptr == '#') continue; |
|
|
|
|
|
if (tmplLine->type == LT_KERNEL && |
|
|
tmplLine->numElements >= 2) { |
|
|
if (!template->multiboot && (needs & NEED_MB)) { |
|
|
/* it's not a multiboot template and this is the kernel |
|
|
* line. Try to be intelligent about inserting the |
|
|
* hypervisor at the same time. |
|
|
*/ |
|
|
if (config->cfi->mbHyperFirst) { |
|
|
/* insert the hypervisor first */ |
|
|
newLine = addLine(new, config->cfi, LT_HYPER, |
|
|
tmplLine->indent, |
|
|
newMBKernel + strlen(prefix)); |
|
|
/* set up for adding the kernel line */ |
|
|
free(tmplLine->indent); |
|
|
tmplLine->indent = strdup(config->secondaryIndent); |
|
|
needs &= ~NEED_MB; |
|
|
} |
|
|
if (needs & NEED_KERNEL) { |
|
|
/* use addLineTmpl to preserve line elements, |
|
|
* otherwise we could just call addLine. Unfortunately |
|
|
* this means making some changes to the template |
|
|
* such as the indent change above and the type |
|
|
* change below. |
|
|
*/ |
|
|
struct keywordTypes * mbm_kw = |
|
|
getKeywordByType(LT_MBMODULE, config->cfi); |
|
|
if (mbm_kw) { |
|
|
tmplLine->type = LT_MBMODULE; |
|
|
free(tmplLine->elements[0].item); |
|
|
tmplLine->elements[0].item = strdup(mbm_kw->key); |
|
|
} |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, |
|
|
newKernelPath + strlen(prefix), config->cfi); |
|
|
needs &= ~NEED_KERNEL; |
|
|
} |
|
|
if (needs & NEED_MB) { /* !mbHyperFirst */ |
|
|
newLine = addLine(new, config->cfi, LT_HYPER, |
|
|
config->secondaryIndent, |
|
|
newMBKernel + strlen(prefix)); |
|
|
needs &= ~NEED_MB; |
|
|
} |
|
|
} else if (needs & NEED_KERNEL) { |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, |
|
|
newKernelPath + strlen(prefix), config->cfi); |
|
|
needs &= ~NEED_KERNEL; |
|
|
} |
|
|
|
|
|
} else if (tmplLine->type == LT_HYPER && |
|
|
tmplLine->numElements >= 2) { |
|
|
if (needs & NEED_MB) { |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, |
|
|
newMBKernel + strlen(prefix), config->cfi); |
|
|
needs &= ~NEED_MB; |
|
|
} |
|
|
|
|
|
} else if (tmplLine->type == LT_MBMODULE && |
|
|
tmplLine->numElements >= 2) { |
|
|
if (new->multiboot) { |
|
|
if (needs & NEED_KERNEL) { |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, |
|
|
newKernelPath + |
|
|
strlen(prefix), config->cfi); |
|
|
needs &= ~NEED_KERNEL; |
|
|
} else if (config->cfi->mbInitRdIsModule && |
|
|
(needs & NEED_INITRD)) { |
|
|
char *initrdVal; |
|
|
initrdVal = getInitrdVal(config, prefix, tmplLine, |
|
|
newKernelInitrd, extraInitrds, |
|
|
extraInitrdCount); |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, |
|
|
initrdVal, config->cfi); |
|
|
free(initrdVal); |
|
|
needs &= ~NEED_INITRD; |
|
|
} |
|
|
} else if (needs & NEED_KERNEL) { |
|
|
/* template is multi but new is not, |
|
|
* insert the kernel in the first module slot |
|
|
*/ |
|
|
tmplLine->type = LT_KERNEL; |
|
|
free(tmplLine->elements[0].item); |
|
|
tmplLine->elements[0].item = |
|
|
strdup(getKeywordByType(LT_KERNEL, config->cfi)->key); |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, |
|
|
newKernelPath + strlen(prefix), config->cfi); |
|
|
needs &= ~NEED_KERNEL; |
|
|
} else if (needs & NEED_INITRD) { |
|
|
char *initrdVal; |
|
|
/* template is multi but new is not, |
|
|
* insert the initrd in the second module slot |
|
|
*/ |
|
|
tmplLine->type = LT_INITRD; |
|
|
free(tmplLine->elements[0].item); |
|
|
tmplLine->elements[0].item = |
|
|
strdup(getKeywordByType(LT_INITRD, config->cfi)->key); |
|
|
initrdVal = getInitrdVal(config, prefix, tmplLine, newKernelInitrd, extraInitrds, extraInitrdCount); |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, initrdVal, config->cfi); |
|
|
free(initrdVal); |
|
|
needs &= ~NEED_INITRD; |
|
|
} |
|
|
|
|
|
} else if (tmplLine->type == LT_INITRD && |
|
|
tmplLine->numElements >= 2) { |
|
|
if (needs & NEED_INITRD && |
|
|
new->multiboot && !template->multiboot && |
|
|
config->cfi->mbInitRdIsModule) { |
|
|
/* make sure we don't insert the module initrd |
|
|
* before the module kernel... if we don't do it here, |
|
|
* it will be inserted following the template. |
|
|
*/ |
|
|
if (!needs & NEED_KERNEL) { |
|
|
char *initrdVal; |
|
|
|
|
|
initrdVal = getInitrdVal(config, prefix, tmplLine, newKernelInitrd, extraInitrds, extraInitrdCount); |
|
|
newLine = addLine(new, config->cfi, LT_MBMODULE, |
|
|
config->secondaryIndent, |
|
|
initrdVal); |
|
|
free(initrdVal); |
|
|
needs &= ~NEED_INITRD; |
|
|
} |
|
|
} else if (needs & NEED_INITRD) { |
|
|
char *initrdVal; |
|
|
initrdVal = getInitrdVal(config, prefix, tmplLine, newKernelInitrd, extraInitrds, extraInitrdCount); |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, initrdVal, config->cfi); |
|
|
free(initrdVal); |
|
|
needs &= ~NEED_INITRD; |
|
|
} |
|
|
|
|
|
} else if (tmplLine->type == LT_MENUENTRY && |
|
|
(needs & NEED_TITLE)) { |
|
|
requote(tmplLine, config->cfi); |
|
|
char *nkt = malloc(strlen(newKernelTitle)+3); |
|
|
strcpy(nkt, "'"); |
|
|
strcat(nkt, newKernelTitle); |
|
|
strcat(nkt, "'"); |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, nkt, config->cfi); |
|
|
free(nkt); |
|
|
needs &= ~NEED_TITLE; |
|
|
} else if (tmplLine->type == LT_TITLE && |
|
|
(needs & NEED_TITLE)) { |
|
|
if (tmplLine->numElements >= 2) { |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, |
|
|
newKernelTitle, config->cfi); |
|
|
needs &= ~NEED_TITLE; |
|
|
} else if (tmplLine->numElements == 1 && |
|
|
config->cfi->titleBracketed) { |
|
|
/* addLineTmpl doesn't handle titleBracketed */ |
|
|
newLine = addLine(new, config->cfi, LT_TITLE, |
|
|
tmplLine->indent, newKernelTitle); |
|
|
needs &= ~NEED_TITLE; |
|
|
} |
|
|
} else if (tmplLine->type == LT_ECHO) { |
|
|
requote(tmplLine, config->cfi); |
|
|
static const char *prefix = "'Loading "; |
|
|
if (tmplLine->numElements > 1 && |
|
|
strstr(tmplLine->elements[1].item, prefix) && |
|
|
masterLine->next && masterLine->next->type == LT_KERNEL) { |
|
|
char *newTitle = malloc(strlen(prefix) + |
|
|
strlen(newKernelTitle) + 2); |
|
|
|
|
|
strcpy(newTitle, prefix); |
|
|
strcat(newTitle, newKernelTitle); |
|
|
strcat(newTitle, "'"); |
|
|
newLine = addLine(new, config->cfi, LT_ECHO, |
|
|
tmplLine->indent, newTitle); |
|
|
free(newTitle); |
|
|
} else { |
|
|
/* pass through other lines from the template */ |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, NULL, |
|
|
config->cfi); |
|
|
} |
|
|
} else { |
|
|
/* pass through other lines from the template */ |
|
|
newLine = addLineTmpl(new, tmplLine, newLine, NULL, config->cfi); |
|
|
} |
|
3294 |
} |
} |
3295 |
|
|
3296 |
} else { |
return newLine; |
3297 |
/* don't have a template, so start the entry with the |
} |
|
* appropriate starting line |
|
|
*/ |
|
|
switch (config->cfi->entryStart) { |
|
|
case LT_KERNEL: |
|
|
if (new->multiboot && config->cfi->mbHyperFirst) { |
|
|
/* fall through to LT_HYPER */ |
|
|
} else { |
|
|
newLine = addLine(new, config->cfi, LT_KERNEL, |
|
|
config->primaryIndent, |
|
|
newKernelPath + strlen(prefix)); |
|
|
needs &= ~NEED_KERNEL; |
|
|
break; |
|
|
} |
|
3298 |
|
|
3299 |
case LT_HYPER: |
/* val may be NULL */ |
3300 |
newLine = addLine(new, config->cfi, LT_HYPER, |
struct singleLine *addLine(struct singleEntry *entry, |
3301 |
config->primaryIndent, |
struct configFileInfo *cfi, |
3302 |
newMBKernel + strlen(prefix)); |
enum lineType_e type, char *defaultIndent, |
3303 |
needs &= ~NEED_MB; |
const char *val) |
3304 |
break; |
{ |
3305 |
|
struct singleLine *line, *prev; |
3306 |
|
struct keywordTypes *kw; |
3307 |
|
struct singleLine tmpl; |
3308 |
|
|
3309 |
case LT_MENUENTRY: { |
/* NB: This function shouldn't allocate items on the heap, rather on |
3310 |
char *nkt = malloc(strlen(newKernelTitle)+3); |
* the stack since it calls addLineTmpl which will make copies. |
3311 |
strcpy(nkt, "'"); |
*/ |
3312 |
strcat(nkt, newKernelTitle); |
if (type == LT_TITLE && cfi->titleBracketed) { |
3313 |
strcat(nkt, "'"); |
/* we're doing a bracketed title (zipl) */ |
3314 |
newLine = addLine(new, config->cfi, LT_MENUENTRY, |
tmpl.type = type; |
3315 |
config->primaryIndent, nkt); |
tmpl.numElements = 1; |
3316 |
free(nkt); |
tmpl.elements = alloca(sizeof(*tmpl.elements)); |
3317 |
needs &= ~NEED_TITLE; |
tmpl.elements[0].item = alloca(strlen(val) + 3); |
3318 |
needs |= NEED_END; |
sprintf(tmpl.elements[0].item, "[%s]", val); |
3319 |
break; |
tmpl.elements[0].indent = ""; |
3320 |
} |
val = NULL; |
3321 |
case LT_TITLE: |
} else if (type == LT_MENUENTRY) { |
3322 |
if( useextlinuxmenu != 0 ){ // We just need useextlinuxmenu to not be zero (set above) |
char *lineend = "--class gnu-linux --class gnu --class os {"; |
3323 |
char * templabel; |
if (!val) { |
3324 |
int x = 0, y = 0; |
fprintf(stderr, |
3325 |
|
"Line type LT_MENUENTRY requires a value\n"); |
3326 |
templabel = strdup(newKernelTitle); |
abort(); |
|
while( templabel[x]){ |
|
|
if( templabel[x] == ' ' ){ |
|
|
y = x; |
|
|
while( templabel[y] ){ |
|
|
templabel[y] = templabel[y+1]; |
|
|
y++; |
|
|
} |
|
|
} |
|
|
x++; |
|
|
} |
|
|
newLine = addLine(new, config->cfi, LT_TITLE, |
|
|
config->primaryIndent, templabel); |
|
|
free(templabel); |
|
|
}else{ |
|
|
newLine = addLine(new, config->cfi, LT_TITLE, |
|
|
config->primaryIndent, newKernelTitle); |
|
3327 |
} |
} |
3328 |
needs &= ~NEED_TITLE; |
kw = getKeywordByType(type, cfi); |
3329 |
break; |
if (!kw) { |
3330 |
|
fprintf(stderr, |
3331 |
|
"Looking up keyword for unknown type %d\n", |
3332 |
|
type); |
3333 |
|
abort(); |
3334 |
|
} |
3335 |
|
tmpl.indent = ""; |
3336 |
|
tmpl.type = type; |
3337 |
|
tmpl.numElements = 3; |
3338 |
|
tmpl.elements = |
3339 |
|
alloca(sizeof(*tmpl.elements) * tmpl.numElements); |
3340 |
|
tmpl.elements[0].item = kw->key; |
3341 |
|
tmpl.elements[0].indent = alloca(2); |
3342 |
|
sprintf(tmpl.elements[0].indent, "%c", kw->nextChar); |
3343 |
|
tmpl.elements[1].item = (char *)val; |
3344 |
|
tmpl.elements[1].indent = alloca(2); |
3345 |
|
sprintf(tmpl.elements[1].indent, "%c", kw->nextChar); |
3346 |
|
tmpl.elements[2].item = alloca(strlen(lineend) + 1); |
3347 |
|
strcpy(tmpl.elements[2].item, lineend); |
3348 |
|
tmpl.elements[2].indent = ""; |
3349 |
|
} else { |
3350 |
|
kw = getKeywordByType(type, cfi); |
3351 |
|
if (!kw) { |
3352 |
|
fprintf(stderr, |
3353 |
|
"Looking up keyword for unknown type %d\n", |
3354 |
|
type); |
3355 |
|
abort(); |
3356 |
|
} |
3357 |
|
tmpl.type = type; |
3358 |
|
tmpl.numElements = val ? 2 : 1; |
3359 |
|
tmpl.elements = |
3360 |
|
alloca(sizeof(*tmpl.elements) * tmpl.numElements); |
3361 |
|
tmpl.elements[0].item = kw->key; |
3362 |
|
tmpl.elements[0].indent = alloca(2); |
3363 |
|
sprintf(tmpl.elements[0].indent, "%c", kw->nextChar); |
3364 |
|
if (val) { |
3365 |
|
tmpl.elements[1].item = (char *)val; |
3366 |
|
tmpl.elements[1].indent = ""; |
3367 |
|
} |
3368 |
|
} |
3369 |
|
|
3370 |
default: |
/* The last non-empty line gives us the indention to us and the line |
3371 |
abort(); |
* to insert after. Note that comments are considered empty lines, |
3372 |
|
* which may not be ideal? If there are no lines or we are looking at |
3373 |
|
* the first line, we use defaultIndent (the first line is normally |
3374 |
|
* indented differently from the rest) */ |
3375 |
|
for (line = entry->lines, prev = NULL; line; line = line->next) { |
3376 |
|
if (line->numElements) |
3377 |
|
prev = line; |
3378 |
|
/* fall back on the last line if prev isn't otherwise set */ |
3379 |
|
if (!line->next && !prev) |
3380 |
|
prev = line; |
3381 |
|
} |
3382 |
|
|
3383 |
|
struct singleLine *menuEntry; |
3384 |
|
menuEntry = getLineByType(LT_MENUENTRY, entry->lines); |
3385 |
|
if (tmpl.type == LT_ENTRY_END) { |
3386 |
|
if (menuEntry) |
3387 |
|
tmpl.indent = menuEntry->indent; |
3388 |
|
else |
3389 |
|
tmpl.indent = defaultIndent ? : ""; |
3390 |
|
} else if (tmpl.type != LT_MENUENTRY) { |
3391 |
|
if (menuEntry) |
3392 |
|
tmpl.indent = "\t"; |
3393 |
|
else if (prev == entry->lines) |
3394 |
|
tmpl.indent = defaultIndent ? : ""; |
3395 |
|
else |
3396 |
|
tmpl.indent = prev->indent; |
3397 |
} |
} |
|
} |
|
3398 |
|
|
3399 |
/* add the remainder of the lines, i.e. those that either |
return addLineTmpl(entry, &tmpl, prev, val, cfi); |
3400 |
* weren't present in the template, or in the case of no template, |
} |
|
* all the lines following the entryStart. |
|
|
*/ |
|
|
if (needs & NEED_TITLE) { |
|
|
newLine = addLine(new, config->cfi, LT_TITLE, |
|
|
config->secondaryIndent, |
|
|
newKernelTitle); |
|
|
needs &= ~NEED_TITLE; |
|
|
} |
|
|
if ((needs & NEED_MB) && config->cfi->mbHyperFirst) { |
|
|
newLine = addLine(new, config->cfi, LT_HYPER, |
|
|
config->secondaryIndent, |
|
|
newMBKernel + strlen(prefix)); |
|
|
needs &= ~NEED_MB; |
|
|
} |
|
|
if (needs & NEED_KERNEL) { |
|
|
newLine = addLine(new, config->cfi, |
|
|
(new->multiboot && getKeywordByType(LT_MBMODULE, |
|
|
config->cfi)) ? |
|
|
LT_MBMODULE : LT_KERNEL, |
|
|
config->secondaryIndent, |
|
|
newKernelPath + strlen(prefix)); |
|
|
needs &= ~NEED_KERNEL; |
|
|
} |
|
|
if (needs & NEED_MB) { |
|
|
newLine = addLine(new, config->cfi, LT_HYPER, |
|
|
config->secondaryIndent, |
|
|
newMBKernel + strlen(prefix)); |
|
|
needs &= ~NEED_MB; |
|
|
} |
|
|
if (needs & NEED_INITRD) { |
|
|
char *initrdVal; |
|
|
initrdVal = getInitrdVal(config, prefix, NULL, newKernelInitrd, extraInitrds, extraInitrdCount); |
|
|
newLine = addLine(new, config->cfi, |
|
|
(new->multiboot && getKeywordByType(LT_MBMODULE, |
|
|
config->cfi)) ? |
|
|
LT_MBMODULE : LT_INITRD, |
|
|
config->secondaryIndent, |
|
|
initrdVal); |
|
|
free(initrdVal); |
|
|
needs &= ~NEED_INITRD; |
|
|
} |
|
|
if (needs & NEED_END) { |
|
|
newLine = addLine(new, config->cfi, LT_ENTRY_END, |
|
|
config->secondaryIndent, NULL); |
|
|
needs &= ~NEED_END; |
|
|
} |
|
3401 |
|
|
3402 |
if (needs) { |
void removeLine(struct singleEntry *entry, struct singleLine *line) |
3403 |
printf(_("grubby: needs=%d, aborting\n"), needs); |
{ |
3404 |
abort(); |
struct singleLine *prev; |
3405 |
} |
int i; |
3406 |
|
|
3407 |
|
for (i = 0; i < line->numElements; i++) { |
3408 |
|
free(line->elements[i].item); |
3409 |
|
free(line->elements[i].indent); |
3410 |
|
} |
3411 |
|
free(line->elements); |
3412 |
|
free(line->indent); |
3413 |
|
|
3414 |
if (updateImage(config, "0", prefix, newKernelArgs, NULL, |
if (line == entry->lines) { |
3415 |
newMBKernelArgs, NULL)) return 1; |
entry->lines = line->next; |
3416 |
|
} else { |
3417 |
|
prev = entry->lines; |
3418 |
|
while (prev->next != line) |
3419 |
|
prev = prev->next; |
3420 |
|
prev->next = line->next; |
3421 |
|
} |
3422 |
|
|
3423 |
return 0; |
free(line); |
3424 |
} |
} |
3425 |
|
|
3426 |
static void traceback(int signum) |
static void requote(struct singleLine *tmplLine, struct configFileInfo *cfi) |
3427 |
{ |
{ |
3428 |
void *array[40]; |
struct singleLine newLine = { |
3429 |
size_t size; |
.indent = tmplLine->indent, |
3430 |
|
.type = tmplLine->type, |
3431 |
|
.next = tmplLine->next, |
3432 |
|
}; |
3433 |
|
int firstQuotedItem = -1; |
3434 |
|
int quoteLen = 0; |
3435 |
|
int j; |
3436 |
|
int element = 0; |
3437 |
|
char *c; |
3438 |
|
|
3439 |
signal(SIGSEGV, SIG_DFL); |
c = malloc(strlen(tmplLine->elements[0].item) + 1); |
3440 |
memset(array, '\0', sizeof (array)); |
strcpy(c, tmplLine->elements[0].item); |
3441 |
size = backtrace(array, 40); |
insertElement(&newLine, c, element++, cfi); |
3442 |
|
free(c); |
3443 |
fprintf(stderr, "grubby recieved SIGSEGV! Backtrace (%ld):\n", |
c = NULL; |
3444 |
(unsigned long)size); |
|
3445 |
backtrace_symbols_fd(array, size, STDERR_FILENO); |
for (j = 1; j < tmplLine->numElements; j++) { |
3446 |
exit(1); |
if (firstQuotedItem == -1) { |
3447 |
} |
quoteLen += strlen(tmplLine->elements[j].item); |
3448 |
|
|
3449 |
int main(int argc, const char ** argv) { |
if (isquote(tmplLine->elements[j].item[0])) { |
3450 |
poptContext optCon; |
firstQuotedItem = j; |
3451 |
const char * grubConfig = NULL; |
quoteLen += |
3452 |
char * outputFile = NULL; |
strlen(tmplLine->elements[j].indent); |
3453 |
int arg = 0; |
} else { |
3454 |
int flags = 0; |
c = malloc(quoteLen + 1); |
3455 |
int badImageOkay = 0; |
strcpy(c, tmplLine->elements[j].item); |
3456 |
int configureGrub2 = 0; |
insertElement(&newLine, c, element++, cfi); |
3457 |
int configureLilo = 0, configureELilo = 0, configureGrub = 0; |
free(c); |
3458 |
int configureYaboot = 0, configureSilo = 0, configureZipl = 0; |
quoteLen = 0; |
3459 |
int configureExtLinux = 0; |
} |
3460 |
int bootloaderProbe = 0; |
} else { |
3461 |
int extraInitrdCount = 0; |
int itemlen = strlen(tmplLine->elements[j].item); |
3462 |
char * updateKernelPath = NULL; |
quoteLen += itemlen; |
3463 |
char * newKernelPath = NULL; |
quoteLen += strlen(tmplLine->elements[j].indent); |
3464 |
char * removeKernelPath = NULL; |
|
3465 |
char * newKernelArgs = NULL; |
if (isquote(tmplLine->elements[j].item[itemlen - 1])) { |
3466 |
char * newKernelInitrd = NULL; |
c = malloc(quoteLen + 1); |
3467 |
char * newKernelTitle = NULL; |
c[0] = '\0'; |
3468 |
char * newKernelVersion = NULL; |
for (int i = firstQuotedItem; i < j + 1; i++) { |
3469 |
char * newMBKernel = NULL; |
strcat(c, tmplLine->elements[i].item); |
3470 |
char * newMBKernelArgs = NULL; |
strcat(c, tmplLine->elements[i].indent); |
3471 |
char * removeMBKernelArgs = NULL; |
} |
3472 |
char * removeMBKernel = NULL; |
insertElement(&newLine, c, element++, cfi); |
3473 |
char * bootPrefix = NULL; |
free(c); |
3474 |
char * defaultKernel = NULL; |
|
3475 |
char * removeArgs = NULL; |
firstQuotedItem = -1; |
3476 |
char * kernelInfo = NULL; |
quoteLen = 0; |
3477 |
char * extraInitrds[MAX_EXTRA_INITRDS] = { NULL }; |
} |
3478 |
const char * chptr = NULL; |
} |
3479 |
struct configFileInfo * cfi = NULL; |
} |
3480 |
struct grubConfig * config; |
while (tmplLine->numElements) |
3481 |
struct singleEntry * template = NULL; |
removeElement(tmplLine, 0); |
3482 |
int copyDefault = 0, makeDefault = 0; |
if (tmplLine->elements) |
3483 |
int displayDefault = 0; |
free(tmplLine->elements); |
3484 |
int displayDefaultIndex = 0; |
|
3485 |
int displayDefaultTitle = 0; |
tmplLine->numElements = newLine.numElements; |
3486 |
struct poptOption options[] = { |
tmplLine->elements = newLine.elements; |
3487 |
{ "add-kernel", 0, POPT_ARG_STRING, &newKernelPath, 0, |
} |
3488 |
_("add an entry for the specified kernel"), _("kernel-path") }, |
|
3489 |
{ "add-multiboot", 0, POPT_ARG_STRING, &newMBKernel, 0, |
static void insertElement(struct singleLine *line, |
3490 |
_("add an entry for the specified multiboot kernel"), NULL }, |
const char *item, int insertHere, |
3491 |
{ "args", 0, POPT_ARG_STRING, &newKernelArgs, 0, |
struct configFileInfo *cfi) |
3492 |
_("default arguments for the new kernel or new arguments for " |
{ |
3493 |
"kernel being updated"), _("args") }, |
struct keywordTypes *kw; |
3494 |
{ "mbargs", 0, POPT_ARG_STRING, &newMBKernelArgs, 0, |
char indent[2] = ""; |
3495 |
_("default arguments for the new multiboot kernel or " |
|
3496 |
"new arguments for multiboot kernel being updated"), NULL }, |
/* sanity check */ |
3497 |
{ "bad-image-okay", 0, 0, &badImageOkay, 0, |
if (insertHere > line->numElements) { |
3498 |
_("don't sanity check images in boot entries (for testing only)"), |
dbgPrintf |
3499 |
NULL }, |
("insertElement() adjusting insertHere from %d to %d\n", |
3500 |
{ "boot-filesystem", 0, POPT_ARG_STRING, &bootPrefix, 0, |
insertHere, line->numElements); |
3501 |
_("filestystem which contains /boot directory (for testing only)"), |
insertHere = line->numElements; |
3502 |
_("bootfs") }, |
} |
3503 |
#if defined(__i386__) || defined(__x86_64__) |
|
3504 |
{ "bootloader-probe", 0, POPT_ARG_NONE, &bootloaderProbe, 0, |
line->elements = realloc(line->elements, (line->numElements + 1) * |
3505 |
_("check if lilo is installed on lilo.conf boot sector") }, |
sizeof(*line->elements)); |
3506 |
#endif |
memmove(&line->elements[insertHere + 1], |
3507 |
{ "config-file", 'c', POPT_ARG_STRING, &grubConfig, 0, |
&line->elements[insertHere], |
3508 |
_("path to grub config file to update (\"-\" for stdin)"), |
(line->numElements - insertHere) * sizeof(*line->elements)); |
3509 |
_("path") }, |
line->elements[insertHere].item = strdup(item); |
3510 |
{ "copy-default", 0, 0, ©Default, 0, |
|
3511 |
_("use the default boot entry as a template for the new entry " |
kw = getKeywordByType(line->type, cfi); |
3512 |
"being added; if the default is not a linux image, or if " |
|
3513 |
"the kernel referenced by the default image does not exist, " |
if (line->numElements == 0) { |
3514 |
"the first linux entry whose kernel does exist is used as the " |
indent[0] = '\0'; |
3515 |
"template"), NULL }, |
} else if (insertHere == 0) { |
3516 |
{ "debug", 0, 0, &debug, 0, |
indent[0] = kw->nextChar; |
3517 |
_("print debugging information for failures") }, |
} else if (kw->separatorChar != '\0') { |
3518 |
{ "default-kernel", 0, 0, &displayDefault, 0, |
indent[0] = kw->separatorChar; |
3519 |
_("display the path of the default kernel") }, |
} else { |
3520 |
{ "default-index", 0, 0, &displayDefaultIndex, 0, |
indent[0] = ' '; |
3521 |
_("display the index of the default kernel") }, |
} |
3522 |
{ "default-title", 0, 0, &displayDefaultTitle, 0, |
|
3523 |
_("display the title of the default kernel") }, |
if (insertHere > 0 && line->elements[insertHere - 1].indent[0] == '\0') { |
3524 |
{ "elilo", 0, POPT_ARG_NONE, &configureELilo, 0, |
/* move the end-of-line forward */ |
3525 |
_("configure elilo bootloader") }, |
line->elements[insertHere].indent = |
3526 |
{ "extlinux", 0, POPT_ARG_NONE, &configureExtLinux, 0, |
line->elements[insertHere - 1].indent; |
3527 |
_("configure extlinux bootloader (from syslinux)") }, |
line->elements[insertHere - 1].indent = strdup(indent); |
3528 |
{ "grub", 0, POPT_ARG_NONE, &configureGrub, 0, |
} else { |
3529 |
_("configure grub bootloader") }, |
line->elements[insertHere].indent = strdup(indent); |
3530 |
{ "grub2", 0, POPT_ARG_NONE, &configureGrub2, 0, |
} |
3531 |
_("configure grub2 bootloader") }, |
|
3532 |
{ "info", 0, POPT_ARG_STRING, &kernelInfo, 0, |
line->numElements++; |
3533 |
_("display boot information for specified kernel"), |
|
3534 |
_("kernel-path") }, |
dbgPrintf("insertElement(%s, '%s%s', %d)\n", |
3535 |
{ "initrd", 0, POPT_ARG_STRING, &newKernelInitrd, 0, |
line->elements[0].item, |
3536 |
_("initrd image for the new kernel"), _("initrd-path") }, |
line->elements[insertHere].item, |
3537 |
{ "extra-initrd", 'i', POPT_ARG_STRING, NULL, 'i', |
line->elements[insertHere].indent, insertHere); |
3538 |
_("auxilliary initrd image for things other than the new kernel"), _("initrd-path") }, |
} |
3539 |
{ "lilo", 0, POPT_ARG_NONE, &configureLilo, 0, |
|
3540 |
_("configure lilo bootloader") }, |
static void removeElement(struct singleLine *line, int removeHere) |
3541 |
{ "make-default", 0, 0, &makeDefault, 0, |
{ |
3542 |
_("make the newly added entry the default boot entry"), NULL }, |
int i; |
3543 |
{ "output-file", 'o', POPT_ARG_STRING, &outputFile, 0, |
|
3544 |
_("path to output updated config file (\"-\" for stdout)"), |
/* sanity check */ |
3545 |
_("path") }, |
if (removeHere >= line->numElements) |
3546 |
{ "remove-args", 0, POPT_ARG_STRING, &removeArgs, 0, |
return; |
3547 |
_("remove kernel arguments"), NULL }, |
|
3548 |
{ "remove-mbargs", 0, POPT_ARG_STRING, &removeMBKernelArgs, 0, |
dbgPrintf("removeElement(%s, %d:%s)\n", line->elements[0].item, |
3549 |
_("remove multiboot kernel arguments"), NULL }, |
removeHere, line->elements[removeHere].item); |
3550 |
{ "remove-kernel", 0, POPT_ARG_STRING, &removeKernelPath, 0, |
|
3551 |
_("remove all entries for the specified kernel"), |
free(line->elements[removeHere].item); |
3552 |
_("kernel-path") }, |
|
3553 |
{ "remove-multiboot", 0, POPT_ARG_STRING, &removeMBKernel, 0, |
if (removeHere > 1) { |
3554 |
_("remove all entries for the specified multiboot kernel"), NULL }, |
/* previous argument gets this argument's post-indentation */ |
3555 |
{ "set-default", 0, POPT_ARG_STRING, &defaultKernel, 0, |
free(line->elements[removeHere - 1].indent); |
3556 |
_("make the first entry referencing the specified kernel " |
line->elements[removeHere - 1].indent = |
3557 |
"the default"), _("kernel-path") }, |
line->elements[removeHere].indent; |
3558 |
{ "silo", 0, POPT_ARG_NONE, &configureSilo, 0, |
} else { |
3559 |
_("configure silo bootloader") }, |
free(line->elements[removeHere].indent); |
3560 |
{ "title", 0, POPT_ARG_STRING, &newKernelTitle, 0, |
} |
3561 |
_("title to use for the new kernel entry"), _("entry-title") }, |
|
3562 |
{ "update-kernel", 0, POPT_ARG_STRING, &updateKernelPath, 0, |
/* now collapse the array, but don't bother to realloc smaller */ |
3563 |
_("updated information for the specified kernel"), |
for (i = removeHere; i < line->numElements - 1; i++) |
3564 |
_("kernel-path") }, |
line->elements[i] = line->elements[i + 1]; |
3565 |
{ "version", 'v', 0, NULL, 'v', |
|
3566 |
_("print the version of this program and exit"), NULL }, |
line->numElements--; |
3567 |
{ "yaboot", 0, POPT_ARG_NONE, &configureYaboot, 0, |
} |
3568 |
_("configure yaboot bootloader") }, |
|
3569 |
{ "zipl", 0, POPT_ARG_NONE, &configureZipl, 0, |
int argMatch(const char *one, const char *two) |
3570 |
_("configure zipl bootloader") }, |
{ |
3571 |
POPT_AUTOHELP |
char *first, *second; |
3572 |
{ 0, 0, 0, 0, 0 } |
char *chptr; |
3573 |
}; |
|
3574 |
|
first = strcpy(alloca(strlen(one) + 1), one); |
3575 |
useextlinuxmenu=0; |
second = strcpy(alloca(strlen(two) + 1), two); |
3576 |
|
|
3577 |
signal(SIGSEGV, traceback); |
chptr = strchr(first, '='); |
3578 |
|
if (chptr) |
3579 |
optCon = poptGetContext("grubby", argc, argv, options, 0); |
*chptr = '\0'; |
3580 |
poptReadDefaultConfig(optCon, 1); |
|
3581 |
|
chptr = strchr(second, '='); |
3582 |
while ((arg = poptGetNextOpt(optCon)) >= 0) { |
if (chptr) |
3583 |
switch (arg) { |
*chptr = '\0'; |
3584 |
case 'v': |
|
3585 |
printf("grubby version %s\n", VERSION); |
return strcmp(first, second); |
3586 |
exit(0); |
} |
3587 |
break; |
|
3588 |
case 'i': |
int updateActualImage(struct grubConfig *cfg, const char *image, |
3589 |
if (extraInitrdCount < MAX_EXTRA_INITRDS) { |
const char *prefix, const char *addArgs, |
3590 |
extraInitrds[extraInitrdCount++] = strdup(poptGetOptArg(optCon)); |
const char *removeArgs, int multibootArgs) |
3591 |
} else { |
{ |
3592 |
fprintf(stderr, _("grubby: extra initrd maximum is %d\n"), extraInitrdCount); |
struct singleEntry *entry; |
3593 |
|
struct singleLine *line, *rootLine; |
3594 |
|
int index = 0; |
3595 |
|
int i, k; |
3596 |
|
const char **newArgs, **oldArgs; |
3597 |
|
const char **arg; |
3598 |
|
int useKernelArgs, useRoot; |
3599 |
|
int firstElement; |
3600 |
|
int *usedElements; |
3601 |
|
int doreplace; |
3602 |
|
|
3603 |
|
if (!image) |
3604 |
|
return 0; |
3605 |
|
|
3606 |
|
if (!addArgs) { |
3607 |
|
newArgs = malloc(sizeof(*newArgs)); |
3608 |
|
*newArgs = NULL; |
3609 |
|
} else { |
3610 |
|
if (poptParseArgvString(addArgs, NULL, &newArgs)) { |
3611 |
|
fprintf(stderr, |
3612 |
|
_("grubby: error separating arguments '%s'\n"), |
3613 |
|
addArgs); |
3614 |
|
return 1; |
3615 |
|
} |
3616 |
|
} |
3617 |
|
|
3618 |
|
if (!removeArgs) { |
3619 |
|
oldArgs = malloc(sizeof(*oldArgs)); |
3620 |
|
*oldArgs = NULL; |
3621 |
|
} else { |
3622 |
|
if (poptParseArgvString(removeArgs, NULL, &oldArgs)) { |
3623 |
|
fprintf(stderr, |
3624 |
|
_("grubby: error separating arguments '%s'\n"), |
3625 |
|
removeArgs); |
3626 |
|
free(newArgs); |
3627 |
|
return 1; |
3628 |
|
} |
3629 |
|
} |
3630 |
|
|
3631 |
|
useKernelArgs = (getKeywordByType(LT_KERNELARGS, cfg->cfi) |
3632 |
|
&& (!multibootArgs || cfg->cfi->mbConcatArgs)); |
3633 |
|
|
3634 |
|
useRoot = (getKeywordByType(LT_ROOT, cfg->cfi) |
3635 |
|
&& !multibootArgs); |
3636 |
|
|
3637 |
|
for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) { |
3638 |
|
|
3639 |
|
if (multibootArgs && !entry->multiboot) |
3640 |
|
continue; |
3641 |
|
|
3642 |
|
/* Determine where to put the args. If this config supports |
3643 |
|
* LT_KERNELARGS, use that. Otherwise use |
3644 |
|
* LT_HYPER/LT_KERNEL/LT_MBMODULE lines. |
3645 |
|
*/ |
3646 |
|
if (useKernelArgs) { |
3647 |
|
line = getLineByType(LT_KERNELARGS, entry->lines); |
3648 |
|
if (!line) { |
3649 |
|
/* no LT_KERNELARGS, need to add it */ |
3650 |
|
line = addLine(entry, cfg->cfi, LT_KERNELARGS, |
3651 |
|
cfg->secondaryIndent, NULL); |
3652 |
|
} |
3653 |
|
firstElement = 1; |
3654 |
|
|
3655 |
|
} else if (multibootArgs) { |
3656 |
|
line = getLineByType(LT_HYPER, entry->lines); |
3657 |
|
if (!line) { |
3658 |
|
/* a multiboot entry without LT_HYPER? */ |
3659 |
|
continue; |
3660 |
|
} |
3661 |
|
firstElement = 2; |
3662 |
|
|
3663 |
|
} else { |
3664 |
|
line = |
3665 |
|
getLineByType(LT_KERNEL | LT_MBMODULE | |
3666 |
|
LT_KERNEL_EFI | LT_KERNEL_16, |
3667 |
|
entry->lines); |
3668 |
|
if (!line) { |
3669 |
|
/* no LT_KERNEL or LT_MBMODULE in this entry? */ |
3670 |
|
continue; |
3671 |
|
} |
3672 |
|
firstElement = 2; |
3673 |
|
} |
3674 |
|
|
3675 |
|
/* handle the elilo case which does: |
3676 |
|
* append="hypervisor args -- kernel args" |
3677 |
|
*/ |
3678 |
|
if (entry->multiboot && cfg->cfi->mbConcatArgs) { |
3679 |
|
/* this is a multiboot entry, make sure there's |
3680 |
|
* -- on the args line |
3681 |
|
*/ |
3682 |
|
for (i = firstElement; i < line->numElements; i++) { |
3683 |
|
if (!strcmp(line->elements[i].item, "--")) |
3684 |
|
break; |
3685 |
|
} |
3686 |
|
if (i == line->numElements) { |
3687 |
|
/* assume all existing args are kernel args, |
3688 |
|
* prepend -- to make it official |
3689 |
|
*/ |
3690 |
|
insertElement(line, "--", firstElement, |
3691 |
|
cfg->cfi); |
3692 |
|
i = firstElement; |
3693 |
|
} |
3694 |
|
if (!multibootArgs) { |
3695 |
|
/* kernel args start after the -- */ |
3696 |
|
firstElement = i + 1; |
3697 |
|
} |
3698 |
|
} else if (cfg->cfi->mbConcatArgs) { |
3699 |
|
/* this is a non-multiboot entry, remove hyper args */ |
3700 |
|
for (i = firstElement; i < line->numElements; i++) { |
3701 |
|
if (!strcmp(line->elements[i].item, "--")) |
3702 |
|
break; |
3703 |
|
} |
3704 |
|
if (i < line->numElements) { |
3705 |
|
/* remove args up to -- */ |
3706 |
|
while (strcmp |
3707 |
|
(line->elements[firstElement].item, |
3708 |
|
"--")) |
3709 |
|
removeElement(line, firstElement); |
3710 |
|
/* remove -- */ |
3711 |
|
removeElement(line, firstElement); |
3712 |
|
} |
3713 |
|
} |
3714 |
|
|
3715 |
|
usedElements = calloc(line->numElements, sizeof(*usedElements)); |
3716 |
|
|
3717 |
|
for (k = 0, arg = newArgs; *arg; arg++, k++) { |
3718 |
|
|
3719 |
|
doreplace = 1; |
3720 |
|
for (i = firstElement; i < line->numElements; i++) { |
3721 |
|
if (multibootArgs && cfg->cfi->mbConcatArgs && |
3722 |
|
!strcmp(line->elements[i].item, "--")) { |
3723 |
|
/* reached the end of hyper args, insert here */ |
3724 |
|
doreplace = 0; |
3725 |
|
break; |
3726 |
|
} |
3727 |
|
if (usedElements[i]) |
3728 |
|
continue; |
3729 |
|
if (!argMatch(line->elements[i].item, *arg)) { |
3730 |
|
usedElements[i] = 1; |
3731 |
|
break; |
3732 |
|
} |
3733 |
|
} |
3734 |
|
|
3735 |
|
if (i < line->numElements && doreplace) { |
3736 |
|
/* direct replacement */ |
3737 |
|
free(line->elements[i].item); |
3738 |
|
line->elements[i].item = strdup(*arg); |
3739 |
|
|
3740 |
|
} else if (useRoot && !strncmp(*arg, "root=/dev/", 10)) { |
3741 |
|
/* root= replacement */ |
3742 |
|
rootLine = getLineByType(LT_ROOT, entry->lines); |
3743 |
|
if (rootLine) { |
3744 |
|
free(rootLine->elements[1].item); |
3745 |
|
rootLine->elements[1].item = |
3746 |
|
strdup(*arg + 5); |
3747 |
|
} else { |
3748 |
|
rootLine = |
3749 |
|
addLine(entry, cfg->cfi, LT_ROOT, |
3750 |
|
cfg->secondaryIndent, |
3751 |
|
*arg + 5); |
3752 |
|
} |
3753 |
|
} |
3754 |
|
|
3755 |
|
else { |
3756 |
|
/* insert/append */ |
3757 |
|
insertElement(line, *arg, i, cfg->cfi); |
3758 |
|
usedElements = |
3759 |
|
realloc(usedElements, |
3760 |
|
line->numElements * |
3761 |
|
sizeof(*usedElements)); |
3762 |
|
memmove(&usedElements[i + 1], &usedElements[i], |
3763 |
|
line->numElements - i - 1); |
3764 |
|
usedElements[i] = 1; |
3765 |
|
|
3766 |
|
/* if we updated a root= here even though |
3767 |
|
* there is a LT_ROOT available we need to |
3768 |
|
* remove the LT_ROOT entry (this will happen |
3769 |
|
* if we switch from a device to a label) */ |
3770 |
|
if (useRoot && !strncmp(*arg, "root=", 5)) { |
3771 |
|
rootLine = |
3772 |
|
getLineByType(LT_ROOT, |
3773 |
|
entry->lines); |
3774 |
|
if (rootLine) |
3775 |
|
removeLine(entry, rootLine); |
3776 |
|
} |
3777 |
|
} |
3778 |
|
} |
3779 |
|
|
3780 |
|
free(usedElements); |
3781 |
|
|
3782 |
|
for (arg = oldArgs; *arg; arg++) { |
3783 |
|
for (i = firstElement; i < line->numElements; i++) { |
3784 |
|
if (multibootArgs && cfg->cfi->mbConcatArgs && |
3785 |
|
!strcmp(line->elements[i].item, "--")) |
3786 |
|
/* reached the end of hyper args, stop here */ |
3787 |
|
break; |
3788 |
|
if (!argMatch(line->elements[i].item, *arg)) { |
3789 |
|
removeElement(line, i); |
3790 |
|
break; |
3791 |
|
} |
3792 |
|
} |
3793 |
|
/* handle removing LT_ROOT line too */ |
3794 |
|
if (useRoot && !strncmp(*arg, "root=", 5)) { |
3795 |
|
rootLine = getLineByType(LT_ROOT, entry->lines); |
3796 |
|
if (rootLine) |
3797 |
|
removeLine(entry, rootLine); |
3798 |
|
} |
3799 |
|
} |
3800 |
|
|
3801 |
|
if (line->numElements == 1) { |
3802 |
|
/* don't need the line at all (note it has to be a |
3803 |
|
LT_KERNELARGS for this to happen */ |
3804 |
|
removeLine(entry, line); |
3805 |
|
} |
3806 |
|
} |
3807 |
|
|
3808 |
|
free(newArgs); |
3809 |
|
free(oldArgs); |
3810 |
|
|
3811 |
|
return 0; |
3812 |
|
} |
3813 |
|
|
3814 |
|
int updateImage(struct grubConfig *cfg, const char *image, |
3815 |
|
const char *prefix, const char *addArgs, |
3816 |
|
const char *removeArgs, |
3817 |
|
const char *addMBArgs, const char *removeMBArgs) |
3818 |
|
{ |
3819 |
|
int rc = 0; |
3820 |
|
|
3821 |
|
if (!image) |
3822 |
|
return rc; |
3823 |
|
|
3824 |
|
/* update the main args first... */ |
3825 |
|
if (addArgs || removeArgs) |
3826 |
|
rc = updateActualImage(cfg, image, prefix, addArgs, removeArgs, |
3827 |
|
0); |
3828 |
|
if (rc) |
3829 |
|
return rc; |
3830 |
|
|
3831 |
|
/* and now any multiboot args */ |
3832 |
|
if (addMBArgs || removeMBArgs) |
3833 |
|
rc = updateActualImage(cfg, image, prefix, addMBArgs, |
3834 |
|
removeMBArgs, 1); |
3835 |
|
return rc; |
3836 |
|
} |
3837 |
|
|
3838 |
|
int addMBInitrd(struct grubConfig *cfg, const char *newMBKernel, |
3839 |
|
const char *image, const char *prefix, const char *initrd, |
3840 |
|
const char *title) |
3841 |
|
{ |
3842 |
|
struct singleEntry *entry; |
3843 |
|
struct singleLine *line, *kernelLine, *endLine = NULL; |
3844 |
|
int index = 0; |
3845 |
|
|
3846 |
|
if (!image) |
3847 |
|
return 0; |
3848 |
|
|
3849 |
|
for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) { |
3850 |
|
kernelLine = getLineByType(LT_MBMODULE, entry->lines); |
3851 |
|
if (!kernelLine) |
3852 |
|
continue; |
3853 |
|
|
3854 |
|
/* if title is supplied, the entry's title must match it. */ |
3855 |
|
if (title) { |
3856 |
|
char *linetitle; |
3857 |
|
|
3858 |
|
line = |
3859 |
|
getLineByType(LT_TITLE | LT_MENUENTRY, |
3860 |
|
entry->lines); |
3861 |
|
if (!line) |
3862 |
|
continue; |
3863 |
|
|
3864 |
|
linetitle = extractTitle(cfg, line); |
3865 |
|
if (!linetitle) |
3866 |
|
continue; |
3867 |
|
if (strcmp(title, linetitle)) { |
3868 |
|
free(linetitle); |
3869 |
|
continue; |
3870 |
|
} |
3871 |
|
free(linetitle); |
3872 |
|
} |
3873 |
|
|
3874 |
|
if (prefix) { |
3875 |
|
int prefixLen = strlen(prefix); |
3876 |
|
if (!strncmp(initrd, prefix, prefixLen)) |
3877 |
|
initrd += prefixLen; |
3878 |
|
} |
3879 |
|
endLine = getLineByType(LT_ENTRY_END, entry->lines); |
3880 |
|
if (endLine) |
3881 |
|
removeLine(entry, endLine); |
3882 |
|
line = |
3883 |
|
addLine(entry, cfg->cfi, |
3884 |
|
preferredLineType(LT_MBMODULE, cfg->cfi), |
3885 |
|
kernelLine->indent, initrd); |
3886 |
|
if (!line) |
3887 |
|
return 1; |
3888 |
|
if (endLine) { |
3889 |
|
line = addLine(entry, cfg->cfi, LT_ENTRY_END, "", NULL); |
3890 |
|
if (!line) |
3891 |
|
return 1; |
3892 |
|
} |
3893 |
|
|
3894 |
|
break; |
3895 |
|
} |
3896 |
|
|
3897 |
|
return 0; |
3898 |
|
} |
3899 |
|
|
3900 |
|
int updateInitrd(struct grubConfig *cfg, const char *image, |
3901 |
|
const char *prefix, const char *initrd, const char *title) |
3902 |
|
{ |
3903 |
|
struct singleEntry *entry; |
3904 |
|
struct singleLine *line, *kernelLine, *endLine = NULL; |
3905 |
|
int index = 0; |
3906 |
|
|
3907 |
|
if (!image) |
3908 |
|
return 0; |
3909 |
|
|
3910 |
|
for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) { |
3911 |
|
kernelLine = |
3912 |
|
getLineByType(LT_KERNEL | LT_KERNEL_EFI | LT_KERNEL_16, |
3913 |
|
entry->lines); |
3914 |
|
if (!kernelLine) |
3915 |
|
continue; |
3916 |
|
|
3917 |
|
/* if title is supplied, the entry's title must match it. */ |
3918 |
|
if (title) { |
3919 |
|
char *linetitle; |
3920 |
|
|
3921 |
|
line = |
3922 |
|
getLineByType(LT_TITLE | LT_MENUENTRY, |
3923 |
|
entry->lines); |
3924 |
|
if (!line) |
3925 |
|
continue; |
3926 |
|
|
3927 |
|
linetitle = extractTitle(cfg, line); |
3928 |
|
if (!linetitle) |
3929 |
|
continue; |
3930 |
|
if (strcmp(title, linetitle)) { |
3931 |
|
free(linetitle); |
3932 |
|
continue; |
3933 |
|
} |
3934 |
|
free(linetitle); |
3935 |
|
} |
3936 |
|
|
3937 |
|
line = |
3938 |
|
getLineByType(LT_INITRD | LT_INITRD_EFI | LT_INITRD_16, |
3939 |
|
entry->lines); |
3940 |
|
if (line) |
3941 |
|
removeLine(entry, line); |
3942 |
|
if (prefix) { |
3943 |
|
int prefixLen = strlen(prefix); |
3944 |
|
if (!strncmp(initrd, prefix, prefixLen)) |
3945 |
|
initrd += prefixLen; |
3946 |
|
} |
3947 |
|
endLine = getLineByType(LT_ENTRY_END, entry->lines); |
3948 |
|
if (endLine) |
3949 |
|
removeLine(entry, endLine); |
3950 |
|
enum lineType_e lt; |
3951 |
|
switch (kernelLine->type) { |
3952 |
|
case LT_KERNEL: |
3953 |
|
lt = LT_INITRD; |
3954 |
|
break; |
3955 |
|
case LT_KERNEL_EFI: |
3956 |
|
lt = LT_INITRD_EFI; |
3957 |
|
break; |
3958 |
|
case LT_KERNEL_16: |
3959 |
|
lt = LT_INITRD_16; |
3960 |
|
break; |
3961 |
|
default: |
3962 |
|
lt = preferredLineType(LT_INITRD, cfg->cfi); |
3963 |
|
} |
3964 |
|
line = addLine(entry, cfg->cfi, lt, kernelLine->indent, initrd); |
3965 |
|
if (!line) |
3966 |
|
return 1; |
3967 |
|
if (endLine) { |
3968 |
|
line = addLine(entry, cfg->cfi, LT_ENTRY_END, "", NULL); |
3969 |
|
if (!line) |
3970 |
|
return 1; |
3971 |
|
} |
3972 |
|
|
3973 |
|
break; |
3974 |
|
} |
3975 |
|
|
3976 |
|
return 0; |
3977 |
|
} |
3978 |
|
|
3979 |
|
int checkDeviceBootloader(const char *device, const unsigned char *boot) |
3980 |
|
{ |
3981 |
|
int fd; |
3982 |
|
unsigned char bootSect[512]; |
3983 |
|
int offset; |
3984 |
|
|
3985 |
|
fd = open(device, O_RDONLY); |
3986 |
|
if (fd < 0) { |
3987 |
|
fprintf(stderr, _("grubby: unable to open %s: %s\n"), |
3988 |
|
device, strerror(errno)); |
3989 |
return 1; |
return 1; |
|
} |
|
|
break; |
|
3990 |
} |
} |
|
} |
|
3991 |
|
|
3992 |
if (arg < -1) { |
if (read(fd, bootSect, 512) != 512) { |
3993 |
fprintf(stderr, _("grubby: bad argument %s: %s\n"), |
fprintf(stderr, _("grubby: unable to read %s: %s\n"), |
3994 |
poptBadOption(optCon, POPT_BADOPTION_NOALIAS), |
device, strerror(errno)); |
3995 |
poptStrerror(arg)); |
return 1; |
3996 |
return 1; |
} |
3997 |
} |
close(fd); |
3998 |
|
|
3999 |
|
/* first three bytes should match, a jmp short should be in there */ |
4000 |
|
if (memcmp(boot, bootSect, 3)) |
4001 |
|
return 0; |
4002 |
|
|
4003 |
|
if (boot[1] == JMP_SHORT_OPCODE) { |
4004 |
|
offset = boot[2] + 2; |
4005 |
|
} else if (boot[1] == 0xe8 || boot[1] == 0xe9) { |
4006 |
|
offset = (boot[3] << 8) + boot[2] + 2; |
4007 |
|
} else if (boot[0] == JMP_SHORT_OPCODE) { |
4008 |
|
offset = boot[1] + 2; |
4009 |
|
/* |
4010 |
|
* it looks like grub, when copying stage1 into the mbr, |
4011 |
|
* patches stage1 right after the JMP location, replacing |
4012 |
|
* other instructions such as JMPs for NOOPs. So, relax the |
4013 |
|
* check a little bit by skipping those different bytes. |
4014 |
|
*/ |
4015 |
|
if ((bootSect[offset + 1] == NOOP_OPCODE) |
4016 |
|
&& (bootSect[offset + 2] == NOOP_OPCODE)) { |
4017 |
|
offset = offset + 3; |
4018 |
|
} |
4019 |
|
} else if (boot[0] == 0xe8 || boot[0] == 0xe9) { |
4020 |
|
offset = (boot[2] << 8) + boot[1] + 2; |
4021 |
|
} else { |
4022 |
|
return 0; |
4023 |
|
} |
4024 |
|
|
4025 |
|
if (memcmp(boot + offset, bootSect + offset, CODE_SEG_SIZE)) |
4026 |
|
return 0; |
4027 |
|
|
4028 |
|
return 2; |
4029 |
|
} |
4030 |
|
|
4031 |
|
int checkLiloOnRaid(char *mdDev, const unsigned char *boot) |
4032 |
|
{ |
4033 |
|
int fd; |
4034 |
|
char buf[65536]; |
4035 |
|
char *end; |
4036 |
|
char *chptr; |
4037 |
|
char *chptr2; |
4038 |
|
int rc; |
4039 |
|
|
4040 |
|
/* it's on raid; we need to parse /proc/mdstat and check all of the |
4041 |
|
*raw* devices listed in there */ |
4042 |
|
|
4043 |
|
if (!strncmp(mdDev, "/dev/", 5)) |
4044 |
|
mdDev += 5; |
4045 |
|
|
4046 |
|
if ((fd = open("/proc/mdstat", O_RDONLY)) < 0) { |
4047 |
|
fprintf(stderr, _("grubby: failed to open /proc/mdstat: %s\n"), |
4048 |
|
strerror(errno)); |
4049 |
|
return 2; |
4050 |
|
} |
4051 |
|
|
4052 |
|
rc = read(fd, buf, sizeof(buf) - 1); |
4053 |
|
if (rc < 0 || rc == (sizeof(buf) - 1)) { |
4054 |
|
fprintf(stderr, _("grubby: failed to read /proc/mdstat: %s\n"), |
4055 |
|
strerror(errno)); |
4056 |
|
close(fd); |
4057 |
|
return 2; |
4058 |
|
} |
4059 |
|
close(fd); |
4060 |
|
buf[rc] = '\0'; |
4061 |
|
|
4062 |
|
chptr = buf; |
4063 |
|
while (*chptr) { |
4064 |
|
end = strchr(chptr, '\n'); |
4065 |
|
if (!end) |
4066 |
|
break; |
4067 |
|
*end = '\0'; |
4068 |
|
|
4069 |
|
if (!strncmp(chptr, mdDev, strlen(mdDev)) && |
4070 |
|
chptr[strlen(mdDev)] == ' ') { |
4071 |
|
|
4072 |
|
/* found the device */ |
4073 |
|
while (*chptr && *chptr != ':') |
4074 |
|
chptr++; |
4075 |
|
chptr++; |
4076 |
|
while (*chptr && isspace(*chptr)) |
4077 |
|
chptr++; |
4078 |
|
|
4079 |
|
/* skip the "active" bit */ |
4080 |
|
while (*chptr && !isspace(*chptr)) |
4081 |
|
chptr++; |
4082 |
|
while (*chptr && isspace(*chptr)) |
4083 |
|
chptr++; |
4084 |
|
|
4085 |
|
/* skip the raid level */ |
4086 |
|
while (*chptr && !isspace(*chptr)) |
4087 |
|
chptr++; |
4088 |
|
while (*chptr && isspace(*chptr)) |
4089 |
|
chptr++; |
4090 |
|
|
4091 |
|
/* everything else is partition stuff */ |
4092 |
|
while (*chptr) { |
4093 |
|
chptr2 = chptr; |
4094 |
|
while (*chptr2 && *chptr2 != '[') |
4095 |
|
chptr2++; |
4096 |
|
if (!*chptr2) |
4097 |
|
break; |
4098 |
|
|
4099 |
|
/* yank off the numbers at the end */ |
4100 |
|
chptr2--; |
4101 |
|
while (isdigit(*chptr2) && chptr2 > chptr) |
4102 |
|
chptr2--; |
4103 |
|
chptr2++; |
4104 |
|
*chptr2 = '\0'; |
4105 |
|
|
4106 |
|
/* Better, now we need the /dev/ back. We're |
4107 |
|
* done with everything before this point, so |
4108 |
|
* we can just put the /dev/ part there. |
4109 |
|
* There will always be room. */ |
4110 |
|
memcpy(chptr - 5, "/dev/", 5); |
4111 |
|
rc = checkDeviceBootloader(chptr - 5, boot); |
4112 |
|
if (rc != 2) { |
4113 |
|
return rc; |
4114 |
|
} |
4115 |
|
|
4116 |
|
chptr = chptr2 + 1; |
4117 |
|
/* skip the [11] bit */ |
4118 |
|
while (*chptr && !isspace(*chptr)) |
4119 |
|
chptr++; |
4120 |
|
/* and move to the next one */ |
4121 |
|
while (*chptr && isspace(*chptr)) |
4122 |
|
chptr++; |
4123 |
|
} |
4124 |
|
|
4125 |
|
/* we're good to go */ |
4126 |
|
return 2; |
4127 |
|
} |
4128 |
|
|
4129 |
|
chptr = end + 1; |
4130 |
|
} |
4131 |
|
|
4132 |
|
fprintf(stderr, |
4133 |
|
_("grubby: raid device /dev/%s not found in /proc/mdstat\n"), |
4134 |
|
mdDev); |
4135 |
|
return 0; |
4136 |
|
} |
4137 |
|
|
4138 |
|
int checkForLilo(struct grubConfig *config) |
4139 |
|
{ |
4140 |
|
int fd; |
4141 |
|
unsigned char boot[512]; |
4142 |
|
struct singleLine *line; |
4143 |
|
|
4144 |
|
for (line = config->theLines; line; line = line->next) |
4145 |
|
if (line->type == LT_BOOT) |
4146 |
|
break; |
4147 |
|
|
4148 |
|
if (!line) { |
4149 |
|
fprintf(stderr, |
4150 |
|
_ |
4151 |
|
("grubby: no boot line found in lilo configuration\n")); |
4152 |
|
return 1; |
4153 |
|
} |
4154 |
|
|
4155 |
|
if (line->numElements != 2) |
4156 |
|
return 1; |
4157 |
|
|
4158 |
|
fd = open("/boot/boot.b", O_RDONLY); |
4159 |
|
if (fd < 0) { |
4160 |
|
fprintf(stderr, _("grubby: unable to open %s: %s\n"), |
4161 |
|
"/boot/boot.b", strerror(errno)); |
4162 |
|
return 1; |
4163 |
|
} |
4164 |
|
|
4165 |
|
if (read(fd, boot, 512) != 512) { |
4166 |
|
fprintf(stderr, _("grubby: unable to read %s: %s\n"), |
4167 |
|
"/boot/boot.b", strerror(errno)); |
4168 |
|
return 1; |
4169 |
|
} |
4170 |
|
close(fd); |
4171 |
|
|
4172 |
|
if (!strncmp("/dev/md", line->elements[1].item, 7)) |
4173 |
|
return checkLiloOnRaid(line->elements[1].item, boot); |
4174 |
|
|
4175 |
|
return checkDeviceBootloader(line->elements[1].item, boot); |
4176 |
|
} |
4177 |
|
|
4178 |
|
int checkForGrub2(struct grubConfig *config) |
4179 |
|
{ |
4180 |
|
if (!access("/etc/grub.d/", R_OK)) |
4181 |
|
return 2; |
4182 |
|
|
|
if ((chptr = poptGetArg(optCon))) { |
|
|
fprintf(stderr, _("grubby: unexpected argument %s\n"), chptr); |
|
4183 |
return 1; |
return 1; |
4184 |
} |
} |
4185 |
|
|
4186 |
|
int checkForGrub(struct grubConfig *config) |
4187 |
|
{ |
4188 |
|
int fd; |
4189 |
|
unsigned char bootSect[512]; |
4190 |
|
char *boot; |
4191 |
|
int onSuse = isSuseSystem(); |
4192 |
|
|
4193 |
|
if (onSuse) { |
4194 |
|
if (parseSuseGrubConf(NULL, &boot)) |
4195 |
|
return 0; |
4196 |
|
} else { |
4197 |
|
if (parseSysconfigGrub(NULL, &boot)) |
4198 |
|
return 0; |
4199 |
|
} |
4200 |
|
|
4201 |
|
/* assume grub is not installed -- not an error condition */ |
4202 |
|
if (!boot) |
4203 |
|
return 0; |
4204 |
|
|
4205 |
|
fd = open("/boot/grub/stage1", O_RDONLY); |
4206 |
|
if (fd < 0) |
4207 |
|
/* this doesn't exist if grub hasn't been installed */ |
4208 |
|
return 0; |
4209 |
|
|
4210 |
|
if (read(fd, bootSect, 512) != 512) { |
4211 |
|
fprintf(stderr, _("grubby: unable to read %s: %s\n"), |
4212 |
|
"/boot/grub/stage1", strerror(errno)); |
4213 |
|
close(fd); |
4214 |
|
return 1; |
4215 |
|
} |
4216 |
|
close(fd); |
4217 |
|
|
4218 |
|
/* The more elaborate checks do not work on SuSE. The checks done |
4219 |
|
* seem to be reasonble (at least for now), so just return success |
4220 |
|
*/ |
4221 |
|
if (onSuse) |
4222 |
|
return 2; |
4223 |
|
|
4224 |
|
return checkDeviceBootloader(boot, bootSect); |
4225 |
|
} |
4226 |
|
|
4227 |
|
int checkForExtLinux(struct grubConfig *config) |
4228 |
|
{ |
4229 |
|
int fd; |
4230 |
|
unsigned char bootSect[512]; |
4231 |
|
char *boot; |
4232 |
|
char executable[] = "/boot/extlinux/extlinux"; |
4233 |
|
|
4234 |
|
printf("entered: checkForExtLinux()\n"); |
4235 |
|
|
4236 |
|
if (parseSysconfigGrub(NULL, &boot)) |
4237 |
|
return 0; |
4238 |
|
|
4239 |
|
/* assume grub is not installed -- not an error condition */ |
4240 |
|
if (!boot) |
4241 |
|
return 0; |
4242 |
|
|
4243 |
|
fd = open(executable, O_RDONLY); |
4244 |
|
if (fd < 0) |
4245 |
|
/* this doesn't exist if grub hasn't been installed */ |
4246 |
|
return 0; |
4247 |
|
|
4248 |
|
if (read(fd, bootSect, 512) != 512) { |
4249 |
|
fprintf(stderr, _("grubby: unable to read %s: %s\n"), |
4250 |
|
executable, strerror(errno)); |
4251 |
|
return 1; |
4252 |
|
} |
4253 |
|
close(fd); |
4254 |
|
|
4255 |
|
return checkDeviceBootloader(boot, bootSect); |
4256 |
|
} |
4257 |
|
|
4258 |
|
int checkForYaboot(struct grubConfig *config) |
4259 |
|
{ |
4260 |
|
/* |
4261 |
|
* This is a simplistic check that we consider good enough for own puporses |
4262 |
|
* |
4263 |
|
* If we were to properly check if yaboot is *installed* we'd need to: |
4264 |
|
* 1) get the system boot device (LT_BOOT) |
4265 |
|
* 2) considering it's a raw filesystem, check if the yaboot binary matches |
4266 |
|
* the content on the boot device |
4267 |
|
* 3) if not, copy the binary to a temporary file and run "addnote" on it |
4268 |
|
* 4) check again if binary and boot device contents match |
4269 |
|
*/ |
4270 |
|
if (!access("/etc/yaboot.conf", R_OK)) |
4271 |
|
return 2; |
4272 |
|
|
|
if ((configureLilo + configureGrub2 + configureGrub + configureELilo + |
|
|
configureYaboot + configureSilo + configureZipl + |
|
|
configureExtLinux ) > 1) { |
|
|
fprintf(stderr, _("grubby: cannot specify multiple bootloaders\n")); |
|
4273 |
return 1; |
return 1; |
4274 |
} else if (bootloaderProbe && grubConfig) { |
} |
4275 |
fprintf(stderr, |
|
4276 |
_("grubby: cannot specify config file with --bootloader-probe\n")); |
int checkForElilo(struct grubConfig *config) |
4277 |
|
{ |
4278 |
|
if (!access("/etc/elilo.conf", R_OK)) |
4279 |
|
return 2; |
4280 |
|
|
4281 |
return 1; |
return 1; |
4282 |
} else if (configureGrub2) { |
} |
4283 |
cfi = &grub2ConfigType; |
|
4284 |
} else if (configureLilo) { |
static char *getRootSpecifier(char *str) |
4285 |
cfi = &liloConfigType; |
{ |
4286 |
} else if (configureGrub) { |
char *idx, *rootspec = NULL; |
4287 |
cfi = &grubConfigType; |
|
4288 |
} else if (configureELilo) { |
if (*str == '(') { |
4289 |
cfi = &eliloConfigType; |
idx = rootspec = strdup(str); |
4290 |
} else if (configureYaboot) { |
while (*idx && (*idx != ')') && (!isspace(*idx))) |
4291 |
cfi = &yabootConfigType; |
idx++; |
4292 |
} else if (configureSilo) { |
*(++idx) = '\0'; |
4293 |
cfi = &siloConfigType; |
} |
4294 |
} else if (configureZipl) { |
return rootspec; |
4295 |
cfi = &ziplConfigType; |
} |
|
} else if (configureExtLinux) { |
|
|
cfi = &extlinuxConfigType; |
|
|
useextlinuxmenu=1; |
|
|
} |
|
4296 |
|
|
4297 |
if (!cfi) { |
static char *getInitrdVal(struct grubConfig *config, |
4298 |
if (grub2FindConfig(&grub2ConfigType)) |
const char *prefix, struct singleLine *tmplLine, |
4299 |
cfi = &grub2ConfigType; |
const char *newKernelInitrd, |
4300 |
|
const char **extraInitrds, int extraInitrdCount) |
4301 |
|
{ |
4302 |
|
char *initrdVal, *end; |
4303 |
|
int i; |
4304 |
|
size_t totalSize; |
4305 |
|
size_t prefixLen; |
4306 |
|
char separatorChar; |
4307 |
|
|
4308 |
|
prefixLen = strlen(prefix); |
4309 |
|
totalSize = strlen(newKernelInitrd) - prefixLen + 1 /* \0 */ ; |
4310 |
|
|
4311 |
|
for (i = 0; i < extraInitrdCount; i++) { |
4312 |
|
totalSize += sizeof(separatorChar); |
4313 |
|
totalSize += strlen(extraInitrds[i]) - prefixLen; |
4314 |
|
} |
4315 |
|
|
4316 |
|
initrdVal = end = malloc(totalSize); |
4317 |
|
|
4318 |
|
end = stpcpy(end, newKernelInitrd + prefixLen); |
4319 |
|
|
4320 |
|
separatorChar = getKeywordByType(LT_INITRD, config->cfi)->separatorChar; |
4321 |
|
for (i = 0; i < extraInitrdCount; i++) { |
4322 |
|
const char *extraInitrd; |
4323 |
|
int j; |
4324 |
|
|
4325 |
|
extraInitrd = extraInitrds[i] + prefixLen; |
4326 |
|
/* Don't add entries that are already there */ |
4327 |
|
if (tmplLine != NULL) { |
4328 |
|
for (j = 2; j < tmplLine->numElements; j++) |
4329 |
|
if (strcmp |
4330 |
|
(extraInitrd, |
4331 |
|
tmplLine->elements[j].item) == 0) |
4332 |
|
break; |
4333 |
|
|
4334 |
|
if (j != tmplLine->numElements) |
4335 |
|
continue; |
4336 |
|
} |
4337 |
|
|
4338 |
|
*end++ = separatorChar; |
4339 |
|
end = stpcpy(end, extraInitrd); |
4340 |
|
} |
4341 |
|
|
4342 |
|
return initrdVal; |
4343 |
|
} |
4344 |
|
|
4345 |
|
int addNewKernel(struct grubConfig *config, struct singleEntry *template, |
4346 |
|
const char *prefix, |
4347 |
|
const char *newKernelPath, const char *newKernelTitle, |
4348 |
|
const char *newKernelArgs, const char *newKernelInitrd, |
4349 |
|
const char **extraInitrds, int extraInitrdCount, |
4350 |
|
const char *newMBKernel, const char *newMBKernelArgs, |
4351 |
|
const char *newDevTreePath, int newIndex) |
4352 |
|
{ |
4353 |
|
struct singleEntry *new, *entry, *prev = NULL; |
4354 |
|
struct singleLine *newLine = NULL, *tmplLine = NULL, *masterLine = NULL; |
4355 |
|
int needs; |
4356 |
|
char *indexs; |
4357 |
|
char *chptr; |
4358 |
|
int rc; |
4359 |
|
|
4360 |
|
if (!newKernelPath) |
4361 |
|
return 0; |
4362 |
|
|
4363 |
|
rc = asprintf(&indexs, "%d", newIndex); |
4364 |
|
if (rc < 0) |
4365 |
|
return 1; |
4366 |
|
|
4367 |
|
/* if the newKernelTitle is too long silently munge it into something |
4368 |
|
* we can live with. truncating is first check, then we'll just mess with |
4369 |
|
* it until it looks better */ |
4370 |
|
if (config->cfi->maxTitleLength && |
4371 |
|
(strlen(newKernelTitle) > config->cfi->maxTitleLength)) { |
4372 |
|
char *buf = alloca(config->cfi->maxTitleLength + 7); |
4373 |
|
char *numBuf = alloca(config->cfi->maxTitleLength + 1); |
4374 |
|
int i = 1; |
4375 |
|
|
4376 |
|
sprintf(buf, "TITLE=%.*s", config->cfi->maxTitleLength, |
4377 |
|
newKernelTitle); |
4378 |
|
while (findEntryByPath(config, buf, NULL, NULL)) { |
4379 |
|
sprintf(numBuf, "%d", i++); |
4380 |
|
strcpy(buf + strlen(buf) - strlen(numBuf), numBuf); |
4381 |
|
} |
4382 |
|
|
4383 |
|
newKernelTitle = buf + 6; |
4384 |
|
} |
4385 |
|
|
4386 |
|
new = malloc(sizeof(*new)); |
4387 |
|
new->skip = 0; |
4388 |
|
new->multiboot = 0; |
4389 |
|
new->lines = NULL; |
4390 |
|
entry = config->entries; |
4391 |
|
for (unsigned int i = 0; i < newIndex; i++) { |
4392 |
|
if (!entry) |
4393 |
|
break; |
4394 |
|
prev = entry; |
4395 |
|
entry = entry->next; |
4396 |
|
} |
4397 |
|
new->next = entry; |
4398 |
|
|
4399 |
|
if (prev) |
4400 |
|
prev->next = new; |
4401 |
else |
else |
4402 |
#ifdef __ia64__ |
config->entries = new; |
|
cfi = &eliloConfigType; |
|
|
#elif __powerpc__ |
|
|
cfi = &yabootConfigType; |
|
|
#elif __sparc__ |
|
|
cfi = &siloConfigType; |
|
|
#elif __s390__ |
|
|
cfi = &ziplConfigType; |
|
|
#elif __s390x__ |
|
|
cfi = &ziplConfigtype; |
|
|
#else |
|
|
cfi = &grubConfigType; |
|
|
#endif |
|
|
} |
|
4403 |
|
|
4404 |
if (!grubConfig) { |
/* copy/update from the template */ |
4405 |
if (cfi->findConfig) |
needs = NEED_KERNEL | NEED_TITLE; |
4406 |
grubConfig = cfi->findConfig(cfi); |
if (newKernelInitrd) |
4407 |
if (!grubConfig) |
needs |= NEED_INITRD; |
4408 |
grubConfig = cfi->defaultConfig; |
if (newMBKernel) { |
4409 |
} |
needs |= NEED_MB; |
4410 |
|
new->multiboot = 1; |
4411 |
|
} |
4412 |
|
if (newDevTreePath && getKeywordByType(LT_DEVTREE, config->cfi)) |
4413 |
|
needs |= NEED_DEVTREE; |
4414 |
|
|
4415 |
|
if (template) { |
4416 |
|
for (masterLine = template->lines; |
4417 |
|
masterLine && (tmplLine = lineDup(masterLine)); |
4418 |
|
lineFree(tmplLine), masterLine = masterLine->next) { |
4419 |
|
dbgPrintf("addNewKernel processing %d\n", |
4420 |
|
tmplLine->type); |
4421 |
|
|
4422 |
|
/* skip comments */ |
4423 |
|
chptr = tmplLine->indent; |
4424 |
|
while (*chptr && isspace(*chptr)) |
4425 |
|
chptr++; |
4426 |
|
if (*chptr == '#') |
4427 |
|
continue; |
4428 |
|
|
4429 |
|
if (iskernel(tmplLine->type) |
4430 |
|
&& tmplLine->numElements >= 2) { |
4431 |
|
if (!template->multiboot && (needs & NEED_MB)) { |
4432 |
|
/* it's not a multiboot template and |
4433 |
|
* this is the kernel line. Try to |
4434 |
|
* be intelligent about inserting the |
4435 |
|
* hypervisor at the same time. |
4436 |
|
*/ |
4437 |
|
if (config->cfi->mbHyperFirst) { |
4438 |
|
/* insert the hypervisor first */ |
4439 |
|
newLine = |
4440 |
|
addLine(new, config->cfi, |
4441 |
|
LT_HYPER, |
4442 |
|
tmplLine->indent, |
4443 |
|
newMBKernel + |
4444 |
|
strlen(prefix)); |
4445 |
|
/* set up for adding the |
4446 |
|
* kernel line */ |
4447 |
|
free(tmplLine->indent); |
4448 |
|
tmplLine->indent = |
4449 |
|
strdup(config-> |
4450 |
|
secondaryIndent); |
4451 |
|
needs &= ~NEED_MB; |
4452 |
|
} |
4453 |
|
if (needs & NEED_KERNEL) { |
4454 |
|
/* use addLineTmpl to |
4455 |
|
* preserve line elements, |
4456 |
|
* otherwise we could just |
4457 |
|
* call addLine. |
4458 |
|
* Unfortunately this means |
4459 |
|
* making some changes to the |
4460 |
|
* template such as the |
4461 |
|
* indent change above and |
4462 |
|
* the type change below. |
4463 |
|
*/ |
4464 |
|
struct keywordTypes *mbm_kw = |
4465 |
|
getKeywordByType |
4466 |
|
(LT_MBMODULE, config->cfi); |
4467 |
|
if (mbm_kw) { |
4468 |
|
tmplLine->type = |
4469 |
|
LT_MBMODULE; |
4470 |
|
free(tmplLine-> |
4471 |
|
elements[0].item); |
4472 |
|
tmplLine->elements[0]. |
4473 |
|
item = |
4474 |
|
strdup(mbm_kw->key); |
4475 |
|
} |
4476 |
|
newLine = |
4477 |
|
addLineTmpl(new, tmplLine, |
4478 |
|
newLine, |
4479 |
|
newKernelPath + |
4480 |
|
strlen(prefix), |
4481 |
|
config->cfi); |
4482 |
|
needs &= ~NEED_KERNEL; |
4483 |
|
} |
4484 |
|
if (needs & NEED_MB) { /* !mbHyperFirst */ |
4485 |
|
newLine = |
4486 |
|
addLine(new, config->cfi, |
4487 |
|
LT_HYPER, |
4488 |
|
config-> |
4489 |
|
secondaryIndent, |
4490 |
|
newMBKernel + |
4491 |
|
strlen(prefix)); |
4492 |
|
needs &= ~NEED_MB; |
4493 |
|
} |
4494 |
|
} else if (needs & NEED_KERNEL) { |
4495 |
|
newLine = |
4496 |
|
addLineTmpl(new, tmplLine, newLine, |
4497 |
|
newKernelPath + |
4498 |
|
strlen(prefix), |
4499 |
|
config->cfi); |
4500 |
|
needs &= ~NEED_KERNEL; |
4501 |
|
} |
4502 |
|
|
4503 |
|
} else if (tmplLine->type == LT_HYPER && |
4504 |
|
tmplLine->numElements >= 2) { |
4505 |
|
if (needs & NEED_MB) { |
4506 |
|
newLine = |
4507 |
|
addLineTmpl(new, tmplLine, newLine, |
4508 |
|
newMBKernel + |
4509 |
|
strlen(prefix), |
4510 |
|
config->cfi); |
4511 |
|
needs &= ~NEED_MB; |
4512 |
|
} |
4513 |
|
|
4514 |
|
} else if (tmplLine->type == LT_MBMODULE && |
4515 |
|
tmplLine->numElements >= 2) { |
4516 |
|
if (new->multiboot) { |
4517 |
|
if (needs & NEED_KERNEL) { |
4518 |
|
newLine = |
4519 |
|
addLineTmpl(new, tmplLine, |
4520 |
|
newLine, |
4521 |
|
newKernelPath + |
4522 |
|
strlen(prefix), |
4523 |
|
config->cfi); |
4524 |
|
needs &= ~NEED_KERNEL; |
4525 |
|
} else if (config->cfi->mbInitRdIsModule |
4526 |
|
&& (needs & NEED_INITRD)) { |
4527 |
|
char *initrdVal; |
4528 |
|
initrdVal = |
4529 |
|
getInitrdVal(config, prefix, |
4530 |
|
tmplLine, |
4531 |
|
newKernelInitrd, |
4532 |
|
extraInitrds, |
4533 |
|
extraInitrdCount); |
4534 |
|
newLine = |
4535 |
|
addLineTmpl(new, tmplLine, |
4536 |
|
newLine, |
4537 |
|
initrdVal, |
4538 |
|
config->cfi); |
4539 |
|
free(initrdVal); |
4540 |
|
needs &= ~NEED_INITRD; |
4541 |
|
} |
4542 |
|
} else if (needs & NEED_KERNEL) { |
4543 |
|
/* template is multi but new is not, |
4544 |
|
* insert the kernel in the first |
4545 |
|
* module slot |
4546 |
|
*/ |
4547 |
|
tmplLine->type = |
4548 |
|
preferredLineType(LT_KERNEL, |
4549 |
|
config->cfi); |
4550 |
|
free(tmplLine->elements[0].item); |
4551 |
|
tmplLine->elements[0].item = |
4552 |
|
strdup(getKeywordByType |
4553 |
|
(tmplLine->type, |
4554 |
|
config->cfi)->key); |
4555 |
|
newLine = |
4556 |
|
addLineTmpl(new, tmplLine, newLine, |
4557 |
|
newKernelPath + |
4558 |
|
strlen(prefix), |
4559 |
|
config->cfi); |
4560 |
|
needs &= ~NEED_KERNEL; |
4561 |
|
} else if (needs & NEED_INITRD) { |
4562 |
|
char *initrdVal; |
4563 |
|
/* template is multi but new is not, |
4564 |
|
* insert the initrd in the second |
4565 |
|
* module slot |
4566 |
|
*/ |
4567 |
|
tmplLine->type = |
4568 |
|
preferredLineType(LT_INITRD, |
4569 |
|
config->cfi); |
4570 |
|
free(tmplLine->elements[0].item); |
4571 |
|
tmplLine->elements[0].item = |
4572 |
|
strdup(getKeywordByType |
4573 |
|
(tmplLine->type, |
4574 |
|
config->cfi)->key); |
4575 |
|
initrdVal = |
4576 |
|
getInitrdVal(config, prefix, |
4577 |
|
tmplLine, |
4578 |
|
newKernelInitrd, |
4579 |
|
extraInitrds, |
4580 |
|
extraInitrdCount); |
4581 |
|
newLine = |
4582 |
|
addLineTmpl(new, tmplLine, newLine, |
4583 |
|
initrdVal, config->cfi); |
4584 |
|
free(initrdVal); |
4585 |
|
needs &= ~NEED_INITRD; |
4586 |
|
} |
4587 |
|
|
4588 |
|
} else if (isinitrd(tmplLine->type) |
4589 |
|
&& tmplLine->numElements >= 2) { |
4590 |
|
if (needs & NEED_INITRD && new->multiboot |
4591 |
|
&& !template->multiboot |
4592 |
|
&& config->cfi->mbInitRdIsModule) { |
4593 |
|
/* make sure we don't insert the |
4594 |
|
* module initrd before the module |
4595 |
|
* kernel... if we don't do it here, |
4596 |
|
* it will be inserted following the |
4597 |
|
* template. |
4598 |
|
*/ |
4599 |
|
if (!needs & NEED_KERNEL) { |
4600 |
|
char *initrdVal; |
4601 |
|
|
4602 |
|
initrdVal = |
4603 |
|
getInitrdVal(config, prefix, |
4604 |
|
tmplLine, |
4605 |
|
newKernelInitrd, |
4606 |
|
extraInitrds, |
4607 |
|
extraInitrdCount); |
4608 |
|
newLine = |
4609 |
|
addLine(new, config->cfi, |
4610 |
|
LT_MBMODULE, |
4611 |
|
config-> |
4612 |
|
secondaryIndent, |
4613 |
|
initrdVal); |
4614 |
|
free(initrdVal); |
4615 |
|
needs &= ~NEED_INITRD; |
4616 |
|
} |
4617 |
|
} else if (needs & NEED_INITRD) { |
4618 |
|
char *initrdVal; |
4619 |
|
initrdVal = |
4620 |
|
getInitrdVal(config, prefix, |
4621 |
|
tmplLine, |
4622 |
|
newKernelInitrd, |
4623 |
|
extraInitrds, |
4624 |
|
extraInitrdCount); |
4625 |
|
newLine = |
4626 |
|
addLineTmpl(new, tmplLine, newLine, |
4627 |
|
initrdVal, config->cfi); |
4628 |
|
free(initrdVal); |
4629 |
|
needs &= ~NEED_INITRD; |
4630 |
|
} |
4631 |
|
|
4632 |
|
} else if (tmplLine->type == LT_MENUENTRY && |
4633 |
|
(needs & NEED_TITLE)) { |
4634 |
|
requote(tmplLine, config->cfi); |
4635 |
|
char *nkt = malloc(strlen(newKernelTitle) + 3); |
4636 |
|
strcpy(nkt, "'"); |
4637 |
|
strcat(nkt, newKernelTitle); |
4638 |
|
strcat(nkt, "'"); |
4639 |
|
newLine = |
4640 |
|
addLineTmpl(new, tmplLine, newLine, nkt, |
4641 |
|
config->cfi); |
4642 |
|
free(nkt); |
4643 |
|
needs &= ~NEED_TITLE; |
4644 |
|
} else if (tmplLine->type == LT_TITLE && |
4645 |
|
(needs & NEED_TITLE)) { |
4646 |
|
if (tmplLine->numElements >= 2) { |
4647 |
|
newLine = |
4648 |
|
addLineTmpl(new, tmplLine, newLine, |
4649 |
|
newKernelTitle, |
4650 |
|
config->cfi); |
4651 |
|
needs &= ~NEED_TITLE; |
4652 |
|
} else if (tmplLine->numElements == 1 && |
4653 |
|
config->cfi->titleBracketed) { |
4654 |
|
/* addLineTmpl doesn't handle |
4655 |
|
* titleBracketed */ |
4656 |
|
newLine = |
4657 |
|
addLine(new, config->cfi, LT_TITLE, |
4658 |
|
tmplLine->indent, |
4659 |
|
newKernelTitle); |
4660 |
|
needs &= ~NEED_TITLE; |
4661 |
|
} |
4662 |
|
} else if (tmplLine->type == LT_ECHO) { |
4663 |
|
requote(tmplLine, config->cfi); |
4664 |
|
static const char *prefix = "'Loading "; |
4665 |
|
if (tmplLine->numElements > 1 && |
4666 |
|
strstr(tmplLine->elements[1].item, prefix) |
4667 |
|
&& masterLine->next |
4668 |
|
&& iskernel(masterLine->next->type)) { |
4669 |
|
char *newTitle = |
4670 |
|
malloc(strlen(prefix) + |
4671 |
|
strlen(newKernelTitle) + 2); |
4672 |
|
|
4673 |
|
strcpy(newTitle, prefix); |
4674 |
|
strcat(newTitle, newKernelTitle); |
4675 |
|
strcat(newTitle, "'"); |
4676 |
|
newLine = |
4677 |
|
addLine(new, config->cfi, LT_ECHO, |
4678 |
|
tmplLine->indent, newTitle); |
4679 |
|
free(newTitle); |
4680 |
|
} else { |
4681 |
|
/* pass through other lines from the |
4682 |
|
* template */ |
4683 |
|
newLine = |
4684 |
|
addLineTmpl(new, tmplLine, newLine, |
4685 |
|
NULL, config->cfi); |
4686 |
|
} |
4687 |
|
} else if (tmplLine->type == LT_DEVTREE && |
4688 |
|
tmplLine->numElements == 2 |
4689 |
|
&& newDevTreePath) { |
4690 |
|
newLine = |
4691 |
|
addLineTmpl(new, tmplLine, newLine, |
4692 |
|
newDevTreePath + strlen(prefix), |
4693 |
|
config->cfi); |
4694 |
|
needs &= ~NEED_DEVTREE; |
4695 |
|
} else if (tmplLine->type == LT_ENTRY_END |
4696 |
|
&& needs & NEED_DEVTREE) { |
4697 |
|
const char *ndtp = newDevTreePath; |
4698 |
|
if (!strncmp |
4699 |
|
(newDevTreePath, prefix, strlen(prefix))) |
4700 |
|
ndtp += strlen(prefix); |
4701 |
|
newLine = addLine(new, config->cfi, LT_DEVTREE, |
4702 |
|
config->secondaryIndent, |
4703 |
|
ndtp); |
4704 |
|
needs &= ~NEED_DEVTREE; |
4705 |
|
newLine = |
4706 |
|
addLineTmpl(new, tmplLine, newLine, NULL, |
4707 |
|
config->cfi); |
4708 |
|
} else { |
4709 |
|
/* pass through other lines from the template */ |
4710 |
|
newLine = |
4711 |
|
addLineTmpl(new, tmplLine, newLine, NULL, |
4712 |
|
config->cfi); |
4713 |
|
} |
4714 |
|
} |
4715 |
|
|
4716 |
|
} else { |
4717 |
|
/* don't have a template, so start the entry with the |
4718 |
|
* appropriate starting line |
4719 |
|
*/ |
4720 |
|
switch (config->cfi->entryStart) { |
4721 |
|
case LT_KERNEL: |
4722 |
|
case LT_KERNEL_EFI: |
4723 |
|
case LT_KERNEL_16: |
4724 |
|
if (new->multiboot && config->cfi->mbHyperFirst) { |
4725 |
|
/* fall through to LT_HYPER */ |
4726 |
|
} else { |
4727 |
|
newLine = addLine(new, config->cfi, |
4728 |
|
preferredLineType(LT_KERNEL, |
4729 |
|
config-> |
4730 |
|
cfi), |
4731 |
|
config->primaryIndent, |
4732 |
|
newKernelPath + |
4733 |
|
strlen(prefix)); |
4734 |
|
needs &= ~NEED_KERNEL; |
4735 |
|
break; |
4736 |
|
} |
4737 |
|
|
4738 |
if (bootloaderProbe && (displayDefault || kernelInfo || newKernelVersion || |
case LT_HYPER: |
4739 |
newKernelPath || removeKernelPath || makeDefault || |
newLine = addLine(new, config->cfi, LT_HYPER, |
4740 |
defaultKernel || displayDefaultIndex || displayDefaultTitle)) { |
config->primaryIndent, |
4741 |
fprintf(stderr, _("grubby: --bootloader-probe may not be used with " |
newMBKernel + strlen(prefix)); |
4742 |
|
needs &= ~NEED_MB; |
4743 |
|
break; |
4744 |
|
|
4745 |
|
case LT_MENUENTRY:{ |
4746 |
|
char *nkt = malloc(strlen(newKernelTitle) + 3); |
4747 |
|
strcpy(nkt, "'"); |
4748 |
|
strcat(nkt, newKernelTitle); |
4749 |
|
strcat(nkt, "'"); |
4750 |
|
newLine = |
4751 |
|
addLine(new, config->cfi, LT_MENUENTRY, |
4752 |
|
config->primaryIndent, nkt); |
4753 |
|
free(nkt); |
4754 |
|
needs &= ~NEED_TITLE; |
4755 |
|
needs |= NEED_END; |
4756 |
|
break; |
4757 |
|
} |
4758 |
|
case LT_TITLE: |
4759 |
|
if (useextlinuxmenu != 0) { // We just need useextlinuxmenu to not be zero (set above) |
4760 |
|
char *templabel; |
4761 |
|
int x = 0, y = 0; |
4762 |
|
|
4763 |
|
templabel = strdup(newKernelTitle); |
4764 |
|
while (templabel[x]) { |
4765 |
|
if (templabel[x] == ' ') { |
4766 |
|
y = x; |
4767 |
|
while (templabel[y]) { |
4768 |
|
templabel[y] = |
4769 |
|
templabel[y + 1]; |
4770 |
|
y++; |
4771 |
|
} |
4772 |
|
} |
4773 |
|
x++; |
4774 |
|
} |
4775 |
|
newLine = addLine(new, config->cfi, LT_TITLE, |
4776 |
|
config->primaryIndent, |
4777 |
|
templabel); |
4778 |
|
free(templabel); |
4779 |
|
} else { |
4780 |
|
newLine = addLine(new, config->cfi, LT_TITLE, |
4781 |
|
config->primaryIndent, |
4782 |
|
newKernelTitle); |
4783 |
|
} |
4784 |
|
needs &= ~NEED_TITLE; |
4785 |
|
break; |
4786 |
|
|
4787 |
|
default: |
4788 |
|
abort(); |
4789 |
|
} |
4790 |
|
} |
4791 |
|
|
4792 |
|
struct singleLine *endLine = NULL; |
4793 |
|
endLine = getLineByType(LT_ENTRY_END, new->lines); |
4794 |
|
if (endLine) { |
4795 |
|
removeLine(new, endLine); |
4796 |
|
needs |= NEED_END; |
4797 |
|
} |
4798 |
|
|
4799 |
|
/* add the remainder of the lines, i.e. those that either |
4800 |
|
* weren't present in the template, or in the case of no template, |
4801 |
|
* all the lines following the entryStart. |
4802 |
|
*/ |
4803 |
|
if (needs & NEED_TITLE) { |
4804 |
|
newLine = addLine(new, config->cfi, LT_TITLE, |
4805 |
|
config->secondaryIndent, newKernelTitle); |
4806 |
|
needs &= ~NEED_TITLE; |
4807 |
|
} |
4808 |
|
if ((needs & NEED_MB) && config->cfi->mbHyperFirst) { |
4809 |
|
newLine = addLine(new, config->cfi, LT_HYPER, |
4810 |
|
config->secondaryIndent, |
4811 |
|
newMBKernel + strlen(prefix)); |
4812 |
|
needs &= ~NEED_MB; |
4813 |
|
} |
4814 |
|
if (needs & NEED_KERNEL) { |
4815 |
|
newLine = addLine(new, config->cfi, |
4816 |
|
(new->multiboot |
4817 |
|
&& getKeywordByType(LT_MBMODULE, |
4818 |
|
config->cfi)) |
4819 |
|
? LT_MBMODULE : preferredLineType(LT_KERNEL, |
4820 |
|
config-> |
4821 |
|
cfi), |
4822 |
|
config->secondaryIndent, |
4823 |
|
newKernelPath + strlen(prefix)); |
4824 |
|
needs &= ~NEED_KERNEL; |
4825 |
|
} |
4826 |
|
if (needs & NEED_MB) { |
4827 |
|
newLine = addLine(new, config->cfi, LT_HYPER, |
4828 |
|
config->secondaryIndent, |
4829 |
|
newMBKernel + strlen(prefix)); |
4830 |
|
needs &= ~NEED_MB; |
4831 |
|
} |
4832 |
|
if (needs & NEED_INITRD) { |
4833 |
|
char *initrdVal; |
4834 |
|
initrdVal = |
4835 |
|
getInitrdVal(config, prefix, NULL, newKernelInitrd, |
4836 |
|
extraInitrds, extraInitrdCount); |
4837 |
|
newLine = |
4838 |
|
addLine(new, config->cfi, |
4839 |
|
(new->multiboot |
4840 |
|
&& getKeywordByType(LT_MBMODULE, config->cfi)) |
4841 |
|
? LT_MBMODULE : preferredLineType(LT_INITRD, |
4842 |
|
config->cfi), |
4843 |
|
config->secondaryIndent, initrdVal); |
4844 |
|
free(initrdVal); |
4845 |
|
needs &= ~NEED_INITRD; |
4846 |
|
} |
4847 |
|
if (needs & NEED_DEVTREE) { |
4848 |
|
newLine = addLine(new, config->cfi, LT_DEVTREE, |
4849 |
|
config->secondaryIndent, newDevTreePath); |
4850 |
|
needs &= ~NEED_DEVTREE; |
4851 |
|
} |
4852 |
|
|
4853 |
|
/* NEEDS_END must be last on bootloaders that need it... */ |
4854 |
|
if (needs & NEED_END) { |
4855 |
|
newLine = addLine(new, config->cfi, LT_ENTRY_END, |
4856 |
|
config->secondaryIndent, NULL); |
4857 |
|
needs &= ~NEED_END; |
4858 |
|
} |
4859 |
|
|
4860 |
|
if (needs) { |
4861 |
|
printf(_("grubby: needs=%d, aborting\n"), needs); |
4862 |
|
abort(); |
4863 |
|
} |
4864 |
|
|
4865 |
|
if (updateImage(config, indexs, prefix, newKernelArgs, NULL, |
4866 |
|
newMBKernelArgs, NULL)) { |
4867 |
|
config->isModified = 1; |
4868 |
|
return 1; |
4869 |
|
} |
4870 |
|
|
4871 |
|
return 0; |
4872 |
|
} |
4873 |
|
|
4874 |
|
int main(int argc, const char **argv) |
4875 |
|
{ |
4876 |
|
poptContext optCon; |
4877 |
|
const char *grubConfig = NULL; |
4878 |
|
char *outputFile = NULL; |
4879 |
|
int arg = 0; |
4880 |
|
int flags = 0; |
4881 |
|
int badImageOkay = 0; |
4882 |
|
int configureGrub2 = 0; |
4883 |
|
int configureLilo = 0, configureELilo = 0, configureGrub = 0; |
4884 |
|
int configureYaboot = 0, configureSilo = 0, configureZipl = 0; |
4885 |
|
int configureExtLinux = 0; |
4886 |
|
int bootloaderProbe = 0; |
4887 |
|
int extraInitrdCount = 0; |
4888 |
|
char *updateKernelPath = NULL; |
4889 |
|
char *newKernelPath = NULL; |
4890 |
|
char *removeKernelPath = NULL; |
4891 |
|
char *newKernelArgs = NULL; |
4892 |
|
char *newKernelInitrd = NULL; |
4893 |
|
char *newKernelTitle = NULL; |
4894 |
|
char *newDevTreePath = NULL; |
4895 |
|
char *newMBKernel = NULL; |
4896 |
|
char *newMBKernelArgs = NULL; |
4897 |
|
int newIndex = 0; |
4898 |
|
char *removeMBKernelArgs = NULL; |
4899 |
|
char *removeMBKernel = NULL; |
4900 |
|
char *bootPrefix = NULL; |
4901 |
|
char *defaultKernel = NULL; |
4902 |
|
char *removeArgs = NULL; |
4903 |
|
char *kernelInfo = NULL; |
4904 |
|
char *extraInitrds[MAX_EXTRA_INITRDS] = { NULL }; |
4905 |
|
char *envPath = NULL; |
4906 |
|
const char *chptr = NULL; |
4907 |
|
struct configFileInfo *cfi = NULL; |
4908 |
|
struct grubConfig *config; |
4909 |
|
struct singleEntry *template = NULL; |
4910 |
|
int copyDefault = 0, makeDefault = 0; |
4911 |
|
int displayDefault = 0; |
4912 |
|
int displayDefaultIndex = 0; |
4913 |
|
int displayDefaultTitle = 0; |
4914 |
|
int defaultIndex = -1; |
4915 |
|
struct poptOption options[] = { |
4916 |
|
{"add-kernel", 0, POPT_ARG_STRING, &newKernelPath, 0, |
4917 |
|
_("add an entry for the specified kernel"), _("kernel-path")}, |
4918 |
|
{"add-multiboot", 0, POPT_ARG_STRING, &newMBKernel, 0, |
4919 |
|
_("add an entry for the specified multiboot kernel"), NULL}, |
4920 |
|
{"args", 0, POPT_ARG_STRING, &newKernelArgs, 0, |
4921 |
|
_("default arguments for the new kernel or new arguments for " |
4922 |
|
"kernel being updated"), _("args")}, |
4923 |
|
{"mbargs", 0, POPT_ARG_STRING, &newMBKernelArgs, 0, |
4924 |
|
_("default arguments for the new multiboot kernel or " |
4925 |
|
"new arguments for multiboot kernel being updated"), NULL}, |
4926 |
|
{"bad-image-okay", 0, 0, &badImageOkay, 0, |
4927 |
|
_ |
4928 |
|
("don't sanity check images in boot entries (for testing only)"), |
4929 |
|
NULL}, |
4930 |
|
{"boot-filesystem", 0, POPT_ARG_STRING, &bootPrefix, 0, |
4931 |
|
_ |
4932 |
|
("filesystem which contains /boot directory (for testing only)"), |
4933 |
|
_("bootfs")}, |
4934 |
|
#if defined(__i386__) || defined(__x86_64__) || defined (__powerpc64__) || defined (__ia64__) |
4935 |
|
{"bootloader-probe", 0, POPT_ARG_NONE, &bootloaderProbe, 0, |
4936 |
|
_("check which bootloader is installed on boot sector")}, |
4937 |
|
#endif |
4938 |
|
{"config-file", 'c', POPT_ARG_STRING, &grubConfig, 0, |
4939 |
|
_("path to grub config file to update (\"-\" for stdin)"), |
4940 |
|
_("path")}, |
4941 |
|
{"copy-default", 0, 0, ©Default, 0, |
4942 |
|
_("use the default boot entry as a template for the new entry " |
4943 |
|
"being added; if the default is not a linux image, or if " |
4944 |
|
"the kernel referenced by the default image does not exist, " |
4945 |
|
"the first linux entry whose kernel does exist is used as the " |
4946 |
|
"template"), NULL}, |
4947 |
|
{"debug", 0, 0, &debug, 0, |
4948 |
|
_("print debugging information for failures")}, |
4949 |
|
{"default-kernel", 0, 0, &displayDefault, 0, |
4950 |
|
_("display the path of the default kernel")}, |
4951 |
|
{"default-index", 0, 0, &displayDefaultIndex, 0, |
4952 |
|
_("display the index of the default kernel")}, |
4953 |
|
{"default-title", 0, 0, &displayDefaultTitle, 0, |
4954 |
|
_("display the title of the default kernel")}, |
4955 |
|
{"devtree", 0, POPT_ARG_STRING, &newDevTreePath, 0, |
4956 |
|
_("device tree file for new stanza"), _("dtb-path")}, |
4957 |
|
{"devtreedir", 0, POPT_ARG_STRING, &newDevTreePath, 0, |
4958 |
|
_("device tree directory for new stanza"), _("dtb-path")}, |
4959 |
|
{"elilo", 0, POPT_ARG_NONE, &configureELilo, 0, |
4960 |
|
_("configure elilo bootloader")}, |
4961 |
|
{"efi", 0, POPT_ARG_NONE, &isEfi, 0, |
4962 |
|
_("force grub2 stanzas to use efi")}, |
4963 |
|
{"env", 0, POPT_ARG_STRING, &envPath, 0, |
4964 |
|
_("path for environment data"), |
4965 |
|
_("path")}, |
4966 |
|
{"extlinux", 0, POPT_ARG_NONE, &configureExtLinux, 0, |
4967 |
|
_("configure extlinux bootloader (from syslinux)")}, |
4968 |
|
{"grub", 0, POPT_ARG_NONE, &configureGrub, 0, |
4969 |
|
_("configure grub bootloader")}, |
4970 |
|
{"grub2", 0, POPT_ARG_NONE, &configureGrub2, 0, |
4971 |
|
_("configure grub2 bootloader")}, |
4972 |
|
{"info", 0, POPT_ARG_STRING, &kernelInfo, 0, |
4973 |
|
_("display boot information for specified kernel"), |
4974 |
|
_("kernel-path")}, |
4975 |
|
{"initrd", 0, POPT_ARG_STRING, &newKernelInitrd, 0, |
4976 |
|
_("initrd image for the new kernel"), _("initrd-path")}, |
4977 |
|
{"extra-initrd", 'i', POPT_ARG_STRING, NULL, 'i', |
4978 |
|
_ |
4979 |
|
("auxiliary initrd image for things other than the new kernel"), |
4980 |
|
_("initrd-path")}, |
4981 |
|
{"lilo", 0, POPT_ARG_NONE, &configureLilo, 0, |
4982 |
|
_("configure lilo bootloader")}, |
4983 |
|
{"make-default", 0, 0, &makeDefault, 0, |
4984 |
|
_("make the newly added entry the default boot entry"), NULL}, |
4985 |
|
{"output-file", 'o', POPT_ARG_STRING, &outputFile, 0, |
4986 |
|
_("path to output updated config file (\"-\" for stdout)"), |
4987 |
|
_("path")}, |
4988 |
|
{"remove-args", 0, POPT_ARG_STRING, &removeArgs, 0, |
4989 |
|
_("remove kernel arguments"), NULL}, |
4990 |
|
{"remove-mbargs", 0, POPT_ARG_STRING, &removeMBKernelArgs, 0, |
4991 |
|
_("remove multiboot kernel arguments"), NULL}, |
4992 |
|
{"remove-kernel", 0, POPT_ARG_STRING, &removeKernelPath, 0, |
4993 |
|
_("remove all entries for the specified kernel"), |
4994 |
|
_("kernel-path")}, |
4995 |
|
{"remove-multiboot", 0, POPT_ARG_STRING, &removeMBKernel, 0, |
4996 |
|
_("remove all entries for the specified multiboot kernel"), |
4997 |
|
NULL}, |
4998 |
|
{"set-default", 0, POPT_ARG_STRING, &defaultKernel, 0, |
4999 |
|
_("make the first entry referencing the specified kernel " |
5000 |
|
"the default"), _("kernel-path")}, |
5001 |
|
{"set-default-index", 0, POPT_ARG_INT, &defaultIndex, 0, |
5002 |
|
_("make the given entry index the default entry"), |
5003 |
|
_("entry-index")}, |
5004 |
|
{"set-index", 0, POPT_ARG_INT, &newIndex, 0, |
5005 |
|
_("use the given index when creating a new entry"), |
5006 |
|
_("entry-index")}, |
5007 |
|
{"silo", 0, POPT_ARG_NONE, &configureSilo, 0, |
5008 |
|
_("configure silo bootloader")}, |
5009 |
|
{"title", 0, POPT_ARG_STRING, &newKernelTitle, 0, |
5010 |
|
_("title to use for the new kernel entry"), _("entry-title")}, |
5011 |
|
{"update-kernel", 0, POPT_ARG_STRING, &updateKernelPath, 0, |
5012 |
|
_("updated information for the specified kernel"), |
5013 |
|
_("kernel-path")}, |
5014 |
|
{"version", 'v', 0, NULL, 'v', |
5015 |
|
_("print the version of this program and exit"), NULL}, |
5016 |
|
{"yaboot", 0, POPT_ARG_NONE, &configureYaboot, 0, |
5017 |
|
_("configure yaboot bootloader")}, |
5018 |
|
{"zipl", 0, POPT_ARG_NONE, &configureZipl, 0, |
5019 |
|
_("configure zipl bootloader")}, |
5020 |
|
POPT_AUTOHELP {0, 0, 0, 0, 0} |
5021 |
|
}; |
5022 |
|
|
5023 |
|
useextlinuxmenu = 0; |
5024 |
|
|
5025 |
|
int i = 0; |
5026 |
|
for (int j = 1; j < argc; j++) |
5027 |
|
i += strlen(argv[j]) + 1; |
5028 |
|
saved_command_line = malloc(i); |
5029 |
|
if (!saved_command_line) { |
5030 |
|
fprintf(stderr, "grubby: %m\n"); |
5031 |
|
exit(1); |
5032 |
|
} |
5033 |
|
saved_command_line[0] = '\0'; |
5034 |
|
for (int j = 1; j < argc; j++) { |
5035 |
|
strcat(saved_command_line, argv[j]); |
5036 |
|
strncat(saved_command_line, j == argc - 1 ? "" : " ", 1); |
5037 |
|
} |
5038 |
|
|
5039 |
|
optCon = poptGetContext("grubby", argc, argv, options, 0); |
5040 |
|
poptReadDefaultConfig(optCon, 1); |
5041 |
|
|
5042 |
|
while ((arg = poptGetNextOpt(optCon)) >= 0) { |
5043 |
|
switch (arg) { |
5044 |
|
case 'v': |
5045 |
|
printf("grubby version %s\n", VERSION); |
5046 |
|
exit(0); |
5047 |
|
break; |
5048 |
|
case 'i': |
5049 |
|
if (extraInitrdCount < MAX_EXTRA_INITRDS) { |
5050 |
|
extraInitrds[extraInitrdCount++] = |
5051 |
|
strdup(poptGetOptArg(optCon)); |
5052 |
|
} else { |
5053 |
|
fprintf(stderr, |
5054 |
|
_ |
5055 |
|
("grubby: extra initrd maximum is %d\n"), |
5056 |
|
extraInitrdCount); |
5057 |
|
return 1; |
5058 |
|
} |
5059 |
|
break; |
5060 |
|
} |
5061 |
|
} |
5062 |
|
|
5063 |
|
if (arg < -1) { |
5064 |
|
fprintf(stderr, _("grubby: bad argument %s: %s\n"), |
5065 |
|
poptBadOption(optCon, POPT_BADOPTION_NOALIAS), |
5066 |
|
poptStrerror(arg)); |
5067 |
|
return 1; |
5068 |
|
} |
5069 |
|
|
5070 |
|
if ((chptr = poptGetArg(optCon))) { |
5071 |
|
fprintf(stderr, _("grubby: unexpected argument %s\n"), chptr); |
5072 |
|
return 1; |
5073 |
|
} |
5074 |
|
|
5075 |
|
if ((configureLilo + configureGrub2 + configureGrub + configureELilo + |
5076 |
|
configureYaboot + configureSilo + configureZipl + |
5077 |
|
configureExtLinux) > 1) { |
5078 |
|
fprintf(stderr, |
5079 |
|
_("grubby: cannot specify multiple bootloaders\n")); |
5080 |
|
return 1; |
5081 |
|
} else if (bootloaderProbe && grubConfig) { |
5082 |
|
fprintf(stderr, |
5083 |
|
_ |
5084 |
|
("grubby: cannot specify config file with --bootloader-probe\n")); |
5085 |
|
return 1; |
5086 |
|
} else if (configureGrub2) { |
5087 |
|
cfi = &grub2ConfigType; |
5088 |
|
if (envPath) |
5089 |
|
cfi->envFile = envPath; |
5090 |
|
} else if (configureLilo) { |
5091 |
|
cfi = &liloConfigType; |
5092 |
|
} else if (configureGrub) { |
5093 |
|
cfi = &grubConfigType; |
5094 |
|
} else if (configureELilo) { |
5095 |
|
cfi = &eliloConfigType; |
5096 |
|
} else if (configureYaboot) { |
5097 |
|
cfi = &yabootConfigType; |
5098 |
|
} else if (configureSilo) { |
5099 |
|
cfi = &siloConfigType; |
5100 |
|
} else if (configureZipl) { |
5101 |
|
cfi = &ziplConfigType; |
5102 |
|
} else if (configureExtLinux) { |
5103 |
|
cfi = &extlinuxConfigType; |
5104 |
|
useextlinuxmenu = 1; |
5105 |
|
} |
5106 |
|
|
5107 |
|
if (!cfi) { |
5108 |
|
if (grub2FindConfig(&grub2ConfigType)) { |
5109 |
|
cfi = &grub2ConfigType; |
5110 |
|
if (envPath) |
5111 |
|
cfi->envFile = envPath; |
5112 |
|
} else |
5113 |
|
#ifdef __ia64__ |
5114 |
|
cfi = &eliloConfigType; |
5115 |
|
#elif __powerpc__ |
5116 |
|
cfi = &yabootConfigType; |
5117 |
|
#elif __sparc__ |
5118 |
|
cfi = &siloConfigType; |
5119 |
|
#elif __s390__ |
5120 |
|
cfi = &ziplConfigType; |
5121 |
|
#elif __s390x__ |
5122 |
|
cfi = &ziplConfigtype; |
5123 |
|
#else |
5124 |
|
cfi = &grubConfigType; |
5125 |
|
#endif |
5126 |
|
} |
5127 |
|
|
5128 |
|
if (!grubConfig) { |
5129 |
|
if (cfi->findConfig) |
5130 |
|
grubConfig = cfi->findConfig(cfi); |
5131 |
|
if (!grubConfig) |
5132 |
|
grubConfig = cfi->defaultConfig; |
5133 |
|
} |
5134 |
|
|
5135 |
|
if (bootloaderProbe && (displayDefault || kernelInfo || |
5136 |
|
newKernelPath || removeKernelPath || makeDefault |
5137 |
|
|| defaultKernel || displayDefaultIndex |
5138 |
|
|| displayDefaultTitle |
5139 |
|
|| (defaultIndex >= 0))) { |
5140 |
|
fprintf(stderr, |
5141 |
|
_("grubby: --bootloader-probe may not be used with " |
5142 |
"specified option")); |
"specified option")); |
5143 |
return 1; |
return 1; |
5144 |
} |
} |
5145 |
|
|
5146 |
if ((displayDefault || kernelInfo) && (newKernelVersion || newKernelPath || |
if ((displayDefault || kernelInfo) && (newKernelPath || |
5147 |
removeKernelPath)) { |
removeKernelPath)) { |
5148 |
fprintf(stderr, _("grubby: --default-kernel and --info may not " |
fprintf(stderr, _("grubby: --default-kernel and --info may not " |
5149 |
"be used when adding or removing kernels\n")); |
"be used when adding or removing kernels\n")); |
5150 |
return 1; |
return 1; |
5151 |
} |
} |
5152 |
|
|
5153 |
if (newKernelPath && !newKernelTitle) { |
if (newKernelPath && !newKernelTitle) { |
5154 |
fprintf(stderr, _("grubby: kernel title must be specified\n")); |
fprintf(stderr, _("grubby: kernel title must be specified\n")); |
5155 |
return 1; |
return 1; |
5156 |
} else if (!newKernelPath && (newKernelTitle || copyDefault || |
} else if (!newKernelPath && (copyDefault || |
5157 |
(newKernelInitrd && !updateKernelPath)|| |
(newKernelInitrd && !updateKernelPath) || |
5158 |
makeDefault || extraInitrdCount > 0)) { |
makeDefault || extraInitrdCount > 0)) { |
5159 |
fprintf(stderr, _("grubby: kernel path expected\n")); |
fprintf(stderr, _("grubby: kernel path expected\n")); |
5160 |
return 1; |
return 1; |
5161 |
} |
} |
5162 |
|
|
5163 |
if (newKernelPath && updateKernelPath) { |
if (newKernelPath && updateKernelPath) { |
5164 |
fprintf(stderr, _("grubby: --add-kernel and --update-kernel may" |
fprintf(stderr, _("grubby: --add-kernel and --update-kernel may" |
5165 |
"not be used together")); |
"not be used together")); |
5166 |
return 1; |
return 1; |
5167 |
} |
} |
5168 |
|
|
5169 |
if (makeDefault && defaultKernel) { |
if (makeDefault && defaultKernel) { |
5170 |
fprintf(stderr, _("grubby: --make-default and --default-kernel " |
fprintf(stderr, _("grubby: --make-default and --default-kernel " |
5171 |
|
"may not be used together\n")); |
5172 |
|
return 1; |
5173 |
|
} else if (defaultKernel && removeKernelPath && |
5174 |
|
!strcmp(defaultKernel, removeKernelPath)) { |
5175 |
|
fprintf(stderr, |
5176 |
|
_("grubby: cannot make removed kernel the default\n")); |
5177 |
|
return 1; |
5178 |
|
} else if (defaultKernel && newKernelPath && |
5179 |
|
!strcmp(defaultKernel, newKernelPath)) { |
5180 |
|
makeDefault = 1; |
5181 |
|
defaultKernel = NULL; |
5182 |
|
} else if (defaultKernel && (defaultIndex >= 0)) { |
5183 |
|
fprintf(stderr, |
5184 |
|
_("grubby: --set-default and --set-default-index " |
5185 |
"may not be used together\n")); |
"may not be used together\n")); |
5186 |
return 1; |
return 1; |
5187 |
} else if (defaultKernel && removeKernelPath && |
} |
|
!strcmp(defaultKernel, removeKernelPath)) { |
|
|
fprintf(stderr, _("grubby: cannot make removed kernel the default\n")); |
|
|
return 1; |
|
|
} else if (defaultKernel && newKernelPath && |
|
|
!strcmp(defaultKernel, newKernelPath)) { |
|
|
makeDefault = 1; |
|
|
defaultKernel = NULL; |
|
|
} |
|
5188 |
|
|
5189 |
if (grubConfig && !strcmp(grubConfig, "-") && !outputFile) { |
if (grubConfig && !strcmp(grubConfig, "-") && !outputFile) { |
5190 |
fprintf(stderr, _("grubby: output file must be specified if stdin " |
fprintf(stderr, |
5191 |
"is used\n")); |
_("grubby: output file must be specified if stdin " |
5192 |
return 1; |
"is used\n")); |
5193 |
} |
return 1; |
5194 |
|
} |
5195 |
|
|
5196 |
if (!removeKernelPath && !newKernelPath && !displayDefault && !defaultKernel |
if (!removeKernelPath && !newKernelPath && !displayDefault |
5197 |
&& !kernelInfo && !bootloaderProbe && !updateKernelPath |
&& !defaultKernel && !kernelInfo && !bootloaderProbe |
5198 |
&& !removeMBKernel && !displayDefaultIndex && !displayDefaultTitle) { |
&& !updateKernelPath && !removeMBKernel && !displayDefaultIndex |
5199 |
fprintf(stderr, _("grubby: no action specified\n")); |
&& !displayDefaultTitle && (defaultIndex == -1)) { |
5200 |
return 1; |
fprintf(stderr, _("grubby: no action specified\n")); |
5201 |
} |
return 1; |
5202 |
|
} |
5203 |
|
|
5204 |
flags |= badImageOkay ? GRUBBY_BADIMAGE_OKAY : 0; |
flags |= badImageOkay ? GRUBBY_BADIMAGE_OKAY : 0; |
5205 |
|
|
5206 |
if (cfi->needsBootPrefix) { |
if (cfi->needsBootPrefix) { |
5207 |
if (!bootPrefix) { |
if (!bootPrefix) { |
5208 |
bootPrefix = findBootPrefix(); |
bootPrefix = findBootPrefix(); |
5209 |
if (!bootPrefix) return 1; |
if (!bootPrefix) |
5210 |
} else { |
return 1; |
5211 |
/* this shouldn't end with a / */ |
} else { |
5212 |
if (bootPrefix[strlen(bootPrefix) - 1] == '/') |
/* this shouldn't end with a / */ |
5213 |
bootPrefix[strlen(bootPrefix) - 1] = '\0'; |
if (bootPrefix[strlen(bootPrefix) - 1] == '/') |
5214 |
|
bootPrefix[strlen(bootPrefix) - 1] = '\0'; |
5215 |
|
} |
5216 |
|
} else { |
5217 |
|
bootPrefix = ""; |
5218 |
} |
} |
|
} else { |
|
|
bootPrefix = ""; |
|
|
} |
|
5219 |
|
|
5220 |
if (!cfi->mbAllowExtraInitRds && |
if (!cfi->mbAllowExtraInitRds && extraInitrdCount > 0) { |
5221 |
extraInitrdCount > 0) { |
fprintf(stderr, |
5222 |
fprintf(stderr, _("grubby: %s doesn't allow multiple initrds\n"), cfi->defaultConfig); |
_("grubby: %s doesn't allow multiple initrds\n"), |
5223 |
return 1; |
cfi->defaultConfig); |
5224 |
} |
return 1; |
5225 |
|
} |
5226 |
|
|
5227 |
if (bootloaderProbe) { |
if (bootloaderProbe) { |
5228 |
int lrc = 0, grc = 0, gr2c = 0, erc = 0; |
int lrc = 0, grc = 0, gr2c = 0, extrc = 0, yrc = 0, erc = 0; |
5229 |
struct grubConfig * lconfig, * gconfig; |
struct grubConfig *lconfig, *gconfig, *yconfig, *econfig; |
5230 |
|
|
5231 |
const char *grub2config = grub2FindConfig(&grub2ConfigType); |
const char *grub2config = grub2FindConfig(&grub2ConfigType); |
5232 |
if (grub2config) { |
if (grub2config) { |
5233 |
gconfig = readConfig(grub2config, &grub2ConfigType); |
gconfig = readConfig(grub2config, &grub2ConfigType); |
5234 |
if (!gconfig) |
if (!gconfig) |
5235 |
gr2c = 1; |
gr2c = 1; |
5236 |
else |
else |
5237 |
gr2c = checkForGrub2(gconfig); |
gr2c = checkForGrub2(gconfig); |
5238 |
} |
} |
|
|
|
|
const char *grubconfig = grubFindConfig(&grubConfigType); |
|
|
if (!access(grubconfig, F_OK)) { |
|
|
gconfig = readConfig(grubconfig, &grubConfigType); |
|
|
if (!gconfig) |
|
|
grc = 1; |
|
|
else |
|
|
grc = checkForGrub(gconfig); |
|
|
} |
|
|
|
|
|
if (!access(liloConfigType.defaultConfig, F_OK)) { |
|
|
lconfig = readConfig(liloConfigType.defaultConfig, &liloConfigType); |
|
|
if (!lconfig) |
|
|
lrc = 1; |
|
|
else |
|
|
lrc = checkForLilo(lconfig); |
|
|
} |
|
|
|
|
|
if (!access(extlinuxConfigType.defaultConfig, F_OK)) { |
|
|
lconfig = readConfig(extlinuxConfigType.defaultConfig, &extlinuxConfigType); |
|
|
if (!lconfig) |
|
|
erc = 1; |
|
|
else |
|
|
erc = checkForExtLinux(lconfig); |
|
|
} |
|
|
|
|
|
if (lrc == 1 || grc == 1 || gr2c == 1) return 1; |
|
|
|
|
|
if (lrc == 2) printf("lilo\n"); |
|
|
if (gr2c == 2) printf("grub2\n"); |
|
|
if (grc == 2) printf("grub\n"); |
|
|
if (erc == 2) printf("extlinux\n"); |
|
5239 |
|
|
5240 |
return 0; |
const char *grubconfig = grubFindConfig(&grubConfigType); |
5241 |
} |
if (!access(grubconfig, F_OK)) { |
5242 |
|
gconfig = readConfig(grubconfig, &grubConfigType); |
5243 |
|
if (!gconfig) |
5244 |
|
grc = 1; |
5245 |
|
else |
5246 |
|
grc = checkForGrub(gconfig); |
5247 |
|
} |
5248 |
|
|
5249 |
config = readConfig(grubConfig, cfi); |
if (!access(liloConfigType.defaultConfig, F_OK)) { |
5250 |
if (!config) return 1; |
lconfig = |
5251 |
|
readConfig(liloConfigType.defaultConfig, |
5252 |
|
&liloConfigType); |
5253 |
|
if (!lconfig) |
5254 |
|
lrc = 1; |
5255 |
|
else |
5256 |
|
lrc = checkForLilo(lconfig); |
5257 |
|
} |
5258 |
|
|
5259 |
if (displayDefault) { |
if (!access(eliloConfigType.defaultConfig, F_OK)) { |
5260 |
struct singleLine * line; |
econfig = readConfig(eliloConfigType.defaultConfig, |
5261 |
struct singleEntry * entry; |
&eliloConfigType); |
5262 |
char * rootspec; |
if (!econfig) |
5263 |
|
erc = 1; |
5264 |
if (config->defaultImage == -1) return 0; |
else |
5265 |
entry = findEntryByIndex(config, config->defaultImage); |
erc = checkForElilo(econfig); |
5266 |
if (!entry) return 0; |
} |
|
if (!suitableImage(entry, bootPrefix, 0, flags)) return 0; |
|
|
|
|
|
line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines); |
|
|
if (!line) return 0; |
|
|
|
|
|
rootspec = getRootSpecifier(line->elements[1].item); |
|
|
printf("%s%s\n", bootPrefix, line->elements[1].item + |
|
|
((rootspec != NULL) ? strlen(rootspec) : 0)); |
|
5267 |
|
|
5268 |
return 0; |
if (!access(extlinuxConfigType.defaultConfig, F_OK)) { |
5269 |
|
lconfig = |
5270 |
|
readConfig(extlinuxConfigType.defaultConfig, |
5271 |
|
&extlinuxConfigType); |
5272 |
|
if (!lconfig) |
5273 |
|
extrc = 1; |
5274 |
|
else |
5275 |
|
extrc = checkForExtLinux(lconfig); |
5276 |
|
} |
5277 |
|
|
5278 |
|
if (!access(yabootConfigType.defaultConfig, F_OK)) { |
5279 |
|
yconfig = readConfig(yabootConfigType.defaultConfig, |
5280 |
|
&yabootConfigType); |
5281 |
|
if (!yconfig) |
5282 |
|
yrc = 1; |
5283 |
|
else |
5284 |
|
yrc = checkForYaboot(yconfig); |
5285 |
|
} |
5286 |
|
|
5287 |
|
if (lrc == 1 || grc == 1 || gr2c == 1 || extrc == 1 || yrc == 1 |
5288 |
|
|| erc == 1) |
5289 |
|
return 1; |
5290 |
|
|
5291 |
|
if (lrc == 2) |
5292 |
|
printf("lilo\n"); |
5293 |
|
if (gr2c == 2) |
5294 |
|
printf("grub2\n"); |
5295 |
|
if (grc == 2) |
5296 |
|
printf("grub\n"); |
5297 |
|
if (extrc == 2) |
5298 |
|
printf("extlinux\n"); |
5299 |
|
if (yrc == 2) |
5300 |
|
printf("yaboot\n"); |
5301 |
|
if (erc == 2) |
5302 |
|
printf("elilo\n"); |
5303 |
|
|
5304 |
} else if (displayDefaultTitle) { |
return 0; |
|
struct singleLine * line; |
|
|
struct singleEntry * entry; |
|
|
|
|
|
if (config->defaultImage == -1) return 0; |
|
|
entry = findEntryByIndex(config, config->defaultImage); |
|
|
if (!entry) return 0; |
|
|
|
|
|
if (!configureGrub2) { |
|
|
line = getLineByType(LT_TITLE, entry->lines); |
|
|
if (!line) return 0; |
|
|
printf("%s\n", line->elements[1].item); |
|
|
|
|
|
} else { |
|
|
char * title; |
|
|
|
|
|
dbgPrintf("This is GRUB2, default title is embeded in menuentry\n"); |
|
|
line = getLineByType(LT_MENUENTRY, entry->lines); |
|
|
if (!line) return 0; |
|
|
title = grub2ExtractTitle(line); |
|
|
if (title) |
|
|
printf("%s\n", title); |
|
5305 |
} |
} |
|
return 0; |
|
5306 |
|
|
5307 |
} else if (displayDefaultIndex) { |
if (grubConfig == NULL) { |
5308 |
if (config->defaultImage == -1) return 0; |
printf("Could not find bootloader configuration file.\n"); |
5309 |
printf("%i\n", config->defaultImage); |
exit(1); |
5310 |
|
} |
|
} else if (kernelInfo) |
|
|
return displayInfo(config, kernelInfo, bootPrefix); |
|
|
|
|
|
if (copyDefault) { |
|
|
template = findTemplate(config, bootPrefix, NULL, 0, flags); |
|
|
if (!template) return 1; |
|
|
} |
|
5311 |
|
|
5312 |
markRemovedImage(config, removeKernelPath, bootPrefix); |
config = readConfig(grubConfig, cfi); |
5313 |
markRemovedImage(config, removeMBKernel, bootPrefix); |
if (!config) |
5314 |
setDefaultImage(config, newKernelPath != NULL, defaultKernel, makeDefault, |
return 1; |
5315 |
bootPrefix, flags); |
|
5316 |
setFallbackImage(config, newKernelPath != NULL); |
if (displayDefault) { |
5317 |
if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs, |
struct singleLine *line; |
5318 |
removeArgs, newMBKernelArgs, removeMBKernelArgs)) return 1; |
struct singleEntry *entry; |
5319 |
if (updateKernelPath && newKernelInitrd) { |
char *rootspec; |
5320 |
if (updateInitrd(config, updateKernelPath, bootPrefix, |
|
5321 |
newKernelInitrd)) return 1; |
if (config->defaultImage == NO_DEFAULT_ENTRY) |
5322 |
} |
return 0; |
5323 |
if (addNewKernel(config, template, bootPrefix, newKernelPath, |
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
5324 |
newKernelTitle, newKernelArgs, newKernelInitrd, |
cfi->defaultIsSaved) |
5325 |
extraInitrds, extraInitrdCount, |
config->defaultImage = FIRST_ENTRY_INDEX; |
5326 |
newMBKernel, newMBKernelArgs)) return 1; |
entry = findEntryByIndex(config, config->defaultImage); |
5327 |
|
if (!entry) |
5328 |
|
return 0; |
5329 |
if (numEntries(config) == 0) { |
if (!suitableImage(entry, bootPrefix, 0, flags)) |
5330 |
fprintf(stderr, _("grubby: doing this would leave no kernel entries. " |
return 0; |
5331 |
"Not writing out new config.\n")); |
|
5332 |
return 1; |
line = |
5333 |
} |
getLineByType(LT_KERNEL | LT_HYPER | LT_KERNEL_EFI | |
5334 |
|
LT_KERNEL_16, entry->lines); |
5335 |
|
if (!line) |
5336 |
|
return 0; |
5337 |
|
|
5338 |
|
rootspec = getRootSpecifier(line->elements[1].item); |
5339 |
|
printf("%s%s\n", bootPrefix, line->elements[1].item + |
5340 |
|
((rootspec != NULL) ? strlen(rootspec) : 0)); |
5341 |
|
|
5342 |
|
return 0; |
5343 |
|
|
5344 |
|
} else if (displayDefaultTitle) { |
5345 |
|
struct singleLine *line; |
5346 |
|
struct singleEntry *entry; |
5347 |
|
|
5348 |
|
if (config->defaultImage == NO_DEFAULT_ENTRY) |
5349 |
|
return 0; |
5350 |
|
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
5351 |
|
cfi->defaultIsSaved) |
5352 |
|
config->defaultImage = FIRST_ENTRY_INDEX; |
5353 |
|
entry = findEntryByIndex(config, config->defaultImage); |
5354 |
|
if (!entry) |
5355 |
|
return 0; |
5356 |
|
|
5357 |
|
if (!configureGrub2) { |
5358 |
|
char *title; |
5359 |
|
line = getLineByType(LT_TITLE, entry->lines); |
5360 |
|
if (!line) |
5361 |
|
return 0; |
5362 |
|
title = extractTitle(config, line); |
5363 |
|
if (!title) |
5364 |
|
return 0; |
5365 |
|
printf("%s\n", title); |
5366 |
|
free(title); |
5367 |
|
} else { |
5368 |
|
char *title; |
5369 |
|
|
5370 |
|
dbgPrintf |
5371 |
|
("This is GRUB2, default title is embeded in menuentry\n"); |
5372 |
|
line = getLineByType(LT_MENUENTRY, entry->lines); |
5373 |
|
if (!line) |
5374 |
|
return 0; |
5375 |
|
title = grub2ExtractTitle(line); |
5376 |
|
if (title) |
5377 |
|
printf("%s\n", title); |
5378 |
|
} |
5379 |
|
return 0; |
5380 |
|
|
5381 |
|
} else if (displayDefaultIndex) { |
5382 |
|
if (config->defaultImage == NO_DEFAULT_ENTRY) |
5383 |
|
return 0; |
5384 |
|
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
5385 |
|
cfi->defaultIsSaved) |
5386 |
|
config->defaultImage = FIRST_ENTRY_INDEX; |
5387 |
|
printf("%i\n", config->defaultImage); |
5388 |
|
return 0; |
5389 |
|
|
5390 |
|
} else if (kernelInfo) |
5391 |
|
return displayInfo(config, kernelInfo, bootPrefix); |
5392 |
|
|
5393 |
|
if (copyDefault) { |
5394 |
|
template = findTemplate(config, bootPrefix, NULL, 0, flags); |
5395 |
|
if (!template) |
5396 |
|
return 1; |
5397 |
|
} |
5398 |
|
|
5399 |
|
markRemovedImage(config, removeKernelPath, bootPrefix); |
5400 |
|
markRemovedImage(config, removeMBKernel, bootPrefix); |
5401 |
|
setDefaultImage(config, newKernelPath != NULL, defaultKernel, |
5402 |
|
makeDefault, bootPrefix, flags, defaultIndex, |
5403 |
|
newIndex); |
5404 |
|
setFallbackImage(config, newKernelPath != NULL); |
5405 |
|
if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs, |
5406 |
|
removeArgs, newMBKernelArgs, removeMBKernelArgs)) |
5407 |
|
return 1; |
5408 |
|
if (updateKernelPath && newKernelInitrd) { |
5409 |
|
if (newMBKernel) { |
5410 |
|
if (addMBInitrd(config, newMBKernel, updateKernelPath, |
5411 |
|
bootPrefix, newKernelInitrd, |
5412 |
|
newKernelTitle)) |
5413 |
|
return 1; |
5414 |
|
} else { |
5415 |
|
if (updateInitrd(config, updateKernelPath, bootPrefix, |
5416 |
|
newKernelInitrd, newKernelTitle)) |
5417 |
|
return 1; |
5418 |
|
} |
5419 |
|
} |
5420 |
|
if (addNewKernel(config, template, bootPrefix, newKernelPath, |
5421 |
|
newKernelTitle, newKernelArgs, newKernelInitrd, |
5422 |
|
(const char **)extraInitrds, extraInitrdCount, |
5423 |
|
newMBKernel, newMBKernelArgs, newDevTreePath, |
5424 |
|
newIndex)) |
5425 |
|
return 1; |
5426 |
|
|
5427 |
|
if (numEntries(config) == 0) { |
5428 |
|
fprintf(stderr, |
5429 |
|
_("grubby: doing this would leave no kernel entries. " |
5430 |
|
"Not writing out new config.\n")); |
5431 |
|
return 1; |
5432 |
|
} |
5433 |
|
|
5434 |
if (!outputFile) |
if (!outputFile) |
5435 |
outputFile = (char *)grubConfig; |
outputFile = (char *)grubConfig; |
5436 |
|
|
5437 |
return writeConfig(config, outputFile, bootPrefix); |
return writeConfig(config, outputFile, bootPrefix); |
5438 |
} |
} |