182 |
"/boot/grub/grub.conf", |
"/boot/grub/grub.conf", |
183 |
"/boot/grub/menu.lst", |
"/boot/grub/menu.lst", |
184 |
"/etc/grub.conf", |
"/etc/grub.conf", |
185 |
|
"/boot/grub2/grub.cfg", |
186 |
|
"/boot/grub2-efi/grub.cfg", |
187 |
NULL |
NULL |
188 |
}; |
}; |
189 |
static int i = -1; |
static int i = -1; |
299 |
return ret; |
return ret; |
300 |
} |
} |
301 |
|
|
302 |
|
static int sPopCount(const char *s, const char *c) |
303 |
|
{ |
304 |
|
int ret = 0; |
305 |
|
if (!s) |
306 |
|
return -1; |
307 |
|
for (int i = 0; s[i] != '\0'; i++) |
308 |
|
for (int j = 0; c[j] != '\0'; j++) |
309 |
|
if (s[i] == c[j]) |
310 |
|
ret++; |
311 |
|
return ret; |
312 |
|
} |
313 |
|
|
314 |
|
static char *shellEscape(const char *s) |
315 |
|
{ |
316 |
|
int l = strlen(s) + sPopCount(s, "'") * 2; |
317 |
|
|
318 |
|
char *ret = calloc(l+1, sizeof (*ret)); |
319 |
|
if (!ret) |
320 |
|
return NULL; |
321 |
|
for (int i = 0, j = 0; s[i] != '\0'; i++, j++) { |
322 |
|
if (s[i] == '\'') |
323 |
|
ret[j++] = '\\'; |
324 |
|
ret[j] = s[i]; |
325 |
|
} |
326 |
|
return ret; |
327 |
|
} |
328 |
|
|
329 |
|
static void unquote(char *s) |
330 |
|
{ |
331 |
|
int l = strlen(s); |
332 |
|
|
333 |
|
if ((s[l-1] == '\'' && s[0] == '\'') || (s[l-1] == '"' && s[0] == '"')) { |
334 |
|
memmove(s, s+1, l-2); |
335 |
|
s[l-2] = '\0'; |
336 |
|
} |
337 |
|
} |
338 |
|
|
339 |
static int grub2SetEnv(struct configFileInfo *info, char *name, char *value) |
static int grub2SetEnv(struct configFileInfo *info, char *name, char *value) |
340 |
{ |
{ |
341 |
char *s = NULL; |
char *s = NULL; |
342 |
int rc = 0; |
int rc = 0; |
343 |
char *envFile = info->envFile ? info->envFile : "/boot/grub2/grubenv"; |
char *envFile = info->envFile ? info->envFile : "/boot/grub2/grubenv"; |
344 |
|
|
345 |
|
unquote(value); |
346 |
|
value = shellEscape(value); |
347 |
|
if (!value) |
348 |
|
return -1; |
349 |
|
|
350 |
rc = asprintf(&s, "grub2-editenv %s set '%s=%s'", envFile, name, value); |
rc = asprintf(&s, "grub2-editenv %s set '%s=%s'", envFile, name, value); |
351 |
|
free(value); |
352 |
if (rc <0) |
if (rc <0) |
353 |
return -1; |
return -1; |
354 |
|
|
1068 |
return 0; |
return 0; |
1069 |
} |
} |
1070 |
|
|
1071 |
|
static int isnumber(const char *s) |
1072 |
|
{ |
1073 |
|
int i; |
1074 |
|
for (i = 0; s[i] != '\0'; i++) |
1075 |
|
if (s[i] < '0' || s[i] > '9') |
1076 |
|
return 0; |
1077 |
|
return i; |
1078 |
|
} |
1079 |
|
|
1080 |
static struct grubConfig * readConfig(const char * inName, |
static struct grubConfig * readConfig(const char * inName, |
1081 |
struct configFileInfo * cfi) { |
struct configFileInfo * cfi) { |
1082 |
int in; |
int in; |
1373 |
char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry"); |
char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry"); |
1374 |
if (defTitle) { |
if (defTitle) { |
1375 |
int index = 0; |
int index = 0; |
1376 |
entry = findEntryByTitle(cfg, defTitle, &index); |
if (isnumber(defTitle)) { |
1377 |
|
index = atoi(defTitle); |
1378 |
|
entry = findEntryByIndex(cfg, index); |
1379 |
|
} else { |
1380 |
|
entry = findEntryByTitle(cfg, defTitle, &index); |
1381 |
|
} |
1382 |
if (entry) |
if (entry) |
1383 |
cfg->defaultImage = index; |
cfg->defaultImage = index; |
1384 |
} |
} |
1427 |
char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry"); |
char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry"); |
1428 |
if (defTitle) { |
if (defTitle) { |
1429 |
int index = 0; |
int index = 0; |
1430 |
entry = findEntryByTitle(cfg, defTitle, &index); |
if (isnumber(defTitle)) { |
1431 |
|
index = atoi(defTitle); |
1432 |
|
entry = findEntryByIndex(cfg, index); |
1433 |
|
} else { |
1434 |
|
entry = findEntryByTitle(cfg, defTitle, &index); |
1435 |
|
} |
1436 |
if (entry) |
if (entry) |
1437 |
cfg->defaultImage = index; |
cfg->defaultImage = index; |
1438 |
} |
} |
2083 |
char *defTitle = cfg->cfi->getEnv(cfg->cfi, "saved_entry"); |
char *defTitle = cfg->cfi->getEnv(cfg->cfi, "saved_entry"); |
2084 |
if (defTitle) { |
if (defTitle) { |
2085 |
int index = 0; |
int index = 0; |
2086 |
entry = findEntryByTitle(cfg, defTitle, &index); |
if (isnumber(defTitle)) { |
2087 |
|
index = atoi(defTitle); |
2088 |
|
entry = findEntryByIndex(cfg, index); |
2089 |
|
} else { |
2090 |
|
entry = findEntryByTitle(cfg, defTitle, &index); |
2091 |
|
} |
2092 |
|
if (entry) |
2093 |
|
cfg->defaultImage = index; |
2094 |
} |
} |
2095 |
} |
} |
2096 |
} else if (cfg->defaultImage > -1) { |
} else if (cfg->defaultImage > -1) { |
3258 |
return rc; |
return rc; |
3259 |
} |
} |
3260 |
|
|
3261 |
|
int addMBInitrd(struct grubConfig * cfg, const char *newMBKernel, |
3262 |
|
const char * image, const char * prefix, const char * initrd) { |
3263 |
|
struct singleEntry * entry; |
3264 |
|
struct singleLine * line, * kernelLine, *endLine = NULL; |
3265 |
|
int index = 0; |
3266 |
|
|
3267 |
|
if (!image) return 0; |
3268 |
|
|
3269 |
|
for (; (entry = findEntryByPath(cfg, newMBKernel, prefix, &index)); index++) { |
3270 |
|
kernelLine = getLineByType(LT_MBMODULE, entry->lines); |
3271 |
|
if (!kernelLine) continue; |
3272 |
|
|
3273 |
|
if (prefix) { |
3274 |
|
int prefixLen = strlen(prefix); |
3275 |
|
if (!strncmp(initrd, prefix, prefixLen)) |
3276 |
|
initrd += prefixLen; |
3277 |
|
} |
3278 |
|
endLine = getLineByType(LT_ENTRY_END, entry->lines); |
3279 |
|
if (endLine) |
3280 |
|
removeLine(entry, endLine); |
3281 |
|
line = addLine(entry, cfg->cfi, preferredLineType(LT_MBMODULE,cfg->cfi), |
3282 |
|
kernelLine->indent, initrd); |
3283 |
|
if (!line) |
3284 |
|
return 1; |
3285 |
|
if (endLine) { |
3286 |
|
line = addLine(entry, cfg->cfi, LT_ENTRY_END, "", NULL); |
3287 |
|
if (!line) |
3288 |
|
return 1; |
3289 |
|
} |
3290 |
|
|
3291 |
|
break; |
3292 |
|
} |
3293 |
|
|
3294 |
|
return 0; |
3295 |
|
} |
3296 |
|
|
3297 |
int updateInitrd(struct grubConfig * cfg, const char * image, |
int updateInitrd(struct grubConfig * cfg, const char * image, |
3298 |
const char * prefix, const char * initrd) { |
const char * prefix, const char * initrd) { |
3299 |
struct singleEntry * entry; |
struct singleEntry * entry; |
4467 |
char * rootspec; |
char * rootspec; |
4468 |
|
|
4469 |
if (config->defaultImage == -1) return 0; |
if (config->defaultImage == -1) return 0; |
4470 |
|
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
4471 |
|
cfi->defaultIsSaved) |
4472 |
|
config->defaultImage = 0; |
4473 |
entry = findEntryByIndex(config, config->defaultImage); |
entry = findEntryByIndex(config, config->defaultImage); |
4474 |
if (!entry) return 0; |
if (!entry) return 0; |
4475 |
if (!suitableImage(entry, bootPrefix, 0, flags)) return 0; |
if (!suitableImage(entry, bootPrefix, 0, flags)) return 0; |
4488 |
struct singleEntry * entry; |
struct singleEntry * entry; |
4489 |
|
|
4490 |
if (config->defaultImage == -1) return 0; |
if (config->defaultImage == -1) return 0; |
4491 |
|
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
4492 |
|
cfi->defaultIsSaved) |
4493 |
|
config->defaultImage = 0; |
4494 |
entry = findEntryByIndex(config, config->defaultImage); |
entry = findEntryByIndex(config, config->defaultImage); |
4495 |
if (!entry) return 0; |
if (!entry) return 0; |
4496 |
|
|
4513 |
|
|
4514 |
} else if (displayDefaultIndex) { |
} else if (displayDefaultIndex) { |
4515 |
if (config->defaultImage == -1) return 0; |
if (config->defaultImage == -1) return 0; |
4516 |
|
if (config->defaultImage == DEFAULT_SAVED_GRUB2 && |
4517 |
|
cfi->defaultIsSaved) |
4518 |
|
config->defaultImage = 0; |
4519 |
printf("%i\n", config->defaultImage); |
printf("%i\n", config->defaultImage); |
4520 |
return 0; |
return 0; |
4521 |
|
|
4535 |
if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs, |
if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs, |
4536 |
removeArgs, newMBKernelArgs, removeMBKernelArgs)) return 1; |
removeArgs, newMBKernelArgs, removeMBKernelArgs)) return 1; |
4537 |
if (updateKernelPath && newKernelInitrd) { |
if (updateKernelPath && newKernelInitrd) { |
4538 |
if (updateInitrd(config, updateKernelPath, bootPrefix, |
if (newMBKernel) { |
4539 |
newKernelInitrd)) return 1; |
if (addMBInitrd(config, newMBKernel, updateKernelPath, |
4540 |
|
bootPrefix, newKernelInitrd)) |
4541 |
|
return 1; |
4542 |
|
} else { |
4543 |
|
if (updateInitrd(config, updateKernelPath, bootPrefix, |
4544 |
|
newKernelInitrd)) |
4545 |
|
return 1; |
4546 |
|
} |
4547 |
} |
} |
4548 |
if (addNewKernel(config, template, bootPrefix, newKernelPath, |
if (addNewKernel(config, template, bootPrefix, newKernelPath, |
4549 |
newKernelTitle, newKernelArgs, newKernelInitrd, |
newKernelTitle, newKernelArgs, newKernelInitrd, |