136 |
typedef const char *(*findConfigFunc)(struct configFileInfo *); |
typedef const char *(*findConfigFunc)(struct configFileInfo *); |
137 |
typedef const int (*writeLineFunc)(struct configFileInfo *, |
typedef const int (*writeLineFunc)(struct configFileInfo *, |
138 |
struct singleLine *line); |
struct singleLine *line); |
139 |
|
typedef char *(*getEnvFunc)(struct configFileInfo *, char *name); |
140 |
|
typedef int (*setEnvFunc)(struct configFileInfo *, char *name, char *value); |
141 |
|
|
142 |
struct configFileInfo { |
struct configFileInfo { |
143 |
char * defaultConfig; |
char * defaultConfig; |
144 |
findConfigFunc findConfig; |
findConfigFunc findConfig; |
145 |
writeLineFunc writeLine; |
writeLineFunc writeLine; |
146 |
|
getEnvFunc getEnv; |
147 |
|
setEnvFunc setEnv; |
148 |
struct keywordTypes * keywords; |
struct keywordTypes * keywords; |
149 |
int caseInsensitive; |
int caseInsensitive; |
150 |
int defaultIsIndex; |
int defaultIsIndex; |
151 |
int defaultIsVariable; |
int defaultIsVariable; |
152 |
int defaultSupportSaved; |
int defaultSupportSaved; |
153 |
|
int defaultIsSaved; |
154 |
enum lineType_e entryStart; |
enum lineType_e entryStart; |
155 |
enum lineType_e entryEnd; |
enum lineType_e entryEnd; |
156 |
int needsBootPrefix; |
int needsBootPrefix; |
162 |
int mbInitRdIsModule; |
int mbInitRdIsModule; |
163 |
int mbConcatArgs; |
int mbConcatArgs; |
164 |
int mbAllowExtraInitRds; |
int mbAllowExtraInitRds; |
165 |
|
char *envFile; |
166 |
}; |
}; |
167 |
|
|
168 |
struct keywordTypes grubKeywords[] = { |
struct keywordTypes grubKeywords[] = { |
267 |
return configFiles[i]; |
return configFiles[i]; |
268 |
} |
} |
269 |
|
|
270 |
|
/* kind of hacky. It'll give the first 1024 bytes, ish. */ |
271 |
|
static char *grub2GetEnv(struct configFileInfo *info, char *name) |
272 |
|
{ |
273 |
|
static char buf[1025]; |
274 |
|
char *s = NULL; |
275 |
|
char *ret = NULL; |
276 |
|
char *envFile = info->envFile ? info->envFile : "/boot/grub2/grubenv"; |
277 |
|
int rc = asprintf(&s, "grub2-editenv %s list | grep '^%s='", envFile, name); |
278 |
|
|
279 |
|
if (rc < 0) |
280 |
|
return NULL; |
281 |
|
|
282 |
|
FILE *f = popen(s, "r"); |
283 |
|
if (!f) |
284 |
|
goto out; |
285 |
|
|
286 |
|
memset(buf, '\0', sizeof (buf)); |
287 |
|
ret = fgets(buf, 1024, f); |
288 |
|
pclose(f); |
289 |
|
|
290 |
|
if (ret) { |
291 |
|
ret += strlen(name) + 1; |
292 |
|
ret[strlen(ret) - 1] = '\0'; |
293 |
|
} |
294 |
|
dbgPrintf("grub2GetEnv(%s): %s\n", name, ret); |
295 |
|
out: |
296 |
|
free(s); |
297 |
|
return ret; |
298 |
|
} |
299 |
|
|
300 |
|
static int sPopCount(const char *s, const char *c) |
301 |
|
{ |
302 |
|
int ret = 0; |
303 |
|
if (!s) |
304 |
|
return -1; |
305 |
|
for (int i = 0; s[i] != '\0'; i++) |
306 |
|
for (int j = 0; c[j] != '\0'; j++) |
307 |
|
if (s[i] == c[j]) |
308 |
|
ret++; |
309 |
|
return ret; |
310 |
|
} |
311 |
|
|
312 |
|
static char *shellEscape(const char *s) |
313 |
|
{ |
314 |
|
int l = strlen(s) + sPopCount(s, "'") * 2; |
315 |
|
|
316 |
|
char *ret = calloc(l+1, sizeof (*ret)); |
317 |
|
if (!ret) |
318 |
|
return NULL; |
319 |
|
for (int i = 0, j = 0; s[i] != '\0'; i++, j++) { |
320 |
|
if (s[i] == '\'') |
321 |
|
ret[j++] = '\\'; |
322 |
|
ret[j] = s[i]; |
323 |
|
} |
324 |
|
return ret; |
325 |
|
} |
326 |
|
|
327 |
|
static void unquote(char *s) |
328 |
|
{ |
329 |
|
int l = strlen(s); |
330 |
|
|
331 |
|
if ((s[l-1] == '\'' && s[0] == '\'') || (s[l-1] == '"' && s[0] == '"')) { |
332 |
|
memmove(s, s+1, l-2); |
333 |
|
s[l-2] = '\0'; |
334 |
|
} |
335 |
|
} |
336 |
|
|
337 |
|
static int grub2SetEnv(struct configFileInfo *info, char *name, char *value) |
338 |
|
{ |
339 |
|
char *s = NULL; |
340 |
|
int rc = 0; |
341 |
|
char *envFile = info->envFile ? info->envFile : "/boot/grub2/grubenv"; |
342 |
|
|
343 |
|
unquote(value); |
344 |
|
value = shellEscape(value); |
345 |
|
if (!value) |
346 |
|
return -1; |
347 |
|
|
348 |
|
rc = asprintf(&s, "grub2-editenv %s set '%s=%s'", envFile, name, value); |
349 |
|
free(value); |
350 |
|
if (rc <0) |
351 |
|
return -1; |
352 |
|
|
353 |
|
dbgPrintf("grub2SetEnv(%s): %s\n", name, s); |
354 |
|
rc = system(s); |
355 |
|
free(s); |
356 |
|
return rc; |
357 |
|
} |
358 |
|
|
359 |
|
/* this is a gigantic hack to avoid clobbering grub2 variables... */ |
360 |
|
static int is_special_grub2_variable(const char *name) |
361 |
|
{ |
362 |
|
if (!strcmp(name,"\"${next_entry}\"")) |
363 |
|
return 1; |
364 |
|
if (!strcmp(name,"\"${prev_saved_entry}\"")) |
365 |
|
return 1; |
366 |
|
return 0; |
367 |
|
} |
368 |
|
|
369 |
int sizeOfSingleLine(struct singleLine * line) { |
int sizeOfSingleLine(struct singleLine * line) { |
370 |
int count = 0; |
int count = 0; |
371 |
|
|
460 |
|
|
461 |
struct configFileInfo grub2ConfigType = { |
struct configFileInfo grub2ConfigType = { |
462 |
.findConfig = grub2FindConfig, |
.findConfig = grub2FindConfig, |
463 |
|
.getEnv = grub2GetEnv, |
464 |
|
.setEnv = grub2SetEnv, |
465 |
.keywords = grub2Keywords, |
.keywords = grub2Keywords, |
466 |
.defaultIsIndex = 1, |
.defaultIsIndex = 1, |
467 |
.defaultSupportSaved = 1, |
.defaultSupportSaved = 1, |
641 |
struct singleEntry * findEntryByPath(struct grubConfig * cfg, |
struct singleEntry * findEntryByPath(struct grubConfig * cfg, |
642 |
const char * path, const char * prefix, |
const char * path, const char * prefix, |
643 |
int * index); |
int * index); |
644 |
|
struct singleEntry * findEntryByTitle(struct grubConfig * cfg, char *title, |
645 |
|
int * index); |
646 |
static int readFile(int fd, char ** bufPtr); |
static int readFile(int fd, char ** bufPtr); |
647 |
static void lineInit(struct singleLine * line); |
static void lineInit(struct singleLine * line); |
648 |
struct singleLine * lineDup(struct singleLine * line); |
struct singleLine * lineDup(struct singleLine * line); |
800 |
/* extract the title from within brackets (for zipl) */ |
/* extract the title from within brackets (for zipl) */ |
801 |
static char * extractTitle(struct singleLine * line) { |
static char * extractTitle(struct singleLine * line) { |
802 |
/* bracketed title... let's extract it (leaks a byte) */ |
/* bracketed title... let's extract it (leaks a byte) */ |
803 |
char * title; |
char * title = NULL; |
804 |
title = strdup(line->elements[0].item); |
if (line->type == LT_TITLE) { |
805 |
title++; |
title = strdup(line->elements[0].item); |
806 |
*(title + strlen(title) - 1) = '\0'; |
title++; |
807 |
|
*(title + strlen(title) - 1) = '\0'; |
808 |
|
} else if (line->type == LT_MENUENTRY) |
809 |
|
title = strdup(line->elements[1].item); |
810 |
|
else |
811 |
|
return NULL; |
812 |
return title; |
return title; |
813 |
} |
} |
814 |
|
|
1150 |
dbgPrintf("\n"); |
dbgPrintf("\n"); |
1151 |
struct keywordTypes *kwType = getKeywordByType(LT_DEFAULT, cfi); |
struct keywordTypes *kwType = getKeywordByType(LT_DEFAULT, cfi); |
1152 |
if (kwType && line->numElements == 3 && |
if (kwType && line->numElements == 3 && |
1153 |
!strcmp(line->elements[1].item, kwType->key)) { |
!strcmp(line->elements[1].item, kwType->key) && |
1154 |
|
!is_special_grub2_variable(line->elements[2].item)) { |
1155 |
dbgPrintf("Line sets default config\n"); |
dbgPrintf("Line sets default config\n"); |
1156 |
cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT; |
cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT; |
1157 |
defaultLine = line; |
defaultLine = line; |
1356 |
if (defaultLine->numElements > 2 && |
if (defaultLine->numElements > 2 && |
1357 |
cfi->defaultSupportSaved && |
cfi->defaultSupportSaved && |
1358 |
!strncmp(defaultLine->elements[2].item,"\"${saved_entry}\"", 16)) { |
!strncmp(defaultLine->elements[2].item,"\"${saved_entry}\"", 16)) { |
1359 |
cfg->defaultImage = DEFAULT_SAVED_GRUB2; |
cfg->cfi->defaultIsSaved = 1; |
1360 |
|
cfg->defaultImage = DEFAULT_SAVED_GRUB2; |
1361 |
|
if (cfg->cfi->getEnv) { |
1362 |
|
char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry"); |
1363 |
|
if (defTitle) { |
1364 |
|
int index = 0; |
1365 |
|
entry = findEntryByTitle(cfg, defTitle, &index); |
1366 |
|
if (entry) |
1367 |
|
cfg->defaultImage = index; |
1368 |
|
} |
1369 |
|
} |
1370 |
} else if (cfi->defaultIsVariable) { |
} else if (cfi->defaultIsVariable) { |
1371 |
char *value = defaultLine->elements[2].item; |
char *value = defaultLine->elements[2].item; |
1372 |
while (*value && (*value == '"' || *value == '\'' || |
while (*value && (*value == '"' || *value == '\'' || |
1407 |
cfg->defaultImage = -1; |
cfg->defaultImage = -1; |
1408 |
} |
} |
1409 |
} |
} |
1410 |
|
} else if (cfg->cfi->defaultIsSaved && cfg->cfi->getEnv) { |
1411 |
|
char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry"); |
1412 |
|
if (defTitle) { |
1413 |
|
int index = 0; |
1414 |
|
entry = findEntryByTitle(cfg, defTitle, &index); |
1415 |
|
if (entry) |
1416 |
|
cfg->defaultImage = index; |
1417 |
|
} |
1418 |
} else { |
} else { |
1419 |
cfg->defaultImage = 0; |
cfg->defaultImage = 0; |
1420 |
} |
} |
1432 |
|
|
1433 |
if (cfg->defaultImage == DEFAULT_SAVED) |
if (cfg->defaultImage == DEFAULT_SAVED) |
1434 |
fprintf(out, "%sdefault%ssaved\n", indent, separator); |
fprintf(out, "%sdefault%ssaved\n", indent, separator); |
1435 |
else if (cfg->defaultImage == DEFAULT_SAVED_GRUB2) |
else if (cfg->cfi->defaultIsSaved) { |
1436 |
fprintf(out, "%sset default=\"${saved_entry}\"\n", indent); |
fprintf(out, "%sset default=\"${saved_entry}\"\n", indent); |
1437 |
else if (cfg->defaultImage > -1) { |
if (cfg->defaultImage >= 0 && cfg->cfi->setEnv) { |
1438 |
|
char *title; |
1439 |
|
entry = findEntryByIndex(cfg, cfg->defaultImage); |
1440 |
|
line = getLineByType(LT_MENUENTRY, entry->lines); |
1441 |
|
if (!line) |
1442 |
|
line = getLineByType(LT_TITLE, entry->lines); |
1443 |
|
if (line) { |
1444 |
|
title = extractTitle(line); |
1445 |
|
if (title) |
1446 |
|
cfg->cfi->setEnv(cfg->cfi, "saved_entry", title); |
1447 |
|
} |
1448 |
|
} |
1449 |
|
} else if (cfg->defaultImage > -1) { |
1450 |
if (cfg->cfi->defaultIsIndex) { |
if (cfg->cfi->defaultIsIndex) { |
1451 |
if (cfg->cfi->defaultIsVariable) { |
if (cfg->cfi->defaultIsVariable) { |
1452 |
fprintf(out, "%sset default=\"%d\"\n", indent, |
fprintf(out, "%sset default=\"%d\"\n", indent, |
1549 |
while (line) { |
while (line) { |
1550 |
if (line->type == LT_SET_VARIABLE && defaultKw && |
if (line->type == LT_SET_VARIABLE && defaultKw && |
1551 |
line->numElements == 3 && |
line->numElements == 3 && |
1552 |
!strcmp(line->elements[1].item, defaultKw->key)) { |
!strcmp(line->elements[1].item, defaultKw->key) && |
1553 |
|
!is_special_grub2_variable(line->elements[2].item)) { |
1554 |
writeDefault(out, line->indent, line->elements[0].indent, cfg); |
writeDefault(out, line->indent, line->elements[0].indent, cfg); |
1555 |
needs &= ~MAIN_DEFAULT; |
needs &= ~MAIN_DEFAULT; |
1556 |
} else if (line->type == LT_DEFAULT) { |
} else if (line->type == LT_DEFAULT) { |
2005 |
return entry; |
return entry; |
2006 |
} |
} |
2007 |
|
|
2008 |
|
struct singleEntry * findEntryByTitle(struct grubConfig * cfg, char *title, |
2009 |
|
int * index) { |
2010 |
|
struct singleEntry * entry; |
2011 |
|
struct singleLine * line; |
2012 |
|
int i; |
2013 |
|
char * newtitle; |
2014 |
|
|
2015 |
|
for (i = 0, entry = cfg->entries; entry; entry = entry->next, i++) { |
2016 |
|
if (index && i < *index) |
2017 |
|
continue; |
2018 |
|
line = getLineByType(LT_TITLE, entry->lines); |
2019 |
|
if (!line) |
2020 |
|
line = getLineByType(LT_MENUENTRY, entry->lines); |
2021 |
|
if (!line) |
2022 |
|
continue; |
2023 |
|
newtitle = grub2ExtractTitle(line); |
2024 |
|
if (!newtitle) |
2025 |
|
continue; |
2026 |
|
if (!strcmp(title, newtitle)) |
2027 |
|
break; |
2028 |
|
} |
2029 |
|
|
2030 |
|
if (!entry) |
2031 |
|
return NULL; |
2032 |
|
|
2033 |
|
if (index) |
2034 |
|
*index = i; |
2035 |
|
return entry; |
2036 |
|
} |
2037 |
|
|
2038 |
struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index) { |
struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index) { |
2039 |
struct singleEntry * entry; |
struct singleEntry * entry; |
2040 |
|
|
2057 |
struct singleEntry * entry, * entry2; |
struct singleEntry * entry, * entry2; |
2058 |
int index; |
int index; |
2059 |
|
|
2060 |
if (cfg->defaultImage > -1) { |
if (cfg->cfi->defaultIsSaved) { |
2061 |
|
if (cfg->cfi->getEnv) { |
2062 |
|
char *defTitle = cfg->cfi->getEnv(cfg->cfi, "saved_entry"); |
2063 |
|
if (defTitle) { |
2064 |
|
int index = 0; |
2065 |
|
entry = findEntryByTitle(cfg, defTitle, &index); |
2066 |
|
} |
2067 |
|
} |
2068 |
|
} else if (cfg->defaultImage > -1) { |
2069 |
entry = findEntryByIndex(cfg, cfg->defaultImage); |
entry = findEntryByIndex(cfg, cfg->defaultImage); |
2070 |
if (entry && suitableImage(entry, prefix, skipRemoved, flags)) { |
if (entry && suitableImage(entry, prefix, skipRemoved, flags)) { |
2071 |
if (indexPtr) *indexPtr = cfg->defaultImage; |
if (indexPtr) *indexPtr = cfg->defaultImage; |
4019 |
char * removeArgs = NULL; |
char * removeArgs = NULL; |
4020 |
char * kernelInfo = NULL; |
char * kernelInfo = NULL; |
4021 |
char * extraInitrds[MAX_EXTRA_INITRDS] = { NULL }; |
char * extraInitrds[MAX_EXTRA_INITRDS] = { NULL }; |
4022 |
|
char * envPath = NULL; |
4023 |
const char * chptr = NULL; |
const char * chptr = NULL; |
4024 |
struct configFileInfo * cfi = NULL; |
struct configFileInfo * cfi = NULL; |
4025 |
struct grubConfig * config; |
struct grubConfig * config; |
4071 |
_("configure elilo bootloader") }, |
_("configure elilo bootloader") }, |
4072 |
{ "efi", 0, POPT_ARG_NONE, &isEfi, 0, |
{ "efi", 0, POPT_ARG_NONE, &isEfi, 0, |
4073 |
_("force grub2 stanzas to use efi") }, |
_("force grub2 stanzas to use efi") }, |
4074 |
|
{ "env", 0, POPT_ARG_STRING, &envPath, 0, |
4075 |
|
_("path for environment data"), |
4076 |
|
_("path") }, |
4077 |
{ "extlinux", 0, POPT_ARG_NONE, &configureExtLinux, 0, |
{ "extlinux", 0, POPT_ARG_NONE, &configureExtLinux, 0, |
4078 |
_("configure extlinux bootloader (from syslinux)") }, |
_("configure extlinux bootloader (from syslinux)") }, |
4079 |
{ "grub", 0, POPT_ARG_NONE, &configureGrub, 0, |
{ "grub", 0, POPT_ARG_NONE, &configureGrub, 0, |
4187 |
return 1; |
return 1; |
4188 |
} else if (configureGrub2) { |
} else if (configureGrub2) { |
4189 |
cfi = &grub2ConfigType; |
cfi = &grub2ConfigType; |
4190 |
|
if (envPath) |
4191 |
|
cfi->envFile = envPath; |
4192 |
} else if (configureLilo) { |
} else if (configureLilo) { |
4193 |
cfi = &liloConfigType; |
cfi = &liloConfigType; |
4194 |
} else if (configureGrub) { |
} else if (configureGrub) { |
4403 |
char * rootspec; |
char * rootspec; |
4404 |
|
|
4405 |
if (config->defaultImage == -1) return 0; |
if (config->defaultImage == -1) return 0; |
4406 |
|
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
4407 |
|
cfi->defaultIsSaved) |
4408 |
|
config->defaultImage = 0; |
4409 |
entry = findEntryByIndex(config, config->defaultImage); |
entry = findEntryByIndex(config, config->defaultImage); |
4410 |
if (!entry) return 0; |
if (!entry) return 0; |
4411 |
if (!suitableImage(entry, bootPrefix, 0, flags)) return 0; |
if (!suitableImage(entry, bootPrefix, 0, flags)) return 0; |
4424 |
struct singleEntry * entry; |
struct singleEntry * entry; |
4425 |
|
|
4426 |
if (config->defaultImage == -1) return 0; |
if (config->defaultImage == -1) return 0; |
4427 |
|
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
4428 |
|
cfi->defaultIsSaved) |
4429 |
|
config->defaultImage = 0; |
4430 |
entry = findEntryByIndex(config, config->defaultImage); |
entry = findEntryByIndex(config, config->defaultImage); |
4431 |
if (!entry) return 0; |
if (!entry) return 0; |
4432 |
|
|
4449 |
|
|
4450 |
} else if (displayDefaultIndex) { |
} else if (displayDefaultIndex) { |
4451 |
if (config->defaultImage == -1) return 0; |
if (config->defaultImage == -1) return 0; |
4452 |
|
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
4453 |
|
cfi->defaultIsSaved) |
4454 |
|
config->defaultImage = 0; |
4455 |
printf("%i\n", config->defaultImage); |
printf("%i\n", config->defaultImage); |
4456 |
return 0; |
return 0; |
4457 |
|
|