Diff of /tags/grubby-8_32/grubby.c

Parent Directory | Revision Log | Patch

	trunk/mkinitrd-magellan/grubby/grubby.c revision 926 by niro, Wed Oct 28 13:29:38 2009 UTC	trunk/grubby/grubby.c revision 1931 by niro, Mon Oct 1 12:06:26 2012 UTC
#	Line 36	Line 36
36	#include <signal.h>	#include <signal.h>
37	#include <blkid/blkid.h>	#include <blkid/blkid.h>
38
39		#ifndef DEBUG
40	#define DEBUG 0	#define DEBUG 0
41		#endif
42
43	#if DEBUG	#if DEBUG
44	#define dbgPrintf(format, args...) fprintf(stderr, format , ## args)	#define dbgPrintf(format, args...) fprintf(stderr, format , ## args)
#	Line 44	Line 46
46	#define dbgPrintf(format, args...)	#define dbgPrintf(format, args...)
47	#endif	#endif
48
49		int debug = 0; /* Currently just for template debugging */
50
51	#define _(A) (A)	#define _(A) (A)
52
53	#define MAX_EXTRA_INITRDS  16 /* code segment checked by --bootloader-probe */	#define MAX_EXTRA_INITRDS  16 /* code segment checked by --bootloader-probe */
54	#define CODE_SEG_SIZE  128 /* code segment checked by --bootloader-probe */	#define CODE_SEG_SIZE  128 /* code segment checked by --bootloader-probe */
55
56		#define NOOP_OPCODE 0x90
57		#define JMP_SHORT_OPCODE 0xeb
58
59	/* comments get lumped in with indention */	/* comments get lumped in with indention */
60	struct lineElement {	struct lineElement {
61	char * item;	char * item;
#	Line 56  struct lineElement {	Line 63  struct lineElement {
63	};	};
64
65	enum lineType_e {	enum lineType_e {
66	LT_WHITESPACE = 1 << 0,	LT_WHITESPACE   = 1 << 0,
67	LT_TITLE      = 1 << 1,	LT_TITLE        = 1 << 1,
68	LT_KERNEL     = 1 << 2,	LT_KERNEL       = 1 << 2,
69	LT_INITRD     = 1 << 3,	LT_INITRD       = 1 << 3,
70	LT_HYPER      = 1 << 4,	LT_HYPER        = 1 << 4,
71	LT_DEFAULT    = 1 << 5,	LT_DEFAULT      = 1 << 5,
72	LT_MBMODULE   = 1 << 6,	LT_MBMODULE     = 1 << 6,
73	LT_ROOT       = 1 << 7,	LT_ROOT         = 1 << 7,
74	LT_FALLBACK   = 1 << 8,	LT_FALLBACK     = 1 << 8,
75	LT_KERNELARGS = 1 << 9,	LT_KERNELARGS   = 1 << 9,
76	LT_BOOT       = 1 << 10,	LT_BOOT         = 1 << 10,
77	LT_BOOTROOT   = 1 << 11,	LT_BOOTROOT     = 1 << 11,
78	LT_LBA        = 1 << 12,	LT_LBA          = 1 << 12,
79	LT_OTHER      = 1 << 13,	LT_OTHER        = 1 << 13,
80	LT_GENERIC    = 1 << 14,	LT_GENERIC      = 1 << 14,
81	LT_UNKNOWN    = 1 << 15,	LT_ECHO    = 1 << 16,
82		LT_MENUENTRY    = 1 << 17,
83		LT_ENTRY_END    = 1 << 18,
84		LT_SET_VARIABLE = 1 << 19,
85		LT_UNKNOWN      = 1 << 20,
86	};	};
87
88	struct singleLine {	struct singleLine {
#	Line 99  struct singleEntry {	Line 110  struct singleEntry {
110	#define NEED_TITLE   (1 << 2)	#define NEED_TITLE   (1 << 2)
111	#define NEED_ARGS    (1 << 3)	#define NEED_ARGS    (1 << 3)
112	#define NEED_MB      (1 << 4)	#define NEED_MB      (1 << 4)
113		#define NEED_END     (1 << 5)
114
115	#define MAIN_DEFAULT    (1 << 0)	#define MAIN_DEFAULT    (1 << 0)
116	#define DEFAULT_SAVED       -2	#define DEFAULT_SAVED       -2
117		#define DEFAULT_SAVED_GRUB2 -3
118
119	struct keywordTypes {	struct keywordTypes {
120	char * key;	char * key;
#	Line 110  struct keywordTypes {	Line 123  struct keywordTypes {
123	char separatorChar;	char separatorChar;
124	};	};
125
126		struct configFileInfo;
127
128		typedef const char *(*findConfigFunc)(struct configFileInfo *);
129		typedef const int (*writeLineFunc)(struct configFileInfo *,
130		struct singleLine *line);
131
132	struct configFileInfo {	struct configFileInfo {
133	char * defaultConfig;	char * defaultConfig;
134		findConfigFunc findConfig;
135		writeLineFunc writeLine;
136	struct keywordTypes * keywords;	struct keywordTypes * keywords;
137	int defaultIsIndex;	int defaultIsIndex;
138		int defaultIsVariable;
139	int defaultSupportSaved;	int defaultSupportSaved;
140	enum lineType_e entrySeparator;	enum lineType_e entryStart;
141		enum lineType_e entryEnd;
142	int needsBootPrefix;	int needsBootPrefix;
143	int argsInQuotes;	int argsInQuotes;
144	int maxTitleLength;	int maxTitleLength;
145	int titleBracketed;	int titleBracketed;
146		int titlePosition;
147	int mbHyperFirst;	int mbHyperFirst;
148	int mbInitRdIsModule;	int mbInitRdIsModule;
149	int mbConcatArgs;	int mbConcatArgs;
#	Line 138  struct keywordTypes grubKeywords[] = {	Line 162  struct keywordTypes grubKeywords[] = {
162	{ NULL,    0, 0 },	{ NULL,    0, 0 },
163	};	};
164
165		const char *grubFindConfig(struct configFileInfo *cfi) {
166		static const char *configFiles[] = {
167		"/boot/grub/grub.conf",
168		"/boot/grub/menu.lst",
169		"/etc/grub.conf",
170		NULL
171		};
172		static int i = -1;
173
174		if (i == -1) {
175		for (i = 0; configFiles[i] != NULL; i++) {
176		dbgPrintf("Checking \"%s\": ", configFiles[i]);
177		if (!access(configFiles[i], R_OK)) {
178		dbgPrintf("found\n");
179		return configFiles[i];
180		}
181		dbgPrintf("not found\n");
182		}
183		}
184		return configFiles[i];
185		}
186
187	struct configFileInfo grubConfigType = {	struct configFileInfo grubConfigType = {
188	"/boot/grub/grub.conf",    /* defaultConfig */	.findConfig = grubFindConfig,
189	grubKeywords,    /* keywords */	.keywords = grubKeywords,
190	1,    /* defaultIsIndex */	.defaultIsIndex = 1,
191	1,    /* defaultSupportSaved */	.defaultSupportSaved = 1,
192	LT_TITLE,    /* entrySeparator */	.entryStart = LT_TITLE,
193	1,    /* needsBootPrefix */	.needsBootPrefix = 1,
194	0,    /* argsInQuotes */	.mbHyperFirst = 1,
195	0,    /* maxTitleLength */	.mbInitRdIsModule = 1,
196	0,                                      /* titleBracketed */	.mbAllowExtraInitRds = 1,
197	1,                                      /* mbHyperFirst */	};
198	1,                                      /* mbInitRdIsModule */
199	0,                                      /* mbConcatArgs */	struct keywordTypes grub2Keywords[] = {
200	1,                                      /* mbAllowExtraInitRds */	{ "menuentry",  LT_MENUENTRY,   ' ' },
201		{ "}",          LT_ENTRY_END,   ' ' },
202		{ "echo",       LT_ECHO,        ' ' },
203		{ "set",        LT_SET_VARIABLE,' ', '=' },
204		{ "root",       LT_BOOTROOT,    ' ' },
205		{ "default",    LT_DEFAULT,     ' ' },
206		{ "fallback",   LT_FALLBACK,    ' ' },
207		{ "linux",      LT_KERNEL,      ' ' },
208		{ "linuxefi",   LT_KERNEL,      ' ' },
209		{ "initrd",     LT_INITRD,      ' ', ' ' },
210		{ "initrdefi",  LT_INITRD,      ' ', ' ' },
211		{ "module",     LT_MBMODULE,    ' ' },
212		{ "kernel",     LT_HYPER,       ' ' },
213		{ NULL, 0, 0 },
214		};
215
216		const char *grub2FindConfig(struct configFileInfo *cfi) {
217		static const char *configFiles[] = {
218		"/boot/grub/grub-efi.cfg",
219		"/boot/grub/grub.cfg",
220		NULL
221		};
222		static int i = -1;
223		static const char *grub_cfg = "/boot/grub/grub.cfg";
224
225		if (i == -1) {
226		for (i = 0; configFiles[i] != NULL; i++) {
227		dbgPrintf("Checking \"%s\": ", configFiles[i]);
228		if (!access(configFiles[i], R_OK)) {
229		dbgPrintf("found\n");
230		return configFiles[i];
231		}
232		}
233		}
234
235		/* Ubuntu renames grub2 to grub, so check for the grub.d directory
236		* that isn't in grub1, and if it exists, return the config file path
237		* that they use. */
238		if (configFiles[i] == NULL && !access("/etc/grub.d/", R_OK)) {
239		dbgPrintf("found\n");
240		return grub_cfg;
241		}
242
243		dbgPrintf("not found\n");
244		return configFiles[i];
245		}
246
247		int sizeOfSingleLine(struct singleLine * line) {
248		int count = 0;
249
250		for (int i = 0; i < line->numElements; i++) {
251		int indentSize = 0;
252
253		count = count + strlen(line->elements[i].item);
254
255		indentSize = strlen(line->elements[i].indent);
256		if (indentSize > 0)
257		count = count + indentSize;
258		else
259		/* be extra safe and add room for whitespaces */
260		count = count + 1;
261		}
262
263		/* room for trailing terminator */
264		count = count + 1;
265
266		return count;
267		}
268
269		static int isquote(char q)
270		{
271		if (q == '\'' || q == '\"')
272		return 1;
273		return 0;
274		}
275
276		char *grub2ExtractTitle(struct singleLine * line) {
277		char * current;
278		char * current_indent;
279		int current_len;
280		int current_indent_len;
281		int i;
282
283		/* bail out if line does not start with menuentry */
284		if (strcmp(line->elements[0].item, "menuentry"))
285		return NULL;
286
287		i = 1;
288		current = line->elements[i].item;
289		current_len = strlen(current);
290
291		/* if second word is quoted, strip the quotes and return single word */
292		if (isquote(*current) && isquote(current[current_len - 1])) {
293		char *tmp;
294
295		tmp = strdup(current);
296		*(tmp + current_len - 1) = '\0';
297		return ++tmp;
298		}
299
300		/* if no quotes, return second word verbatim */
301		if (!isquote(*current))
302		return current;
303
304		/* second element start with a quote, so we have to find the element
305		* whose last character is also quote (assuming it's the closing one) */
306		int resultMaxSize;
307		char * result;
308
309		resultMaxSize = sizeOfSingleLine(line);
310		result = malloc(resultMaxSize);
311		snprintf(result, resultMaxSize, "%s", ++current);
312
313		i++;
314		for (; i < line->numElements; ++i) {
315		current = line->elements[i].item;
316		current_len = strlen(current);
317		current_indent = line->elements[i].indent;
318		current_indent_len = strlen(current_indent);
319
320		strncat(result, current_indent, current_indent_len);
321		if (!isquote(current[current_len-1])) {
322		strncat(result, current, current_len);
323		} else {
324		strncat(result, current, current_len - 1);
325		break;
326		}
327		}
328		return result;
329		}
330
331		struct configFileInfo grub2ConfigType = {
332		.findConfig = grub2FindConfig,
333		.keywords = grub2Keywords,
334		.defaultIsIndex = 1,
335		.defaultSupportSaved = 1,
336		.defaultIsVariable = 1,
337		.entryStart = LT_MENUENTRY,
338		.entryEnd = LT_ENTRY_END,
339		.titlePosition = 1,
340		.needsBootPrefix = 1,
341		.mbHyperFirst = 1,
342		.mbInitRdIsModule = 1,
343		.mbAllowExtraInitRds = 1,
344	};	};
345
346	struct keywordTypes yabootKeywords[] = {	struct keywordTypes yabootKeywords[] = {
#	Line 249  struct keywordTypes extlinuxKeywords[] =	Line 438  struct keywordTypes extlinuxKeywords[] =
438	};	};
439	int useextlinuxmenu;	int useextlinuxmenu;
440	struct configFileInfo eliloConfigType = {	struct configFileInfo eliloConfigType = {
441	"/boot/efi/EFI/redhat/elilo.conf",    /* defaultConfig */	.defaultConfig = "/boot/efi/EFI/redhat/elilo.conf",
442	eliloKeywords,    /* keywords */	.keywords = eliloKeywords,
443	0,    /* defaultIsIndex */	.entryStart = LT_KERNEL,
444	0,    /* defaultSupportSaved */	.needsBootPrefix = 1,
445	LT_KERNEL,    /* entrySeparator */	.argsInQuotes = 1,
446	1,                    /* needsBootPrefix */	.mbConcatArgs = 1,
	1,    /* argsInQuotes */
	0,    /* maxTitleLength */
	0,                                      /* titleBracketed */
	0,                                      /* mbHyperFirst */
	0,                                      /* mbInitRdIsModule */
	1,                                      /* mbConcatArgs */
	0,                                      /* mbAllowExtraInitRds */
447	};	};
448
449	struct configFileInfo liloConfigType = {	struct configFileInfo liloConfigType = {
450	"/etc/lilo.conf",    /* defaultConfig */	.defaultConfig = "/etc/lilo.conf",
451	liloKeywords,    /* keywords */	.keywords = liloKeywords,
452	0,    /* defaultIsIndex */	.entryStart = LT_KERNEL,
453	0,    /* defaultSupportSaved */	.argsInQuotes = 1,
454	LT_KERNEL,    /* entrySeparator */	.maxTitleLength = 15,
	0,    /* needsBootPrefix */
	1,    /* argsInQuotes */
	15,    /* maxTitleLength */
	0,                                      /* titleBracketed */
	0,                                      /* mbHyperFirst */
	0,                                      /* mbInitRdIsModule */
	0,                                      /* mbConcatArgs */
	0,                                      /* mbAllowExtraInitRds */
455	};	};
456
457	struct configFileInfo yabootConfigType = {	struct configFileInfo yabootConfigType = {
458	"/etc/yaboot.conf",    /* defaultConfig */	.defaultConfig = "/etc/yaboot.conf",
459	yabootKeywords,    /* keywords */	.keywords = yabootKeywords,
460	0,    /* defaultIsIndex */	.entryStart = LT_KERNEL,
461	0,    /* defaultSupportSaved */	.needsBootPrefix = 1,
462	LT_KERNEL,    /* entrySeparator */	.argsInQuotes = 1,
463	1,    /* needsBootPrefix */	.maxTitleLength = 15,
464	1,    /* argsInQuotes */	.mbAllowExtraInitRds = 1,
	15,    /* maxTitleLength */
	0,                                      /* titleBracketed */
	0,                                      /* mbHyperFirst */
	0,                                      /* mbInitRdIsModule */
	0,                                      /* mbConcatArgs */
	1,                                      /* mbAllowExtraInitRds */
465	};	};
466
467	struct configFileInfo siloConfigType = {	struct configFileInfo siloConfigType = {
468	"/etc/silo.conf",    /* defaultConfig */	.defaultConfig = "/etc/silo.conf",
469	siloKeywords,    /* keywords */	.keywords = siloKeywords,
470	0,    /* defaultIsIndex */	.entryStart = LT_KERNEL,
471	0,    /* defaultSupportSaved */	.needsBootPrefix = 1,
472	LT_KERNEL,    /* entrySeparator */	.argsInQuotes = 1,
473	1,    /* needsBootPrefix */	.maxTitleLength = 15,
	1,    /* argsInQuotes */
	15,    /* maxTitleLength */
	0,                                      /* titleBracketed */
	0,                                      /* mbHyperFirst */
	0,                                      /* mbInitRdIsModule */
	0,                                      /* mbConcatArgs */
	0,                                      /* mbAllowExtraInitRds */
474	};	};
475
476	struct configFileInfo ziplConfigType = {	struct configFileInfo ziplConfigType = {
477	"/etc/zipl.conf",    /* defaultConfig */	.defaultConfig = "/etc/zipl.conf",
478	ziplKeywords,    /* keywords */	.keywords = ziplKeywords,
479	0,    /* defaultIsIndex */	.entryStart = LT_TITLE,
480	0,    /* defaultSupportSaved */	.argsInQuotes = 1,
481	LT_TITLE,    /* entrySeparator */	.titleBracketed = 1,
	0,    /* needsBootPrefix */
	1,    /* argsInQuotes */
	0,    /* maxTitleLength */
	1,                                      /* titleBracketed */
	0,                                      /* mbHyperFirst */
	0,                                      /* mbInitRdIsModule */
	0,                                      /* mbConcatArgs */
	0,                                      /* mbAllowExtraInitRds */
482	};	};
483
484	struct configFileInfo extlinuxConfigType = {	struct configFileInfo extlinuxConfigType = {
485	"/boot/extlinux/extlinux.conf",         /* defaultConfig */	.defaultConfig = "/boot/extlinux/extlinux.conf",
486	extlinuxKeywords,                       /* keywords */	.keywords = extlinuxKeywords,
487	0,                                      /* defaultIsIndex */	.entryStart = LT_TITLE,
488	0,                                      /* defaultSupportSaved */	.needsBootPrefix = 1,
489	LT_TITLE,                               /* entrySeparator */	.maxTitleLength = 255,
490	1,                                      /* needsBootPrefix */	.mbAllowExtraInitRds = 1,
	0,                                      /* argsInQuotes */
	255,                                    /* maxTitleLength */
	0,                                      /* titleBracketed */
	0,                                      /* mbHyperFirst */
	0,                                      /* mbInitRdIsModule */
	0,                                      /* mbConcatArgs */
	1,                                      /* mbAllowExtraInitRds */
491	};	};
492
493	struct grubConfig {	struct grubConfig {
#	Line 356  struct grubConfig {	Line 502  struct grubConfig {
502	struct configFileInfo * cfi;	struct configFileInfo * cfi;
503	};	};
504
505		blkid_cache blkid;
506
507	struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index);	struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index);
508	struct singleEntry * findEntryByPath(struct grubConfig * cfg,	struct singleEntry * findEntryByPath(struct grubConfig * cfg,
509	const char * path, const char * prefix,	const char * path, const char * prefix,
#	Line 369  static int lineWrite(FILE * out, struct	Line 517  static int lineWrite(FILE * out, struct
517	static int getNextLine(char ** bufPtr, struct singleLine * line,	static int getNextLine(char ** bufPtr, struct singleLine * line,
518	struct configFileInfo * cfi);	struct configFileInfo * cfi);
519	static char * getRootSpecifier(char * str);	static char * getRootSpecifier(char * str);
520		static void requote(struct singleLine *line, struct configFileInfo * cfi);
521	static void insertElement(struct singleLine * line,	static void insertElement(struct singleLine * line,
522	const char * item, int insertHere,	const char * item, int insertHere,
523	struct configFileInfo * cfi);	struct configFileInfo * cfi);
#	Line 425  static char * sdupprintf(const char *for	Line 574  static char * sdupprintf(const char *for
574
575	static struct keywordTypes * getKeywordByType(enum lineType_e type,	static struct keywordTypes * getKeywordByType(enum lineType_e type,
576	struct configFileInfo * cfi) {	struct configFileInfo * cfi) {
577	struct keywordTypes * kw;	for (struct keywordTypes *kw = cfi->keywords; kw->key; kw++) {
	for (kw = cfi->keywords; kw->key; kw++) {
578	if (kw->type == type)	if (kw->type == type)
579	return kw;	return kw;
580	}	}
581	return NULL;	return NULL;
582	}	}
583
584	static char * getpathbyspec(char *device) {	static char *getKeyByType(enum lineType_e type, struct configFileInfo * cfi) {
585	static blkid_cache blkid;	struct keywordTypes *kt = getKeywordByType(type, cfi);
586		if (kt)
587		return kt->key;
588		return "unknown";
589		}
590
591		static char * getpathbyspec(char *device) {
592	if (!blkid)	if (!blkid)
593	blkid_get_cache(&blkid, NULL);	blkid_get_cache(&blkid, NULL);
594
595	return blkid_get_devname(blkid, device, NULL);	return blkid_get_devname(blkid, device, NULL);
596	}	}
597
598		static char * getuuidbydev(char *device) {
599		if (!blkid)
600		blkid_get_cache(&blkid, NULL);
601
602		return blkid_get_tag_value(blkid, "UUID", device);
603		}
604
605	static enum lineType_e getTypeByKeyword(char * keyword,	static enum lineType_e getTypeByKeyword(char * keyword,
606	struct configFileInfo * cfi) {	struct configFileInfo * cfi) {
607	struct keywordTypes * kw;	for (struct keywordTypes *kw = cfi->keywords; kw->key; kw++) {
	for (kw = cfi->keywords; kw->key; kw++) {
608	if (!strcmp(keyword, kw->key))	if (!strcmp(keyword, kw->key))
609	return kw->type;	return kw->type;
610	}	}
#	Line 477  static int isBracketedTitle(struct singl	Line 636  static int isBracketedTitle(struct singl
636	return 0;	return 0;
637	}	}
638
639	static int isEntrySeparator(struct singleLine * line,	static int isEntryStart(struct singleLine * line,
640	struct configFileInfo * cfi) {	struct configFileInfo * cfi) {
641	return line->type == cfi->entrySeparator || line->type == LT_OTHER ||	return line->type == cfi->entryStart || line->type == LT_OTHER ||
642	(cfi->titleBracketed && isBracketedTitle(line));	(cfi->titleBracketed && isBracketedTitle(line));
643	}	}
644
#	Line 532  static void lineInit(struct singleLine *	Line 691  static void lineInit(struct singleLine *
691	}	}
692
693	struct singleLine * lineDup(struct singleLine * line) {	struct singleLine * lineDup(struct singleLine * line) {
	int i;
694	struct singleLine * newLine = malloc(sizeof(*newLine));	struct singleLine * newLine = malloc(sizeof(*newLine));
695
696	newLine->indent = strdup(line->indent);	newLine->indent = strdup(line->indent);
#	Line 542  struct singleLine * lineDup(struct singl	Line 700  struct singleLine * lineDup(struct singl
700	newLine->elements = malloc(sizeof(*newLine->elements) *	newLine->elements = malloc(sizeof(*newLine->elements) *
701	newLine->numElements);	newLine->numElements);
702
703	for (i = 0; i < newLine->numElements; i++) {	for (int i = 0; i < newLine->numElements; i++) {
704	newLine->elements[i].indent = strdup(line->elements[i].indent);	newLine->elements[i].indent = strdup(line->elements[i].indent);
705	newLine->elements[i].item = strdup(line->elements[i].item);	newLine->elements[i].item = strdup(line->elements[i].item);
706	}	}
#	Line 551  struct singleLine * lineDup(struct singl	Line 709  struct singleLine * lineDup(struct singl
709	}	}
710
711	static void lineFree(struct singleLine * line) {	static void lineFree(struct singleLine * line) {
	int i;
712	if (line->indent) free(line->indent);	if (line->indent) free(line->indent);
713
714	for (i = 0; i < line->numElements; i++) {	for (int i = 0; i < line->numElements; i++) {
715	free(line->elements[i].item);	free(line->elements[i].item);
716	free(line->elements[i].indent);	free(line->elements[i].indent);
717	}	}
#	Line 566  static void lineFree(struct singleLine *	Line 722  static void lineFree(struct singleLine *
722
723	static int lineWrite(FILE * out, struct singleLine * line,	static int lineWrite(FILE * out, struct singleLine * line,
724	struct configFileInfo * cfi) {	struct configFileInfo * cfi) {
	int i;
725	if (fprintf(out, "%s", line->indent) == -1) return -1;	if (fprintf(out, "%s", line->indent) == -1) return -1;
726
727	for (i = 0; i < line->numElements; i++) {	for (int i = 0; i < line->numElements; i++) {
728		/* Need to handle this, because we strip the quotes from
729		* menuentry when read it. */
730		if (line->type == LT_MENUENTRY && i == 1) {
731		if(!isquote(*line->elements[i].item))
732		fprintf(out, "\'%s\'", line->elements[i].item);
733		else
734		fprintf(out, "%s", line->elements[i].item);
735		fprintf(out, "%s", line->elements[i].indent);
736
737		continue;
738		}
739
740	if (i == 1 && line->type == LT_KERNELARGS && cfi->argsInQuotes)	if (i == 1 && line->type == LT_KERNELARGS && cfi->argsInQuotes)
741	if (fputc('"', out) == EOF) return -1;	if (fputc('"', out) == EOF) return -1;
742
#	Line 664  static int getNextLine(char ** bufPtr, s	Line 830  static int getNextLine(char ** bufPtr, s
830	if (*line->elements[0].item == '#') {	if (*line->elements[0].item == '#') {
831	char * fullLine;	char * fullLine;
832	int len;	int len;
	int i;
833
834	len = strlen(line->indent);	len = strlen(line->indent);
835	for (i = 0; i < line->numElements; i++)	for (int i = 0; i < line->numElements; i++)
836	len += strlen(line->elements[i].item) +	len += strlen(line->elements[i].item) +
837	strlen(line->elements[i].indent);	strlen(line->elements[i].indent);
838
#	Line 676  static int getNextLine(char ** bufPtr, s	Line 841  static int getNextLine(char ** bufPtr, s
841	free(line->indent);	free(line->indent);
842	line->indent = fullLine;	line->indent = fullLine;
843
844	for (i = 0; i < line->numElements; i++) {	for (int i = 0; i < line->numElements; i++) {
845	strcat(fullLine, line->elements[i].item);	strcat(fullLine, line->elements[i].item);
846	strcat(fullLine, line->elements[i].indent);	strcat(fullLine, line->elements[i].indent);
847	free(line->elements[i].item);	free(line->elements[i].item);
#	Line 695  static int getNextLine(char ** bufPtr, s	Line 860  static int getNextLine(char ** bufPtr, s
860	* elements up more	* elements up more
861	*/	*/
862	if (!isspace(kw->separatorChar)) {	if (!isspace(kw->separatorChar)) {
	int i;
863	char indent[2] = "";	char indent[2] = "";
864	indent[0] = kw->separatorChar;	indent[0] = kw->separatorChar;
865	for (i = 1; i < line->numElements; i++) {	for (int i = 1; i < line->numElements; i++) {
866	char *p;	char *p;
	int j;
867	int numNewElements;	int numNewElements;
868
869	numNewElements = 0;	numNewElements = 0;
#	Line 716  static int getNextLine(char ** bufPtr, s	Line 879  static int getNextLine(char ** bufPtr, s
879	sizeof(*line->elements) * elementsAlloced);	sizeof(*line->elements) * elementsAlloced);
880	}	}
881
882	for (j = line->numElements; j > i; j--) {	for (int j = line->numElements; j > i; j--) {
883	line->elements[j + numNewElements] = line->elements[j];	line->elements[j + numNewElements] = line->elements[j];
884	}	}
885	line->numElements += numNewElements;	line->numElements += numNewElements;
#	Line 729  static int getNextLine(char ** bufPtr, s	Line 892  static int getNextLine(char ** bufPtr, s
892	break;	break;
893	}	}
894
895	free(line->elements[i].indent);	line->elements[i + 1].indent = line->elements[i].indent;
896	line->elements[i].indent = strdup(indent);	line->elements[i].indent = strdup(indent);
897	*p++ = '\0';	*p++ = '\0';
898	i++;	i++;
899	line->elements[i].item = strdup(p);	line->elements[i].item = strdup(p);
	line->elements[i].indent = strdup("");
	p = line->elements[i].item;
900	}	}
901	}	}
902	}	}
#	Line 756  static struct grubConfig * readConfig(co	Line 917  static struct grubConfig * readConfig(co
917	struct singleLine * last = NULL, * line, * defaultLine = NULL;	struct singleLine * last = NULL, * line, * defaultLine = NULL;
918	char * end;	char * end;
919	struct singleEntry * entry = NULL;	struct singleEntry * entry = NULL;
920	int i, len;	int len;
921	char * buf;	char * buf;
922
923	if (!strcmp(inName, "-")) {	if (!strcmp(inName, "-")) {
#	Line 802  static struct grubConfig * readConfig(co	Line 963  static struct grubConfig * readConfig(co
963	cfg->secondaryIndent = strdup(line->indent);	cfg->secondaryIndent = strdup(line->indent);
964	}	}
965
966	if (isEntrySeparator(line, cfi)) {	if (isEntryStart(line, cfi) || (cfg->entries && !sawEntry)) {
967	sawEntry = 1;	sawEntry = 1;
968	if (!entry) {	if (!entry) {
969	cfg->entries = malloc(sizeof(*entry));	cfg->entries = malloc(sizeof(*entry));
#	Line 818  static struct grubConfig * readConfig(co	Line 979  static struct grubConfig * readConfig(co
979	entry->next = NULL;	entry->next = NULL;
980	}	}
981
982	if (line->type == LT_DEFAULT && line->numElements == 2) {	if (line->type == LT_SET_VARIABLE) {
983		dbgPrintf("found 'set' command (%d elements): ", line->numElements);
984		dbgPrintf("%s", line->indent);
985		for (int i = 0; i < line->numElements; i++)
986		dbgPrintf("\"%s\"%s", line->elements[i].item, line->elements[i].indent);
987		dbgPrintf("\n");
988		struct keywordTypes *kwType = getKeywordByType(LT_DEFAULT, cfi);
989		if (kwType && line->numElements == 3 &&
990		!strcmp(line->elements[1].item, kwType->key)) {
991		dbgPrintf("Line sets default config\n");
992		cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;
993		defaultLine = line;
994		}
995		} else if (line->type == LT_DEFAULT && line->numElements == 2) {
996	cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;	cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;
997	defaultLine = line;	defaultLine = line;
998
#	Line 836  static struct grubConfig * readConfig(co	Line 1010  static struct grubConfig * readConfig(co
1010	* This only applies to grub, but that's the only place we	* This only applies to grub, but that's the only place we
1011	* should find LT_MBMODULE lines anyway.	* should find LT_MBMODULE lines anyway.
1012	*/	*/
1013	struct singleLine * l;	for (struct singleLine *l = entry->lines; l; l = l->next) {
	for (l = entry->lines; l; l = l->next) {
1014	if (l->type == LT_HYPER)	if (l->type == LT_HYPER)
1015	break;	break;
1016	else if (l->type == LT_KERNEL) {	else if (l->type == LT_KERNEL) {
#	Line 857  static struct grubConfig * readConfig(co	Line 1030  static struct grubConfig * readConfig(co
1030	} else if (line->type == LT_TITLE && line->numElements > 1) {	} else if (line->type == LT_TITLE && line->numElements > 1) {
1031	/* make the title a single argument (undoing our parsing) */	/* make the title a single argument (undoing our parsing) */
1032	len = 0;	len = 0;
1033	for (i = 1; i < line->numElements; i++) {	for (int i = 1; i < line->numElements; i++) {
1034	len += strlen(line->elements[i].item);	len += strlen(line->elements[i].item);
1035	len += strlen(line->elements[i].indent);	len += strlen(line->elements[i].indent);
1036	}	}
1037	buf = malloc(len + 1);	buf = malloc(len + 1);
1038	*buf = '\0';	*buf = '\0';
1039
1040	for (i = 1; i < line->numElements; i++) {	for (int i = 1; i < line->numElements; i++) {
1041	strcat(buf, line->elements[i].item);	strcat(buf, line->elements[i].item);
1042	free(line->elements[i].item);	free(line->elements[i].item);
1043
#	Line 878  static struct grubConfig * readConfig(co	Line 1051  static struct grubConfig * readConfig(co
1051	line->elements[line->numElements - 1].indent;	line->elements[line->numElements - 1].indent;
1052	line->elements[1].item = buf;	line->elements[1].item = buf;
1053	line->numElements = 2;	line->numElements = 2;
1054		} else if (line->type == LT_MENUENTRY && line->numElements > 3) {
1055		/* let --remove-kernel="TITLE=what" work */
1056		len = 0;
1057		char *extras;
1058		char *title;
1059
1060		for (int i = 1; i < line->numElements; i++) {
1061		len += strlen(line->elements[i].item);
1062		len += strlen(line->elements[i].indent);
1063		}
1064		buf = malloc(len + 1);
1065		*buf = '\0';
1066
1067		/* allocate mem for extra flags. */
1068		extras = malloc(len + 1);
1069		*extras = '\0';
1070
1071		/* get title. */
1072		for (int i = 0; i < line->numElements; i++) {
1073		if (!strcmp(line->elements[i].item, "menuentry"))
1074		continue;
1075		if (isquote(*line->elements[i].item))
1076		title = line->elements[i].item + 1;
1077		else
1078		title = line->elements[i].item;
1079
1080		len = strlen(title);
1081		if (isquote(title[len-1])) {
1082		strncat(buf, title,len-1);
1083		break;
1084		} else {
1085		strcat(buf, title);
1086		strcat(buf, line->elements[i].indent);
1087		}
1088		}
1089
1090		/* get extras */
1091		int count = 0;
1092		for (int i = 0; i < line->numElements; i++) {
1093		if (count >= 2) {
1094		strcat(extras, line->elements[i].item);
1095		strcat(extras, line->elements[i].indent);
1096		}
1097
1098		if (!strcmp(line->elements[i].item, "menuentry"))
1099		continue;
1100
1101		/* count ' or ", there should be two in menuentry line. */
1102		if (isquote(*line->elements[i].item))
1103		count++;
1104
1105		len = strlen(line->elements[i].item);
1106
1107		if (isquote(line->elements[i].item[len -1]))
1108		count++;
1109
1110		/* ok, we get the final ' or ", others are extras. */
1111		}
1112		line->elements[1].indent =
1113		line->elements[line->numElements - 2].indent;
1114		line->elements[1].item = buf;
1115		line->elements[2].indent =
1116		line->elements[line->numElements - 2].indent;
1117		line->elements[2].item = extras;
1118		line->numElements = 3;
1119	} else if (line->type == LT_KERNELARGS && cfi->argsInQuotes) {	} else if (line->type == LT_KERNELARGS && cfi->argsInQuotes) {
1120	/* Strip off any " which may be present; they'll be put back	/* Strip off any " which may be present; they'll be put back
1121	on write. This is one of the few (the only?) places that grubby	on write. This is one of the few (the only?) places that grubby
#	Line 887  static struct grubConfig * readConfig(co	Line 1124  static struct grubConfig * readConfig(co
1124	if (line->numElements >= 2) {	if (line->numElements >= 2) {
1125	int last, len;	int last, len;
1126
1127	if (*line->elements[1].item == '"')	if (isquote(*line->elements[1].item))
1128	memmove(line->elements[1].item, line->elements[1].item + 1,	memmove(line->elements[1].item, line->elements[1].item + 1,
1129	strlen(line->elements[1].item + 1) + 1);	strlen(line->elements[1].item + 1) + 1);
1130
1131	last = line->numElements - 1;	last = line->numElements - 1;
1132	len = strlen(line->elements[last].item) - 1;	len = strlen(line->elements[last].item) - 1;
1133	if (line->elements[last].item[len] == '"')	if (isquote(line->elements[last].item[len]))
1134	line->elements[last].item[len] = '\0';	line->elements[last].item[len] = '\0';
1135	}	}
1136	}	}
#	Line 931  static struct grubConfig * readConfig(co	Line 1168  static struct grubConfig * readConfig(co
1168	entry->lines = line;	entry->lines = line;
1169	else	else
1170	last->next = line;	last->next = line;
1171	dbgPrintf("readConfig added %d to %p\n", line->type, entry);	dbgPrintf("readConfig added %s to %p\n", getKeyByType(line->type, cfi), entry);
1172
1173		/* we could have seen this outside of an entry... if so, we
1174		* ignore it like any other line we don't grok */
1175		if (line->type == LT_ENTRY_END && sawEntry)
1176		sawEntry = 0;
1177	} else {	} else {
1178	if (!cfg->theLines)	if (!cfg->theLines)
1179	cfg->theLines = line;	cfg->theLines = line;
1180	else	else
1181	last->next = line;	last->next = line;
1182	dbgPrintf("readConfig added %d to cfg\n", line->type);	dbgPrintf("readConfig added %s to cfg\n", getKeyByType(line->type, cfi));
1183	}	}
1184
1185	last = line;	last = line;
#	Line 945  static struct grubConfig * readConfig(co	Line 1187  static struct grubConfig * readConfig(co
1187
1188	free(incoming);	free(incoming);
1189
1190		dbgPrintf("defaultLine is %s\n", defaultLine ? "set" : "unset");
1191	if (defaultLine) {	if (defaultLine) {
1192	if (cfi->defaultSupportSaved &&	if (defaultLine->numElements > 2 &&
1193		cfi->defaultSupportSaved &&
1194		!strncmp(defaultLine->elements[2].item,"\"${saved_entry}\"", 16)) {
1195		cfg->defaultImage = DEFAULT_SAVED_GRUB2;
1196		} else if (cfi->defaultIsVariable) {
1197		char *value = defaultLine->elements[2].item;
1198		while (*value && (*value == '"' || *value == '\'' ||
1199		*value == ' ' || *value == '\t'))
1200		value++;
1201		cfg->defaultImage = strtol(value, &end, 10);
1202		while (*end && (*end == '"' || *end == '\'' ||
1203		*end == ' ' || *end == '\t'))
1204		end++;
1205		if (*end) cfg->defaultImage = -1;
1206		} else if (cfi->defaultSupportSaved &&
1207	!strncmp(defaultLine->elements[1].item, "saved", 5)) {	!strncmp(defaultLine->elements[1].item, "saved", 5)) {
1208	cfg->defaultImage = DEFAULT_SAVED;	cfg->defaultImage = DEFAULT_SAVED;
1209	} else if (cfi->defaultIsIndex) {	} else if (cfi->defaultIsIndex) {
1210	cfg->defaultImage = strtol(defaultLine->elements[1].item, &end, 10);	cfg->defaultImage = strtol(defaultLine->elements[1].item, &end, 10);
1211	if (*end) cfg->defaultImage = -1;	if (*end) cfg->defaultImage = -1;
1212	} else if (defaultLine->numElements >= 2) {	} else if (defaultLine->numElements >= 2) {
1213	i = 0;	int i = 0;
1214	while ((entry = findEntryByIndex(cfg, i))) {	while ((entry = findEntryByIndex(cfg, i))) {
1215	for (line = entry->lines; line; line = line->next)	for (line = entry->lines; line; line = line->next)
1216	if (line->type == LT_TITLE) break;	if (line->type == LT_TITLE) break;
#	Line 993  static void writeDefault(FILE * out, cha	Line 1250  static void writeDefault(FILE * out, cha
1250
1251	if (cfg->defaultImage == DEFAULT_SAVED)	if (cfg->defaultImage == DEFAULT_SAVED)
1252	fprintf(out, "%sdefault%ssaved\n", indent, separator);	fprintf(out, "%sdefault%ssaved\n", indent, separator);
1253		else if (cfg->defaultImage == DEFAULT_SAVED_GRUB2)
1254		fprintf(out, "%sset default=\"${saved_entry}\"\n", indent);
1255	else if (cfg->defaultImage > -1) {	else if (cfg->defaultImage > -1) {
1256	if (cfg->cfi->defaultIsIndex) {	if (cfg->cfi->defaultIsIndex) {
1257	fprintf(out, "%sdefault%s%d\n", indent, separator,	if (cfg->cfi->defaultIsVariable) {
1258	cfg->defaultImage);	fprintf(out, "%sset default=\"%d\"\n", indent,
1259		cfg->defaultImage);
1260		} else {
1261		fprintf(out, "%sdefault%s%d\n", indent, separator,
1262		cfg->defaultImage);
1263		}
1264	} else {	} else {
1265	int image = cfg->defaultImage;	int image = cfg->defaultImage;
1266
#	Line 1048  static int writeConfig(struct grubConfig	Line 1312  static int writeConfig(struct grubConfig
1312
1313	/* most likely the symlink is relative, so change our	/* most likely the symlink is relative, so change our
1314	directory to the dir of the symlink */	directory to the dir of the symlink */
1315	rc = chdir(dirname(strdupa(outName)));	char *dir = strdupa(outName);
1316		rc = chdir(dirname(dir));
1317		free(dir);
1318	do {	do {
1319	buf = alloca(len + 1);	buf = alloca(len + 1);
1320	rc = readlink(basename(outName), buf, len);	rc = readlink(basename(outName), buf, len);
#	Line 1086  static int writeConfig(struct grubConfig	Line 1352  static int writeConfig(struct grubConfig
1352	}	}
1353
1354	line = cfg->theLines;	line = cfg->theLines;
1355		struct keywordTypes *defaultKw = getKeywordByType(LT_DEFAULT, cfg->cfi);
1356	while (line) {	while (line) {
1357	if (line->type == LT_DEFAULT) {	if (line->type == LT_SET_VARIABLE && defaultKw &&
1358		line->numElements == 3 &&
1359		!strcmp(line->elements[1].item, defaultKw->key)) {
1360		writeDefault(out, line->indent, line->elements[0].indent, cfg);
1361		needs &= ~MAIN_DEFAULT;
1362		} else if (line->type == LT_DEFAULT) {
1363	writeDefault(out, line->indent, line->elements[0].indent, cfg);	writeDefault(out, line->indent, line->elements[0].indent, cfg);
1364	needs &= ~MAIN_DEFAULT;	needs &= ~MAIN_DEFAULT;
1365	} else if (line->type == LT_FALLBACK) {	} else if (line->type == LT_FALLBACK) {
#	Line 1155  static int numEntries(struct grubConfig	Line 1427  static int numEntries(struct grubConfig
1427	return i;	return i;
1428	}	}
1429
1430		static char *findDiskForRoot()
1431		{
1432		int fd;
1433		char buf[65536];
1434		char *devname;
1435		char *chptr;
1436		int rc;
1437
1438		if ((fd = open(_PATH_MOUNTED, O_RDONLY)) < 0) {
1439		fprintf(stderr, "grubby: failed to open %s: %s\n",
1440		_PATH_MOUNTED, strerror(errno));
1441		return NULL;
1442		}
1443
1444		rc = read(fd, buf, sizeof(buf) - 1);
1445		if (rc <= 0) {
1446		fprintf(stderr, "grubby: failed to read %s: %s\n",
1447		_PATH_MOUNTED, strerror(errno));
1448		close(fd);
1449		return NULL;
1450		}
1451		close(fd);
1452		buf[rc] = '\0';
1453		chptr = buf;
1454
1455		char *foundanswer = NULL;
1456
1457		while (chptr && chptr != buf+rc) {
1458		devname = chptr;
1459
1460		/*
1461		* The first column of a mtab entry is the device, but if the entry is a
1462		* special device it won't start with /, so move on to the next line.
1463		*/
1464		if (*devname != '/') {
1465		chptr = strchr(chptr, '\n');
1466		if (chptr)
1467		chptr++;
1468		continue;
1469		}
1470
1471		/* Seek to the next space */
1472		chptr = strchr(chptr, ' ');
1473		if (!chptr) {
1474		fprintf(stderr, "grubby: error parsing %s: %s\n",
1475		_PATH_MOUNTED, strerror(errno));
1476		return NULL;
1477		}
1478
1479		/*
1480		* The second column of a mtab entry is the mount point, we are looking
1481		* for '/' obviously.
1482		*/
1483		if (*(++chptr) == '/' && *(++chptr) == ' ') {
1484		/* remember the last / entry in mtab */
1485		foundanswer = devname;
1486		}
1487
1488		/* Next line */
1489		chptr = strchr(chptr, '\n');
1490		if (chptr)
1491		chptr++;
1492		}
1493
1494		/* Return the last / entry found */
1495		if (foundanswer) {
1496		chptr = strchr(foundanswer, ' ');
1497		*chptr = '\0';
1498		return strdup(foundanswer);
1499		}
1500
1501		return NULL;
1502		}
1503
1504		void printEntry(struct singleEntry * entry) {
1505		int i;
1506		struct singleLine * line;
1507
1508		for (line = entry->lines; line; line = line->next) {
1509		fprintf(stderr, "DBG: %s", line->indent);
1510		for (i = 0; i < line->numElements; i++) {
1511		/* Need to handle this, because we strip the quotes from
1512		* menuentry when read it. */
1513		if (line->type == LT_MENUENTRY && i == 1) {
1514		if(!isquote(*line->elements[i].item))
1515		fprintf(stderr, "\'%s\'", line->elements[i].item);
1516		else
1517		fprintf(stderr, "%s", line->elements[i].item);
1518		fprintf(stderr, "%s", line->elements[i].indent);
1519
1520		continue;
1521		}
1522
1523		fprintf(stderr, "%s%s",
1524		line->elements[i].item, line->elements[i].indent);
1525		}
1526		fprintf(stderr, "\n");
1527		}
1528		}
1529
1530		void notSuitablePrintf(struct singleEntry * entry, const char *fmt, ...)
1531		{
1532		va_list argp;
1533
1534		if (!debug)
1535		return;
1536
1537		va_start(argp, fmt);
1538		fprintf(stderr, "DBG: Image entry failed: ");
1539		vfprintf(stderr, fmt, argp);
1540		printEntry(entry);
1541		va_end(argp);
1542		}
1543
1544		#define beginswith(s, c) ((s) && (s)[0] == (c))
1545
1546		static int endswith(const char *s, char c)
1547		{
1548		int slen;
1549
1550		if (!s || !s[0])
1551		return 0;
1552		slen = strlen(s) - 1;
1553
1554		return s[slen] == c;
1555		}
1556
1557	int suitableImage(struct singleEntry * entry, const char * bootPrefix,	int suitableImage(struct singleEntry * entry, const char * bootPrefix,
1558	int skipRemoved, int flags) {	int skipRemoved, int flags) {
1559	struct singleLine * line;	struct singleLine * line;
1560	char * fullName;	char * fullName;
1561	int i;	int i;
	struct stat sb, sb2;
1562	char * dev;	char * dev;
1563	char * rootspec;	char * rootspec;
1564		char * rootdev;
1565
1566	if (skipRemoved && entry->skip) return 0;	if (skipRemoved && entry->skip) {
1567		notSuitablePrintf(entry, "marked to skip\n");
1568		return 0;
1569		}
1570
1571	line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines);	line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines);
1572	if (!line || line->numElements < 2) return 0;	if (!line) {
1573		notSuitablePrintf(entry, "no line found\n");
1574		return 0;
1575		}
1576		if (line->numElements < 2) {
1577		notSuitablePrintf(entry, "line has only %d elements\n",
1578		line->numElements);
1579		return 0;
1580		}
1581
1582	if (flags & GRUBBY_BADIMAGE_OKAY) return 1;	if (flags & GRUBBY_BADIMAGE_OKAY) return 1;
1583
1584	fullName = alloca(strlen(bootPrefix) +	fullName = alloca(strlen(bootPrefix) +
1585	strlen(line->elements[1].item) + 1);	strlen(line->elements[1].item) + 1);
1586	rootspec = getRootSpecifier(line->elements[1].item);	rootspec = getRootSpecifier(line->elements[1].item);
1587	sprintf(fullName, "%s%s", bootPrefix,	int rootspec_offset = rootspec ? strlen(rootspec) : 0;
1588	line->elements[1].item + (rootspec ? strlen(rootspec) : 0));	int hasslash = endswith(bootPrefix, '/') ||
1589	if (access(fullName, R_OK)) return 0;	beginswith(line->elements[1].item + rootspec_offset, '/');
1590		sprintf(fullName, "%s%s%s", bootPrefix, hasslash ? "" : "/",
1591		line->elements[1].item + rootspec_offset);
1592		if (access(fullName, R_OK)) {
1593		notSuitablePrintf(entry, "access to %s failed\n", fullName);
1594		return 0;
1595		}
1596	for (i = 2; i < line->numElements; i++)	for (i = 2; i < line->numElements; i++)
1597	if (!strncasecmp(line->elements[i].item, "root=", 5)) break;	if (!strncasecmp(line->elements[i].item, "root=", 5)) break;
1598	if (i < line->numElements) {	if (i < line->numElements) {
#	Line 1195  int suitableImage(struct singleEntry * e	Line 1610  int suitableImage(struct singleEntry * e
1610	line = getLineByType(LT_KERNELARGS|LT_MBMODULE, entry->lines);	line = getLineByType(LT_KERNELARGS|LT_MBMODULE, entry->lines);
1611
1612	/* failed to find one */	/* failed to find one */
1613	if (!line) return 0;	if (!line) {
1614		notSuitablePrintf(entry, "no line found\n");
1615		return 0;
1616		}
1617
1618	for (i = 1; i < line->numElements; i++)	for (i = 1; i < line->numElements; i++)
1619	if (!strncasecmp(line->elements[i].item, "root=", 5)) break;	if (!strncasecmp(line->elements[i].item, "root=", 5)) break;
1620	if (i < line->numElements)	if (i < line->numElements)
1621	dev = line->elements[i].item + 5;	dev = line->elements[i].item + 5;
1622	else {	else {
1623		notSuitablePrintf(entry, "no root= entry found\n");
1624	/* it failed too...  can't find root= */	/* it failed too...  can't find root= */
1625	return 0;	return 0;
1626	}	}
#	Line 1209  int suitableImage(struct singleEntry * e	Line 1628  int suitableImage(struct singleEntry * e
1628	}	}
1629
1630	dev = getpathbyspec(dev);	dev = getpathbyspec(dev);
1631	if (!dev)	if (!getpathbyspec(dev)) {
1632		notSuitablePrintf(entry, "can't find blkid entry for %s\n", dev);
1633	return 0;	return 0;
1634		} else
1635		dev = getpathbyspec(dev);
1636
1637	i = stat(dev, &sb);	rootdev = findDiskForRoot();
1638	if (i)	if (!rootdev) {
1639		notSuitablePrintf(entry, "can't find root device\n");
1640	return 0;	return 0;
1641		}
1642
1643	stat("/", &sb2);	if (!getuuidbydev(rootdev) || !getuuidbydev(dev)) {
1644		notSuitablePrintf(entry, "uuid missing: rootdev %s, dev %s\n",
1645		getuuidbydev(rootdev), getuuidbydev(dev));
1646		free(rootdev);
1647		return 0;
1648		}
1649
1650	if (sb.st_rdev != sb2.st_dev)	if (strcmp(getuuidbydev(rootdev), getuuidbydev(dev))) {
1651		notSuitablePrintf(entry, "uuid mismatch: rootdev %s, dev %s\n",
1652		getuuidbydev(rootdev), getuuidbydev(dev));
1653		free(rootdev);
1654	return 0;	return 0;
1655		}
1656
1657		free(rootdev);
1658
1659	return 1;	return 1;
1660	}	}
#	Line 1301  struct singleEntry * findEntryByPath(str	Line 1736  struct singleEntry * findEntryByPath(str
1736
1737	if (!strncmp(kernel, "TITLE=", 6)) {	if (!strncmp(kernel, "TITLE=", 6)) {
1738	prefix = "";	prefix = "";
1739	checkType = LT_TITLE;	checkType = LT_TITLE|LT_MENUENTRY;
1740	kernel += 6;	kernel += 6;
1741	}	}
1742
#	Line 1317  struct singleEntry * findEntryByPath(str	Line 1752  struct singleEntry * findEntryByPath(str
1752	checkType, line);	checkType, line);
1753	if (!line) break;  /* not found in this entry */	if (!line) break;  /* not found in this entry */
1754
1755	if (line && line->numElements >= 2) {	if (line && line->type != LT_MENUENTRY &&
1756		line->numElements >= 2) {
1757	rootspec = getRootSpecifier(line->elements[1].item);	rootspec = getRootSpecifier(line->elements[1].item);
1758	if (!strcmp(line->elements[1].item +	if (!strcmp(line->elements[1].item +
1759	((rootspec != NULL) ? strlen(rootspec) : 0),	((rootspec != NULL) ? strlen(rootspec) : 0),
1760	kernel + strlen(prefix)))	kernel + strlen(prefix)))
1761	break;	break;
1762	}	}
1763		if(line->type == LT_MENUENTRY &&
1764		!strcmp(line->elements[1].item, kernel))
1765		break;
1766	}	}
1767
1768	/* make sure this entry has a kernel identifier; this skips	/* make sure this entry has a kernel identifier; this skips
#	Line 1415  void markRemovedImage(struct grubConfig	Line 1854  void markRemovedImage(struct grubConfig
1854	const char * prefix) {	const char * prefix) {
1855	struct singleEntry * entry;	struct singleEntry * entry;
1856
1857	if (!image) return;	if (!image)
1858		return;
1859
1860		/* check and see if we're removing the default image */
1861		if (isdigit(*image)) {
1862		entry = findEntryByPath(cfg, image, prefix, NULL);
1863		if(entry)
1864		entry->skip = 1;
1865		return;
1866		}
1867
1868	while ((entry = findEntryByPath(cfg, image, prefix, NULL)))	while ((entry = findEntryByPath(cfg, image, prefix, NULL)))
1869	entry->skip = 1;	entry->skip = 1;
#	Line 1423  void markRemovedImage(struct grubConfig	Line 1871  void markRemovedImage(struct grubConfig
1871
1872	void setDefaultImage(struct grubConfig * config, int hasNew,	void setDefaultImage(struct grubConfig * config, int hasNew,
1873	const char * defaultKernelPath, int newIsDefault,	const char * defaultKernelPath, int newIsDefault,
1874	const char * prefix, int flags) {	const char * prefix, int flags, int index) {
1875	struct singleEntry * entry, * entry2, * newDefault;	struct singleEntry * entry, * entry2, * newDefault;
1876	int i, j;	int i, j;
1877
1878	if (newIsDefault) {	if (newIsDefault) {
1879	config->defaultImage = 0;	config->defaultImage = 0;
1880	return;	return;
1881		} else if ((index >= 0) && config->cfi->defaultIsIndex) {
1882		if (findEntryByIndex(config, index))
1883		config->defaultImage = index;
1884		else
1885		config->defaultImage = -1;
1886		return;
1887	} else if (defaultKernelPath) {	} else if (defaultKernelPath) {
1888	i = 0;	i = 0;
1889	if (findEntryByPath(config, defaultKernelPath, prefix, &i)) {	if (findEntryByPath(config, defaultKernelPath, prefix, &i)) {
#	Line 1442  void setDefaultImage(struct grubConfig *	Line 1896  void setDefaultImage(struct grubConfig *
1896
1897	/* defaultImage now points to what we'd like to use, but before any order	/* defaultImage now points to what we'd like to use, but before any order
1898	changes */	changes */
1899	if (config->defaultImage == DEFAULT_SAVED)	if ((config->defaultImage == DEFAULT_SAVED) ||
1900		(config->defaultImage == DEFAULT_SAVED_GRUB2))
1901	/* default is set to saved, we don't want to change it */	/* default is set to saved, we don't want to change it */
1902	return;	return;
1903
#	Line 1509  void displayEntry(struct singleEntry * e	Line 1964  void displayEntry(struct singleEntry * e
1964	return;	return;
1965	}	}
1966
1967	printf("kernel=%s\n", line->elements[1].item);	if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))
1968		printf("kernel=%s\n", line->elements[1].item);
1969		else
1970		printf("kernel=%s%s\n", prefix, line->elements[1].item);
1971
1972	if (line->numElements >= 3) {	if (line->numElements >= 3) {
1973	printf("args=\"");	printf("args=\"");
#	Line 1568  void displayEntry(struct singleEntry * e	Line 2026  void displayEntry(struct singleEntry * e
2026	line = getLineByType(LT_INITRD, entry->lines);	line = getLineByType(LT_INITRD, entry->lines);
2027
2028	if (line && line->numElements >= 2) {	if (line && line->numElements >= 2) {
2029	printf("initrd=%s", prefix);	if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))
2030		printf("initrd=");
2031		else
2032		printf("initrd=%s", prefix);
2033
2034	for (i = 1; i < line->numElements; i++)	for (i = 1; i < line->numElements; i++)
2035	printf("%s%s", line->elements[i].item, line->elements[i].indent);	printf("%s%s", line->elements[i].item, line->elements[i].indent);
2036	printf("\n");	printf("\n");
2037	}	}
2038
2039		line = getLineByType(LT_TITLE, entry->lines);
2040		if (line) {
2041		printf("title=%s\n", line->elements[1].item);
2042		} else {
2043		char * title;
2044		line = getLineByType(LT_MENUENTRY, entry->lines);
2045		title = grub2ExtractTitle(line);
2046		if (title)
2047		printf("title=%s\n", title);
2048		}
2049		}
2050
2051		int isSuseSystem(void) {
2052		const char * path;
2053		const static char default_path[] = "/etc/SuSE-release";
2054
2055		if ((path = getenv("GRUBBY_SUSE_RELEASE")) == NULL)
2056		path = default_path;
2057
2058		if (!access(path, R_OK))
2059		return 1;
2060		return 0;
2061		}
2062
2063		int isSuseGrubConf(const char * path) {
2064		FILE * grubConf;
2065		char * line = NULL;
2066		size_t len = 0, res = 0;
2067
2068		grubConf = fopen(path, "r");
2069		if (!grubConf) {
2070		dbgPrintf("Could not open SuSE configuration file '%s'\n", path);
2071		return 0;
2072		}
2073
2074		while ((res = getline(&line, &len, grubConf)) != -1) {
2075		if (!strncmp(line, "setup", 5)) {
2076		fclose(grubConf);
2077		free(line);
2078		return 1;
2079		}
2080		}
2081
2082		dbgPrintf("SuSE configuration file '%s' does not appear to be valid\n",
2083		path);
2084
2085		fclose(grubConf);
2086		free(line);
2087		return 0;
2088		}
2089
2090		int suseGrubConfGetLba(const char * path, int * lbaPtr) {
2091		FILE * grubConf;
2092		char * line = NULL;
2093		size_t res = 0, len = 0;
2094
2095		if (!path) return 1;
2096		if (!lbaPtr) return 1;
2097
2098		grubConf = fopen(path, "r");
2099		if (!grubConf) return 1;
2100
2101		while ((res = getline(&line, &len, grubConf)) != -1) {
2102		if (line[res - 1] == '\n')
2103		line[res - 1] = '\0';
2104		else if (len > res)
2105		line[res] = '\0';
2106		else {
2107		line = realloc(line, res + 1);
2108		line[res] = '\0';
2109		}
2110
2111		if (!strncmp(line, "setup", 5)) {
2112		if (strstr(line, "--force-lba")) {
2113		*lbaPtr = 1;
2114		} else {
2115		*lbaPtr = 0;
2116		}
2117		dbgPrintf("lba: %i\n", *lbaPtr);
2118		break;
2119		}
2120		}
2121
2122		free(line);
2123		fclose(grubConf);
2124		return 0;
2125		}
2126
2127		int suseGrubConfGetInstallDevice(const char * path, char ** devicePtr) {
2128		FILE * grubConf;
2129		char * line = NULL;
2130		size_t res = 0, len = 0;
2131		char * lastParamPtr = NULL;
2132		char * secLastParamPtr = NULL;
2133		char installDeviceNumber = '\0';
2134		char * bounds = NULL;
2135
2136		if (!path) return 1;
2137		if (!devicePtr) return 1;
2138
2139		grubConf = fopen(path, "r");
2140		if (!grubConf) return 1;
2141
2142		while ((res = getline(&line, &len, grubConf)) != -1) {
2143		if (strncmp(line, "setup", 5))
2144		continue;
2145
2146		if (line[res - 1] == '\n')
2147		line[res - 1] = '\0';
2148		else if (len > res)
2149		line[res] = '\0';
2150		else {
2151		line = realloc(line, res + 1);
2152		line[res] = '\0';
2153		}
2154
2155		lastParamPtr = bounds = line + res;
2156
2157		/* Last parameter in grub may be an optional IMAGE_DEVICE */
2158		while (!isspace(*lastParamPtr))
2159		lastParamPtr--;
2160		lastParamPtr++;
2161
2162		secLastParamPtr = lastParamPtr - 2;
2163		dbgPrintf("lastParamPtr: %s\n", lastParamPtr);
2164
2165		if (lastParamPtr + 3 > bounds) {
2166		dbgPrintf("lastParamPtr going over boundary");
2167		fclose(grubConf);
2168		free(line);
2169		return 1;
2170		}
2171		if (!strncmp(lastParamPtr, "(hd", 3))
2172		lastParamPtr += 3;
2173		dbgPrintf("lastParamPtr: %c\n", *lastParamPtr);
2174
2175		/*
2176		* Second last parameter will decide wether last parameter is
2177		* an IMAGE_DEVICE or INSTALL_DEVICE
2178		*/
2179		while (!isspace(*secLastParamPtr))
2180		secLastParamPtr--;
2181		secLastParamPtr++;
2182
2183		if (secLastParamPtr + 3 > bounds) {
2184		dbgPrintf("secLastParamPtr going over boundary");
2185		fclose(grubConf);
2186		free(line);
2187		return 1;
2188		}
2189		dbgPrintf("secLastParamPtr: %s\n", secLastParamPtr);
2190		if (!strncmp(secLastParamPtr, "(hd", 3)) {
2191		secLastParamPtr += 3;
2192		dbgPrintf("secLastParamPtr: %c\n", *secLastParamPtr);
2193		installDeviceNumber = *secLastParamPtr;
2194		} else {
2195		installDeviceNumber = *lastParamPtr;
2196		}
2197
2198		*devicePtr = malloc(6);
2199		snprintf(*devicePtr, 6, "(hd%c)", installDeviceNumber);
2200		dbgPrintf("installDeviceNumber: %c\n", installDeviceNumber);
2201		fclose(grubConf);
2202		free(line);
2203		return 0;
2204		}
2205
2206		free(line);
2207		fclose(grubConf);
2208		return 1;
2209		}
2210
2211		int grubGetBootFromDeviceMap(const char * device,
2212		char ** bootPtr) {
2213		FILE * deviceMap;
2214		char * line = NULL;
2215		size_t res = 0, len = 0;
2216		char * devicePtr;
2217		char * bounds = NULL;
2218		const char * path;
2219		const static char default_path[] = "/boot/grub/device.map";
2220
2221		if (!device) return 1;
2222		if (!bootPtr) return 1;
2223
2224		if ((path = getenv("GRUBBY_GRUB_DEVICE_MAP")) == NULL)
2225		path = default_path;
2226
2227		dbgPrintf("opening grub device.map file from: %s\n", path);
2228		deviceMap = fopen(path, "r");
2229		if (!deviceMap)
2230		return 1;
2231
2232		while ((res = getline(&line, &len, deviceMap)) != -1) {
2233		if (!strncmp(line, "#", 1))
2234		continue;
2235
2236		if (line[res - 1] == '\n')
2237		line[res - 1] = '\0';
2238		else if (len > res)
2239		line[res] = '\0';
2240		else {
2241		line = realloc(line, res + 1);
2242		line[res] = '\0';
2243		}
2244
2245		devicePtr = line;
2246		bounds = line + res;
2247
2248		while ((isspace(*line) && ((devicePtr + 1) <= bounds)))
2249		devicePtr++;
2250		dbgPrintf("device: %s\n", devicePtr);
2251
2252		if (!strncmp(devicePtr, device, strlen(device))) {
2253		devicePtr += strlen(device);
2254		while (isspace(*devicePtr) && ((devicePtr + 1) <= bounds))
2255		devicePtr++;
2256
2257		*bootPtr = strdup(devicePtr);
2258		break;
2259		}
2260		}
2261
2262		free(line);
2263		fclose(deviceMap);
2264		return 0;
2265		}
2266
2267		int suseGrubConfGetBoot(const char * path, char ** bootPtr) {
2268		char * grubDevice;
2269
2270		if (suseGrubConfGetInstallDevice(path, &grubDevice))
2271		dbgPrintf("error looking for grub installation device\n");
2272		else
2273		dbgPrintf("grubby installation device: %s\n", grubDevice);
2274
2275		if (grubGetBootFromDeviceMap(grubDevice, bootPtr))
2276		dbgPrintf("error looking for grub boot device\n");
2277		else
2278		dbgPrintf("grubby boot device: %s\n", *bootPtr);
2279
2280		free(grubDevice);
2281		return 0;
2282		}
2283
2284		int parseSuseGrubConf(int * lbaPtr, char ** bootPtr) {
2285		/*
2286		* This SuSE grub configuration file at this location is not your average
2287		* grub configuration file, but instead the grub commands used to setup
2288		* grub on that system.
2289		*/
2290		const char * path;
2291		const static char default_path[] = "/etc/grub.conf";
2292
2293		if ((path = getenv("GRUBBY_SUSE_GRUB_CONF")) == NULL)
2294		path = default_path;
2295
2296		if (!isSuseGrubConf(path)) return 1;
2297
2298		if (lbaPtr) {
2299		*lbaPtr = 0;
2300		if (suseGrubConfGetLba(path, lbaPtr))
2301		return 1;
2302		}
2303
2304		if (bootPtr) {
2305		*bootPtr = NULL;
2306		suseGrubConfGetBoot(path, bootPtr);
2307		}
2308
2309		return 0;
2310	}	}
2311
2312	int parseSysconfigGrub(int * lbaPtr, char ** bootPtr) {	int parseSysconfigGrub(int * lbaPtr, char ** bootPtr) {
#	Line 1623  int parseSysconfigGrub(int * lbaPtr, cha	Line 2357  int parseSysconfigGrub(int * lbaPtr, cha
2357	}	}
2358
2359	void dumpSysconfigGrub(void) {	void dumpSysconfigGrub(void) {
2360	char * boot;	char * boot = NULL;
2361	int lba;	int lba;
2362
2363	if (!parseSysconfigGrub(&lba, &boot)) {	if (isSuseSystem()) {
2364	if (lba) printf("lba\n");	if (parseSuseGrubConf(&lba, &boot)) {
2365	if (boot) printf("boot=%s\n", boot);	free(boot);
2366		return;
2367		}
2368		} else {
2369		if (parseSysconfigGrub(&lba, &boot)) {
2370		free(boot);
2371		return;
2372		}
2373		}
2374
2375		if (lba) printf("lba\n");
2376		if (boot) {
2377		printf("boot=%s\n", boot);
2378		free(boot);
2379	}	}
2380	}	}
2381
#	Line 1733  struct singleLine *  addLine(struct sing	Line 2480  struct singleLine *  addLine(struct sing
2480	sprintf(tmpl.elements[0].item, "[%s]", val);	sprintf(tmpl.elements[0].item, "[%s]", val);
2481	tmpl.elements[0].indent = "";	tmpl.elements[0].indent = "";
2482	val = NULL;	val = NULL;
2483		} else if (type == LT_MENUENTRY) {
2484		char *lineend = "--class gnu-linux --class gnu --class os {";
2485		if (!val) {
2486		fprintf(stderr, "Line type LT_MENUENTRY requires a value\n");
2487		abort();
2488		}
2489		kw = getKeywordByType(type, cfi);
2490		if (!kw) {
2491		fprintf(stderr, "Looking up keyword for unknown type %d\n", type);
2492		abort();
2493		}
2494		tmpl.indent = "";
2495		tmpl.type = type;
2496		tmpl.numElements = 3;
2497		tmpl.elements = alloca(sizeof(*tmpl.elements) * tmpl.numElements);
2498		tmpl.elements[0].item = kw->key;
2499		tmpl.elements[0].indent = alloca(2);
2500		sprintf(tmpl.elements[0].indent, "%c", kw->nextChar);
2501		tmpl.elements[1].item = (char *)val;
2502		tmpl.elements[1].indent = alloca(2);
2503		sprintf(tmpl.elements[1].indent, "%c", kw->nextChar);
2504		tmpl.elements[2].item = alloca(strlen(lineend)+1);
2505		strcpy(tmpl.elements[2].item, lineend);
2506		tmpl.elements[2].indent = "";
2507	} else {	} else {
2508	kw = getKeywordByType(type, cfi);	kw = getKeywordByType(type, cfi);
2509	if (!kw) abort();	if (!kw) {
2510		fprintf(stderr, "Looking up keyword for unknown type %d\n", type);
2511		abort();
2512		}
2513	tmpl.type = type;	tmpl.type = type;
2514	tmpl.numElements = val ? 2 : 1;	tmpl.numElements = val ? 2 : 1;
2515	tmpl.elements = alloca(sizeof(*tmpl.elements) * tmpl.numElements);	tmpl.elements = alloca(sizeof(*tmpl.elements) * tmpl.numElements);
#	Line 1759  struct singleLine *  addLine(struct sing	Line 2533  struct singleLine *  addLine(struct sing
2533	if (!line->next && !prev) prev = line;	if (!line->next && !prev) prev = line;
2534	}	}
2535
2536	if (prev == entry->lines)	struct singleLine *menuEntry;
2537	tmpl.indent = defaultIndent ?: "";	menuEntry = getLineByType(LT_MENUENTRY, entry->lines);
2538	else	if (tmpl.type == LT_ENTRY_END) {
2539	tmpl.indent = prev->indent;	if (menuEntry)
2540		tmpl.indent = menuEntry->indent;
2541		else
2542		tmpl.indent = defaultIndent ?: "";
2543		} else if (tmpl.type != LT_MENUENTRY) {
2544		if (menuEntry)
2545		tmpl.indent = "\t";
2546		else if (prev == entry->lines)
2547		tmpl.indent = defaultIndent ?: "";
2548		else
2549		tmpl.indent = prev->indent;
2550		}
2551
2552	return addLineTmpl(entry, &tmpl, prev, val, cfi);	return addLineTmpl(entry, &tmpl, prev, val, cfi);
2553	}	}
#	Line 1789  void removeLine(struct singleEntry * ent	Line 2574  void removeLine(struct singleEntry * ent
2574	free(line);	free(line);
2575	}	}
2576
2577		static void requote(struct singleLine *tmplLine, struct configFileInfo * cfi)
2578		{
2579		struct singleLine newLine = {
2580		.indent = tmplLine->indent,
2581		.type = tmplLine->type,
2582		.next = tmplLine->next,
2583		};
2584		int firstQuotedItem = -1;
2585		int quoteLen = 0;
2586		int j;
2587		int element = 0;
2588		char *c;
2589
2590		c = malloc(strlen(tmplLine->elements[0].item) + 1);
2591		strcpy(c, tmplLine->elements[0].item);
2592		insertElement(&newLine, c, element++, cfi);
2593		free(c);
2594		c = NULL;
2595
2596		for (j = 1; j < tmplLine->numElements; j++) {
2597		if (firstQuotedItem == -1) {
2598		quoteLen += strlen(tmplLine->elements[j].item);
2599
2600		if (isquote(tmplLine->elements[j].item[0])) {
2601		firstQuotedItem = j;
2602		quoteLen += strlen(tmplLine->elements[j].indent);
2603		} else {
2604		c = malloc(quoteLen + 1);
2605		strcpy(c, tmplLine->elements[j].item);
2606		insertElement(&newLine, c, element++, cfi);
2607		free(c);
2608		quoteLen = 0;
2609		}
2610		} else {
2611		int itemlen = strlen(tmplLine->elements[j].item);
2612		quoteLen += itemlen;
2613		quoteLen += strlen(tmplLine->elements[j].indent);
2614
2615		if (isquote(tmplLine->elements[j].item[itemlen - 1])) {
2616		c = malloc(quoteLen + 1);
2617		c[0] = '\0';
2618		for (int i = firstQuotedItem; i < j+1; i++) {
2619		strcat(c, tmplLine->elements[i].item);
2620		strcat(c, tmplLine->elements[i].indent);
2621		}
2622		insertElement(&newLine, c, element++, cfi);
2623		free(c);
2624
2625		firstQuotedItem = -1;
2626		quoteLen = 0;
2627		}
2628		}
2629		}
2630		while (tmplLine->numElements)
2631		removeElement(tmplLine, 0);
2632		if (tmplLine->elements)
2633		free(tmplLine->elements);
2634
2635		tmplLine->numElements = newLine.numElements;
2636		tmplLine->elements = newLine.elements;
2637		}
2638
2639	static void insertElement(struct singleLine * line,	static void insertElement(struct singleLine * line,
2640	const char * item, int insertHere,	const char * item, int insertHere,
2641	struct configFileInfo * cfi)	struct configFileInfo * cfi)
#	Line 1895  int updateActualImage(struct grubConfig	Line 2742  int updateActualImage(struct grubConfig
2742	const char ** arg;	const char ** arg;
2743	int useKernelArgs, useRoot;	int useKernelArgs, useRoot;
2744	int firstElement;	int firstElement;
2745	int *usedElements, *usedArgs;	int *usedElements;
2746	int doreplace;	int doreplace;
2747
2748	if (!image) return 0;	if (!image) return 0;
#	Line 1930  int updateActualImage(struct grubConfig	Line 2777  int updateActualImage(struct grubConfig
2777	useRoot = (getKeywordByType(LT_ROOT, cfg->cfi)	useRoot = (getKeywordByType(LT_ROOT, cfg->cfi)
2778	&& !multibootArgs);	&& !multibootArgs);
2779
	for (k = 0, arg = newArgs; *arg; arg++, k++) ;
	usedArgs = calloc(k, sizeof(*usedArgs));
2780	for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {	for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {
2781
2782	if (multibootArgs && !entry->multiboot)	if (multibootArgs && !entry->multiboot)
#	Line 2008  int updateActualImage(struct grubConfig	Line 2852  int updateActualImage(struct grubConfig
2852	usedElements = calloc(line->numElements, sizeof(*usedElements));	usedElements = calloc(line->numElements, sizeof(*usedElements));
2853
2854	for (k = 0, arg = newArgs; *arg; arg++, k++) {	for (k = 0, arg = newArgs; *arg; arg++, k++) {
	if (usedArgs[k]) continue;
2855
2856	doreplace = 1;	doreplace = 1;
2857	for (i = firstElement; i < line->numElements; i++) {	for (i = firstElement; i < line->numElements; i++) {
#	Line 2023  int updateActualImage(struct grubConfig	Line 2866  int updateActualImage(struct grubConfig
2866	continue;	continue;
2867	if (!argMatch(line->elements[i].item, *arg)) {	if (!argMatch(line->elements[i].item, *arg)) {
2868	usedElements[i]=1;	usedElements[i]=1;
	usedArgs[k]=1;
2869	break;	break;
2870	}	}
2871	}	}
#	Line 2093  int updateActualImage(struct grubConfig	Line 2935  int updateActualImage(struct grubConfig
2935	}	}
2936	}	}
2937
	free(usedArgs);
2938	free(newArgs);	free(newArgs);
2939	free(oldArgs);	free(oldArgs);
2940
#	Line 2119  int updateImage(struct grubConfig * cfg,	Line 2960  int updateImage(struct grubConfig * cfg,
2960	return rc;	return rc;
2961	}	}
2962
2963		int updateInitrd(struct grubConfig * cfg, const char * image,
2964		const char * prefix, const char * initrd) {
2965		struct singleEntry * entry;
2966		struct singleLine * line, * kernelLine, *endLine = NULL;
2967		int index = 0;
2968
2969		if (!image) return 0;
2970
2971		for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {
2972		kernelLine = getLineByType(LT_KERNEL, entry->lines);
2973		if (!kernelLine) continue;
2974
2975		line = getLineByType(LT_INITRD, entry->lines);
2976		if (line)
2977		removeLine(entry, line);
2978		if (prefix) {
2979		int prefixLen = strlen(prefix);
2980		if (!strncmp(initrd, prefix, prefixLen))
2981		initrd += prefixLen;
2982		}
2983		endLine = getLineByType(LT_ENTRY_END, entry->lines);
2984		if (endLine)
2985		removeLine(entry, endLine);
2986		line = addLine(entry, cfg->cfi, LT_INITRD, kernelLine->indent, initrd);
2987		if (!line)
2988		return 1;
2989		if (endLine) {
2990		line = addLine(entry, cfg->cfi, LT_ENTRY_END, "", NULL);
2991		if (!line)
2992		return 1;
2993		}
2994
2995		break;
2996		}
2997
2998		return 0;
2999		}
3000
3001	int checkDeviceBootloader(const char * device, const unsigned char * boot) {	int checkDeviceBootloader(const char * device, const unsigned char * boot) {
3002	int fd;	int fd;
3003	unsigned char bootSect[512];	unsigned char bootSect[512];
#	Line 2142  int checkDeviceBootloader(const char * d	Line 3021  int checkDeviceBootloader(const char * d
3021	if (memcmp(boot, bootSect, 3))	if (memcmp(boot, bootSect, 3))
3022	return 0;	return 0;
3023
3024	if (boot[1] == 0xeb) {	if (boot[1] == JMP_SHORT_OPCODE) {
3025	offset = boot[2] + 2;	offset = boot[2] + 2;
3026	} else if (boot[1] == 0xe8 || boot[1] == 0xe9) {	} else if (boot[1] == 0xe8 || boot[1] == 0xe9) {
3027	offset = (boot[3] << 8) + boot[2] + 2;	offset = (boot[3] << 8) + boot[2] + 2;
3028	} else if (boot[0] == 0xeb) {	} else if (boot[0] == JMP_SHORT_OPCODE) {
3029	offset = boot[1] + 2;	offset = boot[1] + 2;
3030		/*
3031		* it looks like grub, when copying stage1 into the mbr, patches stage1
3032		* right after the JMP location, replacing other instructions such as
3033		* JMPs for NOOPs. So, relax the check a little bit by skipping those
3034		* different bytes.
3035		*/
3036		if ((bootSect[offset + 1] == NOOP_OPCODE)
3037		&& (bootSect[offset + 2] == NOOP_OPCODE)) {
3038		offset = offset + 3;
3039		}
3040	} else if (boot[0] == 0xe8 || boot[0] == 0xe9) {	} else if (boot[0] == 0xe8 || boot[0] == 0xe9) {
3041	offset = (boot[2] << 8) + boot[1] + 2;	offset = (boot[2] << 8) + boot[1] + 2;
3042	} else {	} else {
#	Line 2289  int checkForLilo(struct grubConfig * con	Line 3178  int checkForLilo(struct grubConfig * con
3178	return checkDeviceBootloader(line->elements[1].item, boot);	return checkDeviceBootloader(line->elements[1].item, boot);
3179	}	}
3180
3181		int checkForGrub2(struct grubConfig * config) {
3182		if (!access("/etc/grub.d/", R_OK))
3183		return 2;
3184
3185		return 1;
3186		}
3187
3188	int checkForGrub(struct grubConfig * config) {	int checkForGrub(struct grubConfig * config) {
3189	int fd;	int fd;
3190	unsigned char bootSect[512];	unsigned char bootSect[512];
3191	char * boot;	char * boot;
3192		int onSuse = isSuseSystem();
3193
3194	if (parseSysconfigGrub(NULL, &boot))
3195	return 0;	if (onSuse) {
3196		if (parseSuseGrubConf(NULL, &boot))
3197		return 0;
3198		} else {
3199		if (parseSysconfigGrub(NULL, &boot))
3200		return 0;
3201		}
3202
3203	/* assume grub is not installed -- not an error condition */	/* assume grub is not installed -- not an error condition */
3204	if (!boot)	if (!boot)
#	Line 2314  int checkForGrub(struct grubConfig * con	Line 3217  int checkForGrub(struct grubConfig * con
3217	}	}
3218	close(fd);	close(fd);
3219
3220		/* The more elaborate checks do not work on SuSE. The checks done
3221		* seem to be reasonble (at least for now), so just return success
3222		*/
3223		if (onSuse)
3224		return 2;
3225
3226	return checkDeviceBootloader(boot, bootSect);	return checkDeviceBootloader(boot, bootSect);
3227	}	}
3228
#	Line 2347  int checkForExtLinux(struct grubConfig *	Line 3256  int checkForExtLinux(struct grubConfig *
3256	return checkDeviceBootloader(boot, bootSect);	return checkDeviceBootloader(boot, bootSect);
3257	}	}
3258
3259		int checkForYaboot(struct grubConfig * config) {
3260		/*
3261		* This is a simplistic check that we consider good enough for own puporses
3262		*
3263		* If we were to properly check if yaboot is *installed* we'd need to:
3264		* 1) get the system boot device (LT_BOOT)
3265		* 2) considering it's a raw filesystem, check if the yaboot binary matches
3266		*    the content on the boot device
3267		* 3) if not, copy the binary to a temporary file and run "addnote" on it
3268		* 4) check again if binary and boot device contents match
3269		*/
3270		if (!access("/etc/yaboot.conf", R_OK))
3271		return 2;
3272
3273		return 1;
3274		}
3275
3276		int checkForElilo(struct grubConfig * config) {
3277		if (!access("/etc/elilo.conf", R_OK))
3278		return 2;
3279
3280		return 1;
3281		}
3282
3283	static char * getRootSpecifier(char * str) {	static char * getRootSpecifier(char * str) {
3284	char * idx, * rootspec = NULL;	char * idx, * rootspec = NULL;
3285
#	Line 2361  static char * getRootSpecifier(char * st	Line 3294  static char * getRootSpecifier(char * st
3294	static char * getInitrdVal(struct grubConfig * config,	static char * getInitrdVal(struct grubConfig * config,
3295	const char * prefix, struct singleLine *tmplLine,	const char * prefix, struct singleLine *tmplLine,
3296	const char * newKernelInitrd,	const char * newKernelInitrd,
3297	char ** extraInitrds, int extraInitrdCount)	const char ** extraInitrds, int extraInitrdCount)
3298	{	{
3299	char *initrdVal, *end;	char *initrdVal, *end;
3300	int i;	int i;
#	Line 2406  static char * getInitrdVal(struct grubCo	Line 3339  static char * getInitrdVal(struct grubCo
3339
3340	int addNewKernel(struct grubConfig * config, struct singleEntry * template,	int addNewKernel(struct grubConfig * config, struct singleEntry * template,
3341	const char * prefix,	const char * prefix,
3342	char * newKernelPath, char * newKernelTitle,	const char * newKernelPath, const char * newKernelTitle,
3343	char * newKernelArgs, char * newKernelInitrd,	const char * newKernelArgs, const char * newKernelInitrd,
3344	char ** extraInitrds, int extraInitrdCount,	const char ** extraInitrds, int extraInitrdCount,
3345	char * newMBKernel, char * newMBKernelArgs) {	const char * newMBKernel, const char * newMBKernelArgs) {
3346	struct singleEntry * new;	struct singleEntry * new;
3347	struct singleLine * newLine = NULL, * tmplLine = NULL, * masterLine = NULL;	struct singleLine * newLine = NULL, * tmplLine = NULL, * masterLine = NULL;
3348	int needs;	int needs;
#	Line 2464  int addNewKernel(struct grubConfig * con	Line 3397  int addNewKernel(struct grubConfig * con
3397	if (*chptr == '#') continue;	if (*chptr == '#') continue;
3398
3399	if (tmplLine->type == LT_KERNEL &&	if (tmplLine->type == LT_KERNEL &&
3400	tmplLine->numElements >= 2) {	tmplLine->numElements >= 2) {
3401	if (!template->multiboot && (needs & NEED_MB)) {	if (!template->multiboot && (needs & NEED_MB)) {
3402	/* it's not a multiboot template and this is the kernel	/* it's not a multiboot template and this is the kernel
3403	* line.  Try to be intelligent about inserting the	* line.  Try to be intelligent about inserting the
#	Line 2590  int addNewKernel(struct grubConfig * con	Line 3523  int addNewKernel(struct grubConfig * con
3523	needs &= ~NEED_INITRD;	needs &= ~NEED_INITRD;
3524	}	}
3525
3526		} else if (tmplLine->type == LT_MENUENTRY &&
3527		(needs & NEED_TITLE)) {
3528		requote(tmplLine, config->cfi);
3529		char *nkt = malloc(strlen(newKernelTitle)+3);
3530		strcpy(nkt, "'");
3531		strcat(nkt, newKernelTitle);
3532		strcat(nkt, "'");
3533		newLine = addLineTmpl(new, tmplLine, newLine, nkt, config->cfi);
3534		free(nkt);
3535		needs &= ~NEED_TITLE;
3536	} else if (tmplLine->type == LT_TITLE &&	} else if (tmplLine->type == LT_TITLE &&
3537	(needs & NEED_TITLE)) {	(needs & NEED_TITLE)) {
3538	if (tmplLine->numElements >= 2) {	if (tmplLine->numElements >= 2) {
#	Line 2603  int addNewKernel(struct grubConfig * con	Line 3546  int addNewKernel(struct grubConfig * con
3546	tmplLine->indent, newKernelTitle);	tmplLine->indent, newKernelTitle);
3547	needs &= ~NEED_TITLE;	needs &= ~NEED_TITLE;
3548	}	}
3549		} else if (tmplLine->type == LT_ECHO) {
3550		requote(tmplLine, config->cfi);
3551		static const char *prefix = "'Loading ";
3552		if (tmplLine->numElements > 1 &&
3553		strstr(tmplLine->elements[1].item, prefix) &&
3554		masterLine->next && masterLine->next->type == LT_KERNEL) {
3555		char *newTitle = malloc(strlen(prefix) +
3556		strlen(newKernelTitle) + 2);
3557
3558		strcpy(newTitle, prefix);
3559		strcat(newTitle, newKernelTitle);
3560		strcat(newTitle, "'");
3561		newLine = addLine(new, config->cfi, LT_ECHO,
3562		tmplLine->indent, newTitle);
3563		free(newTitle);
3564		} else {
3565		/* pass through other lines from the template */
3566		newLine = addLineTmpl(new, tmplLine, newLine, NULL,
3567		config->cfi);
3568		}
3569	} else {	} else {
3570	/* pass through other lines from the template */	/* pass through other lines from the template */
3571	newLine = addLineTmpl(new, tmplLine, newLine, NULL, config->cfi);	newLine = addLineTmpl(new, tmplLine, newLine, NULL, config->cfi);
#	Line 2614  int addNewKernel(struct grubConfig * con	Line 3576  int addNewKernel(struct grubConfig * con
3576	/* don't have a template, so start the entry with the	/* don't have a template, so start the entry with the
3577	* appropriate starting line	* appropriate starting line
3578	*/	*/
3579	switch (config->cfi->entrySeparator) {	switch (config->cfi->entryStart) {
3580	case LT_KERNEL:	case LT_KERNEL:
3581	if (new->multiboot && config->cfi->mbHyperFirst) {	if (new->multiboot && config->cfi->mbHyperFirst) {
3582	/* fall through to LT_HYPER */	/* fall through to LT_HYPER */
#	Line 2633  int addNewKernel(struct grubConfig * con	Line 3595  int addNewKernel(struct grubConfig * con
3595	needs &= ~NEED_MB;	needs &= ~NEED_MB;
3596	break;	break;
3597
3598		case LT_MENUENTRY: {
3599		char *nkt = malloc(strlen(newKernelTitle)+3);
3600		strcpy(nkt, "'");
3601		strcat(nkt, newKernelTitle);
3602		strcat(nkt, "'");
3603		newLine = addLine(new, config->cfi, LT_MENUENTRY,
3604		config->primaryIndent, nkt);
3605		free(nkt);
3606		needs &= ~NEED_TITLE;
3607		needs |= NEED_END;
3608		break;
3609		}
3610	case LT_TITLE:	case LT_TITLE:
3611	if( useextlinuxmenu != 0 ){ // We just need useextlinuxmenu to not be zero (set above)	if( useextlinuxmenu != 0 ){ // We just need useextlinuxmenu to not be zero (set above)
3612	char * templabel;	char * templabel;
#	Line 2666  int addNewKernel(struct grubConfig * con	Line 3640  int addNewKernel(struct grubConfig * con
3640
3641	/* add the remainder of the lines, i.e. those that either	/* add the remainder of the lines, i.e. those that either
3642	* weren't present in the template, or in the case of no template,	* weren't present in the template, or in the case of no template,
3643	* all the lines following the entrySeparator.	* all the lines following the entryStart.
3644	*/	*/
3645	if (needs & NEED_TITLE) {	if (needs & NEED_TITLE) {
3646	newLine = addLine(new, config->cfi, LT_TITLE,	newLine = addLine(new, config->cfi, LT_TITLE,
#	Line 2707  int addNewKernel(struct grubConfig * con	Line 3681  int addNewKernel(struct grubConfig * con
3681	free(initrdVal);	free(initrdVal);
3682	needs &= ~NEED_INITRD;	needs &= ~NEED_INITRD;
3683	}	}
3684		if (needs & NEED_END) {
3685		newLine = addLine(new, config->cfi, LT_ENTRY_END,
3686		config->secondaryIndent, NULL);
3687		needs &= ~NEED_END;
3688		}
3689
3690	if (needs) {	if (needs) {
3691	printf(_("grubby: needs=%d, aborting\n"), needs);	printf(_("grubby: needs=%d, aborting\n"), needs);
#	Line 2736  static void traceback(int signum)	Line 3715  static void traceback(int signum)
3715
3716	int main(int argc, const char ** argv) {	int main(int argc, const char ** argv) {
3717	poptContext optCon;	poptContext optCon;
3718	char * grubConfig = NULL;	const char * grubConfig = NULL;
3719	char * outputFile = NULL;	char * outputFile = NULL;
3720	int arg = 0;	int arg = 0;
3721	int flags = 0;	int flags = 0;
3722	int badImageOkay = 0;	int badImageOkay = 0;
3723		int configureGrub2 = 0;
3724	int configureLilo = 0, configureELilo = 0, configureGrub = 0;	int configureLilo = 0, configureELilo = 0, configureGrub = 0;
3725	int configureYaboot = 0, configureSilo = 0, configureZipl = 0;	int configureYaboot = 0, configureSilo = 0, configureZipl = 0;
3726	int configureExtLinux = 0;	int configureExtLinux = 0;
#	Line 2768  int main(int argc, const char ** argv) {	Line 3748  int main(int argc, const char ** argv) {
3748	struct singleEntry * template = NULL;	struct singleEntry * template = NULL;
3749	int copyDefault = 0, makeDefault = 0;	int copyDefault = 0, makeDefault = 0;
3750	int displayDefault = 0;	int displayDefault = 0;
3751		int displayDefaultIndex = 0;
3752		int displayDefaultTitle = 0;
3753		int defaultIndex = -1;
3754	struct poptOption options[] = {	struct poptOption options[] = {
3755	{ "add-kernel", 0, POPT_ARG_STRING, &newKernelPath, 0,	{ "add-kernel", 0, POPT_ARG_STRING, &newKernelPath, 0,
3756	_("add an entry for the specified kernel"), _("kernel-path") },	_("add an entry for the specified kernel"), _("kernel-path") },
#	Line 2785  int main(int argc, const char ** argv) {	Line 3768  int main(int argc, const char ** argv) {
3768	{ "boot-filesystem", 0, POPT_ARG_STRING, &bootPrefix, 0,	{ "boot-filesystem", 0, POPT_ARG_STRING, &bootPrefix, 0,
3769	_("filestystem which contains /boot directory (for testing only)"),	_("filestystem which contains /boot directory (for testing only)"),
3770	_("bootfs") },	_("bootfs") },
3771	#if defined(__i386__) || defined(__x86_64__)	#if defined(__i386__) || defined(__x86_64__) || defined (__powerpc64__) || defined (__ia64__)
3772	{ "bootloader-probe", 0, POPT_ARG_NONE, &bootloaderProbe, 0,	{ "bootloader-probe", 0, POPT_ARG_NONE, &bootloaderProbe, 0,
3773	_("check if lilo is installed on lilo.conf boot sector") },	_("check which bootloader is installed on boot sector") },
3774	#endif	#endif
3775	{ "config-file", 'c', POPT_ARG_STRING, &grubConfig, 0,	{ "config-file", 'c', POPT_ARG_STRING, &grubConfig, 0,
3776	_("path to grub config file to update (\"-\" for stdin)"),	_("path to grub config file to update (\"-\" for stdin)"),
#	Line 2798  int main(int argc, const char ** argv) {	Line 3781  int main(int argc, const char ** argv) {
3781	"the kernel referenced by the default image does not exist, "	"the kernel referenced by the default image does not exist, "
3782	"the first linux entry whose kernel does exist is used as the "	"the first linux entry whose kernel does exist is used as the "
3783	"template"), NULL },	"template"), NULL },
3784		{ "debug", 0, 0, &debug, 0,
3785		_("print debugging information for failures") },
3786	{ "default-kernel", 0, 0, &displayDefault, 0,	{ "default-kernel", 0, 0, &displayDefault, 0,
3787	_("display the path of the default kernel") },	_("display the path of the default kernel") },
3788		{ "default-index", 0, 0, &displayDefaultIndex, 0,
3789		_("display the index of the default kernel") },
3790		{ "default-title", 0, 0, &displayDefaultTitle, 0,
3791		_("display the title of the default kernel") },
3792	{ "elilo", 0, POPT_ARG_NONE, &configureELilo, 0,	{ "elilo", 0, POPT_ARG_NONE, &configureELilo, 0,
3793	_("configure elilo bootloader") },	_("configure elilo bootloader") },
3794	{ "extlinux", 0, POPT_ARG_NONE, &configureExtLinux, 0,	{ "extlinux", 0, POPT_ARG_NONE, &configureExtLinux, 0,
3795	_("configure extlinux bootloader (from syslinux)") },	_("configure extlinux bootloader (from syslinux)") },
3796	{ "grub", 0, POPT_ARG_NONE, &configureGrub, 0,	{ "grub", 0, POPT_ARG_NONE, &configureGrub, 0,
3797	_("configure grub bootloader") },	_("configure grub bootloader") },
3798		{ "grub2", 0, POPT_ARG_NONE, &configureGrub2, 0,
3799		_("configure grub2 bootloader") },
3800	{ "info", 0, POPT_ARG_STRING, &kernelInfo, 0,	{ "info", 0, POPT_ARG_STRING, &kernelInfo, 0,
3801	_("display boot information for specified kernel"),	_("display boot information for specified kernel"),
3802	_("kernel-path") },	_("kernel-path") },
#	Line 2832  int main(int argc, const char ** argv) {	Line 3823  int main(int argc, const char ** argv) {
3823	{ "set-default", 0, POPT_ARG_STRING, &defaultKernel, 0,	{ "set-default", 0, POPT_ARG_STRING, &defaultKernel, 0,
3824	_("make the first entry referencing the specified kernel "	_("make the first entry referencing the specified kernel "
3825	"the default"), _("kernel-path") },	"the default"), _("kernel-path") },
3826		{ "set-default-index", 0, POPT_ARG_INT, &defaultIndex, 0,
3827		_("make the given entry index the default entry"),
3828		_("entry-index") },
3829	{ "silo", 0, POPT_ARG_NONE, &configureSilo, 0,	{ "silo", 0, POPT_ARG_NONE, &configureSilo, 0,
3830	_("configure silo bootloader") },	_("configure silo bootloader") },
3831	{ "title", 0, POPT_ARG_STRING, &newKernelTitle, 0,	{ "title", 0, POPT_ARG_STRING, &newKernelTitle, 0,
#	Line 2885  int main(int argc, const char ** argv) {	Line 3879  int main(int argc, const char ** argv) {
3879	return 1;	return 1;
3880	}	}
3881
3882	if ((configureLilo + configureGrub + configureELilo +	if ((configureLilo + configureGrub2 + configureGrub + configureELilo +
3883	configureYaboot + configureSilo + configureZipl +	configureYaboot + configureSilo + configureZipl +
3884	configureExtLinux ) > 1) {	configureExtLinux ) > 1) {
3885	fprintf(stderr, _("grubby: cannot specify multiple bootloaders\n"));	fprintf(stderr, _("grubby: cannot specify multiple bootloaders\n"));
#	Line 2894  int main(int argc, const char ** argv) {	Line 3888  int main(int argc, const char ** argv) {
3888	fprintf(stderr,	fprintf(stderr,
3889	_("grubby: cannot specify config file with --bootloader-probe\n"));	_("grubby: cannot specify config file with --bootloader-probe\n"));
3890	return 1;	return 1;
3891		} else if (configureGrub2) {
3892		cfi = &grub2ConfigType;
3893	} else if (configureLilo) {	} else if (configureLilo) {
3894	cfi = &liloConfigType;	cfi = &liloConfigType;
3895	} else if (configureGrub) {	} else if (configureGrub) {
#	Line 2912  int main(int argc, const char ** argv) {	Line 3908  int main(int argc, const char ** argv) {
3908	}	}
3909
3910	if (!cfi) {	if (!cfi) {
3911		if (grub2FindConfig(&grub2ConfigType))
3912		cfi = &grub2ConfigType;
3913		else
3914	#ifdef __ia64__	#ifdef __ia64__
3915	cfi = &eliloConfigType;	cfi = &eliloConfigType;
3916	#elif __powerpc__	#elif __powerpc__
3917	cfi = &yabootConfigType;	cfi = &yabootConfigType;
3918	#elif __sparc__	#elif __sparc__
3919	cfi = &siloConfigType;	cfi = &siloConfigType;
3920	#elif __s390__	#elif __s390__
3921	cfi = &ziplConfigType;	cfi = &ziplConfigType;
3922	#elif __s390x__	#elif __s390x__
3923	cfi = &ziplConfigtype;	cfi = &ziplConfigtype;
3924	#else	#else
3925	cfi = &grubConfigType;	cfi = &grubConfigType;
3926	#endif	#endif
3927	}	}
3928
3929	if (!grubConfig)	if (!grubConfig) {
3930	grubConfig = cfi->defaultConfig;	if (cfi->findConfig)
3931		grubConfig = cfi->findConfig(cfi);
3932		if (!grubConfig)
3933		grubConfig = cfi->defaultConfig;
3934		}
3935
3936	if (bootloaderProbe && (displayDefault || kernelInfo || newKernelVersion ||	if (bootloaderProbe && (displayDefault || kernelInfo || newKernelVersion ||
3937	newKernelPath || removeKernelPath || makeDefault ||	newKernelPath || removeKernelPath || makeDefault ||
3938	defaultKernel)) {	defaultKernel || displayDefaultIndex || displayDefaultTitle ||
3939		(defaultIndex >= 0))) {
3940	fprintf(stderr, _("grubby: --bootloader-probe may not be used with "	fprintf(stderr, _("grubby: --bootloader-probe may not be used with "
3941	"specified option"));	"specified option"));
3942	return 1;	return 1;
#	Line 2948  int main(int argc, const char ** argv) {	Line 3952  int main(int argc, const char ** argv) {
3952	if (newKernelPath && !newKernelTitle) {	if (newKernelPath && !newKernelTitle) {
3953	fprintf(stderr, _("grubby: kernel title must be specified\n"));	fprintf(stderr, _("grubby: kernel title must be specified\n"));
3954	return 1;	return 1;
3955	} else if (!newKernelPath && (newKernelTitle  || newKernelInitrd ||	} else if (!newKernelPath && (newKernelTitle  || copyDefault ||
3956	newKernelInitrd || copyDefault     ||	(newKernelInitrd && !updateKernelPath)||
3957	makeDefault || extraInitrdCount > 0)) {	makeDefault || extraInitrdCount > 0)) {
3958	fprintf(stderr, _("grubby: kernel path expected\n"));	fprintf(stderr, _("grubby: kernel path expected\n"));
3959	return 1;	return 1;
#	Line 2974  int main(int argc, const char ** argv) {	Line 3978  int main(int argc, const char ** argv) {
3978	makeDefault = 1;	makeDefault = 1;
3979	defaultKernel = NULL;	defaultKernel = NULL;
3980	}	}
3981		else if (defaultKernel && (defaultIndex >= 0)) {
3982		fprintf(stderr, _("grubby: --set-default and --set-default-index "
3983		"may not be used together\n"));
3984		return 1;
3985		}
3986
3987	if (!strcmp(grubConfig, "-") && !outputFile) {	if (grubConfig && !strcmp(grubConfig, "-") && !outputFile) {
3988	fprintf(stderr, _("grubby: output file must be specified if stdin "	fprintf(stderr, _("grubby: output file must be specified if stdin "
3989	"is used\n"));	"is used\n"));
3990	return 1;	return 1;
3991	}	}
3992
3993	if (!removeKernelPath && !newKernelPath && !displayDefault && !defaultKernel	if (!removeKernelPath && !newKernelPath && !displayDefault && !defaultKernel
3994	&& !kernelInfo && !bootloaderProbe && !updateKernelPath	&& !kernelInfo && !bootloaderProbe && !updateKernelPath
3995	&& !removeMBKernel) {	&& !removeMBKernel && !displayDefaultIndex && !displayDefaultTitle
3996		&& (defaultIndex == -1)) {
3997	fprintf(stderr, _("grubby: no action specified\n"));	fprintf(stderr, _("grubby: no action specified\n"));
3998	return 1;	return 1;
3999	}	}
#	Line 3010  int main(int argc, const char ** argv) {	Line 4020  int main(int argc, const char ** argv) {
4020	}	}
4021
4022	if (bootloaderProbe) {	if (bootloaderProbe) {
4023	int lrc = 0, grc = 0, erc = 0;	int lrc = 0, grc = 0, gr2c = 0, extrc = 0, yrc = 0, erc = 0;
4024	struct grubConfig * lconfig, * gconfig;	struct grubConfig * lconfig, * gconfig, * yconfig, * econfig;
4025
4026		const char *grub2config = grub2FindConfig(&grub2ConfigType);
4027		if (grub2config) {
4028		gconfig = readConfig(grub2config, &grub2ConfigType);
4029		if (!gconfig)
4030		gr2c = 1;
4031		else
4032		gr2c = checkForGrub2(gconfig);
4033		}
4034
4035	if (!access(grubConfigType.defaultConfig, F_OK)) {	const char *grubconfig = grubFindConfig(&grubConfigType);
4036	gconfig = readConfig(grubConfigType.defaultConfig, &grubConfigType);	if (!access(grubconfig, F_OK)) {
4037		gconfig = readConfig(grubconfig, &grubConfigType);
4038	if (!gconfig)	if (!gconfig)
4039	grc = 1;	grc = 1;
4040	else	else
#	Line 3029  int main(int argc, const char ** argv) {	Line 4049  int main(int argc, const char ** argv) {
4049	lrc = checkForLilo(lconfig);	lrc = checkForLilo(lconfig);
4050	}	}
4051
4052		if (!access(eliloConfigType.defaultConfig, F_OK)) {
4053		econfig = readConfig(eliloConfigType.defaultConfig,
4054		&eliloConfigType);
4055		if (!econfig)
4056		erc = 1;
4057		else
4058		erc = checkForElilo(econfig);
4059		}
4060
4061	if (!access(extlinuxConfigType.defaultConfig, F_OK)) {	if (!access(extlinuxConfigType.defaultConfig, F_OK)) {
4062	lconfig = readConfig(extlinuxConfigType.defaultConfig, &extlinuxConfigType);	lconfig = readConfig(extlinuxConfigType.defaultConfig, &extlinuxConfigType);
4063	if (!lconfig)	if (!lconfig)
4064	erc = 1;	extrc = 1;
4065	else	else
4066	erc = checkForExtLinux(lconfig);	extrc = checkForExtLinux(lconfig);
4067	}	}
4068
4069	if (lrc == 1 || grc == 1) return 1;
4070		if (!access(yabootConfigType.defaultConfig, F_OK)) {
4071		yconfig = readConfig(yabootConfigType.defaultConfig,
4072		&yabootConfigType);
4073		if (!yconfig)
4074		yrc = 1;
4075		else
4076		yrc = checkForYaboot(yconfig);
4077		}
4078
4079		if (lrc == 1 || grc == 1 || gr2c == 1 || extrc == 1 || yrc == 1 ||
4080		erc == 1)
4081		return 1;
4082
4083	if (lrc == 2) printf("lilo\n");	if (lrc == 2) printf("lilo\n");
4084		if (gr2c == 2) printf("grub2\n");
4085	if (grc == 2) printf("grub\n");	if (grc == 2) printf("grub\n");
4086	if (erc == 2) printf("extlinux\n");	if (extrc == 2) printf("extlinux\n");
4087		if (yrc == 2) printf("yaboot\n");
4088		if (erc == 2) printf("elilo\n");
4089
4090	return 0;	return 0;
4091	}	}
#	Line 3067  int main(int argc, const char ** argv) {	Line 4111  int main(int argc, const char ** argv) {
4111	((rootspec != NULL) ? strlen(rootspec) : 0));	((rootspec != NULL) ? strlen(rootspec) : 0));
4112
4113	return 0;	return 0;
4114
4115		} else if (displayDefaultTitle) {
4116		struct singleLine * line;
4117		struct singleEntry * entry;
4118
4119		if (config->defaultImage == -1) return 0;
4120		entry = findEntryByIndex(config, config->defaultImage);
4121		if (!entry) return 0;
4122
4123		if (!configureGrub2) {
4124		line = getLineByType(LT_TITLE, entry->lines);
4125		if (!line) return 0;
4126		printf("%s\n", line->elements[1].item);
4127
4128		} else {
4129		char * title;
4130
4131		dbgPrintf("This is GRUB2, default title is embeded in menuentry\n");
4132		line = getLineByType(LT_MENUENTRY, entry->lines);
4133		if (!line) return 0;
4134		title = grub2ExtractTitle(line);
4135		if (title)
4136		printf("%s\n", title);
4137		}
4138		return 0;
4139
4140		} else if (displayDefaultIndex) {
4141		if (config->defaultImage == -1) return 0;
4142		printf("%i\n", config->defaultImage);
4143
4144	} else if (kernelInfo)	} else if (kernelInfo)
4145	return displayInfo(config, kernelInfo, bootPrefix);	return displayInfo(config, kernelInfo, bootPrefix);
4146
#	Line 3078  int main(int argc, const char ** argv) {	Line 4152  int main(int argc, const char ** argv) {
4152	markRemovedImage(config, removeKernelPath, bootPrefix);	markRemovedImage(config, removeKernelPath, bootPrefix);
4153	markRemovedImage(config, removeMBKernel, bootPrefix);	markRemovedImage(config, removeMBKernel, bootPrefix);
4154	setDefaultImage(config, newKernelPath != NULL, defaultKernel, makeDefault,	setDefaultImage(config, newKernelPath != NULL, defaultKernel, makeDefault,
4155	bootPrefix, flags);	bootPrefix, flags, defaultIndex);
4156	setFallbackImage(config, newKernelPath != NULL);	setFallbackImage(config, newKernelPath != NULL);
4157	if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs,	if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs,
4158	removeArgs, newMBKernelArgs, removeMBKernelArgs)) return 1;	removeArgs, newMBKernelArgs, removeMBKernelArgs)) return 1;
4159		if (updateKernelPath && newKernelInitrd) {
4160		if (updateInitrd(config, updateKernelPath, bootPrefix,
4161		newKernelInitrd)) return 1;
4162		}
4163	if (addNewKernel(config, template, bootPrefix, newKernelPath,	if (addNewKernel(config, template, bootPrefix, newKernelPath,
4164	newKernelTitle, newKernelArgs, newKernelInitrd,	newKernelTitle, newKernelArgs, newKernelInitrd,
4165	extraInitrds, extraInitrdCount,	(const char **)extraInitrds, extraInitrdCount,
4166	newMBKernel, newMBKernelArgs)) return 1;	newMBKernel, newMBKernelArgs)) return 1;
4167
4168
#	Line 3095  int main(int argc, const char ** argv) {	Line 4173  int main(int argc, const char ** argv) {
4173	}	}
4174
4175	if (!outputFile)	if (!outputFile)
4176	outputFile = grubConfig;	outputFile = (char *)grubConfig;
4177
4178	return writeConfig(config, outputFile, bootPrefix);	return writeConfig(config, outputFile, bootPrefix);
4179	}	}

---

Colored Diff Long Colored Diff Full Colored Diff Unidiff Context Diff Side by Side

Legend: Removed from v.926   changed lines   Added in v.1931