rpms/grubby
6
0
Fork 0
Code Issues Pull Requests Releases Activity
280685e9ea
grubby/SOURCES/0014-Lindent-dammit.patch

9240 lines
254 KiB
Diff
Raw Normal View History

import grubby-8.40-34.el8
2019-05-07 11:56:13 +00:00
From 446e0e6f0179373ccb2073df62d5552c44a0f36a Mon Sep 17 00:00:00 2001
From: Peter Jones <pjones@redhat.com>
Date: Tue, 18 Aug 2015 10:43:39 -0400
Subject: [PATCH 14/55] Lindent, dammit.
Signed-off-by: Peter Jones <pjones@redhat.com>
---
grubby.c | 8666 ++++++++++++++++++++++++++++--------------------------
1 file changed, 4560 insertions(+), 4106 deletions(-)
diff --git a/grubby.c b/grubby.c
index 70477ba14ee..fe6595b8386 100644
--- a/grubby.c
+++ b/grubby.c
@@ -48,7 +48,7 @@
#define dbgPrintf(format, args...)
#endif
-int debug = 0; /* Currently just for template debugging */
+int debug = 0; /* Currently just for template debugging */
#define _(A) (A)
@@ -70,51 +70,51 @@ char *saved_command_line = NULL;
/* comments get lumped in with indention */
struct lineElement {
- char * item;
- char * indent;
+ char *item;
+ char *indent;
};
-enum lineType_e {
- LT_WHITESPACE = 1 << 0,
- LT_TITLE = 1 << 1,
- LT_KERNEL = 1 << 2,
- LT_INITRD = 1 << 3,
- LT_HYPER = 1 << 4,
- LT_DEFAULT = 1 << 5,
- LT_MBMODULE = 1 << 6,
- LT_ROOT = 1 << 7,
- LT_FALLBACK = 1 << 8,
- LT_KERNELARGS = 1 << 9,
- LT_BOOT = 1 << 10,
- LT_BOOTROOT = 1 << 11,
- LT_LBA = 1 << 12,
- LT_OTHER = 1 << 13,
- LT_GENERIC = 1 << 14,
- LT_ECHO = 1 << 16,
- LT_MENUENTRY = 1 << 17,
- LT_ENTRY_END = 1 << 18,
- LT_SET_VARIABLE = 1 << 19,
- LT_KERNEL_EFI = 1 << 20,
- LT_INITRD_EFI = 1 << 21,
- LT_KERNEL_16 = 1 << 22,
- LT_INITRD_16 = 1 << 23,
- LT_DEVTREE = 1 << 24,
- LT_UNKNOWN = 1 << 25,
+enum lineType_e {
+ LT_WHITESPACE = 1 << 0,
+ LT_TITLE = 1 << 1,
+ LT_KERNEL = 1 << 2,
+ LT_INITRD = 1 << 3,
+ LT_HYPER = 1 << 4,
+ LT_DEFAULT = 1 << 5,
+ LT_MBMODULE = 1 << 6,
+ LT_ROOT = 1 << 7,
+ LT_FALLBACK = 1 << 8,
+ LT_KERNELARGS = 1 << 9,
+ LT_BOOT = 1 << 10,
+ LT_BOOTROOT = 1 << 11,
+ LT_LBA = 1 << 12,
+ LT_OTHER = 1 << 13,
+ LT_GENERIC = 1 << 14,
+ LT_ECHO = 1 << 16,
+ LT_MENUENTRY = 1 << 17,
+ LT_ENTRY_END = 1 << 18,
+ LT_SET_VARIABLE = 1 << 19,
+ LT_KERNEL_EFI = 1 << 20,
+ LT_INITRD_EFI = 1 << 21,
+ LT_KERNEL_16 = 1 << 22,
+ LT_INITRD_16 = 1 << 23,
+ LT_DEVTREE = 1 << 24,
+ LT_UNKNOWN = 1 << 25,
};
struct singleLine {
- char * indent;
- int numElements;
- struct lineElement * elements;
- struct singleLine * next;
- enum lineType_e type;
+ char *indent;
+ int numElements;
+ struct lineElement *elements;
+ struct singleLine *next;
+ enum lineType_e type;
};
struct singleEntry {
- struct singleLine * lines;
- int skip;
- int multiboot;
- struct singleEntry * next;
+ struct singleLine *lines;
+ int skip;
+ int multiboot;
+ struct singleEntry *next;
};
#define GRUBBY_BADIMAGE_OKAY (1 << 0)
@@ -135,1723 +135,1849 @@ struct singleEntry {
#define DEFAULT_SAVED_GRUB2 -3
struct keywordTypes {
- char * key;
- enum lineType_e type;
- char nextChar;
- char separatorChar;
+ char *key;
+ enum lineType_e type;
+ char nextChar;
+ char separatorChar;
};
struct configFileInfo;
-typedef const char *(*findConfigFunc)(struct configFileInfo *);
-typedef const int (*writeLineFunc)(struct configFileInfo *,
- struct singleLine *line);
-typedef char *(*getEnvFunc)(struct configFileInfo *, char *name);
-typedef int (*setEnvFunc)(struct configFileInfo *, char *name, char *value);
+typedef const char *(*findConfigFunc) (struct configFileInfo *);
+typedef const int (*writeLineFunc) (struct configFileInfo *,
+ struct singleLine * line);
+typedef char *(*getEnvFunc) (struct configFileInfo *, char *name);
+typedef int (*setEnvFunc) (struct configFileInfo *, char *name, char *value);
struct configFileInfo {
- char * defaultConfig;
- findConfigFunc findConfig;
- writeLineFunc writeLine;
- getEnvFunc getEnv;
- setEnvFunc setEnv;
- struct keywordTypes * keywords;
- int caseInsensitive;
- int defaultIsIndex;
- int defaultIsVariable;
- int defaultSupportSaved;
- int defaultIsSaved;
- int defaultIsUnquoted;
- enum lineType_e entryStart;
- enum lineType_e entryEnd;
- int needsBootPrefix;
- int argsInQuotes;
- int maxTitleLength;
- int titleBracketed;
- int titlePosition;
- int mbHyperFirst;
- int mbInitRdIsModule;
- int mbConcatArgs;
- int mbAllowExtraInitRds;
- char *envFile;
+ char *defaultConfig;
+ findConfigFunc findConfig;
+ writeLineFunc writeLine;
+ getEnvFunc getEnv;
+ setEnvFunc setEnv;
+ struct keywordTypes *keywords;
+ int caseInsensitive;
+ int defaultIsIndex;
+ int defaultIsVariable;
+ int defaultSupportSaved;
+ int defaultIsSaved;
+ int defaultIsUnquoted;
+ enum lineType_e entryStart;
+ enum lineType_e entryEnd;
+ int needsBootPrefix;
+ int argsInQuotes;
+ int maxTitleLength;
+ int titleBracketed;
+ int titlePosition;
+ int mbHyperFirst;
+ int mbInitRdIsModule;
+ int mbConcatArgs;
+ int mbAllowExtraInitRds;
+ char *envFile;
};
struct keywordTypes grubKeywords[] = {
- { "title", LT_TITLE, ' ' },
- { "root", LT_BOOTROOT, ' ' },
- { "default", LT_DEFAULT, ' ' },
- { "fallback", LT_FALLBACK, ' ' },
- { "kernel", LT_KERNEL, ' ' },
- { "initrd", LT_INITRD, ' ', ' ' },
- { "module", LT_MBMODULE, ' ' },
- { "kernel", LT_HYPER, ' ' },
- { NULL, 0, 0 },
+ {"title", LT_TITLE, ' '},
+ {"root", LT_BOOTROOT, ' '},
+ {"default", LT_DEFAULT, ' '},
+ {"fallback", LT_FALLBACK, ' '},
+ {"kernel", LT_KERNEL, ' '},
+ {"initrd", LT_INITRD, ' ', ' '},
+ {"module", LT_MBMODULE, ' '},
+ {"kernel", LT_HYPER, ' '},
+ {NULL, 0, 0},
};
-const char *grubFindConfig(struct configFileInfo *cfi) {
- static const char *configFiles[] = {
- "/boot/grub/menu.lst",
- "/etc/grub.conf",
- NULL
- };
- static int i = -1;
+const char *grubFindConfig(struct configFileInfo *cfi)
+{
+ static const char *configFiles[] = {
+ "/boot/grub/menu.lst",
+ "/etc/grub.conf",
+ NULL
+ };
+ static int i = -1;
- if (i == -1) {
- for (i = 0; configFiles[i] != NULL; i++) {
- dbgPrintf("Checking \"%s\": ", configFiles[i]);
- if (!access(configFiles[i], R_OK)) {
- dbgPrintf("found\n");
- return configFiles[i];
- }
- dbgPrintf("not found\n");
+ if (i == -1) {
+ for (i = 0; configFiles[i] != NULL; i++) {
+ dbgPrintf("Checking \"%s\": ", configFiles[i]);
+ if (!access(configFiles[i], R_OK)) {
+ dbgPrintf("found\n");
+ return configFiles[i];
+ }
+ dbgPrintf("not found\n");
+ }
}
- }
- return configFiles[i];
+ return configFiles[i];
}
struct configFileInfo grubConfigType = {
- .findConfig = grubFindConfig,
- .keywords = grubKeywords,
- .defaultIsIndex = 1,
- .defaultSupportSaved = 1,
- .entryStart = LT_TITLE,
- .needsBootPrefix = 1,
- .mbHyperFirst = 1,
- .mbInitRdIsModule = 1,
- .mbAllowExtraInitRds = 1,
- .titlePosition = 1,
+ .findConfig = grubFindConfig,
+ .keywords = grubKeywords,
+ .defaultIsIndex = 1,
+ .defaultSupportSaved = 1,
+ .entryStart = LT_TITLE,
+ .needsBootPrefix = 1,
+ .mbHyperFirst = 1,
+ .mbInitRdIsModule = 1,
+ .mbAllowExtraInitRds = 1,
+ .titlePosition = 1,
};
struct keywordTypes grub2Keywords[] = {
- { "menuentry", LT_MENUENTRY, ' ' },
- { "}", LT_ENTRY_END, ' ' },
- { "echo", LT_ECHO, ' ' },
- { "set", LT_SET_VARIABLE,' ', '=' },
- { "root", LT_BOOTROOT, ' ' },
- { "default", LT_DEFAULT, ' ' },
- { "fallback", LT_FALLBACK, ' ' },
- { "linux", LT_KERNEL, ' ' },
- { "linuxefi", LT_KERNEL_EFI, ' ' },
- { "linux16", LT_KERNEL_16, ' ' },
- { "initrd", LT_INITRD, ' ', ' ' },
- { "initrdefi", LT_INITRD_EFI, ' ', ' ' },
- { "initrd16", LT_INITRD_16, ' ', ' ' },
- { "module", LT_MBMODULE, ' ' },
- { "kernel", LT_HYPER, ' ' },
- { "devicetree", LT_DEVTREE, ' ' },
- { NULL, 0, 0 },
+ {"menuentry", LT_MENUENTRY, ' '},
+ {"}", LT_ENTRY_END, ' '},
+ {"echo", LT_ECHO, ' '},
+ {"set", LT_SET_VARIABLE, ' ', '='},
+ {"root", LT_BOOTROOT, ' '},
+ {"default", LT_DEFAULT, ' '},
+ {"fallback", LT_FALLBACK, ' '},
+ {"linux", LT_KERNEL, ' '},
+ {"linuxefi", LT_KERNEL_EFI, ' '},
+ {"linux16", LT_KERNEL_16, ' '},
+ {"initrd", LT_INITRD, ' ', ' '},
+ {"initrdefi", LT_INITRD_EFI, ' ', ' '},
+ {"initrd16", LT_INITRD_16, ' ', ' '},
+ {"module", LT_MBMODULE, ' '},
+ {"kernel", LT_HYPER, ' '},
+ {"devicetree", LT_DEVTREE, ' '},
+ {NULL, 0, 0},
};
-const char *grub2FindConfig(struct configFileInfo *cfi) {
- static const char *configFiles[] = {
- "/etc/grub2-efi.cfg",
- "/etc/grub2.cfg",
- "/boot/grub2/grub.cfg",
- "/boot/grub2-efi/grub.cfg",
- NULL
- };
- static int i = -1;
- static const char *grub_cfg = "/boot/grub/grub.cfg";
- int rc = -1;
+const char *grub2FindConfig(struct configFileInfo *cfi)
+{
+ static const char *configFiles[] = {
+ "/etc/grub2-efi.cfg",
+ "/etc/grub2.cfg",
+ "/boot/grub2/grub.cfg",
+ "/boot/grub2-efi/grub.cfg",
+ NULL
+ };
+ static int i = -1;
+ static const char *grub_cfg = "/boot/grub/grub.cfg";
+ int rc = -1;
- if (i == -1) {
- for (i = 0; configFiles[i] != NULL; i++) {
- dbgPrintf("Checking \"%s\": ", configFiles[i]);
- if ((rc = access(configFiles[i], R_OK))) {
- if (errno == EACCES) {
- printf("Unable to access bootloader configuration file "
- "\"%s\": %m\n", configFiles[i]);
- exit(1);
+ if (i == -1) {
+ for (i = 0; configFiles[i] != NULL; i++) {
+ dbgPrintf("Checking \"%s\": ", configFiles[i]);
+ if ((rc = access(configFiles[i], R_OK))) {
+ if (errno == EACCES) {
+ printf
+ ("Unable to access bootloader configuration file "
+ "\"%s\": %m\n", configFiles[i]);
+ exit(1);
+ }
+ continue;
+ } else {
+ dbgPrintf("found\n");
+ return configFiles[i];
+ }
}
- continue;
- } else {
+ }
+
+ /* Ubuntu renames grub2 to grub, so check for the grub.d directory
+ * that isn't in grub1, and if it exists, return the config file path
+ * that they use. */
+ if (configFiles[i] == NULL && !access("/etc/grub.d/", R_OK)) {
dbgPrintf("found\n");
- return configFiles[i];
- }
+ return grub_cfg;
}
- }
- /* Ubuntu renames grub2 to grub, so check for the grub.d directory
- * that isn't in grub1, and if it exists, return the config file path
- * that they use. */
- if (configFiles[i] == NULL && !access("/etc/grub.d/", R_OK)) {
- dbgPrintf("found\n");
- return grub_cfg;
- }
-
- dbgPrintf("not found\n");
- return configFiles[i];
+ dbgPrintf("not found\n");
+ return configFiles[i];
}
/* kind of hacky. It'll give the first 1024 bytes, ish. */
static char *grub2GetEnv(struct configFileInfo *info, char *name)
{
- static char buf[1025];
- char *s = NULL;
- char *ret = NULL;
- char *envFile = info->envFile ? info->envFile : "/boot/grub2/grubenv";
- int rc = asprintf(&s, "grub2-editenv %s list | grep '^%s='", envFile, name);
+ static char buf[1025];
+ char *s = NULL;
+ char *ret = NULL;
+ char *envFile = info->envFile ? info->envFile : "/boot/grub2/grubenv";
+ int rc =
+ asprintf(&s, "grub2-editenv %s list | grep '^%s='", envFile, name);
- if (rc < 0)
- return NULL;
+ if (rc < 0)
+ return NULL;
- FILE *f = popen(s, "r");
- if (!f)
- goto out;
+ FILE *f = popen(s, "r");
+ if (!f)
+ goto out;
- memset(buf, '\0', sizeof (buf));
- ret = fgets(buf, 1024, f);
- pclose(f);
+ memset(buf, '\0', sizeof(buf));
+ ret = fgets(buf, 1024, f);
+ pclose(f);
- if (ret) {
- ret += strlen(name) + 1;
- ret[strlen(ret) - 1] = '\0';
- }
- dbgPrintf("grub2GetEnv(%s): %s\n", name, ret);
+ if (ret) {
+ ret += strlen(name) + 1;
+ ret[strlen(ret) - 1] = '\0';
+ }
+ dbgPrintf("grub2GetEnv(%s): %s\n", name, ret);
out:
- free(s);
- return ret;
+ free(s);
+ return ret;
}
static int sPopCount(const char *s, const char *c)
{
- int ret = 0;
- if (!s)
- return -1;
- for (int i = 0; s[i] != '\0'; i++)
- for (int j = 0; c[j] != '\0'; j++)
- if (s[i] == c[j])
- ret++;
- return ret;
+ int ret = 0;
+ if (!s)
+ return -1;
+ for (int i = 0; s[i] != '\0'; i++)
+ for (int j = 0; c[j] != '\0'; j++)
+ if (s[i] == c[j])
+ ret++;
+ return ret;
}
static char *shellEscape(const char *s)
{
- int l = strlen(s) + sPopCount(s, "'") * 2;
+ int l = strlen(s) + sPopCount(s, "'") * 2;
- char *ret = calloc(l+1, sizeof (*ret));
- if (!ret)
- return NULL;
- for (int i = 0, j = 0; s[i] != '\0'; i++, j++) {
- if (s[i] == '\'')
- ret[j++] = '\\';
- ret[j] = s[i];
- }
- return ret;
+ char *ret = calloc(l + 1, sizeof(*ret));
+ if (!ret)
+ return NULL;
+ for (int i = 0, j = 0; s[i] != '\0'; i++, j++) {
+ if (s[i] == '\'')
+ ret[j++] = '\\';
+ ret[j] = s[i];
+ }
+ return ret;
}
static void unquote(char *s)
{
- int l = strlen(s);
+ int l = strlen(s);
- if ((s[l-1] == '\'' && s[0] == '\'') || (s[l-1] == '"' && s[0] == '"')) {
- memmove(s, s+1, l-2);
- s[l-2] = '\0';
- }
+ if ((s[l - 1] == '\'' && s[0] == '\'')
+ || (s[l - 1] == '"' && s[0] == '"')) {
+ memmove(s, s + 1, l - 2);
+ s[l - 2] = '\0';
+ }
}
static int grub2SetEnv(struct configFileInfo *info, char *name, char *value)
{
- char *s = NULL;
- int rc = 0;
- char *envFile = info->envFile ? info->envFile : "/boot/grub2/grubenv";
+ char *s = NULL;
+ int rc = 0;
+ char *envFile = info->envFile ? info->envFile : "/boot/grub2/grubenv";
- unquote(value);
- value = shellEscape(value);
- if (!value)
- return -1;
+ unquote(value);
+ value = shellEscape(value);
+ if (!value)
+ return -1;
- rc = asprintf(&s, "grub2-editenv %s set '%s=%s'", envFile, name, value);
- free(value);
- if (rc <0)
- return -1;
+ rc = asprintf(&s, "grub2-editenv %s set '%s=%s'", envFile, name, value);
+ free(value);
+ if (rc < 0)
+ return -1;
- dbgPrintf("grub2SetEnv(%s): %s\n", name, s);
- rc = system(s);
- free(s);
- return rc;
+ dbgPrintf("grub2SetEnv(%s): %s\n", name, s);
+ rc = system(s);
+ free(s);
+ return rc;
}
/* this is a gigantic hack to avoid clobbering grub2 variables... */
static int is_special_grub2_variable(const char *name)
{
- if (!strcmp(name,"\"${next_entry}\""))
- return 1;
- if (!strcmp(name,"\"${prev_saved_entry}\""))
- return 1;
- return 0;
+ if (!strcmp(name, "\"${next_entry}\""))
+ return 1;
+ if (!strcmp(name, "\"${prev_saved_entry}\""))
+ return 1;
+ return 0;
}
-int sizeOfSingleLine(struct singleLine * line) {
- int count = 0;
+int sizeOfSingleLine(struct singleLine *line)
+{
+ int count = 0;
- for (int i = 0; i < line->numElements; i++) {
- int indentSize = 0;
+ for (int i = 0; i < line->numElements; i++) {
+ int indentSize = 0;
- count = count + strlen(line->elements[i].item);
+ count = count + strlen(line->elements[i].item);
- indentSize = strlen(line->elements[i].indent);
- if (indentSize > 0)
- count = count + indentSize;
- else
- /* be extra safe and add room for whitespaces */
- count = count + 1;
- }
+ indentSize = strlen(line->elements[i].indent);
+ if (indentSize > 0)
+ count = count + indentSize;
+ else
+ /* be extra safe and add room for whitespaces */
+ count = count + 1;
+ }
- /* room for trailing terminator */
- count = count + 1;
+ /* room for trailing terminator */
+ count = count + 1;
- return count;
+ return count;
}
static int isquote(char q)
{
- if (q == '\'' || q == '\"')
- return 1;
- return 0;
+ if (q == '\'' || q == '\"')
+ return 1;
+ return 0;
}
-static int iskernel(enum lineType_e type) {
- return (type == LT_KERNEL || type == LT_KERNEL_EFI || type == LT_KERNEL_16);
+static int iskernel(enum lineType_e type)
+{
+ return (type == LT_KERNEL || type == LT_KERNEL_EFI
+ || type == LT_KERNEL_16);
}
-static int isinitrd(enum lineType_e type) {
- return (type == LT_INITRD || type == LT_INITRD_EFI || type == LT_INITRD_16);
+static int isinitrd(enum lineType_e type)
+{
+ return (type == LT_INITRD || type == LT_INITRD_EFI
+ || type == LT_INITRD_16);
}
-char *grub2ExtractTitle(struct singleLine * line) {
- char * current;
- char * current_indent;
- int current_len;
- int current_indent_len;
- int i;
+char *grub2ExtractTitle(struct singleLine *line)
+{
+ char *current;
+ char *current_indent;
+ int current_len;
+ int current_indent_len;
+ int i;
- /* bail out if line does not start with menuentry */
- if (strcmp(line->elements[0].item, "menuentry"))
- return NULL;
+ /* bail out if line does not start with menuentry */
+ if (strcmp(line->elements[0].item, "menuentry"))
+ return NULL;
- i = 1;
- current = line->elements[i].item;
- current_len = strlen(current);
-
- /* if second word is quoted, strip the quotes and return single word */
- if (isquote(*current) && isquote(current[current_len - 1])) {
- char *tmp;
-
- tmp = strdup(current+1);
- if (!tmp)
- return NULL;
- tmp[strlen(tmp)-1] = '\0';
- return tmp;
- }
-
- /* if no quotes, return second word verbatim */
- if (!isquote(*current))
- return current;
-
- /* second element start with a quote, so we have to find the element
- * whose last character is also quote (assuming it's the closing one) */
- int resultMaxSize;
- char * result;
- /* need to ensure that ' does not match " as we search */
- char quote_char = *current;
-
- resultMaxSize = sizeOfSingleLine(line);
- result = malloc(resultMaxSize);
- snprintf(result, resultMaxSize, "%s", ++current);
-
- i++;
- for (; i < line->numElements; ++i) {
+ i = 1;
current = line->elements[i].item;
current_len = strlen(current);
- current_indent = line->elements[i].indent;
- current_indent_len = strlen(current_indent);
-
- strncat(result, current_indent, current_indent_len);
- if (current[current_len-1] != quote_char) {
- strncat(result, current, current_len);
- } else {
- strncat(result, current, current_len - 1);
- break;
+
+ /* if second word is quoted, strip the quotes and return single word */
+ if (isquote(*current) && isquote(current[current_len - 1])) {
+ char *tmp;
+
+ tmp = strdup(current + 1);
+ if (!tmp)
+ return NULL;
+ tmp[strlen(tmp) - 1] = '\0';
+ return tmp;
+ }
+
+ /* if no quotes, return second word verbatim */
+ if (!isquote(*current))
+ return current;
+
+ /* second element start with a quote, so we have to find the element
+ * whose last character is also quote (assuming it's the closing one) */
+ int resultMaxSize;
+ char *result;
+ /* need to ensure that ' does not match " as we search */
+ char quote_char = *current;
+
+ resultMaxSize = sizeOfSingleLine(line);
+ result = malloc(resultMaxSize);
+ snprintf(result, resultMaxSize, "%s", ++current);
+
+ i++;
+ for (; i < line->numElements; ++i) {
+ current = line->elements[i].item;
+ current_len = strlen(current);
+ current_indent = line->elements[i].indent;
+ current_indent_len = strlen(current_indent);
+
+ strncat(result, current_indent, current_indent_len);
+ if (current[current_len - 1] != quote_char) {
+ strncat(result, current, current_len);
+ } else {
+ strncat(result, current, current_len - 1);
+ break;
+ }
}
- }
- return result;
+ return result;
}
struct configFileInfo grub2ConfigType = {
- .findConfig = grub2FindConfig,
- .getEnv = grub2GetEnv,
- .setEnv = grub2SetEnv,
- .keywords = grub2Keywords,
- .defaultIsIndex = 1,
- .defaultSupportSaved = 1,
- .defaultIsVariable = 1,
- .entryStart = LT_MENUENTRY,
- .entryEnd = LT_ENTRY_END,
- .titlePosition = 1,
- .needsBootPrefix = 1,
- .mbHyperFirst = 1,
- .mbInitRdIsModule = 1,
- .mbAllowExtraInitRds = 1,
+ .findConfig = grub2FindConfig,
+ .getEnv = grub2GetEnv,
+ .setEnv = grub2SetEnv,
+ .keywords = grub2Keywords,
+ .defaultIsIndex = 1,
+ .defaultSupportSaved = 1,
+ .defaultIsVariable = 1,
+ .entryStart = LT_MENUENTRY,
+ .entryEnd = LT_ENTRY_END,
+ .titlePosition = 1,
+ .needsBootPrefix = 1,
+ .mbHyperFirst = 1,
+ .mbInitRdIsModule = 1,
+ .mbAllowExtraInitRds = 1,
};
struct keywordTypes yabootKeywords[] = {
- { "label", LT_TITLE, '=' },
- { "root", LT_ROOT, '=' },
- { "default", LT_DEFAULT, '=' },
- { "image", LT_KERNEL, '=' },
- { "bsd", LT_GENERIC, '=' },
- { "macos", LT_GENERIC, '=' },
- { "macosx", LT_GENERIC, '=' },
- { "magicboot", LT_GENERIC, '=' },
- { "darwin", LT_GENERIC, '=' },
- { "timeout", LT_GENERIC, '=' },
- { "install", LT_GENERIC, '=' },
- { "fstype", LT_GENERIC, '=' },
- { "hfstype", LT_GENERIC, '=' },
- { "delay", LT_GENERIC, '=' },
- { "defaultos", LT_GENERIC, '=' },
- { "init-message", LT_GENERIC, '=' },
- { "enablecdboot", LT_GENERIC, ' ' },
- { "enableofboot", LT_GENERIC, ' ' },
- { "enablenetboot", LT_GENERIC, ' ' },
- { "nonvram", LT_GENERIC, ' ' },
- { "hide", LT_GENERIC, ' ' },
- { "protect", LT_GENERIC, ' ' },
- { "nobless", LT_GENERIC, ' ' },
- { "nonvram", LT_GENERIC, ' ' },
- { "brokenosx", LT_GENERIC, ' ' },
- { "usemount", LT_GENERIC, ' ' },
- { "mntpoint", LT_GENERIC, '=' },
- { "partition", LT_GENERIC, '=' },
- { "device", LT_GENERIC, '=' },
- { "fstype", LT_GENERIC, '=' },
- { "initrd", LT_INITRD, '=', ';' },
- { "append", LT_KERNELARGS, '=' },
- { "boot", LT_BOOT, '=' },
- { "lba", LT_LBA, ' ' },
- { NULL, 0, 0 },
+ {"label", LT_TITLE, '='},
+ {"root", LT_ROOT, '='},
+ {"default", LT_DEFAULT, '='},
+ {"image", LT_KERNEL, '='},
+ {"bsd", LT_GENERIC, '='},
+ {"macos", LT_GENERIC, '='},
+ {"macosx", LT_GENERIC, '='},
+ {"magicboot", LT_GENERIC, '='},
+ {"darwin", LT_GENERIC, '='},
+ {"timeout", LT_GENERIC, '='},
+ {"install", LT_GENERIC, '='},
+ {"fstype", LT_GENERIC, '='},
+ {"hfstype", LT_GENERIC, '='},
+ {"delay", LT_GENERIC, '='},
+ {"defaultos", LT_GENERIC, '='},
+ {"init-message", LT_GENERIC, '='},
+ {"enablecdboot", LT_GENERIC, ' '},
+ {"enableofboot", LT_GENERIC, ' '},
+ {"enablenetboot", LT_GENERIC, ' '},
+ {"nonvram", LT_GENERIC, ' '},
+ {"hide", LT_GENERIC, ' '},
+ {"protect", LT_GENERIC, ' '},
+ {"nobless", LT_GENERIC, ' '},
+ {"nonvram", LT_GENERIC, ' '},
+ {"brokenosx", LT_GENERIC, ' '},
+ {"usemount", LT_GENERIC, ' '},
+ {"mntpoint", LT_GENERIC, '='},
+ {"partition", LT_GENERIC, '='},
+ {"device", LT_GENERIC, '='},
+ {"fstype", LT_GENERIC, '='},
+ {"initrd", LT_INITRD, '=', ';'},
+ {"append", LT_KERNELARGS, '='},
+ {"boot", LT_BOOT, '='},
+ {"lba", LT_LBA, ' '},
+ {NULL, 0, 0},
};
struct keywordTypes liloKeywords[] = {
- { "label", LT_TITLE, '=' },
- { "root", LT_ROOT, '=' },
- { "default", LT_DEFAULT, '=' },
- { "image", LT_KERNEL, '=' },
- { "other", LT_OTHER, '=' },
- { "initrd", LT_INITRD, '=' },
- { "append", LT_KERNELARGS, '=' },
- { "boot", LT_BOOT, '=' },
- { "lba", LT_LBA, ' ' },
- { NULL, 0, 0 },
+ {"label", LT_TITLE, '='},
+ {"root", LT_ROOT, '='},
+ {"default", LT_DEFAULT, '='},
+ {"image", LT_KERNEL, '='},
+ {"other", LT_OTHER, '='},
+ {"initrd", LT_INITRD, '='},
+ {"append", LT_KERNELARGS, '='},
+ {"boot", LT_BOOT, '='},
+ {"lba", LT_LBA, ' '},
+ {NULL, 0, 0},
};
struct keywordTypes eliloKeywords[] = {
- { "label", LT_TITLE, '=' },
- { "root", LT_ROOT, '=' },
- { "default", LT_DEFAULT, '=' },
- { "image", LT_KERNEL, '=' },
- { "initrd", LT_INITRD, '=' },
- { "append", LT_KERNELARGS, '=' },
- { "vmm", LT_HYPER, '=' },
- { NULL, 0, 0 },
+ {"label", LT_TITLE, '='},
+ {"root", LT_ROOT, '='},
+ {"default", LT_DEFAULT, '='},
+ {"image", LT_KERNEL, '='},
+ {"initrd", LT_INITRD, '='},
+ {"append", LT_KERNELARGS, '='},
+ {"vmm", LT_HYPER, '='},
+ {NULL, 0, 0},
};
struct keywordTypes siloKeywords[] = {
- { "label", LT_TITLE, '=' },
- { "root", LT_ROOT, '=' },
- { "default", LT_DEFAULT, '=' },
- { "image", LT_KERNEL, '=' },
- { "other", LT_OTHER, '=' },
- { "initrd", LT_INITRD, '=' },
- { "append", LT_KERNELARGS, '=' },
- { "boot", LT_BOOT, '=' },
- { NULL, 0, 0 },
+ {"label", LT_TITLE, '='},
+ {"root", LT_ROOT, '='},
+ {"default", LT_DEFAULT, '='},
+ {"image", LT_KERNEL, '='},
+ {"other", LT_OTHER, '='},
+ {"initrd", LT_INITRD, '='},
+ {"append", LT_KERNELARGS, '='},
+ {"boot", LT_BOOT, '='},
+ {NULL, 0, 0},
};
struct keywordTypes ziplKeywords[] = {
- { "target", LT_BOOTROOT, '=' },
- { "image", LT_KERNEL, '=' },
- { "ramdisk", LT_INITRD, '=' },
- { "parameters", LT_KERNELARGS, '=' },
- { "default", LT_DEFAULT, '=' },
- { NULL, 0, 0 },
+ {"target", LT_BOOTROOT, '='},
+ {"image", LT_KERNEL, '='},
+ {"ramdisk", LT_INITRD, '='},
+ {"parameters", LT_KERNELARGS, '='},
+ {"default", LT_DEFAULT, '='},
+ {NULL, 0, 0},
};
struct keywordTypes extlinuxKeywords[] = {
- { "label", LT_TITLE, ' ' },
- { "root", LT_ROOT, ' ' },
- { "default", LT_DEFAULT, ' ' },
- { "kernel", LT_KERNEL, ' ' },
- { "initrd", LT_INITRD, ' ', ',' },
- { "append", LT_KERNELARGS, ' ' },
- { "prompt", LT_UNKNOWN, ' ' },
- { "fdt", LT_DEVTREE, ' ' },
- { "fdtdir", LT_DEVTREE, ' ' },
- { NULL, 0, 0 },
+ {"label", LT_TITLE, ' '},
+ {"root", LT_ROOT, ' '},
+ {"default", LT_DEFAULT, ' '},
+ {"kernel", LT_KERNEL, ' '},
+ {"initrd", LT_INITRD, ' ', ','},
+ {"append", LT_KERNELARGS, ' '},
+ {"prompt", LT_UNKNOWN, ' '},
+ {"fdt", LT_DEVTREE, ' '},
+ {"fdtdir", LT_DEVTREE, ' '},
+ {NULL, 0, 0},
};
+
int useextlinuxmenu;
struct configFileInfo eliloConfigType = {
- .defaultConfig = "/boot/efi/EFI/redhat/elilo.conf",
- .keywords = eliloKeywords,
- .entryStart = LT_KERNEL,
- .needsBootPrefix = 1,
- .argsInQuotes = 1,
- .mbConcatArgs = 1,
- .titlePosition = 1,
+ .defaultConfig = "/boot/efi/EFI/redhat/elilo.conf",
+ .keywords = eliloKeywords,
+ .entryStart = LT_KERNEL,
+ .needsBootPrefix = 1,
+ .argsInQuotes = 1,
+ .mbConcatArgs = 1,
+ .titlePosition = 1,
};
struct configFileInfo liloConfigType = {
- .defaultConfig = "/etc/lilo.conf",
- .keywords = liloKeywords,
- .entryStart = LT_KERNEL,
- .argsInQuotes = 1,
- .maxTitleLength = 15,
- .titlePosition = 1,
+ .defaultConfig = "/etc/lilo.conf",
+ .keywords = liloKeywords,
+ .entryStart = LT_KERNEL,
+ .argsInQuotes = 1,
+ .maxTitleLength = 15,
+ .titlePosition = 1,
};
struct configFileInfo yabootConfigType = {
- .defaultConfig = "/etc/yaboot.conf",
- .keywords = yabootKeywords,
- .entryStart = LT_KERNEL,
- .needsBootPrefix = 1,
- .argsInQuotes = 1,
- .maxTitleLength = 15,
- .mbAllowExtraInitRds = 1,
- .titlePosition = 1,
+ .defaultConfig = "/etc/yaboot.conf",
+ .keywords = yabootKeywords,
+ .entryStart = LT_KERNEL,
+ .needsBootPrefix = 1,
+ .argsInQuotes = 1,
+ .maxTitleLength = 15,
+ .mbAllowExtraInitRds = 1,
+ .titlePosition = 1,
};
struct configFileInfo siloConfigType = {
- .defaultConfig = "/etc/silo.conf",
- .keywords = siloKeywords,
- .entryStart = LT_KERNEL,
- .needsBootPrefix = 1,
- .argsInQuotes = 1,
- .maxTitleLength = 15,
- .titlePosition = 1,
+ .defaultConfig = "/etc/silo.conf",
+ .keywords = siloKeywords,
+ .entryStart = LT_KERNEL,
+ .needsBootPrefix = 1,
+ .argsInQuotes = 1,
+ .maxTitleLength = 15,
+ .titlePosition = 1,
};
struct configFileInfo ziplConfigType = {
- .defaultConfig = "/etc/zipl.conf",
- .keywords = ziplKeywords,
- .entryStart = LT_TITLE,
- .argsInQuotes = 1,
- .titleBracketed = 1,
+ .defaultConfig = "/etc/zipl.conf",
+ .keywords = ziplKeywords,
+ .entryStart = LT_TITLE,
+ .argsInQuotes = 1,
+ .titleBracketed = 1,
};
struct configFileInfo extlinuxConfigType = {
- .defaultConfig = "/boot/extlinux/extlinux.conf",
- .keywords = extlinuxKeywords,
- .caseInsensitive = 1,
- .entryStart = LT_TITLE,
- .needsBootPrefix = 1,
- .maxTitleLength = 255,
- .mbAllowExtraInitRds = 1,
- .defaultIsUnquoted = 1,
- .titlePosition = 1,
+ .defaultConfig = "/boot/extlinux/extlinux.conf",
+ .keywords = extlinuxKeywords,
+ .caseInsensitive = 1,
+ .entryStart = LT_TITLE,
+ .needsBootPrefix = 1,
+ .maxTitleLength = 255,
+ .mbAllowExtraInitRds = 1,
+ .defaultIsUnquoted = 1,
+ .titlePosition = 1,
};
struct grubConfig {
- struct singleLine * theLines;
- struct singleEntry * entries;
- char * primaryIndent;
- char * secondaryIndent;
- int defaultImage; /* -1 if none specified -- this value is
- * written out, overriding original */
- int fallbackImage; /* just like defaultImage */
- int flags;
- struct configFileInfo * cfi;
+ struct singleLine *theLines;
+ struct singleEntry *entries;
+ char *primaryIndent;
+ char *secondaryIndent;
+ int defaultImage; /* -1 if none specified -- this value is
+ * written out, overriding original */
+ int fallbackImage; /* just like defaultImage */
+ int flags;
+ struct configFileInfo *cfi;
};
blkid_cache blkid;
-struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index);
-struct singleEntry * findEntryByPath(struct grubConfig * cfg,
- const char * path, const char * prefix,
- int * index);
-struct singleEntry * findEntryByTitle(struct grubConfig * cfg, char *title,
- int * index);
-static int readFile(int fd, char ** bufPtr);
-static void lineInit(struct singleLine * line);
-struct singleLine * lineDup(struct singleLine * line);
-static void lineFree(struct singleLine * line);
-static int lineWrite(FILE * out, struct singleLine * line,
- struct configFileInfo * cfi);
-static int getNextLine(char ** bufPtr, struct singleLine * line,
- struct configFileInfo * cfi);
-static char * getRootSpecifier(char * str);
-static void requote(struct singleLine *line, struct configFileInfo * cfi);
-static void insertElement(struct singleLine * line,
- const char * item, int insertHere,
- struct configFileInfo * cfi);
-static void removeElement(struct singleLine * line, int removeHere);
-static struct keywordTypes * getKeywordByType(enum lineType_e type,
- struct configFileInfo * cfi);
-static enum lineType_e getTypeByKeyword(char * keyword,
- struct configFileInfo * cfi);
-static struct singleLine * getLineByType(enum lineType_e type,
- struct singleLine * line);
-static int checkForExtLinux(struct grubConfig * config);
-struct singleLine * addLineTmpl(struct singleEntry * entry,
- struct singleLine * tmplLine,
- struct singleLine * prevLine,
- const char * val,
- struct configFileInfo * cfi);
-struct singleLine * addLine(struct singleEntry * entry,
- struct configFileInfo * cfi,
- enum lineType_e type, char * defaultIndent,
- const char * val);
+struct singleEntry *findEntryByIndex(struct grubConfig *cfg, int index);
+struct singleEntry *findEntryByPath(struct grubConfig *cfg,
+ const char *path, const char *prefix,
+ int *index);
+struct singleEntry *findEntryByTitle(struct grubConfig *cfg, char *title,
+ int *index);
+static int readFile(int fd, char **bufPtr);
+static void lineInit(struct singleLine *line);
+struct singleLine *lineDup(struct singleLine *line);
+static void lineFree(struct singleLine *line);
+static int lineWrite(FILE * out, struct singleLine *line,
+ struct configFileInfo *cfi);
+static int getNextLine(char **bufPtr, struct singleLine *line,
+ struct configFileInfo *cfi);
+static char *getRootSpecifier(char *str);
+static void requote(struct singleLine *line, struct configFileInfo *cfi);
+static void insertElement(struct singleLine *line,
+ const char *item, int insertHere,
+ struct configFileInfo *cfi);
+static void removeElement(struct singleLine *line, int removeHere);
+static struct keywordTypes *getKeywordByType(enum lineType_e type,
+ struct configFileInfo *cfi);
+static enum lineType_e getTypeByKeyword(char *keyword,
+ struct configFileInfo *cfi);
+static struct singleLine *getLineByType(enum lineType_e type,
+ struct singleLine *line);
+static int checkForExtLinux(struct grubConfig *config);
+struct singleLine *addLineTmpl(struct singleEntry *entry,
+ struct singleLine *tmplLine,
+ struct singleLine *prevLine,
+ const char *val, struct configFileInfo *cfi);
+struct singleLine *addLine(struct singleEntry *entry,
+ struct configFileInfo *cfi,
+ enum lineType_e type, char *defaultIndent,
+ const char *val);
-static char * sdupprintf(const char *format, ...)
+static char *sdupprintf(const char *format, ...)
#ifdef __GNUC__
- __attribute__ ((format (printf, 1, 2)));
+ __attribute__ ((format(printf, 1, 2)));
#else
- ;
+;
#endif
-static char * sdupprintf(const char *format, ...) {
- char *buf = NULL;
- char c;
- va_list args;
- size_t size = 0;
- va_start(args, format);
-
- /* XXX requires C99 vsnprintf behavior */
- size = vsnprintf(&c, 1, format, args) + 1;
- if (size == -1) {
- printf("ERROR: vsnprintf behavior is not C99\n");
- abort();
- }
+static char *sdupprintf(const char *format, ...)
+{
+ char *buf = NULL;
+ char c;
+ va_list args;
+ size_t size = 0;
+ va_start(args, format);
- va_end(args);
- va_start(args, format);
+ /* XXX requires C99 vsnprintf behavior */
+ size = vsnprintf(&c, 1, format, args) + 1;
+ if (size == -1) {
+ printf("ERROR: vsnprintf behavior is not C99\n");
+ abort();
+ }
- buf = malloc(size);
- if (buf == NULL)
- return NULL;
- vsnprintf(buf, size, format, args);
- va_end (args);
+ va_end(args);
+ va_start(args, format);
- return buf;
+ buf = malloc(size);
+ if (buf == NULL)
+ return NULL;
+ vsnprintf(buf, size, format, args);
+ va_end(args);
+
+ return buf;
}
static enum lineType_e preferredLineType(enum lineType_e type,
- struct configFileInfo *cfi) {
- if (isEfi && cfi == &grub2ConfigType) {
- switch (type) {
- case LT_KERNEL:
- return isEfiOnly ? LT_KERNEL : LT_KERNEL_EFI;
- case LT_INITRD:
- return isEfiOnly ? LT_INITRD : LT_INITRD_EFI;
- default:
- return type;
- }
+ struct configFileInfo *cfi)
+{
+ if (isEfi && cfi == &grub2ConfigType) {
+ switch (type) {
+ case LT_KERNEL:
+ return isEfiOnly ? LT_KERNEL : LT_KERNEL_EFI;
+ case LT_INITRD:
+ return isEfiOnly ? LT_INITRD : LT_INITRD_EFI;
+ default:
+ return type;
+ }
#if defined(__i386__) || defined(__x86_64__)
- } else if (cfi == &grub2ConfigType) {
- switch (type) {
- case LT_KERNEL:
- return LT_KERNEL_16;
- case LT_INITRD:
- return LT_INITRD_16;
- default:
- return type;
- }
+ } else if (cfi == &grub2ConfigType) {
+ switch (type) {
+ case LT_KERNEL:
+ return LT_KERNEL_16;
+ case LT_INITRD:
+ return LT_INITRD_16;
+ default:
+ return type;
+ }
#endif
- }
- return type;
+ }
+ return type;
}
-static struct keywordTypes * getKeywordByType(enum lineType_e type,
- struct configFileInfo * cfi) {
- for (struct keywordTypes *kw = cfi->keywords; kw->key; kw++) {
- if (kw->type == type)
- return kw;
- }
- return NULL;
+static struct keywordTypes *getKeywordByType(enum lineType_e type,
+ struct configFileInfo *cfi)
+{
+ for (struct keywordTypes * kw = cfi->keywords; kw->key; kw++) {
+ if (kw->type == type)
+ return kw;
+ }
+ return NULL;
}
-static char *getKeyByType(enum lineType_e type, struct configFileInfo * cfi) {
- struct keywordTypes *kt = getKeywordByType(type, cfi);
- if (kt)
- return kt->key;
- return "unknown";
+static char *getKeyByType(enum lineType_e type, struct configFileInfo *cfi)
+{
+ struct keywordTypes *kt = getKeywordByType(type, cfi);
+ if (kt)
+ return kt->key;
+ return "unknown";
}
-static char * getpathbyspec(char *device) {
- if (!blkid)
- blkid_get_cache(&blkid, NULL);
+static char *getpathbyspec(char *device)
+{
+ if (!blkid)
+ blkid_get_cache(&blkid, NULL);
- return blkid_get_devname(blkid, device, NULL);
+ return blkid_get_devname(blkid, device, NULL);
}
-static char * getuuidbydev(char *device) {
- if (!blkid)
- blkid_get_cache(&blkid, NULL);
+static char *getuuidbydev(char *device)
+{
+ if (!blkid)
+ blkid_get_cache(&blkid, NULL);
- return blkid_get_tag_value(blkid, "UUID", device);
+ return blkid_get_tag_value(blkid, "UUID", device);
}
-static enum lineType_e getTypeByKeyword(char * keyword,
- struct configFileInfo * cfi) {
- for (struct keywordTypes *kw = cfi->keywords; kw->key; kw++) {
- if (cfi->caseInsensitive) {
- if (!strcasecmp(keyword, kw->key))
- return kw->type;
- } else {
- if (!strcmp(keyword, kw->key))
- return kw->type;
+static enum lineType_e getTypeByKeyword(char *keyword,
+ struct configFileInfo *cfi)
+{
+ for (struct keywordTypes * kw = cfi->keywords; kw->key; kw++) {
+ if (cfi->caseInsensitive) {
+ if (!strcasecmp(keyword, kw->key))
+ return kw->type;
+ } else {
+ if (!strcmp(keyword, kw->key))
+ return kw->type;
+ }
}
- }
- return LT_UNKNOWN;
+ return LT_UNKNOWN;
}
-static struct singleLine * getLineByType(enum lineType_e type,
- struct singleLine * line) {
- dbgPrintf("getLineByType(%d): ", type);
- for (; line; line = line->next) {
- dbgPrintf("%d:%s ", line->type,
- line->numElements ? line->elements[0].item : "(empty)");
- if (line->type & type) break;
- }
- dbgPrintf(line ? "\n" : " (failed)\n");
- return line;
+static struct singleLine *getLineByType(enum lineType_e type,
+ struct singleLine *line)
+{
+ dbgPrintf("getLineByType(%d): ", type);
+ for (; line; line = line->next) {
+ dbgPrintf("%d:%s ", line->type,
+ line->numElements ? line->elements[0].
+ item : "(empty)");
+ if (line->type & type)
+ break;
+ }
+ dbgPrintf(line ? "\n" : " (failed)\n");
+ return line;
}
-static int isBracketedTitle(struct singleLine * line) {
- if (line->numElements == 1 && *line->elements[0].item == '[') {
- int len = strlen(line->elements[0].item);
- if (*(line->elements[0].item + len - 1) == ']') {
- /* FIXME: this is a hack... */
- if (strcmp(line->elements[0].item, "[defaultboot]")) {
- return 1;
- }
- }
- }
- return 0;
+static int isBracketedTitle(struct singleLine *line)
+{
+ if (line->numElements == 1 && *line->elements[0].item == '[') {
+ int len = strlen(line->elements[0].item);
+ if (*(line->elements[0].item + len - 1) == ']') {
+ /* FIXME: this is a hack... */
+ if (strcmp(line->elements[0].item, "[defaultboot]")) {
+ return 1;
+ }
+ }
+ }
+ return 0;
}
-static int isEntryStart(struct singleLine * line,
- struct configFileInfo * cfi) {
- return line->type == cfi->entryStart || line->type == LT_OTHER ||
- (cfi->titleBracketed && isBracketedTitle(line));
+static int isEntryStart(struct singleLine *line, struct configFileInfo *cfi)
+{
+ return line->type == cfi->entryStart || line->type == LT_OTHER ||
+ (cfi->titleBracketed && isBracketedTitle(line));
}
/* extract the title from within brackets (for zipl) */
-static char * extractTitle(struct grubConfig *cfg, struct singleLine * line) {
- /* bracketed title... let's extract it */
- char * title = NULL;
- if (line->type == LT_TITLE) {
- char *tmp = line->elements[cfg->cfi->titlePosition].item;
- if (cfg->cfi->titleBracketed) {
- tmp++;
- title = strdup(tmp);
- *(title + strlen(title) - 1) = '\0';
- } else {
- title = strdup(tmp);
- }
- } else if (line->type == LT_MENUENTRY)
- title = strdup(line->elements[1].item);
- else
- return NULL;
- return title;
-}
-
-static int readFile(int fd, char ** bufPtr) {
- int alloced = 0, size = 0, i = 0;
- char * buf = NULL;
-
- do {
- size += i;
- if ((size + 1024) > alloced) {
- alloced += 4096;
- buf = realloc(buf, alloced + 1);
- }
- } while ((i = read(fd, buf + size, 1024)) > 0);
-
- if (i < 0) {
- fprintf(stderr, _("error reading input: %s\n"), strerror(errno));
- free(buf);
- return 1;
- }
-
- buf = realloc(buf, size + 2);
- if (size == 0)
- buf[size++] = '\n';
- else
- if (buf[size - 1] != '\n')
- buf[size++] = '\n';
- buf[size] = '\0';
-
- *bufPtr = buf;
-
- return 0;
-}
-
-static void lineInit(struct singleLine * line) {
- line->indent = NULL;
- line->elements = NULL;
- line->numElements = 0;
- line->next = NULL;
-}
-
-struct singleLine * lineDup(struct singleLine * line) {
- struct singleLine * newLine = malloc(sizeof(*newLine));
-
- newLine->indent = strdup(line->indent);
- newLine->next = NULL;
- newLine->type = line->type;
- newLine->numElements = line->numElements;
- newLine->elements = malloc(sizeof(*newLine->elements) *
- newLine->numElements);
-
- for (int i = 0; i < newLine->numElements; i++) {
- newLine->elements[i].indent = strdup(line->elements[i].indent);
- newLine->elements[i].item = strdup(line->elements[i].item);
- }
-
- return newLine;
-}
-
-static void lineFree(struct singleLine * line) {
- if (line->indent) free(line->indent);
-
- for (int i = 0; i < line->numElements; i++) {
- free(line->elements[i].item);
- free(line->elements[i].indent);
- }
-
- if (line->elements) free(line->elements);
- lineInit(line);
+static char *extractTitle(struct grubConfig *cfg, struct singleLine *line)
+{
+ /* bracketed title... let's extract it */
+ char *title = NULL;
+ if (line->type == LT_TITLE) {
+ char *tmp = line->elements[cfg->cfi->titlePosition].item;
+ if (cfg->cfi->titleBracketed) {
+ tmp++;
+ title = strdup(tmp);
+ *(title + strlen(title) - 1) = '\0';
+ } else {
+ title = strdup(tmp);
+ }
+ } else if (line->type == LT_MENUENTRY)
+ title = strdup(line->elements[1].item);
+ else
+ return NULL;
+ return title;
}
-static int lineWrite(FILE * out, struct singleLine * line,
- struct configFileInfo * cfi) {
- if (fprintf(out, "%s", line->indent) == -1) return -1;
+static int readFile(int fd, char **bufPtr)
+{
+ int alloced = 0, size = 0, i = 0;
+ char *buf = NULL;
- for (int i = 0; i < line->numElements; i++) {
- /* Need to handle this, because we strip the quotes from
- * menuentry when read it. */
- if (line->type == LT_MENUENTRY && i == 1) {
- if(!isquote(*line->elements[i].item)) {
- int substring = 0;
- /* If the line contains nested quotes, we did not strip
- * the "interna" quotes and we must use the right quotes
- * again when writing the updated file. */
- for (int j = i; j < line->numElements; j++) {
- if (strchr(line->elements[i].item, '\'') != NULL) {
- substring = 1;
- fprintf(out, "\"%s\"", line->elements[i].item);
- break;
- }
+ do {
+ size += i;
+ if ((size + 1024) > alloced) {
+ alloced += 4096;
+ buf = realloc(buf, alloced + 1);
}
- if (!substring)
- fprintf(out, "\'%s\'", line->elements[i].item);
- } else {
- fprintf(out, "%s", line->elements[i].item);
- }
- fprintf(out, "%s", line->elements[i].indent);
+ } while ((i = read(fd, buf + size, 1024)) > 0);
- continue;
+ if (i < 0) {
+ fprintf(stderr, _("error reading input: %s\n"),
+ strerror(errno));
+ free(buf);
+ return 1;
}
- if (i == 1 && line->type == LT_KERNELARGS && cfi->argsInQuotes)
- if (fputc('"', out) == EOF) return -1;
-
- if (fprintf(out, "%s", line->elements[i].item) == -1) return -1;
- if (i < line->numElements - 1)
- if (fprintf(out, "%s", line->elements[i].indent) == -1) return -1;
- }
+ buf = realloc(buf, size + 2);
+ if (size == 0)
+ buf[size++] = '\n';
+ else if (buf[size - 1] != '\n')
+ buf[size++] = '\n';
+ buf[size] = '\0';
- if (line->type == LT_KERNELARGS && cfi->argsInQuotes)
- if (fputc('"', out) == EOF) return -1;
+ *bufPtr = buf;
- if (fprintf(out, "\n") == -1) return -1;
-
- return 0;
+ return 0;
}
-/* we've guaranteed that the buffer ends w/ \n\0 */
-static int getNextLine(char ** bufPtr, struct singleLine * line,
- struct configFileInfo * cfi) {
- char * end;
- char * start = *bufPtr;
- char * chptr;
- int elementsAlloced = 0;
- struct lineElement * element;
- int first = 1;
-
- lineFree(line);
-
- end = strchr(start, '\n');
- *end = '\0';
- *bufPtr = end + 1;
-
- for (chptr = start; *chptr && isspace(*chptr); chptr++) ;
-
- line->indent = strndup(start, chptr - start);
- start = chptr;
-
- while (start < end) {
- /* we know !isspace(*start) */
+static void lineInit(struct singleLine *line)
+{
+ line->indent = NULL;
+ line->elements = NULL;
+ line->numElements = 0;
+ line->next = NULL;
+}
- if (elementsAlloced == line->numElements) {
- elementsAlloced += 5;
- line->elements = realloc(line->elements,
- sizeof(*line->elements) * elementsAlloced);
- }
+struct singleLine *lineDup(struct singleLine *line)
+{
+ struct singleLine *newLine = malloc(sizeof(*newLine));
- element = line->elements + line->numElements;
+ newLine->indent = strdup(line->indent);
+ newLine->next = NULL;
+ newLine->type = line->type;
+ newLine->numElements = line->numElements;
+ newLine->elements = malloc(sizeof(*newLine->elements) *
+ newLine->numElements);
- chptr = start;
- while (*chptr && !isspace(*chptr)) {
- if (first && *chptr == '=') break;
- chptr++;
+ for (int i = 0; i < newLine->numElements; i++) {
+ newLine->elements[i].indent = strdup(line->elements[i].indent);
+ newLine->elements[i].item = strdup(line->elements[i].item);
}
- element->item = strndup(start, chptr - start);
- start = chptr;
-
- /* lilo actually accepts the pathological case of append = " foo " */
- if (*start == '=')
- chptr = start + 1;
- else
- chptr = start;
-
- do {
- for (; *chptr && isspace(*chptr); chptr++);
- if (*chptr == '=')
- chptr = chptr + 1;
- } while (isspace(*chptr));
-
- element->indent = strndup(start, chptr - start);
- start = chptr;
- line->numElements++;
- first = 0;
- }
-
- if (!line->numElements)
- line->type = LT_WHITESPACE;
- else {
- line->type = getTypeByKeyword(line->elements[0].item, cfi);
- if (line->type == LT_UNKNOWN) {
- /* zipl does [title] instead of something reasonable like all
- * the other boot loaders. kind of ugly */
- if (cfi->titleBracketed && isBracketedTitle(line)) {
- line->type = LT_TITLE;
- }
-
- /* this is awkward, but we need to be able to handle keywords
- that begin with a # (specifically for #boot in grub.conf),
- but still make comments lines with no elements (everything
- stored in the indent */
- if (*line->elements[0].item == '#') {
- char * fullLine;
- int len;
-
- len = strlen(line->indent);
- for (int i = 0; i < line->numElements; i++)
- len += strlen(line->elements[i].item) +
- strlen(line->elements[i].indent);
+ return newLine;
+}
- fullLine = malloc(len + 1);
- strcpy(fullLine, line->indent);
+static void lineFree(struct singleLine *line)
+{
+ if (line->indent)
free(line->indent);
- line->indent = fullLine;
- for (int i = 0; i < line->numElements; i++) {
- strcat(fullLine, line->elements[i].item);
- strcat(fullLine, line->elements[i].indent);
- free(line->elements[i].item);
- free(line->elements[i].indent);
+ for (int i = 0; i < line->numElements; i++) {
+ free(line->elements[i].item);
+ free(line->elements[i].indent);
+ }
+
+ if (line->elements)
+ free(line->elements);
+ lineInit(line);
+}
+
+static int lineWrite(FILE * out, struct singleLine *line,
+ struct configFileInfo *cfi)
+{
+ if (fprintf(out, "%s", line->indent) == -1)
+ return -1;
+
+ for (int i = 0; i < line->numElements; i++) {
+ /* Need to handle this, because we strip the quotes from
+ * menuentry when read it. */
+ if (line->type == LT_MENUENTRY && i == 1) {
+ if (!isquote(*line->elements[i].item)) {
+ int substring = 0;
+ /* If the line contains nested quotes, we did
+ * not strip the "interna" quotes and we must
+ * use the right quotes again when writing
+ * the updated file. */
+ for (int j = i; j < line->numElements; j++) {
+ if (strchr(line->elements[i].item, '\'')
+ != NULL) {
+ substring = 1;
+ fprintf(out, "\"%s\"",
+ line->elements[i].item);
+ break;
+ }
+ }
+ if (!substring)
+ fprintf(out, "\'%s\'",
+ line->elements[i].item);
+ } else {
+ fprintf(out, "%s", line->elements[i].item);
+ }
+ fprintf(out, "%s", line->elements[i].indent);
+
+ continue;
+ }
+
+ if (i == 1 && line->type == LT_KERNELARGS && cfi->argsInQuotes)
+ if (fputc('"', out) == EOF)
+ return -1;
+
+ if (fprintf(out, "%s", line->elements[i].item) == -1)
+ return -1;
+ if (i < line->numElements - 1)
+ if (fprintf(out, "%s", line->elements[i].indent) == -1)
+ return -1;
+ }
+
+ if (line->type == LT_KERNELARGS && cfi->argsInQuotes)
+ if (fputc('"', out) == EOF)
+ return -1;
+
+ if (fprintf(out, "\n") == -1)
+ return -1;
+
+ return 0;
+}
+
+/* we've guaranteed that the buffer ends w/ \n\0 */
+static int getNextLine(char **bufPtr, struct singleLine *line,
+ struct configFileInfo *cfi)
+{
+ char *end;
+ char *start = *bufPtr;
+ char *chptr;
+ int elementsAlloced = 0;
+ struct lineElement *element;
+ int first = 1;
+
+ lineFree(line);
+
+ end = strchr(start, '\n');
+ *end = '\0';
+ *bufPtr = end + 1;
+
+ for (chptr = start; *chptr && isspace(*chptr); chptr++) ;
+
+ line->indent = strndup(start, chptr - start);
+ start = chptr;
+
+ while (start < end) {
+ /* we know !isspace(*start) */
+
+ if (elementsAlloced == line->numElements) {
+ elementsAlloced += 5;
+ line->elements = realloc(line->elements,
+ sizeof(*line->elements) *
+ elementsAlloced);
+ }
+
+ element = line->elements + line->numElements;
+
+ chptr = start;
+ while (*chptr && !isspace(*chptr)) {
+ if (first && *chptr == '=')
+ break;
+ chptr++;
}
+ element->item = strndup(start, chptr - start);
+ start = chptr;
+
+ /* lilo actually accepts the pathological case of
+ * append = " foo " */
+ if (*start == '=')
+ chptr = start + 1;
+ else
+ chptr = start;
+
+ do {
+ for (; *chptr && isspace(*chptr); chptr++) ;
+ if (*chptr == '=')
+ chptr = chptr + 1;
+ } while (isspace(*chptr));
+
+ element->indent = strndup(start, chptr - start);
+ start = chptr;
+
+ line->numElements++;
+ first = 0;
+ }
+ if (!line->numElements)
line->type = LT_WHITESPACE;
- line->numElements = 0;
- }
- } else {
- struct keywordTypes *kw;
-
- kw = getKeywordByType(line->type, cfi);
-
- /* space isn't the only separator, we need to split
- * elements up more
- */
- if (!isspace(kw->separatorChar)) {
- char indent[2] = "";
- indent[0] = kw->separatorChar;
- for (int i = 1; i < line->numElements; i++) {
- char *p;
- int numNewElements;
-
- numNewElements = 0;
- p = line->elements[i].item;
- while (*p != '\0') {
- if (*p == kw->separatorChar)
- numNewElements++;
- p++;
- }
- if (line->numElements + numNewElements >= elementsAlloced) {
- elementsAlloced += numNewElements + 5;
- line->elements = realloc(line->elements,
- sizeof(*line->elements) * elementsAlloced);
+ else {
+ line->type = getTypeByKeyword(line->elements[0].item, cfi);
+ if (line->type == LT_UNKNOWN) {
+ /* zipl does [title] instead of something reasonable
+ * like all the other boot loaders. kind of ugly */
+ if (cfi->titleBracketed && isBracketedTitle(line)) {
+ line->type = LT_TITLE;
}
- for (int j = line->numElements; j > i; j--) {
- line->elements[j + numNewElements] = line->elements[j];
- }
- line->numElements += numNewElements;
+ /* this is awkward, but we need to be able to handle
+ * keywords that begin with a # (specifically for
+ * #boot in grub.conf), but still make comments lines
+ * with no elements (everything stored in the indent
+ */
+ if (*line->elements[0].item == '#') {
+ char *fullLine;
+ int len;
+
+ len = strlen(line->indent);
+ for (int i = 0; i < line->numElements; i++)
+ len += strlen(line->elements[i].item) +
+ strlen(line->elements[i].indent);
- p = line->elements[i].item;
- while (*p != '\0') {
+ fullLine = malloc(len + 1);
+ strcpy(fullLine, line->indent);
+ free(line->indent);
+ line->indent = fullLine;
- while (*p != kw->separatorChar && *p != '\0') p++;
- if (*p == '\0') {
- break;
+ for (int i = 0; i < line->numElements; i++) {
+ strcat(fullLine,
+ line->elements[i].item);
+ strcat(fullLine,
+ line->elements[i].indent);
+ free(line->elements[i].item);
+ free(line->elements[i].indent);
}
- line->elements[i + 1].indent = line->elements[i].indent;
- line->elements[i].indent = strdup(indent);
- *p++ = '\0';
- i++;
- line->elements[i].item = strdup(p);
+ line->type = LT_WHITESPACE;
+ line->numElements = 0;
+ }
+ } else {
+ struct keywordTypes *kw;
+
+ kw = getKeywordByType(line->type, cfi);
+
+ /* space isn't the only separator, we need to split
+ * elements up more
+ */
+ if (!isspace(kw->separatorChar)) {
+ char indent[2] = "";
+ indent[0] = kw->separatorChar;
+ for (int i = 1; i < line->numElements; i++) {
+ char *p;
+ int numNewElements;
+
+ numNewElements = 0;
+ p = line->elements[i].item;
+ while (*p != '\0') {
+ if (*p == kw->separatorChar)
+ numNewElements++;
+ p++;
+ }
+ if (line->numElements +
+ numNewElements >= elementsAlloced) {
+ elementsAlloced +=
+ numNewElements + 5;
+ line->elements =
+ realloc(line->elements,
+ sizeof(*line->
+ elements) *
+ elementsAlloced);
+ }
+
+ for (int j = line->numElements; j > i;
+ j--) {
+ line->elements[j +
+ numNewElements] =
+ line->elements[j];
+ }
+ line->numElements += numNewElements;
+
+ p = line->elements[i].item;
+ while (*p != '\0') {
+
+ while (*p != kw->separatorChar
+ && *p != '\0')
+ p++;
+ if (*p == '\0') {
+ break;
+ }
+
+ line->elements[i + 1].indent =
+ line->elements[i].indent;
+ line->elements[i].indent =
+ strdup(indent);
+ *p++ = '\0';
+ i++;
+ line->elements[i].item =
+ strdup(p);
+ }
+ }
}
- }
}
}
- }
- return 0;
+ return 0;
}
static int isnumber(const char *s)
{
- int i;
- for (i = 0; s[i] != '\0'; i++)
- if (s[i] < '0' || s[i] > '9')
- return 0;
- return i;
+ int i;
+ for (i = 0; s[i] != '\0'; i++)
+ if (s[i] < '0' || s[i] > '9')
+ return 0;
+ return i;
}
-static struct grubConfig * readConfig(const char * inName,
- struct configFileInfo * cfi) {
- int in;
- char * incoming = NULL, * head;
- int rc;
- int sawEntry = 0;
- int movedLine = 0;
- struct grubConfig * cfg;
- struct singleLine * last = NULL, * line, * defaultLine = NULL;
- char * end;
- struct singleEntry * entry = NULL;
- int len;
- char * buf;
-
- if (inName == NULL) {
- printf("Could not find bootloader configuration\n");
- exit(1);
- } else if (!strcmp(inName, "-")) {
- in = 0;
- } else {
- if ((in = open(inName, O_RDONLY)) < 0) {
- fprintf(stderr, _("error opening %s for read: %s\n"),
- inName, strerror(errno));
- return NULL;
- }
- }
-
- rc = readFile(in, &incoming);
- close(in);
- if (rc) return NULL;
-
- head = incoming;
- cfg = malloc(sizeof(*cfg));
- cfg->primaryIndent = strdup("");
- cfg->secondaryIndent = strdup("\t");
- cfg->flags = GRUB_CONFIG_NO_DEFAULT;
- cfg->cfi = cfi;
- cfg->theLines = NULL;
- cfg->entries = NULL;
- cfg->fallbackImage = 0;
-
- /* copy everything we have */
- while (*head) {
- line = malloc(sizeof(*line));
- lineInit(line);
-
- if (getNextLine(&head, line, cfi)) {
- free(line);
- /* XXX memory leak of everything in cfg */
- return NULL;
- }
-
- if (!sawEntry && line->numElements) {
- free(cfg->primaryIndent);
- cfg->primaryIndent = strdup(line->indent);
- } else if (line->numElements) {
- free(cfg->secondaryIndent);
- cfg->secondaryIndent = strdup(line->indent);
- }
-
- if (isEntryStart(line, cfi) || (cfg->entries && !sawEntry)) {
- sawEntry = 1;
- if (!entry) {
- cfg->entries = malloc(sizeof(*entry));
- entry = cfg->entries;
- } else {
- entry->next = malloc(sizeof(*entry));
- entry = entry->next;
- }
-
- entry->skip = 0;
- entry->multiboot = 0;
- entry->lines = NULL;
- entry->next = NULL;
+static struct grubConfig *readConfig(const char *inName,
+ struct configFileInfo *cfi)
+{
+ int in;
+ char *incoming = NULL, *head;
+ int rc;
+ int sawEntry = 0;
+ int movedLine = 0;
+ struct grubConfig *cfg;
+ struct singleLine *last = NULL, *line, *defaultLine = NULL;
+ char *end;
+ struct singleEntry *entry = NULL;
+ int len;
+ char *buf;
+
+ if (inName == NULL) {
+ printf("Could not find bootloader configuration\n");
+ exit(1);
+ } else if (!strcmp(inName, "-")) {
+ in = 0;
+ } else {
+ if ((in = open(inName, O_RDONLY)) < 0) {
+ fprintf(stderr, _("error opening %s for read: %s\n"),
+ inName, strerror(errno));
+ return NULL;
+ }
}
- if (line->type == LT_SET_VARIABLE) {
- dbgPrintf("found 'set' command (%d elements): ", line->numElements);
- dbgPrintf("%s", line->indent);
- for (int i = 0; i < line->numElements; i++)
- dbgPrintf("\"%s\"%s", line->elements[i].item, line->elements[i].indent);
- dbgPrintf("\n");
- struct keywordTypes *kwType = getKeywordByType(LT_DEFAULT, cfi);
- if (kwType && line->numElements == 3 &&
- !strcmp(line->elements[1].item, kwType->key) &&
- !is_special_grub2_variable(line->elements[2].item)) {
- dbgPrintf("Line sets default config\n");
- cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;
- defaultLine = line;
- }
-
- } else if (iskernel(line->type)) {
- /* if by some freak chance this is multiboot and the "module"
- * lines came earlier in the template, make sure to use LT_HYPER
- * instead of LT_KERNEL now
- */
- if (entry && entry->multiboot)
- line->type = LT_HYPER;
-
- } else if (line->type == LT_MBMODULE) {
- /* go back and fix the LT_KERNEL line to indicate LT_HYPER
- * instead, now that we know this is a multiboot entry.
- * This only applies to grub, but that's the only place we
- * should find LT_MBMODULE lines anyway.
- */
- for (struct singleLine *l = entry->lines; l; l = l->next) {
- if (l->type == LT_HYPER)
- break;
- else if (iskernel(l->type)) {
- l->type = LT_HYPER;
- break;
- }
- }
- entry->multiboot = 1;
-
- } else if (line->type == LT_HYPER) {
- entry->multiboot = 1;
-
- } else if (line->type == LT_FALLBACK && line->numElements == 2) {
- cfg->fallbackImage = strtol(line->elements[1].item, &end, 10);
- if (*end) cfg->fallbackImage = -1;
-
- } else if ((line->type == LT_DEFAULT && cfi->defaultIsUnquoted) ||
- (line->type == LT_TITLE && line->numElements > 1)) {
- /* make the title/default a single argument (undoing our parsing) */
- len = 0;
- for (int i = 1; i < line->numElements; i++) {
- len += strlen(line->elements[i].item);
- len += strlen(line->elements[i].indent);
- }
- buf = malloc(len + 1);
- *buf = '\0';
-
- for (int i = 1; i < line->numElements; i++) {
- strcat(buf, line->elements[i].item);
- free(line->elements[i].item);
-
- if ((i + 1) != line->numElements) {
- strcat(buf, line->elements[i].indent);
- free(line->elements[i].indent);
- }
- }
-
- line->elements[1].indent =
- line->elements[line->numElements - 1].indent;
- line->elements[1].item = buf;
- line->numElements = 2;
- } else if (line->type == LT_MENUENTRY && line->numElements > 3) {
- /* let --remove-kernel="TITLE=what" work */
- len = 0;
- char *extras;
- char *title;
- /* initially unseen value */
- char quote_char = '\0';
-
- for (int i = 1; i < line->numElements; i++) {
- len += strlen(line->elements[i].item);
- len += strlen(line->elements[i].indent);
- }
- buf = malloc(len + 1);
- *buf = '\0';
-
- /* allocate mem for extra flags. */
- extras = malloc(len + 1);
- *extras = '\0';
-
- /* get title. */
- for (int i = 0; i < line->numElements; i++) {
- if (!strcmp(line->elements[i].item, "menuentry"))
- continue;
- if (isquote(*line->elements[i].item) && quote_char == '\0') {
- /* ensure we properly pair off quotes */
- quote_char = *line->elements[i].item;
- title = line->elements[i].item + 1;
- } else {
- title = line->elements[i].item;
+ rc = readFile(in, &incoming);
+ close(in);
+ if (rc)
+ return NULL;
+
+ head = incoming;
+ cfg = malloc(sizeof(*cfg));
+ cfg->primaryIndent = strdup("");
+ cfg->secondaryIndent = strdup("\t");
+ cfg->flags = GRUB_CONFIG_NO_DEFAULT;
+ cfg->cfi = cfi;
+ cfg->theLines = NULL;
+ cfg->entries = NULL;
+ cfg->fallbackImage = 0;
+
+ /* copy everything we have */
+ while (*head) {
+ line = malloc(sizeof(*line));
+ lineInit(line);
+
+ if (getNextLine(&head, line, cfi)) {
+ free(line);
+ /* XXX memory leak of everything in cfg */
+ return NULL;
}
- len = strlen(title);
- if (title[len-1] == quote_char) {
- strncat(buf, title,len-1);
- break;
- } else {
- strcat(buf, title);
- strcat(buf, line->elements[i].indent);
+ if (!sawEntry && line->numElements) {
+ free(cfg->primaryIndent);
+ cfg->primaryIndent = strdup(line->indent);
+ } else if (line->numElements) {
+ free(cfg->secondaryIndent);
+ cfg->secondaryIndent = strdup(line->indent);
}
- }
-
- /* get extras */
- int count = 0;
- quote_char = '\0';
- for (int i = 0; i < line->numElements; i++) {
- if (count >= 2) {
- strcat(extras, line->elements[i].item);
- strcat(extras, line->elements[i].indent);
- }
-
- if (!strcmp(line->elements[i].item, "menuentry"))
- continue;
- /* count ' or ", there should be two in menuentry line. */
- if (isquote(*line->elements[i].item) && quote_char == '\0') {
- /* ensure we properly pair off quotes */
- quote_char = *line->elements[i].item;
- count++;
+ if (isEntryStart(line, cfi) || (cfg->entries && !sawEntry)) {
+ sawEntry = 1;
+ if (!entry) {
+ cfg->entries = malloc(sizeof(*entry));
+ entry = cfg->entries;
+ } else {
+ entry->next = malloc(sizeof(*entry));
+ entry = entry->next;
+ }
+
+ entry->skip = 0;
+ entry->multiboot = 0;
+ entry->lines = NULL;
+ entry->next = NULL;
}
- len = strlen(line->elements[i].item);
-
- if (line->elements[i].item[len -1] == quote_char)
- count++;
-
- /* ok, we get the final ' or ", others are extras. */
- }
- line->elements[1].indent =
- line->elements[line->numElements - 2].indent;
- line->elements[1].item = buf;
- line->elements[2].indent =
- line->elements[line->numElements - 2].indent;
- line->elements[2].item = extras;
- line->numElements = 3;
- } else if (line->type == LT_KERNELARGS && cfi->argsInQuotes) {
- /* Strip off any " which may be present; they'll be put back
- on write. This is one of the few (the only?) places that grubby
- canonicalizes the output */
-
- if (line->numElements >= 2) {
- int last, len;
-
- if (isquote(*line->elements[1].item))
- memmove(line->elements[1].item, line->elements[1].item + 1,
- strlen(line->elements[1].item + 1) + 1);
-
- last = line->numElements - 1;
- len = strlen(line->elements[last].item) - 1;
- if (isquote(line->elements[last].item[len]))
- line->elements[last].item[len] = '\0';
- }
- }
+ if (line->type == LT_SET_VARIABLE) {
+ dbgPrintf("found 'set' command (%d elements): ",
+ line->numElements);
+ dbgPrintf("%s", line->indent);
+ for (int i = 0; i < line->numElements; i++)
+ dbgPrintf("\"%s\"%s", line->elements[i].item,
+ line->elements[i].indent);
+ dbgPrintf("\n");
+ struct keywordTypes *kwType =
+ getKeywordByType(LT_DEFAULT, cfi);
+ if (kwType && line->numElements == 3
+ && !strcmp(line->elements[1].item, kwType->key)
+ && !is_special_grub2_variable(line->elements[2].
+ item)) {
+ dbgPrintf("Line sets default config\n");
+ cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;
+ defaultLine = line;
+ }
+
+ } else if (iskernel(line->type)) {
+ /* if by some freak chance this is multiboot and the
+ * "module" lines came earlier in the template, make
+ * sure to use LT_HYPER instead of LT_KERNEL now
+ */
+ if (entry && entry->multiboot)
+ line->type = LT_HYPER;
+
+ } else if (line->type == LT_MBMODULE) {
+ /* go back and fix the LT_KERNEL line to indicate
+ * LT_HYPER instead, now that we know this is a
+ * multiboot entry. This only applies to grub, but
+ * that's the only place we should find LT_MBMODULE
+ * lines anyway.
+ */
+ for (struct singleLine * l = entry->lines; l;
+ l = l->next) {
+ if (l->type == LT_HYPER)
+ break;
+ else if (iskernel(l->type)) {
+ l->type = LT_HYPER;
+ break;
+ }
+ }
+ entry->multiboot = 1;
+
+ } else if (line->type == LT_HYPER) {
+ entry->multiboot = 1;
+
+ } else if (line->type == LT_FALLBACK && line->numElements == 2) {
+ cfg->fallbackImage =
+ strtol(line->elements[1].item, &end, 10);
+ if (*end)
+ cfg->fallbackImage = -1;
+
+ } else if ((line->type == LT_DEFAULT && cfi->defaultIsUnquoted)
+ || (line->type == LT_TITLE
+ && line->numElements > 1)) {
+ /* make the title/default a single argument (undoing
+ * our parsing) */
+ len = 0;
+ for (int i = 1; i < line->numElements; i++) {
+ len += strlen(line->elements[i].item);
+ len += strlen(line->elements[i].indent);
+ }
+ buf = malloc(len + 1);
+ *buf = '\0';
+
+ for (int i = 1; i < line->numElements; i++) {
+ strcat(buf, line->elements[i].item);
+ free(line->elements[i].item);
+
+ if ((i + 1) != line->numElements) {
+ strcat(buf, line->elements[i].indent);
+ free(line->elements[i].indent);
+ }
+ }
+
+ line->elements[1].indent =
+ line->elements[line->numElements - 1].indent;
+ line->elements[1].item = buf;
+ line->numElements = 2;
+ } else if (line->type == LT_MENUENTRY && line->numElements > 3) {
+ /* let --remove-kernel="TITLE=what" work */
+ len = 0;
+ char *extras;
+ char *title;
+ /* initially unseen value */
+ char quote_char = '\0';
+
+ for (int i = 1; i < line->numElements; i++) {
+ len += strlen(line->elements[i].item);
+ len += strlen(line->elements[i].indent);
+ }
+ buf = malloc(len + 1);
+ *buf = '\0';
+
+ /* allocate mem for extra flags. */
+ extras = malloc(len + 1);
+ *extras = '\0';
+
+ /* get title. */
+ for (int i = 0; i < line->numElements; i++) {
+ if (!strcmp
+ (line->elements[i].item, "menuentry"))
+ continue;
+ if (isquote(*line->elements[i].item)
+ && quote_char == '\0') {
+ /* ensure we properly pair off quotes */
+ quote_char = *line->elements[i].item;
+ title = line->elements[i].item + 1;
+ } else {
+ title = line->elements[i].item;
+ }
+
+ len = strlen(title);
+ if (title[len - 1] == quote_char) {
+ strncat(buf, title, len - 1);
+ break;
+ } else {
+ strcat(buf, title);
+ strcat(buf, line->elements[i].indent);
+ }
+ }
+
+ /* get extras */
+ int count = 0;
+ quote_char = '\0';
+ for (int i = 0; i < line->numElements; i++) {
+ if (count >= 2) {
+ strcat(extras, line->elements[i].item);
+ strcat(extras,
+ line->elements[i].indent);
+ }
+
+ if (!strcmp
+ (line->elements[i].item, "menuentry"))
+ continue;
+
+ /* count ' or ", there should be two in menuentry line. */
+ if (isquote(*line->elements[i].item)
+ && quote_char == '\0') {
+ /* ensure we properly pair off quotes */
+ quote_char = *line->elements[i].item;
+ count++;
+ }
+
+ len = strlen(line->elements[i].item);
+
+ if (line->elements[i].item[len - 1] ==
+ quote_char)
+ count++;
+
+ /* ok, we get the final ' or ", others are extras. */
+ }
+ line->elements[1].indent =
+ line->elements[line->numElements - 2].indent;
+ line->elements[1].item = buf;
+ line->elements[2].indent =
+ line->elements[line->numElements - 2].indent;
+ line->elements[2].item = extras;
+ line->numElements = 3;
+ } else if (line->type == LT_KERNELARGS && cfi->argsInQuotes) {
+ /* Strip off any " which may be present; they'll be
+ * put back on write. This is one of the few (the
+ * only?) places that grubby canonicalizes the output
+ */
+ if (line->numElements >= 2) {
+ int last, len;
+
+ if (isquote(*line->elements[1].item))
+ memmove(line->elements[1].item,
+ line->elements[1].item + 1,
+ strlen(line->elements[1].item +
+ 1) + 1);
+
+ last = line->numElements - 1;
+ len = strlen(line->elements[last].item) - 1;
+ if (isquote(line->elements[last].item[len]))
+ line->elements[last].item[len] = '\0';
+ }
+ }
- if (line->type == LT_DEFAULT && line->numElements == 2) {
- cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;
- defaultLine = line;
- }
+ if (line->type == LT_DEFAULT && line->numElements == 2) {
+ cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;
+ defaultLine = line;
+ }
- /* If we find a generic config option which should live at the
- top of the file, move it there. Old versions of grubby were
- probably responsible for putting new images in the wrong
- place in front of it anyway. */
- if (sawEntry && line->type == LT_GENERIC) {
- struct singleLine **l = &cfg->theLines;
- struct singleLine **last_nonws = &cfg->theLines;
- while (*l) {
- if ((*l)->type != LT_WHITESPACE)
- last_nonws = &((*l)->next);
- l = &((*l)->next);
+ /* If we find a generic config option which should live at the
+ top of the file, move it there. Old versions of grubby were
+ probably responsible for putting new images in the wrong
+ place in front of it anyway. */
+ if (sawEntry && line->type == LT_GENERIC) {
+ struct singleLine **l = &cfg->theLines;
+ struct singleLine **last_nonws = &cfg->theLines;
+ while (*l) {
+ if ((*l)->type != LT_WHITESPACE)
+ last_nonws = &((*l)->next);
+ l = &((*l)->next);
+ }
+ line->next = *last_nonws;
+ *last_nonws = line;
+ movedLine = 1;
+ continue; /* without setting 'last' */
}
- line->next = *last_nonws;
- *last_nonws = line;
- movedLine = 1;
- continue; /* without setting 'last' */
- }
- /* If a second line of whitespace happens after a generic option
- which was moved, drop it. */
- if (movedLine && line->type == LT_WHITESPACE && last->type == LT_WHITESPACE) {
- lineFree(line);
- free(line);
+ /* If a second line of whitespace happens after a generic
+ * option which was moved, drop it. */
+ if (movedLine && line->type == LT_WHITESPACE
+ && last->type == LT_WHITESPACE) {
+ lineFree(line);
+ free(line);
+ movedLine = 0;
+ continue;
+ }
movedLine = 0;
- continue;
- }
- movedLine = 0;
- if (sawEntry) {
- if (!entry->lines)
- entry->lines = line;
- else
- last->next = line;
- dbgPrintf("readConfig added %s to %p\n", getKeyByType(line->type, cfi), entry);
+ if (sawEntry) {
+ if (!entry->lines)
+ entry->lines = line;
+ else
+ last->next = line;
+ dbgPrintf("readConfig added %s to %p\n",
+ getKeyByType(line->type, cfi), entry);
+
+ /* we could have seen this outside of an entry... if
+ * so, we ignore it like any other line we don't grok
+ */
+ if (line->type == LT_ENTRY_END && sawEntry)
+ sawEntry = 0;
+ } else {
+ if (!cfg->theLines)
+ cfg->theLines = line;
+ else
+ last->next = line;
+ dbgPrintf("readConfig added %s to cfg\n",
+ getKeyByType(line->type, cfi));
+ }
- /* we could have seen this outside of an entry... if so, we
- * ignore it like any other line we don't grok */
- if (line->type == LT_ENTRY_END && sawEntry)
- sawEntry = 0;
- } else {
- if (!cfg->theLines)
- cfg->theLines = line;
- else
- last->next = line;
- dbgPrintf("readConfig added %s to cfg\n", getKeyByType(line->type, cfi));
+ last = line;
}
- last = line;
- }
+ free(incoming);
+
+ dbgPrintf("defaultLine is %s\n", defaultLine ? "set" : "unset");
+ if (defaultLine) {
+ if (defaultLine->numElements > 2 &&
+ cfi->defaultSupportSaved &&
+ !strncmp(defaultLine->elements[2].item,
+ "\"${saved_entry}\"", 16)) {
+ cfg->cfi->defaultIsSaved = 1;
+ cfg->defaultImage = DEFAULT_SAVED_GRUB2;
+ if (cfg->cfi->getEnv) {
+ char *defTitle =
+ cfi->getEnv(cfg->cfi, "saved_entry");
+ if (defTitle) {
+ int index = 0;
+ if (isnumber(defTitle)) {
+ index = atoi(defTitle);
+ entry =
+ findEntryByIndex(cfg,
+ index);
+ } else {
+ entry =
+ findEntryByTitle(cfg,
+ defTitle,
+ &index);
+ }
+ if (entry)
+ cfg->defaultImage = index;
+ }
+ }
+ } else if (cfi->defaultIsVariable) {
+ char *value = defaultLine->elements[2].item;
+ while (*value && (*value == '"' || *value == '\'' ||
+ *value == ' ' || *value == '\t'))
+ value++;
+ cfg->defaultImage = strtol(value, &end, 10);
+ while (*end && (*end == '"' || *end == '\'' ||
+ *end == ' ' || *end == '\t'))
+ end++;
+ if (*end)
+ cfg->defaultImage = -1;
+ } else if (cfi->defaultSupportSaved &&
+ !strncmp(defaultLine->elements[1].item, "saved",
+ 5)) {
+ cfg->defaultImage = DEFAULT_SAVED;
+ } else if (cfi->defaultIsIndex) {
+ cfg->defaultImage =
+ strtol(defaultLine->elements[1].item, &end, 10);
+ if (*end)
+ cfg->defaultImage = -1;
+ } else if (defaultLine->numElements >= 2) {
+ int i = 0;
+ while ((entry = findEntryByIndex(cfg, i))) {
+ for (line = entry->lines; line;
+ line = line->next)
+ if (line->type == LT_TITLE)
+ break;
- free(incoming);
+ if (!cfi->titleBracketed) {
+ if (line && (line->numElements >= 2) &&
+ !strcmp(defaultLine->elements[1].
+ item,
+ line->elements[1].item))
+ break;
+ } else if (line) {
+ if (!strcmp
+ (defaultLine->elements[1].item,
+ extractTitle(cfg, line)))
+ break;
+ }
+ i++;
+ entry = NULL;
+ }
- dbgPrintf("defaultLine is %s\n", defaultLine ? "set" : "unset");
- if (defaultLine) {
- if (defaultLine->numElements > 2 &&
- cfi->defaultSupportSaved &&
- !strncmp(defaultLine->elements[2].item,"\"${saved_entry}\"", 16)) {
- cfg->cfi->defaultIsSaved = 1;
- cfg->defaultImage = DEFAULT_SAVED_GRUB2;
- if (cfg->cfi->getEnv) {
- char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry");
- if (defTitle) {
+ if (entry) {
+ cfg->defaultImage = i;
+ } else {
+ cfg->defaultImage = -1;
+ }
+ }
+ } else if (cfg->cfi->defaultIsSaved && cfg->cfi->getEnv) {
+ char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry");
+ if (defTitle) {
int index = 0;
if (isnumber(defTitle)) {
- index = atoi(defTitle);
- entry = findEntryByIndex(cfg, index);
+ index = atoi(defTitle);
+ entry = findEntryByIndex(cfg, index);
} else {
- entry = findEntryByTitle(cfg, defTitle, &index);
+ entry = findEntryByTitle(cfg, defTitle, &index);
}
if (entry)
- cfg->defaultImage = index;
- }
+ cfg->defaultImage = index;
}
- } else if (cfi->defaultIsVariable) {
- char *value = defaultLine->elements[2].item;
- while (*value && (*value == '"' || *value == '\'' ||
- *value == ' ' || *value == '\t'))
- value++;
- cfg->defaultImage = strtol(value, &end, 10);
- while (*end && (*end == '"' || *end == '\'' ||
- *end == ' ' || *end == '\t'))
- end++;
- if (*end) cfg->defaultImage = -1;
- } else if (cfi->defaultSupportSaved &&
- !strncmp(defaultLine->elements[1].item, "saved", 5)) {
- cfg->defaultImage = DEFAULT_SAVED;
- } else if (cfi->defaultIsIndex) {
- cfg->defaultImage = strtol(defaultLine->elements[1].item, &end, 10);
- if (*end) cfg->defaultImage = -1;
- } else if (defaultLine->numElements >= 2) {
- int i = 0;
- while ((entry = findEntryByIndex(cfg, i))) {
- for (line = entry->lines; line; line = line->next)
- if (line->type == LT_TITLE) break;
-
- if (!cfi->titleBracketed) {
- if (line && (line->numElements >= 2) &&
- !strcmp(defaultLine->elements[1].item,
- line->elements[1].item)) break;
- } else if (line) {
- if (!strcmp(defaultLine->elements[1].item,
- extractTitle(cfg, line))) break;
- }
- i++;
- entry = NULL;
- }
-
- if (entry){
- cfg->defaultImage = i;
- }else{
- cfg->defaultImage = -1;
- }
- }
- } else if (cfg->cfi->defaultIsSaved && cfg->cfi->getEnv) {
- char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry");
- if (defTitle) {
- int index = 0;
- if (isnumber(defTitle)) {
- index = atoi(defTitle);
- entry = findEntryByIndex(cfg, index);
- } else {
- entry = findEntryByTitle(cfg, defTitle, &index);
- }
- if (entry)
- cfg->defaultImage = index;
- }
- } else {
- cfg->defaultImage = 0;
- }
-
- return cfg;
-}
-
-static void writeDefault(FILE * out, char * indent,
- char * separator, struct grubConfig * cfg) {
- struct singleEntry * entry;
- struct singleLine * line;
- int i;
-
- if (!cfg->defaultImage && cfg->flags == GRUB_CONFIG_NO_DEFAULT)
- return;
-
- if (cfg->defaultImage == DEFAULT_SAVED)
- fprintf(out, "%sdefault%ssaved\n", indent, separator);
- else if (cfg->cfi->defaultIsSaved) {
- fprintf(out, "%sset default=\"${saved_entry}\"\n", indent);
- if (cfg->defaultImage >= 0 && cfg->cfi->setEnv) {
- char *title;
- entry = findEntryByIndex(cfg, cfg->defaultImage);
- line = getLineByType(LT_MENUENTRY, entry->lines);
- if (!line)
- line = getLineByType(LT_TITLE, entry->lines);
- if (line) {
- title = extractTitle(cfg, line);
- if (title)
- cfg->cfi->setEnv(cfg->cfi, "saved_entry", title);
- }
- }
- } else if (cfg->defaultImage > -1) {
- if (cfg->cfi->defaultIsIndex) {
- if (cfg->cfi->defaultIsVariable) {
- fprintf(out, "%sset default=\"%d\"\n", indent,
- cfg->defaultImage);
- } else {
- fprintf(out, "%sdefault%s%d\n", indent, separator,
- cfg->defaultImage);
- }
} else {
- int image = cfg->defaultImage;
+ cfg->defaultImage = 0;
+ }
- entry = cfg->entries;
- while (entry && entry->skip)
- entry = entry->next;
+ return cfg;
+}
- i = 0;
- while (entry && i < image) {
- entry = entry->next;
+static void writeDefault(FILE * out, char *indent,
+ char *separator, struct grubConfig *cfg)
+{
+ struct singleEntry *entry;
+ struct singleLine *line;
+ int i;
- while (entry && entry->skip)
- entry = entry->next;
- i++;
- }
-
- if (!entry)
+ if (!cfg->defaultImage && cfg->flags == GRUB_CONFIG_NO_DEFAULT)
return;
- line = getLineByType(LT_TITLE, entry->lines);
-
- if (line && line->numElements >= 2)
- fprintf(out, "%sdefault%s%s\n", indent, separator,
- line->elements[1].item);
- else if (line && (line->numElements == 1) &&
- cfg->cfi->titleBracketed) {
- char *title = extractTitle(cfg, line);
- if (title) {
- fprintf(out, "%sdefault%s%s\n", indent, separator, title);
- free(title);
+ if (cfg->defaultImage == DEFAULT_SAVED)
+ fprintf(out, "%sdefault%ssaved\n", indent, separator);
+ else if (cfg->cfi->defaultIsSaved) {
+ fprintf(out, "%sset default=\"${saved_entry}\"\n", indent);
+ if (cfg->defaultImage >= 0 && cfg->cfi->setEnv) {
+ char *title;
+ entry = findEntryByIndex(cfg, cfg->defaultImage);
+ line = getLineByType(LT_MENUENTRY, entry->lines);
+ if (!line)
+ line = getLineByType(LT_TITLE, entry->lines);
+ if (line) {
+ title = extractTitle(cfg, line);
+ if (title)
+ cfg->cfi->setEnv(cfg->cfi,
+ "saved_entry", title);
+ }
+ }
+ } else if (cfg->defaultImage > -1) {
+ if (cfg->cfi->defaultIsIndex) {
+ if (cfg->cfi->defaultIsVariable) {
+ fprintf(out, "%sset default=\"%d\"\n", indent,
+ cfg->defaultImage);
+ } else {
+ fprintf(out, "%sdefault%s%d\n", indent,
+ separator, cfg->defaultImage);
+ }
+ } else {
+ int image = cfg->defaultImage;
+
+ entry = cfg->entries;
+ while (entry && entry->skip)
+ entry = entry->next;
+
+ i = 0;
+ while (entry && i < image) {
+ entry = entry->next;
+
+ while (entry && entry->skip)
+ entry = entry->next;
+ i++;
+ }
+
+ if (!entry)
+ return;
+
+ line = getLineByType(LT_TITLE, entry->lines);
+
+ if (line && line->numElements >= 2)
+ fprintf(out, "%sdefault%s%s\n", indent,
+ separator, line->elements[1].item);
+ else if (line && (line->numElements == 1)
+ && cfg->cfi->titleBracketed) {
+ char *title = extractTitle(cfg, line);
+ if (title) {
+ fprintf(out, "%sdefault%s%s\n", indent,
+ separator, title);
+ free(title);
+ }
+ }
}
- }
}
- }
}
-static int writeConfig(struct grubConfig * cfg, char * outName,
- const char * prefix) {
- FILE * out;
- struct singleLine * line;
- struct singleEntry * entry;
- char * tmpOutName;
- int needs = MAIN_DEFAULT;
- struct stat sb;
- int i;
-
- if (!strcmp(outName, "-")) {
- out = stdout;
- tmpOutName = NULL;
- } else {
- if (!lstat(outName, &sb) && S_ISLNK(sb.st_mode)) {
- char * buf;
- int len = 256;
- int rc;
-
- /* most likely the symlink is relative, so change our
- directory to the dir of the symlink */
- char *dir = strdupa(outName);
- rc = chdir(dirname(dir));
- do {
- buf = alloca(len + 1);
- rc = readlink(basename(outName), buf, len);
- if (rc == len) len += 256;
- } while (rc == len);
-
- if (rc < 0) {
- fprintf(stderr, _("grubby: error readlink link %s: %s\n"),
- outName, strerror(errno));
- return 1;
- }
-
- outName = buf;
- outName[rc] = '\0';
- }
-
- tmpOutName = alloca(strlen(outName) + 2);
- sprintf(tmpOutName, "%s-", outName);
- out = fopen(tmpOutName, "w");
- if (!out) {
- fprintf(stderr, _("grubby: error creating %s: %s\n"), tmpOutName,
- strerror(errno));
- return 1;
- }
-
- if (!stat(outName, &sb)) {
- if (chmod(tmpOutName, sb.st_mode & ~(S_IFMT))) {
- fprintf(stderr, _("grubby: error setting perms on %s: %s\n"),
- tmpOutName, strerror(errno));
- fclose(out);
- unlink(tmpOutName);
- return 1;
- }
- }
- }
-
- line = cfg->theLines;
- struct keywordTypes *defaultKw = getKeywordByType(LT_DEFAULT, cfg->cfi);
- while (line) {
- if (line->type == LT_SET_VARIABLE && defaultKw &&
- line->numElements == 3 &&
- !strcmp(line->elements[1].item, defaultKw->key) &&
- !is_special_grub2_variable(line->elements[2].item)) {
- writeDefault(out, line->indent, line->elements[0].indent, cfg);
- needs &= ~MAIN_DEFAULT;
- } else if (line->type == LT_DEFAULT) {
- writeDefault(out, line->indent, line->elements[0].indent, cfg);
- needs &= ~MAIN_DEFAULT;
- } else if (line->type == LT_FALLBACK) {
- if (cfg->fallbackImage > -1)
- fprintf(out, "%s%s%s%d\n", line->indent,
- line->elements[0].item, line->elements[0].indent,
- cfg->fallbackImage);
+static int writeConfig(struct grubConfig *cfg, char *outName,
+ const char *prefix)
+{
+ FILE *out;
+ struct singleLine *line;
+ struct singleEntry *entry;
+ char *tmpOutName;
+ int needs = MAIN_DEFAULT;
+ struct stat sb;
+ int i;
+
+ if (!strcmp(outName, "-")) {
+ out = stdout;
+ tmpOutName = NULL;
} else {
- if (lineWrite(out, line, cfg->cfi) == -1) {
- fprintf(stderr, _("grubby: error writing %s: %s\n"),
- tmpOutName, strerror(errno));
- fclose(out);
- unlink(tmpOutName);
- return 1;
- }
+ if (!lstat(outName, &sb) && S_ISLNK(sb.st_mode)) {
+ char *buf;
+ int len = 256;
+ int rc;
+
+ /* most likely the symlink is relative, so change our
+ directory to the dir of the symlink */
+ char *dir = strdupa(outName);
+ rc = chdir(dirname(dir));
+ do {
+ buf = alloca(len + 1);
+ rc = readlink(basename(outName), buf, len);
+ if (rc == len)
+ len += 256;
+ } while (rc == len);
+
+ if (rc < 0) {
+ fprintf(stderr,
+ _
+ ("grubby: error readlink link %s: %s\n"),
+ outName, strerror(errno));
+ return 1;
+ }
+
+ outName = buf;
+ outName[rc] = '\0';
+ }
+
+ tmpOutName = alloca(strlen(outName) + 2);
+ sprintf(tmpOutName, "%s-", outName);
+ out = fopen(tmpOutName, "w");
+ if (!out) {
+ fprintf(stderr, _("grubby: error creating %s: %s\n"),
+ tmpOutName, strerror(errno));
+ return 1;
+ }
+
+ if (!stat(outName, &sb)) {
+ if (chmod(tmpOutName, sb.st_mode & ~(S_IFMT))) {
+ fprintf(stderr,
+ _
+ ("grubby: error setting perms on %s: %s\n"),
+ tmpOutName, strerror(errno));
+ fclose(out);
+ unlink(tmpOutName);
+ return 1;
+ }
+ }
}
- line = line->next;
- }
-
- if (needs & MAIN_DEFAULT) {
- writeDefault(out, cfg->primaryIndent, "=", cfg);
- needs &= ~MAIN_DEFAULT;
- }
-
- i = 0;
- while ((entry = findEntryByIndex(cfg, i++))) {
- if (entry->skip) continue;
-
- line = entry->lines;
+ line = cfg->theLines;
+ struct keywordTypes *defaultKw = getKeywordByType(LT_DEFAULT, cfg->cfi);
while (line) {
- if (lineWrite(out, line, cfg->cfi) == -1) {
- fprintf(stderr, _("grubby: error writing %s: %s\n"),
- tmpOutName, strerror(errno));
- fclose(out);
- unlink(tmpOutName);
- return 1;
- }
- line = line->next;
+ if (line->type == LT_SET_VARIABLE && defaultKw &&
+ line->numElements == 3 &&
+ !strcmp(line->elements[1].item, defaultKw->key) &&
+ !is_special_grub2_variable(line->elements[2].item)) {
+ writeDefault(out, line->indent,
+ line->elements[0].indent, cfg);
+ needs &= ~MAIN_DEFAULT;
+ } else if (line->type == LT_DEFAULT) {
+ writeDefault(out, line->indent,
+ line->elements[0].indent, cfg);
+ needs &= ~MAIN_DEFAULT;
+ } else if (line->type == LT_FALLBACK) {
+ if (cfg->fallbackImage > -1)
+ fprintf(out, "%s%s%s%d\n", line->indent,
+ line->elements[0].item,
+ line->elements[0].indent,
+ cfg->fallbackImage);
+ } else {
+ if (lineWrite(out, line, cfg->cfi) == -1) {
+ fprintf(stderr,
+ _("grubby: error writing %s: %s\n"),
+ tmpOutName, strerror(errno));
+ fclose(out);
+ unlink(tmpOutName);
+ return 1;
+ }
+ }
+
+ line = line->next;
+ }
+
+ if (needs & MAIN_DEFAULT) {
+ writeDefault(out, cfg->primaryIndent, "=", cfg);
+ needs &= ~MAIN_DEFAULT;
+ }
+
+ i = 0;
+ while ((entry = findEntryByIndex(cfg, i++))) {
+ if (entry->skip)
+ continue;
+
+ line = entry->lines;
+ while (line) {
+ if (lineWrite(out, line, cfg->cfi) == -1) {
+ fprintf(stderr,
+ _("grubby: error writing %s: %s\n"),
+ tmpOutName, strerror(errno));
+ fclose(out);
+ unlink(tmpOutName);
+ return 1;
+ }
+ line = line->next;
+ }
}
- }
- if (tmpOutName) {
- if (rename(tmpOutName, outName)) {
- fprintf(stderr, _("grubby: error moving %s to %s: %s\n"),
- tmpOutName, outName, strerror(errno));
- unlink(outName);
- return 1;
+ if (tmpOutName) {
+ if (rename(tmpOutName, outName)) {
+ fprintf(stderr,
+ _("grubby: error moving %s to %s: %s\n"),
+ tmpOutName, outName, strerror(errno));
+ unlink(outName);
+ return 1;
+ }
}
- }
- return 0;
+ return 0;
}
-static int numEntries(struct grubConfig *cfg) {
- int i = 0;
- struct singleEntry * entry;
+static int numEntries(struct grubConfig *cfg)
+{
+ int i = 0;
+ struct singleEntry *entry;
- entry = cfg->entries;
- while (entry) {
- if (!entry->skip)
- i++;
- entry = entry->next;
- }
- return i;
+ entry = cfg->entries;
+ while (entry) {
+ if (!entry->skip)
+ i++;
+ entry = entry->next;
+ }
+ return i;
}
static char *findDiskForRoot()
{
- int fd;
- char buf[65536];
- char *devname;
- char *chptr;
- int rc;
-
- if ((fd = open(_PATH_MOUNTED, O_RDONLY)) < 0) {
- fprintf(stderr, "grubby: failed to open %s: %s\n",
- _PATH_MOUNTED, strerror(errno));
- return NULL;
- }
-
- rc = read(fd, buf, sizeof(buf) - 1);
- if (rc <= 0) {
- fprintf(stderr, "grubby: failed to read %s: %s\n",
- _PATH_MOUNTED, strerror(errno));
- close(fd);
- return NULL;
- }
- close(fd);
- buf[rc] = '\0';
- chptr = buf;
-
- char *foundanswer = NULL;
-
- while (chptr && chptr != buf+rc) {
- devname = chptr;
-
- /*
- * The first column of a mtab entry is the device, but if the entry is a
- * special device it won't start with /, so move on to the next line.
- */
- if (*devname != '/') {
- chptr = strchr(chptr, '\n');
- if (chptr)
- chptr++;
- continue;
- }
-
- /* Seek to the next space */
- chptr = strchr(chptr, ' ');
- if (!chptr) {
- fprintf(stderr, "grubby: error parsing %s: %s\n",
- _PATH_MOUNTED, strerror(errno));
- return NULL;
- }
-
- /*
- * The second column of a mtab entry is the mount point, we are looking
- * for '/' obviously.
- */
- if (*(++chptr) == '/' && *(++chptr) == ' ') {
- /* remember the last / entry in mtab */
- foundanswer = devname;
- }
-
- /* Next line */
- chptr = strchr(chptr, '\n');
- if (chptr)
- chptr++;
- }
-
- /* Return the last / entry found */
- if (foundanswer) {
- chptr = strchr(foundanswer, ' ');
- *chptr = '\0';
- return strdup(foundanswer);
- }
-
- return NULL;
+ int fd;
+ char buf[65536];
+ char *devname;
+ char *chptr;
+ int rc;
+
+ if ((fd = open(_PATH_MOUNTED, O_RDONLY)) < 0) {
+ fprintf(stderr, "grubby: failed to open %s: %s\n",
+ _PATH_MOUNTED, strerror(errno));
+ return NULL;
+ }
+
+ rc = read(fd, buf, sizeof(buf) - 1);
+ if (rc <= 0) {
+ fprintf(stderr, "grubby: failed to read %s: %s\n",
+ _PATH_MOUNTED, strerror(errno));
+ close(fd);
+ return NULL;
+ }
+ close(fd);
+ buf[rc] = '\0';
+ chptr = buf;
+
+ char *foundanswer = NULL;
+
+ while (chptr && chptr != buf + rc) {
+ devname = chptr;
+
+ /*
+ * The first column of a mtab entry is the device, but if the
+ * entry is a special device it won't start with /, so move
+ * on to the next line.
+ */
+ if (*devname != '/') {
+ chptr = strchr(chptr, '\n');
+ if (chptr)
+ chptr++;
+ continue;
+ }
+
+ /* Seek to the next space */
+ chptr = strchr(chptr, ' ');
+ if (!chptr) {
+ fprintf(stderr, "grubby: error parsing %s: %s\n",
+ _PATH_MOUNTED, strerror(errno));
+ return NULL;
+ }
+
+ /*
+ * The second column of a mtab entry is the mount point, we
+ * are looking for '/' obviously.
+ */
+ if (*(++chptr) == '/' && *(++chptr) == ' ') {
+ /* remember the last / entry in mtab */
+ foundanswer = devname;
+ }
+
+ /* Next line */
+ chptr = strchr(chptr, '\n');
+ if (chptr)
+ chptr++;
+ }
+
+ /* Return the last / entry found */
+ if (foundanswer) {
+ chptr = strchr(foundanswer, ' ');
+ *chptr = '\0';
+ return strdup(foundanswer);
+ }
+
+ return NULL;
}
-void printEntry(struct singleEntry * entry, FILE *f) {
- int i;
- struct singleLine * line;
-
- for (line = entry->lines; line; line = line->next) {
- log_message(f, "DBG: %s", line->indent);
- for (i = 0; i < line->numElements; i++) {
- /* Need to handle this, because we strip the quotes from
- * menuentry when read it. */
- if (line->type == LT_MENUENTRY && i == 1) {
- if(!isquote(*line->elements[i].item))
- log_message(f, "\'%s\'", line->elements[i].item);
- else
- log_message(f, "%s", line->elements[i].item);
- log_message(f, "%s", line->elements[i].indent);
-
- continue;
- }
-
- log_message(f, "%s%s",
- line->elements[i].item, line->elements[i].indent);
+void printEntry(struct singleEntry *entry, FILE * f)
+{
+ int i;
+ struct singleLine *line;
+
+ for (line = entry->lines; line; line = line->next) {
+ log_message(f, "DBG: %s", line->indent);
+ for (i = 0; i < line->numElements; i++) {
+ /* Need to handle this, because we strip the quotes from
+ * menuentry when read it. */
+ if (line->type == LT_MENUENTRY && i == 1) {
+ if (!isquote(*line->elements[i].item))
+ log_message(f, "\'%s\'",
+ line->elements[i].item);
+ else
+ log_message(f, "%s",
+ line->elements[i].item);
+ log_message(f, "%s", line->elements[i].indent);
+
+ continue;
+ }
+
+ log_message(f, "%s%s",
+ line->elements[i].item,
+ line->elements[i].indent);
+ }
+ log_message(f, "\n");
}
- log_message(f, "\n");
- }
}
-void notSuitablePrintf(struct singleEntry * entry, int okay, const char *fmt, ...)
+void notSuitablePrintf(struct singleEntry *entry, int okay, const char *fmt,
+ ...)
{
- static int once;
- va_list argp, argq;
-
- va_start(argp, fmt);
-
- va_copy(argq, argp);
- if (!once) {
- log_time(NULL);
- log_message(NULL, "command line: %s\n", saved_command_line);
- }
- log_message(NULL, "DBG: Image entry %s: ", okay ? "succeeded" : "failed");
- log_vmessage(NULL, fmt, argq);
-
- printEntry(entry, NULL);
- va_end(argq);
-
- if (!debug) {
+ static int once;
+ va_list argp, argq;
+
+ va_start(argp, fmt);
+
+ va_copy(argq, argp);
+ if (!once) {
+ log_time(NULL);
+ log_message(NULL, "command line: %s\n", saved_command_line);
+ }
+ log_message(NULL, "DBG: Image entry %s: ",
+ okay ? "succeeded" : "failed");
+ log_vmessage(NULL, fmt, argq);
+
+ printEntry(entry, NULL);
+ va_end(argq);
+
+ if (!debug) {
+ once = 1;
+ va_end(argp);
+ return;
+ }
+
+ if (okay) {
+ va_end(argp);
+ return;
+ }
+
+ if (!once)
+ log_message(stderr, "DBG: command line: %s\n",
+ saved_command_line);
once = 1;
- va_end(argp);
- return;
- }
-
- if (okay) {
+ fprintf(stderr, "DBG: Image entry failed: ");
+ vfprintf(stderr, fmt, argp);
+ printEntry(entry, stderr);
va_end(argp);
- return;
- }
-
- if (!once)
- log_message(stderr, "DBG: command line: %s\n", saved_command_line);
- once = 1;
- fprintf(stderr, "DBG: Image entry failed: ");
- vfprintf(stderr, fmt, argp);
- printEntry(entry, stderr);
- va_end(argp);
}
#define beginswith(s, c) ((s) && (s)[0] == (c))
@@ -1867,281 +1993,318 @@ static int endswith(const char *s, char c)
return s[slen] == c;
}
-int suitableImage(struct singleEntry * entry, const char * bootPrefix,
- int skipRemoved, int flags) {
- struct singleLine * line;
- char * fullName;
- int i;
- char * dev;
- char * rootspec;
- char * rootdev;
+int suitableImage(struct singleEntry *entry, const char *bootPrefix,
+ int skipRemoved, int flags)
+{
+ struct singleLine *line;
+ char *fullName;
+ int i;
+ char *dev;
+ char *rootspec;
+ char *rootdev;
- if (skipRemoved && entry->skip) {
- notSuitablePrintf(entry, 0, "marked to skip\n");
- return 0;
- }
+ if (skipRemoved && entry->skip) {
+ notSuitablePrintf(entry, 0, "marked to skip\n");
+ return 0;
+ }
- line = getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
- if (!line) {
- notSuitablePrintf(entry, 0, "no line found\n");
- return 0;
- }
- if (line->numElements < 2) {
- notSuitablePrintf(entry, 0, "line has only %d elements\n",
- line->numElements);
- return 0;
- }
-
- if (flags & GRUBBY_BADIMAGE_OKAY) {
- notSuitablePrintf(entry, 1, "\n");
- return 1;
- }
-
- fullName = alloca(strlen(bootPrefix) +
- strlen(line->elements[1].item) + 1);
- rootspec = getRootSpecifier(line->elements[1].item);
- int rootspec_offset = rootspec ? strlen(rootspec) : 0;
- int hasslash = endswith(bootPrefix, '/') ||
- beginswith(line->elements[1].item + rootspec_offset, '/');
- sprintf(fullName, "%s%s%s", bootPrefix, hasslash ? "" : "/",
- line->elements[1].item + rootspec_offset);
- if (access(fullName, R_OK)) {
- notSuitablePrintf(entry, 0, "access to %s failed\n", fullName);
- return 0;
- }
- for (i = 2; i < line->numElements; i++)
- if (!strncasecmp(line->elements[i].item, "root=", 5)) break;
- if (i < line->numElements) {
- dev = line->elements[i].item + 5;
- } else {
- /* look for a lilo style LT_ROOT line */
- line = getLineByType(LT_ROOT, entry->lines);
-
- if (line && line->numElements >= 2) {
- dev = line->elements[1].item;
- } else {
- /* didn't succeed in finding a LT_ROOT, let's try LT_KERNELARGS.
- * grub+multiboot uses LT_MBMODULE for the args, so check that too.
- */
- line = getLineByType(LT_KERNELARGS|LT_MBMODULE, entry->lines);
-
- /* failed to find one */
- if (!line) {
+ line =
+ getLineByType(LT_KERNEL | LT_HYPER | LT_KERNEL_EFI | LT_KERNEL_16,
+ entry->lines);
+ if (!line) {
notSuitablePrintf(entry, 0, "no line found\n");
return 0;
- }
+ }
+ if (line->numElements < 2) {
+ notSuitablePrintf(entry, 0, "line has only %d elements\n",
+ line->numElements);
+ return 0;
+ }
+
+ if (flags & GRUBBY_BADIMAGE_OKAY) {
+ notSuitablePrintf(entry, 1, "\n");
+ return 1;
+ }
+
+ fullName = alloca(strlen(bootPrefix) +
+ strlen(line->elements[1].item) + 1);
+ rootspec = getRootSpecifier(line->elements[1].item);
+ int rootspec_offset = rootspec ? strlen(rootspec) : 0;
+ int hasslash = endswith(bootPrefix, '/') ||
+ beginswith(line->elements[1].item + rootspec_offset, '/');
+ sprintf(fullName, "%s%s%s", bootPrefix, hasslash ? "" : "/",
+ line->elements[1].item + rootspec_offset);
+ if (access(fullName, R_OK)) {
+ notSuitablePrintf(entry, 0, "access to %s failed\n", fullName);
+ return 0;
+ }
+ for (i = 2; i < line->numElements; i++)
+ if (!strncasecmp(line->elements[i].item, "root=", 5))
+ break;
+ if (i < line->numElements) {
+ dev = line->elements[i].item + 5;
+ } else {
+ /* look for a lilo style LT_ROOT line */
+ line = getLineByType(LT_ROOT, entry->lines);
+
+ if (line && line->numElements >= 2) {
+ dev = line->elements[1].item;
+ } else {
+ /* didn't succeed in finding a LT_ROOT, let's try
+ * LT_KERNELARGS. grub+multiboot uses LT_MBMODULE
+ * for the args, so check that too.
+ */
+ line =
+ getLineByType(LT_KERNELARGS | LT_MBMODULE,
+ entry->lines);
+
+ /* failed to find one */
+ if (!line) {
+ notSuitablePrintf(entry, 0, "no line found\n");
+ return 0;
+ }
- for (i = 1; i < line->numElements; i++)
- if (!strncasecmp(line->elements[i].item, "root=", 5)) break;
- if (i < line->numElements)
- dev = line->elements[i].item + 5;
- else {
- notSuitablePrintf(entry, 0, "no root= entry found\n");
- /* it failed too... can't find root= */
- return 0;
- }
+ for (i = 1; i < line->numElements; i++)
+ if (!strncasecmp
+ (line->elements[i].item, "root=", 5))
+ break;
+ if (i < line->numElements)
+ dev = line->elements[i].item + 5;
+ else {
+ notSuitablePrintf(entry, 0,
+ "no root= entry found\n");
+ /* it failed too... can't find root= */
+ return 0;
+ }
+ }
}
- }
- dev = getpathbyspec(dev);
- if (!getpathbyspec(dev)) {
- notSuitablePrintf(entry, 0, "can't find blkid entry for %s\n", dev);
- return 0;
- } else
dev = getpathbyspec(dev);
+ if (!getpathbyspec(dev)) {
+ notSuitablePrintf(entry, 0, "can't find blkid entry for %s\n",
+ dev);
+ return 0;
+ } else
+ dev = getpathbyspec(dev);
- rootdev = findDiskForRoot();
- if (!rootdev) {
- notSuitablePrintf(entry, 0, "can't find root device\n");
- return 0;
- }
+ rootdev = findDiskForRoot();
+ if (!rootdev) {
+ notSuitablePrintf(entry, 0, "can't find root device\n");
+ return 0;
+ }
- if (!getuuidbydev(rootdev) || !getuuidbydev(dev)) {
- notSuitablePrintf(entry, 0, "uuid missing: rootdev %s, dev %s\n",
- getuuidbydev(rootdev), getuuidbydev(dev));
- free(rootdev);
- return 0;
- }
+ if (!getuuidbydev(rootdev) || !getuuidbydev(dev)) {
+ notSuitablePrintf(entry, 0,
+ "uuid missing: rootdev %s, dev %s\n",
+ getuuidbydev(rootdev), getuuidbydev(dev));
+ free(rootdev);
+ return 0;
+ }
+
+ if (strcmp(getuuidbydev(rootdev), getuuidbydev(dev))) {
+ notSuitablePrintf(entry, 0,
+ "uuid mismatch: rootdev %s, dev %s\n",
+ getuuidbydev(rootdev), getuuidbydev(dev));
+ free(rootdev);
+ return 0;
+ }
- if (strcmp(getuuidbydev(rootdev), getuuidbydev(dev))) {
- notSuitablePrintf(entry, 0, "uuid mismatch: rootdev %s, dev %s\n",
- getuuidbydev(rootdev), getuuidbydev(dev));
free(rootdev);
- return 0;
- }
+ notSuitablePrintf(entry, 1, "\n");
- free(rootdev);
- notSuitablePrintf(entry, 1, "\n");
-
- return 1;
+ return 1;
}
/* returns the first match on or after the one pointed to by index (if index
is not NULL) which is not marked as skip */
-struct singleEntry * findEntryByPath(struct grubConfig * config,
- const char * kernel, const char * prefix,
- int * index) {
- struct singleEntry * entry = NULL;
- struct singleLine * line;
- int i;
- char * chptr;
- char * rootspec = NULL;
- enum lineType_e checkType = LT_KERNEL;
-
- if (isdigit(*kernel)) {
- int * indexVars = alloca(sizeof(*indexVars) * strlen(kernel));
-
- i = 0;
- indexVars[i] = strtol(kernel, &chptr, 10);
- while (*chptr == ',') {
- i++;
- kernel = chptr + 1;
- indexVars[i] = strtol(kernel, &chptr, 10);
- }
+struct singleEntry *findEntryByPath(struct grubConfig *config,
+ const char *kernel, const char *prefix,
+ int *index)
+{
+ struct singleEntry *entry = NULL;
+ struct singleLine *line;
+ int i;
+ char *chptr;
+ char *rootspec = NULL;
+ enum lineType_e checkType = LT_KERNEL;
- if (*chptr) {
- /* can't parse it, bail */
- return NULL;
- }
+ if (isdigit(*kernel)) {
+ int *indexVars = alloca(sizeof(*indexVars) * strlen(kernel));
+
+ i = 0;
+ indexVars[i] = strtol(kernel, &chptr, 10);
+ while (*chptr == ',') {
+ i++;
+ kernel = chptr + 1;
+ indexVars[i] = strtol(kernel, &chptr, 10);
+ }
+
+ if (*chptr) {
+ /* can't parse it, bail */
+ return NULL;
+ }
+
+ indexVars[i + 1] = -1;
+
+ i = 0;
+ if (index) {
+ while (i < *index) {
+ i++;
+ if (indexVars[i] == -1)
+ return NULL;
+ }
+ }
+
+ entry = findEntryByIndex(config, indexVars[i]);
+ if (!entry)
+ return NULL;
- indexVars[i + 1] = -1;
+ line =
+ getLineByType(LT_KERNEL | LT_HYPER | LT_KERNEL_EFI |
+ LT_KERNEL_16, entry->lines);
+ if (!line)
+ return NULL;
- i = 0;
- if (index) {
- while (i < *index) {
- i++;
- if (indexVars[i] == -1) return NULL;
- }
+ if (index)
+ *index = indexVars[i];
+ return entry;
}
- entry = findEntryByIndex(config, indexVars[i]);
- if (!entry) return NULL;
+ if (!strcmp(kernel, "DEFAULT")) {
+ if (index && *index > config->defaultImage) {
+ entry = NULL;
+ } else {
+ entry = findEntryByIndex(config, config->defaultImage);
+ if (entry && entry->skip)
+ entry = NULL;
+ else if (index)
+ *index = config->defaultImage;
+ }
+ } else if (!strcmp(kernel, "ALL")) {
+ if (index)
+ i = *index;
+ else
+ i = 0;
- line = getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
- if (!line) return NULL;
+ while ((entry = findEntryByIndex(config, i))) {
+ if (!entry->skip)
+ break;
+ i++;
+ }
- if (index) *index = indexVars[i];
- return entry;
- }
-
- if (!strcmp(kernel, "DEFAULT")) {
- if (index && *index > config->defaultImage) {
- entry = NULL;
+ if (entry && index)
+ *index = i;
} else {
- entry = findEntryByIndex(config, config->defaultImage);
- if (entry && entry->skip)
- entry = NULL;
- else if (index)
- *index = config->defaultImage;
- }
- } else if (!strcmp(kernel, "ALL")) {
- if (index)
- i = *index;
- else
- i = 0;
-
- while ((entry = findEntryByIndex(config, i))) {
- if (!entry->skip) break;
- i++;
- }
+ if (index)
+ i = *index;
+ else
+ i = 0;
+
+ if (!strncmp(kernel, "TITLE=", 6)) {
+ prefix = "";
+ checkType = LT_TITLE | LT_MENUENTRY;
+ kernel += 6;
+ }
+
+ for (entry = findEntryByIndex(config, i); entry;
+ entry = entry->next, i++) {
+ if (entry->skip)
+ continue;
+
+ dbgPrintf("findEntryByPath looking for %d %s in %p\n",
+ checkType, kernel, entry);
- if (entry && index)
- *index = i;
- } else {
- if (index)
- i = *index;
- else
- i = 0;
-
- if (!strncmp(kernel, "TITLE=", 6)) {
- prefix = "";
- checkType = LT_TITLE|LT_MENUENTRY;
- kernel += 6;
+ /* check all the lines matching checkType */
+ for (line = entry->lines; line; line = line->next) {
+ enum lineType_e ct = checkType;
+ if (entry->multiboot && checkType == LT_KERNEL)
+ ct = LT_KERNEL | LT_KERNEL_EFI |
+ LT_MBMODULE | LT_HYPER |
+ LT_KERNEL_16;
+ else if (checkType & LT_KERNEL)
+ ct = checkType | LT_KERNEL_EFI |
+ LT_KERNEL_16;
+ line = getLineByType(ct, line);
+ if (!line)
+ break; /* not found in this entry */
+
+ if (line && line->type != LT_MENUENTRY &&
+ line->numElements >= 2) {
+ rootspec =
+ getRootSpecifier(line->elements[1].
+ item);
+ if (!strcmp
+ (line->elements[1].item +
+ ((rootspec !=
+ NULL) ? strlen(rootspec) : 0),
+ kernel + strlen(prefix)))
+ break;
+ }
+ if (line->type == LT_MENUENTRY &&
+ !strcmp(line->elements[1].item, kernel))
+ break;
+ }
+
+ /* make sure this entry has a kernel identifier; this skips
+ * non-Linux boot entries (could find netbsd etc, though, which is
+ * unfortunate)
+ */
+ if (line
+ && getLineByType(LT_KERNEL | LT_HYPER |
+ LT_KERNEL_EFI | LT_KERNEL_16,
+ entry->lines))
+ break; /* found 'im! */
+ }
+
+ if (index)
+ *index = i;
}
- for (entry = findEntryByIndex(config, i); entry; entry = entry->next, i++) {
- if (entry->skip) continue;
+ return entry;
+}
- dbgPrintf("findEntryByPath looking for %d %s in %p\n", checkType, kernel, entry);
+struct singleEntry *findEntryByTitle(struct grubConfig *cfg, char *title,
+ int *index)
+{
+ struct singleEntry *entry;
+ struct singleLine *line;
+ int i;
+ char *newtitle;
- /* check all the lines matching checkType */
- for (line = entry->lines; line; line = line->next) {
- enum lineType_e ct = checkType;
- if (entry->multiboot && checkType == LT_KERNEL)
- ct = LT_KERNEL|LT_KERNEL_EFI|LT_MBMODULE|LT_HYPER|LT_KERNEL_16;
- else if (checkType & LT_KERNEL)
- ct = checkType | LT_KERNEL_EFI | LT_KERNEL_16;
- line = getLineByType(ct, line);
+ for (i = 0, entry = cfg->entries; entry; entry = entry->next, i++) {
+ if (index && i < *index)
+ continue;
+ line = getLineByType(LT_TITLE, entry->lines);
+ if (!line)
+ line = getLineByType(LT_MENUENTRY, entry->lines);
if (!line)
- break; /* not found in this entry */
-
- if (line && line->type != LT_MENUENTRY &&
- line->numElements >= 2) {
- rootspec = getRootSpecifier(line->elements[1].item);
- if (!strcmp(line->elements[1].item +
- ((rootspec != NULL) ? strlen(rootspec) : 0),
- kernel + strlen(prefix)))
+ continue;
+ newtitle = grub2ExtractTitle(line);
+ if (!newtitle)
+ continue;
+ if (!strcmp(title, newtitle))
break;
- }
- if(line->type == LT_MENUENTRY &&
- !strcmp(line->elements[1].item, kernel))
- break;
- }
-
- /* make sure this entry has a kernel identifier; this skips
- * non-Linux boot entries (could find netbsd etc, though, which is
- * unfortunate)
- */
- if (line && getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines))
- break; /* found 'im! */
}
- if (index) *index = i;
- }
+ if (!entry)
+ return NULL;
- return entry;
+ if (index)
+ *index = i;
+ return entry;
}
-struct singleEntry * findEntryByTitle(struct grubConfig * cfg, char *title,
- int * index) {
- struct singleEntry * entry;
- struct singleLine * line;
- int i;
- char * newtitle;
-
- for (i = 0, entry = cfg->entries; entry; entry = entry->next, i++) {
- if (index && i < *index)
- continue;
- line = getLineByType(LT_TITLE, entry->lines);
- if (!line)
- line = getLineByType(LT_MENUENTRY, entry->lines);
- if (!line)
- continue;
- newtitle = grub2ExtractTitle(line);
- if (!newtitle)
- continue;
- if (!strcmp(title, newtitle))
- break;
- }
+struct singleEntry *findEntryByIndex(struct grubConfig *cfg, int index)
+{
+ struct singleEntry *entry;
- if (!entry)
- return NULL;
-
- if (index)
- *index = i;
- return entry;
-}
-
-struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index) {
- struct singleEntry * entry;
-
- entry = cfg->entries;
- while (index && entry) {
- entry = entry->next;
- index--;
- }
+ entry = cfg->entries;
+ while (index && entry) {
+ entry = entry->next;
+ index--;
+ }
- return entry;
+ return entry;
}
/* Find a good template to use for the new kernel. An entry is
@@ -2149,2595 +2312,2886 @@ struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index) {
* is going to be removed). Try and use the default entry, but
* if that doesn't work just take the first. If we can't find one,
* bail. */
-struct singleEntry * findTemplate(struct grubConfig * cfg, const char * prefix,
- int * indexPtr, int skipRemoved, int flags) {
- struct singleEntry * entry, * entry2;
- int index;
+struct singleEntry *findTemplate(struct grubConfig *cfg, const char *prefix,
+ int *indexPtr, int skipRemoved, int flags)
+{
+ struct singleEntry *entry, *entry2;
+ int index;
- if (cfg->cfi->defaultIsSaved) {
- if (cfg->cfi->getEnv) {
- char *defTitle = cfg->cfi->getEnv(cfg->cfi, "saved_entry");
- if (defTitle) {
- int index = 0;
- if (isnumber(defTitle)) {
- index = atoi(defTitle);
- entry = findEntryByIndex(cfg, index);
- } else {
- entry = findEntryByTitle(cfg, defTitle, &index);
+ if (cfg->cfi->defaultIsSaved) {
+ if (cfg->cfi->getEnv) {
+ char *defTitle =
+ cfg->cfi->getEnv(cfg->cfi, "saved_entry");
+ if (defTitle) {
+ int index = 0;
+ if (isnumber(defTitle)) {
+ index = atoi(defTitle);
+ entry = findEntryByIndex(cfg, index);
+ } else {
+ entry =
+ findEntryByTitle(cfg, defTitle,
+ &index);
+ }
+ if (entry
+ && suitableImage(entry, prefix, skipRemoved,
+ flags)) {
+ cfg->defaultImage = index;
+ if (indexPtr)
+ *indexPtr = index;
+ return entry;
+ }
+ }
}
+ } else if (cfg->defaultImage > -1) {
+ entry = findEntryByIndex(cfg, cfg->defaultImage);
if (entry && suitableImage(entry, prefix, skipRemoved, flags)) {
- cfg->defaultImage = index;
- if (indexPtr)
- *indexPtr = index;
- return entry;
+ if (indexPtr)
+ *indexPtr = cfg->defaultImage;
+ return entry;
}
- }
- }
- } else if (cfg->defaultImage > -1) {
- entry = findEntryByIndex(cfg, cfg->defaultImage);
- if (entry && suitableImage(entry, prefix, skipRemoved, flags)) {
- if (indexPtr) *indexPtr = cfg->defaultImage;
- return entry;
- }
- }
-
- index = 0;
- while ((entry = findEntryByIndex(cfg, index))) {
- if (suitableImage(entry, prefix, skipRemoved, flags)) {
- int j;
- for (j = 0; j < index; j++) {
- entry2 = findEntryByIndex(cfg, j);
- if (entry2->skip) index--;
- }
- if (indexPtr) *indexPtr = index;
-
- return entry;
}
- index++;
- }
+ index = 0;
+ while ((entry = findEntryByIndex(cfg, index))) {
+ if (suitableImage(entry, prefix, skipRemoved, flags)) {
+ int j;
+ for (j = 0; j < index; j++) {
+ entry2 = findEntryByIndex(cfg, j);
+ if (entry2->skip)
+ index--;
+ }
+ if (indexPtr)
+ *indexPtr = index;
+
+ return entry;
+ }
+
+ index++;
+ }
- fprintf(stderr, _("grubby fatal error: unable to find a suitable template\n"));
+ fprintf(stderr,
+ _("grubby fatal error: unable to find a suitable template\n"));
- return NULL;
+ return NULL;
}
-char * findBootPrefix(void) {
- struct stat sb, sb2;
+char *findBootPrefix(void)
+{
+ struct stat sb, sb2;
- stat("/", &sb);
+ stat("/", &sb);
#ifdef __ia64__
- stat("/boot/efi/EFI/redhat/", &sb2);
+ stat("/boot/efi/EFI/redhat/", &sb2);
#else
- stat("/boot", &sb2);
+ stat("/boot", &sb2);
#endif
- if (sb.st_dev == sb2.st_dev)
- return strdup("");
+ if (sb.st_dev == sb2.st_dev)
+ return strdup("");
#ifdef __ia64__
- return strdup("/boot/efi/EFI/redhat/");
+ return strdup("/boot/efi/EFI/redhat/");
#else
- return strdup("/boot");
+ return strdup("/boot");
#endif
}
-void markRemovedImage(struct grubConfig * cfg, const char * image,
- const char * prefix) {
- struct singleEntry * entry;
+void markRemovedImage(struct grubConfig *cfg, const char *image,
+ const char *prefix)
+{
+ struct singleEntry *entry;
- if (!image)
- return;
+ if (!image)
+ return;
- /* check and see if we're removing the default image */
- if (isdigit(*image)) {
- entry = findEntryByPath(cfg, image, prefix, NULL);
- if(entry)
- entry->skip = 1;
- return;
- }
+ /* check and see if we're removing the default image */
+ if (isdigit(*image)) {
+ entry = findEntryByPath(cfg, image, prefix, NULL);
+ if (entry)
+ entry->skip = 1;
+ return;
+ }
- while ((entry = findEntryByPath(cfg, image, prefix, NULL)))
- entry->skip = 1;
+ while ((entry = findEntryByPath(cfg, image, prefix, NULL)))
+ entry->skip = 1;
}
-void setDefaultImage(struct grubConfig * config, int hasNew,
- const char * defaultKernelPath, int newIsDefault,
- const char * prefix, int flags, int index) {
- struct singleEntry * entry, * entry2, * newDefault;
- int i, j;
-
- if (newIsDefault) {
- config->defaultImage = 0;
- return;
- } else if ((index >= 0) && config->cfi->defaultIsIndex) {
- if (findEntryByIndex(config, index))
- config->defaultImage = index;
+void setDefaultImage(struct grubConfig *config, int hasNew,
+ const char *defaultKernelPath, int newIsDefault,
+ const char *prefix, int flags, int index)
+{
+ struct singleEntry *entry, *entry2, *newDefault;
+ int i, j;
+
+ if (newIsDefault) {
+ config->defaultImage = 0;
+ return;
+ } else if ((index >= 0) && config->cfi->defaultIsIndex) {
+ if (findEntryByIndex(config, index))
+ config->defaultImage = index;
+ else
+ config->defaultImage = -1;
+ return;
+ } else if (defaultKernelPath) {
+ i = 0;
+ if (findEntryByPath(config, defaultKernelPath, prefix, &i)) {
+ config->defaultImage = i;
+ } else {
+ config->defaultImage = -1;
+ return;
+ }
+ }
+
+ /* defaultImage now points to what we'd like to use, but before any
+ * order changes */
+ if ((config->defaultImage == DEFAULT_SAVED) ||
+ (config->defaultImage == DEFAULT_SAVED_GRUB2))
+ /* default is set to saved, we don't want to change it */
+ return;
+
+ if (config->defaultImage > -1)
+ entry = findEntryByIndex(config, config->defaultImage);
else
- config->defaultImage = -1;
- return;
- } else if (defaultKernelPath) {
- i = 0;
- if (findEntryByPath(config, defaultKernelPath, prefix, &i)) {
- config->defaultImage = i;
+ entry = NULL;
+
+ if (entry && !entry->skip) {
+ /* we can preserve the default */
+ if (hasNew)
+ config->defaultImage++;
+
+ /* count the number of entries erased before this one */
+ for (j = 0; j < config->defaultImage; j++) {
+ entry2 = findEntryByIndex(config, j);
+ if (entry2->skip)
+ config->defaultImage--;
+ }
+ } else if (hasNew) {
+ config->defaultImage = 0;
} else {
- config->defaultImage = -1;
- return;
+ /* Either we just erased the default (or the default line was
+ * bad to begin with) and didn't put a new one in. We'll use
+ * the first valid image. */
+ newDefault =
+ findTemplate(config, prefix, &config->defaultImage, 1,
+ flags);
+ if (!newDefault)
+ config->defaultImage = -1;
}
- }
+}
- /* defaultImage now points to what we'd like to use, but before any order
- changes */
- if ((config->defaultImage == DEFAULT_SAVED) ||
- (config->defaultImage == DEFAULT_SAVED_GRUB2))
- /* default is set to saved, we don't want to change it */
- return;
+void setFallbackImage(struct grubConfig *config, int hasNew)
+{
+ struct singleEntry *entry, *entry2;
+ int j;
- if (config->defaultImage > -1)
- entry = findEntryByIndex(config, config->defaultImage);
- else
- entry = NULL;
+ if (config->fallbackImage == -1)
+ return;
+
+ entry = findEntryByIndex(config, config->fallbackImage);
+ if (!entry || entry->skip) {
+ config->fallbackImage = -1;
+ return;
+ }
- if (entry && !entry->skip) {
- /* we can preserve the default */
if (hasNew)
- config->defaultImage++;
-
+ config->fallbackImage++;
+
/* count the number of entries erased before this one */
- for (j = 0; j < config->defaultImage; j++) {
- entry2 = findEntryByIndex(config, j);
- if (entry2->skip) config->defaultImage--;
+ for (j = 0; j < config->fallbackImage; j++) {
+ entry2 = findEntryByIndex(config, j);
+ if (entry2->skip)
+ config->fallbackImage--;
}
- } else if (hasNew) {
- config->defaultImage = 0;
- } else {
- /* Either we just erased the default (or the default line was bad
- * to begin with) and didn't put a new one in. We'll use the first
- * valid image. */
- newDefault = findTemplate(config, prefix, &config->defaultImage, 1,
- flags);
- if (!newDefault)
- config->defaultImage = -1;
- }
}
-void setFallbackImage(struct grubConfig * config, int hasNew) {
- struct singleEntry * entry, * entry2;
- int j;
-
- if (config->fallbackImage == -1) return;
-
- entry = findEntryByIndex(config, config->fallbackImage);
- if (!entry || entry->skip) {
- config->fallbackImage = -1;
- return;
- }
-
- if (hasNew)
- config->fallbackImage++;
-
- /* count the number of entries erased before this one */
- for (j = 0; j < config->fallbackImage; j++) {
- entry2 = findEntryByIndex(config, j);
- if (entry2->skip) config->fallbackImage--;
- }
-}
-
-void displayEntry(struct singleEntry * entry, const char * prefix, int index) {
- struct singleLine * line;
- char * root = NULL;
- int i;
- int j;
-
- printf("index=%d\n", index);
-
- line = getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
- if (!line) {
- printf("non linux entry\n");
- return;
- }
-
- if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))
- printf("kernel=%s\n", line->elements[1].item);
- else
- printf("kernel=%s%s\n", prefix, line->elements[1].item);
-
- if (line->numElements >= 3) {
- printf("args=\"");
- i = 2;
- while (i < line->numElements) {
- if (!strncmp(line->elements[i].item, "root=", 5)) {
- root = line->elements[i].item + 5;
- } else {
- printf("%s%s", line->elements[i].item,
- line->elements[i].indent);
- }
-
- i++;
+void displayEntry(struct singleEntry *entry, const char *prefix, int index)
+{
+ struct singleLine *line;
+ char *root = NULL;
+ int i;
+ int j;
+
+ printf("index=%d\n", index);
+
+ line =
+ getLineByType(LT_KERNEL | LT_HYPER | LT_KERNEL_EFI | LT_KERNEL_16,
+ entry->lines);
+ if (!line) {
+ printf("non linux entry\n");
+ return;
}
- printf("\"\n");
- } else {
- line = getLineByType(LT_KERNELARGS, entry->lines);
- if (line) {
- char * s;
-
- printf("args=\"");
- i = 1;
- while (i < line->numElements) {
- if (!strncmp(line->elements[i].item, "root=", 5)) {
- root = line->elements[i].item + 5;
- } else {
- s = line->elements[i].item;
-
- printf("%s%s", s, line->elements[i].indent);
- }
-
- i++;
- }
-
- s = line->elements[i - 1].indent;
- printf("\"\n");
- }
- }
-
- if (!root) {
- line = getLineByType(LT_ROOT, entry->lines);
- if (line && line->numElements >= 2)
- root=line->elements[1].item;
- }
-
- if (root) {
- char * s = alloca(strlen(root) + 1);
-
- strcpy(s, root);
- if (s[strlen(s) - 1] == '"')
- s[strlen(s) - 1] = '\0';
- /* make sure the root doesn't have a trailing " */
- printf("root=%s\n", s);
- }
-
- line = getLineByType(LT_INITRD|LT_INITRD_EFI|LT_INITRD_16, entry->lines);
- if (line && line->numElements >= 2) {
if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))
- printf("initrd=");
+ printf("kernel=%s\n", line->elements[1].item);
else
- printf("initrd=%s", prefix);
-
- for (i = 1; i < line->numElements; i++)
- printf("%s%s", line->elements[i].item, line->elements[i].indent);
- printf("\n");
- }
-
- line = getLineByType(LT_TITLE, entry->lines);
- if (line) {
- printf("title=%s\n", line->elements[1].item);
- } else {
- char * title;
- line = getLineByType(LT_MENUENTRY, entry->lines);
+ printf("kernel=%s%s\n", prefix, line->elements[1].item);
+
+ if (line->numElements >= 3) {
+ printf("args=\"");
+ i = 2;
+ while (i < line->numElements) {
+ if (!strncmp(line->elements[i].item, "root=", 5)) {
+ root = line->elements[i].item + 5;
+ } else {
+ printf("%s%s", line->elements[i].item,
+ line->elements[i].indent);
+ }
+
+ i++;
+ }
+ printf("\"\n");
+ } else {
+ line = getLineByType(LT_KERNELARGS, entry->lines);
+ if (line) {
+ char *s;
+
+ printf("args=\"");
+ i = 1;
+ while (i < line->numElements) {
+ if (!strncmp
+ (line->elements[i].item, "root=", 5)) {
+ root = line->elements[i].item + 5;
+ } else {
+ s = line->elements[i].item;
+
+ printf("%s%s", s,
+ line->elements[i].indent);
+ }
+
+ i++;
+ }
+
+ s = line->elements[i - 1].indent;
+ printf("\"\n");
+ }
+ }
+
+ if (!root) {
+ line = getLineByType(LT_ROOT, entry->lines);
+ if (line && line->numElements >= 2)
+ root = line->elements[1].item;
+ }
+
+ if (root) {
+ char *s = alloca(strlen(root) + 1);
+
+ strcpy(s, root);
+ if (s[strlen(s) - 1] == '"')
+ s[strlen(s) - 1] = '\0';
+ /* make sure the root doesn't have a trailing " */
+ printf("root=%s\n", s);
+ }
+
+ line =
+ getLineByType(LT_INITRD | LT_INITRD_EFI | LT_INITRD_16,
+ entry->lines);
+
+ if (line && line->numElements >= 2) {
+ if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))
+ printf("initrd=");
+ else
+ printf("initrd=%s", prefix);
+
+ for (i = 1; i < line->numElements; i++)
+ printf("%s%s", line->elements[i].item,
+ line->elements[i].indent);
+ printf("\n");
+ }
+
+ line = getLineByType(LT_TITLE, entry->lines);
if (line) {
- title = grub2ExtractTitle(line);
- if (title)
- printf("title=%s\n", title);
+ printf("title=%s\n", line->elements[1].item);
+ } else {
+ char *title;
+ line = getLineByType(LT_MENUENTRY, entry->lines);
+ if (line) {
+ title = grub2ExtractTitle(line);
+ if (title)
+ printf("title=%s\n", title);
+ }
}
- }
-
- for (j = 0, line = entry->lines; line; line = line->next) {
- if ((line->type & LT_MBMODULE) && line->numElements >= 2) {
- if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))
- printf("mbmodule%d=", j);
- else
- printf("mbmodule%d=%s", j, prefix);
-
- for (i = 1; i < line->numElements; i++)
- printf("%s%s", line->elements[i].item, line->elements[i].indent);
- printf("\n");
- j++;
+
+ for (j = 0, line = entry->lines; line; line = line->next) {
+ if ((line->type & LT_MBMODULE) && line->numElements >= 2) {
+ if (!strncmp
+ (prefix, line->elements[1].item, strlen(prefix)))
+ printf("mbmodule%d=", j);
+ else
+ printf("mbmodule%d=%s", j, prefix);
+
+ for (i = 1; i < line->numElements; i++)
+ printf("%s%s", line->elements[i].item,
+ line->elements[i].indent);
+ printf("\n");
+ j++;
+ }
}
- }
}
-int isSuseSystem(void) {
- const char * path;
- const static char default_path[] = "/etc/SuSE-release";
+int isSuseSystem(void)
+{
+ const char *path;
+ const static char default_path[] = "/etc/SuSE-release";
+
+ if ((path = getenv("GRUBBY_SUSE_RELEASE")) == NULL)
+ path = default_path;
+
+ if (!access(path, R_OK))
+ return 1;
+ return 0;
+}
+
+int isSuseGrubConf(const char *path)
+{
+ FILE *grubConf;
+ char *line = NULL;
+ size_t len = 0, res = 0;
+
+ grubConf = fopen(path, "r");
+ if (!grubConf) {
+ dbgPrintf("Could not open SuSE configuration file '%s'\n",
+ path);
+ return 0;
+ }
+
+ while ((res = getline(&line, &len, grubConf)) != -1) {
+ if (!strncmp(line, "setup", 5)) {
+ fclose(grubConf);
+ free(line);
+ return 1;
+ }
+ }
+
+ dbgPrintf("SuSE configuration file '%s' does not appear to be valid\n",
+ path);
+
+ fclose(grubConf);
+ free(line);
+ return 0;
+}
+
+int suseGrubConfGetLba(const char *path, int *lbaPtr)
+{
+ FILE *grubConf;
+ char *line = NULL;
+ size_t res = 0, len = 0;
+
+ if (!path)
+ return 1;
+ if (!lbaPtr)
+ return 1;
+
+ grubConf = fopen(path, "r");
+ if (!grubConf)
+ return 1;
+
+ while ((res = getline(&line, &len, grubConf)) != -1) {
+ if (line[res - 1] == '\n')
+ line[res - 1] = '\0';
+ else if (len > res)
+ line[res] = '\0';
+ else {
+ line = realloc(line, res + 1);
+ line[res] = '\0';
+ }
- if ((path = getenv("GRUBBY_SUSE_RELEASE")) == NULL)
- path = default_path;
+ if (!strncmp(line, "setup", 5)) {
+ if (strstr(line, "--force-lba")) {
+ *lbaPtr = 1;
+ } else {
+ *lbaPtr = 0;
+ }
+ dbgPrintf("lba: %i\n", *lbaPtr);
+ break;
+ }
+ }
+
+ free(line);
+ fclose(grubConf);
+ return 0;
+}
+
+int suseGrubConfGetInstallDevice(const char *path, char **devicePtr)
+{
+ FILE *grubConf;
+ char *line = NULL;
+ size_t res = 0, len = 0;
+ char *lastParamPtr = NULL;
+ char *secLastParamPtr = NULL;
+ char installDeviceNumber = '\0';
+ char *bounds = NULL;
+
+ if (!path)
+ return 1;
+ if (!devicePtr)
+ return 1;
+
+ grubConf = fopen(path, "r");
+ if (!grubConf)
+ return 1;
+
+ while ((res = getline(&line, &len, grubConf)) != -1) {
+ if (strncmp(line, "setup", 5))
+ continue;
+
+ if (line[res - 1] == '\n')
+ line[res - 1] = '\0';
+ else if (len > res)
+ line[res] = '\0';
+ else {
+ line = realloc(line, res + 1);
+ line[res] = '\0';
+ }
+
+ lastParamPtr = bounds = line + res;
+
+ /* Last parameter in grub may be an optional IMAGE_DEVICE */
+ while (!isspace(*lastParamPtr))
+ lastParamPtr--;
+ lastParamPtr++;
+
+ secLastParamPtr = lastParamPtr - 2;
+ dbgPrintf("lastParamPtr: %s\n", lastParamPtr);
+
+ if (lastParamPtr + 3 > bounds) {
+ dbgPrintf("lastParamPtr going over boundary");
+ fclose(grubConf);
+ free(line);
+ return 1;
+ }
+ if (!strncmp(lastParamPtr, "(hd", 3))
+ lastParamPtr += 3;
+ dbgPrintf("lastParamPtr: %c\n", *lastParamPtr);
+
+ /*
+ * Second last parameter will decide wether last parameter is
+ * an IMAGE_DEVICE or INSTALL_DEVICE
+ */
+ while (!isspace(*secLastParamPtr))
+ secLastParamPtr--;
+ secLastParamPtr++;
+
+ if (secLastParamPtr + 3 > bounds) {
+ dbgPrintf("secLastParamPtr going over boundary");
+ fclose(grubConf);
+ free(line);
+ return 1;
+ }
+ dbgPrintf("secLastParamPtr: %s\n", secLastParamPtr);
+ if (!strncmp(secLastParamPtr, "(hd", 3)) {
+ secLastParamPtr += 3;
+ dbgPrintf("secLastParamPtr: %c\n", *secLastParamPtr);
+ installDeviceNumber = *secLastParamPtr;
+ } else {
+ installDeviceNumber = *lastParamPtr;
+ }
+
+ *devicePtr = malloc(6);
+ snprintf(*devicePtr, 6, "(hd%c)", installDeviceNumber);
+ dbgPrintf("installDeviceNumber: %c\n", installDeviceNumber);
+ fclose(grubConf);
+ free(line);
+ return 0;
+ }
- if (!access(path, R_OK))
+ free(line);
+ fclose(grubConf);
return 1;
- return 0;
}
-int isSuseGrubConf(const char * path) {
- FILE * grubConf;
- char * line = NULL;
- size_t len = 0, res = 0;
-
- grubConf = fopen(path, "r");
- if (!grubConf) {
- dbgPrintf("Could not open SuSE configuration file '%s'\n", path);
- return 0;
- }
-
- while ((res = getline(&line, &len, grubConf)) != -1) {
- if (!strncmp(line, "setup", 5)) {
- fclose(grubConf);
- free(line);
- return 1;
- }
- }
-
- dbgPrintf("SuSE configuration file '%s' does not appear to be valid\n",
- path);
-
- fclose(grubConf);
- free(line);
- return 0;
-}
-
-int suseGrubConfGetLba(const char * path, int * lbaPtr) {
- FILE * grubConf;
- char * line = NULL;
- size_t res = 0, len = 0;
-
- if (!path) return 1;
- if (!lbaPtr) return 1;
-
- grubConf = fopen(path, "r");
- if (!grubConf) return 1;
-
- while ((res = getline(&line, &len, grubConf)) != -1) {
- if (line[res - 1] == '\n')
- line[res - 1] = '\0';
- else if (len > res)
- line[res] = '\0';
- else {
- line = realloc(line, res + 1);
- line[res] = '\0';
+int grubGetBootFromDeviceMap(const char *device, char **bootPtr)
+{
+ FILE *deviceMap;
+ char *line = NULL;
+ size_t res = 0, len = 0;
+ char *devicePtr;
+ char *bounds = NULL;
+ const char *path;
+ const static char default_path[] = "/boot/grub/device.map";
+
+ if (!device)
+ return 1;
+ if (!bootPtr)
+ return 1;
+
+ if ((path = getenv("GRUBBY_GRUB_DEVICE_MAP")) == NULL)
+ path = default_path;
+
+ dbgPrintf("opening grub device.map file from: %s\n", path);
+ deviceMap = fopen(path, "r");
+ if (!deviceMap)
+ return 1;
+
+ while ((res = getline(&line, &len, deviceMap)) != -1) {
+ if (!strncmp(line, "#", 1))
+ continue;
+
+ if (line[res - 1] == '\n')
+ line[res - 1] = '\0';
+ else if (len > res)
+ line[res] = '\0';
+ else {
+ line = realloc(line, res + 1);
+ line[res] = '\0';
+ }
+
+ devicePtr = line;
+ bounds = line + res;
+
+ while ((isspace(*line) && ((devicePtr + 1) <= bounds)))
+ devicePtr++;
+ dbgPrintf("device: %s\n", devicePtr);
+
+ if (!strncmp(devicePtr, device, strlen(device))) {
+ devicePtr += strlen(device);
+ while (isspace(*devicePtr)
+ && ((devicePtr + 1) <= bounds))
+ devicePtr++;
+
+ *bootPtr = strdup(devicePtr);
+ break;
+ }
}
- if (!strncmp(line, "setup", 5)) {
- if (strstr(line, "--force-lba")) {
- *lbaPtr = 1;
- } else {
- *lbaPtr = 0;
- }
- dbgPrintf("lba: %i\n", *lbaPtr);
- break;
- }
- }
-
- free(line);
- fclose(grubConf);
- return 0;
+ free(line);
+ fclose(deviceMap);
+ return 0;
}
-int suseGrubConfGetInstallDevice(const char * path, char ** devicePtr) {
- FILE * grubConf;
- char * line = NULL;
- size_t res = 0, len = 0;
- char * lastParamPtr = NULL;
- char * secLastParamPtr = NULL;
- char installDeviceNumber = '\0';
- char * bounds = NULL;
-
- if (!path) return 1;
- if (!devicePtr) return 1;
-
- grubConf = fopen(path, "r");
- if (!grubConf) return 1;
-
- while ((res = getline(&line, &len, grubConf)) != -1) {
- if (strncmp(line, "setup", 5))
- continue;
-
- if (line[res - 1] == '\n')
- line[res - 1] = '\0';
- else if (len > res)
- line[res] = '\0';
- else {
- line = realloc(line, res + 1);
- line[res] = '\0';
- }
-
- lastParamPtr = bounds = line + res;
+int suseGrubConfGetBoot(const char *path, char **bootPtr)
+{
+ char *grubDevice;
- /* Last parameter in grub may be an optional IMAGE_DEVICE */
- while (!isspace(*lastParamPtr))
- lastParamPtr--;
- lastParamPtr++;
+ if (suseGrubConfGetInstallDevice(path, &grubDevice))
+ dbgPrintf("error looking for grub installation device\n");
+ else
+ dbgPrintf("grubby installation device: %s\n", grubDevice);
- secLastParamPtr = lastParamPtr - 2;
- dbgPrintf("lastParamPtr: %s\n", lastParamPtr);
+ if (grubGetBootFromDeviceMap(grubDevice, bootPtr))
+ dbgPrintf("error looking for grub boot device\n");
+ else
+ dbgPrintf("grubby boot device: %s\n", *bootPtr);
- if (lastParamPtr + 3 > bounds) {
- dbgPrintf("lastParamPtr going over boundary");
- fclose(grubConf);
- free(line);
- return 1;
- }
- if (!strncmp(lastParamPtr, "(hd", 3))
- lastParamPtr += 3;
- dbgPrintf("lastParamPtr: %c\n", *lastParamPtr);
+ free(grubDevice);
+ return 0;
+}
+int parseSuseGrubConf(int *lbaPtr, char **bootPtr)
+{
/*
- * Second last parameter will decide wether last parameter is
- * an IMAGE_DEVICE or INSTALL_DEVICE
+ * This SuSE grub configuration file at this location is not your
+ * average grub configuration file, but instead the grub commands
+ * used to setup grub on that system.
*/
- while (!isspace(*secLastParamPtr))
- secLastParamPtr--;
- secLastParamPtr++;
-
- if (secLastParamPtr + 3 > bounds) {
- dbgPrintf("secLastParamPtr going over boundary");
- fclose(grubConf);
- free(line);
- return 1;
- }
- dbgPrintf("secLastParamPtr: %s\n", secLastParamPtr);
- if (!strncmp(secLastParamPtr, "(hd", 3)) {
- secLastParamPtr += 3;
- dbgPrintf("secLastParamPtr: %c\n", *secLastParamPtr);
- installDeviceNumber = *secLastParamPtr;
- } else {
- installDeviceNumber = *lastParamPtr;
- }
+ const char *path;
+ const static char default_path[] = "/etc/grub.conf";
- *devicePtr = malloc(6);
- snprintf(*devicePtr, 6, "(hd%c)", installDeviceNumber);
- dbgPrintf("installDeviceNumber: %c\n", installDeviceNumber);
- fclose(grubConf);
- free(line);
- return 0;
- }
+ if ((path = getenv("GRUBBY_SUSE_GRUB_CONF")) == NULL)
+ path = default_path;
- free(line);
- fclose(grubConf);
- return 1;
-}
+ if (!isSuseGrubConf(path))
+ return 1;
-int grubGetBootFromDeviceMap(const char * device,
- char ** bootPtr) {
- FILE * deviceMap;
- char * line = NULL;
- size_t res = 0, len = 0;
- char * devicePtr;
- char * bounds = NULL;
- const char * path;
- const static char default_path[] = "/boot/grub/device.map";
-
- if (!device) return 1;
- if (!bootPtr) return 1;
-
- if ((path = getenv("GRUBBY_GRUB_DEVICE_MAP")) == NULL)
- path = default_path;
-
- dbgPrintf("opening grub device.map file from: %s\n", path);
- deviceMap = fopen(path, "r");
- if (!deviceMap)
- return 1;
-
- while ((res = getline(&line, &len, deviceMap)) != -1) {
- if (!strncmp(line, "#", 1))
- continue;
-
- if (line[res - 1] == '\n')
- line[res - 1] = '\0';
- else if (len > res)
- line[res] = '\0';
- else {
- line = realloc(line, res + 1);
- line[res] = '\0';
+ if (lbaPtr) {
+ *lbaPtr = 0;
+ if (suseGrubConfGetLba(path, lbaPtr))
+ return 1;
}
- devicePtr = line;
- bounds = line + res;
-
- while ((isspace(*line) && ((devicePtr + 1) <= bounds)))
- devicePtr++;
- dbgPrintf("device: %s\n", devicePtr);
-
- if (!strncmp(devicePtr, device, strlen(device))) {
- devicePtr += strlen(device);
- while (isspace(*devicePtr) && ((devicePtr + 1) <= bounds))
- devicePtr++;
-
- *bootPtr = strdup(devicePtr);
- break;
+ if (bootPtr) {
+ *bootPtr = NULL;
+ suseGrubConfGetBoot(path, bootPtr);
}
- }
- free(line);
- fclose(deviceMap);
- return 0;
+ return 0;
}
-int suseGrubConfGetBoot(const char * path, char ** bootPtr) {
- char * grubDevice;
-
- if (suseGrubConfGetInstallDevice(path, &grubDevice))
- dbgPrintf("error looking for grub installation device\n");
- else
- dbgPrintf("grubby installation device: %s\n", grubDevice);
+int parseSysconfigGrub(int *lbaPtr, char **bootPtr)
+{
+ FILE *in;
+ char buf[1024];
+ char *chptr;
+ char *start;
+ char *param;
+
+ in = fopen("/etc/sysconfig/grub", "r");
+ if (!in)
+ return 1;
+
+ if (lbaPtr)
+ *lbaPtr = 0;
+ if (bootPtr)
+ *bootPtr = NULL;
+
+ while (fgets(buf, sizeof(buf), in)) {
+ start = buf;
+ while (isspace(*start))
+ start++;
+ if (*start == '#')
+ continue;
+
+ chptr = strchr(start, '=');
+ if (!chptr)
+ continue;
+ chptr--;
+ while (*chptr && isspace(*chptr))
+ chptr--;
+ chptr++;
+ *chptr = '\0';
+
+ param = chptr + 1;
+ while (*param && isspace(*param))
+ param++;
+ if (*param == '=') {
+ param++;
+ while (*param && isspace(*param))
+ param++;
+ }
+
+ chptr = param;
+ while (*chptr && !isspace(*chptr))
+ chptr++;
+ *chptr = '\0';
+
+ if (!strcmp(start, "forcelba") && !strcmp(param, "1") && lbaPtr)
+ *lbaPtr = 1;
+ else if (!strcmp(start, "boot") && bootPtr)
+ *bootPtr = strdup(param);
+ }
- if (grubGetBootFromDeviceMap(grubDevice, bootPtr))
- dbgPrintf("error looking for grub boot device\n");
- else
- dbgPrintf("grubby boot device: %s\n", *bootPtr);
+ fclose(in);
- free(grubDevice);
- return 0;
+ return 0;
}
-int parseSuseGrubConf(int * lbaPtr, char ** bootPtr) {
- /*
- * This SuSE grub configuration file at this location is not your average
- * grub configuration file, but instead the grub commands used to setup
- * grub on that system.
- */
- const char * path;
- const static char default_path[] = "/etc/grub.conf";
-
- if ((path = getenv("GRUBBY_SUSE_GRUB_CONF")) == NULL)
- path = default_path;
-
- if (!isSuseGrubConf(path)) return 1;
-
- if (lbaPtr) {
- *lbaPtr = 0;
- if (suseGrubConfGetLba(path, lbaPtr))
- return 1;
- }
+void dumpSysconfigGrub(void)
+{
+ char *boot = NULL;
+ int lba;
- if (bootPtr) {
- *bootPtr = NULL;
- suseGrubConfGetBoot(path, bootPtr);
- }
-
- return 0;
-}
-
-int parseSysconfigGrub(int * lbaPtr, char ** bootPtr) {
- FILE * in;
- char buf[1024];
- char * chptr;
- char * start;
- char * param;
-
- in = fopen("/etc/sysconfig/grub", "r");
- if (!in) return 1;
-
- if (lbaPtr) *lbaPtr = 0;
- if (bootPtr) *bootPtr = NULL;
-
- while (fgets(buf, sizeof(buf), in)) {
- start = buf;
- while (isspace(*start)) start++;
- if (*start == '#') continue;
-
- chptr = strchr(start, '=');
- if (!chptr) continue;
- chptr--;
- while (*chptr && isspace(*chptr)) chptr--;
- chptr++;
- *chptr = '\0';
-
- param = chptr + 1;
- while (*param && isspace(*param)) param++;
- if (*param == '=') {
- param++;
- while (*param && isspace(*param)) param++;
+ if (isSuseSystem()) {
+ if (parseSuseGrubConf(&lba, &boot)) {
+ free(boot);
+ return;
+ }
+ } else {
+ if (parseSysconfigGrub(&lba, &boot)) {
+ free(boot);
+ return;
+ }
}
- chptr = param;
- while (*chptr && !isspace(*chptr)) chptr++;
- *chptr = '\0';
-
- if (!strcmp(start, "forcelba") && !strcmp(param, "1") && lbaPtr)
- *lbaPtr = 1;
- else if (!strcmp(start, "boot") && bootPtr)
- *bootPtr = strdup(param);
- }
-
- fclose(in);
-
- return 0;
+ if (lba)
+ printf("lba\n");
+ if (boot) {
+ printf("boot=%s\n", boot);
+ free(boot);
+ }
}
-void dumpSysconfigGrub(void) {
- char * boot = NULL;
- int lba;
+int displayInfo(struct grubConfig *config, char *kernel, const char *prefix)
+{
+ int i = 0;
+ struct singleEntry *entry;
+ struct singleLine *line;
- if (isSuseSystem()) {
- if (parseSuseGrubConf(&lba, &boot)) {
- free(boot);
- return;
- }
- } else {
- if (parseSysconfigGrub(&lba, &boot)) {
- free(boot);
- return;
+ entry = findEntryByPath(config, kernel, prefix, &i);
+ if (!entry) {
+ fprintf(stderr, _("grubby: kernel not found\n"));
+ return 1;
}
- }
-
- if (lba) printf("lba\n");
- if (boot) {
- printf("boot=%s\n", boot);
- free(boot);
- }
-}
-int displayInfo(struct grubConfig * config, char * kernel,
- const char * prefix) {
- int i = 0;
- struct singleEntry * entry;
- struct singleLine * line;
-
- entry = findEntryByPath(config, kernel, prefix, &i);
- if (!entry) {
- fprintf(stderr, _("grubby: kernel not found\n"));
- return 1;
- }
-
- /* this is a horrible hack to support /etc/sysconfig/grub; there must
- be a better way */
- if (config->cfi == &grubConfigType) {
- dumpSysconfigGrub();
- } else {
- line = getLineByType(LT_BOOT, config->theLines);
- if (line && line->numElements >= 1) {
- printf("boot=%s\n", line->elements[1].item);
+ /* this is a horrible hack to support /etc/sysconfig/grub; there must
+ be a better way */
+ if (config->cfi == &grubConfigType) {
+ dumpSysconfigGrub();
+ } else {
+ line = getLineByType(LT_BOOT, config->theLines);
+ if (line && line->numElements >= 1) {
+ printf("boot=%s\n", line->elements[1].item);
+ }
+
+ line = getLineByType(LT_LBA, config->theLines);
+ if (line)
+ printf("lba\n");
}
- line = getLineByType(LT_LBA, config->theLines);
- if (line) printf("lba\n");
- }
-
- displayEntry(entry, prefix, i);
-
- i++;
- while ((entry = findEntryByPath(config, kernel, prefix, &i))) {
displayEntry(entry, prefix, i);
+
i++;
- }
+ while ((entry = findEntryByPath(config, kernel, prefix, &i))) {
+ displayEntry(entry, prefix, i);
+ i++;
+ }
- return 0;
+ return 0;
}
-struct singleLine * addLineTmpl(struct singleEntry * entry,
- struct singleLine * tmplLine,
- struct singleLine * prevLine,
- const char * val,
- struct configFileInfo * cfi)
+struct singleLine *addLineTmpl(struct singleEntry *entry,
+ struct singleLine *tmplLine,
+ struct singleLine *prevLine,
+ const char *val, struct configFileInfo *cfi)
{
- struct singleLine * newLine = lineDup(tmplLine);
+ struct singleLine *newLine = lineDup(tmplLine);
- if (isEfi && cfi == &grub2ConfigType) {
- enum lineType_e old = newLine->type;
- newLine->type = preferredLineType(newLine->type, cfi);
- if (old != newLine->type)
- newLine->elements[0].item = getKeyByType(newLine->type, cfi);
- }
-
- if (val) {
- /* override the inherited value with our own.
- * This is a little weak because it only applies to elements[1]
- */
- if (newLine->numElements > 1)
- removeElement(newLine, 1);
- insertElement(newLine, val, 1, cfi);
-
- /* but try to keep the rootspec from the template... sigh */
- if (tmplLine->type & (LT_HYPER|LT_KERNEL|LT_MBMODULE|LT_INITRD|LT_KERNEL_EFI|LT_INITRD_EFI|LT_KERNEL_16|LT_INITRD_16)) {
- char * rootspec = getRootSpecifier(tmplLine->elements[1].item);
- if (rootspec != NULL) {
- free(newLine->elements[1].item);
- newLine->elements[1].item =
- sdupprintf("%s%s", rootspec, val);
- }
+ if (isEfi && cfi == &grub2ConfigType) {
+ enum lineType_e old = newLine->type;
+ newLine->type = preferredLineType(newLine->type, cfi);
+ if (old != newLine->type)
+ newLine->elements[0].item =
+ getKeyByType(newLine->type, cfi);
}
- }
-
- dbgPrintf("addLineTmpl(%s)\n", newLine->numElements ?
- newLine->elements[0].item : "");
-
- if (!entry->lines) {
- /* first one on the list */
- entry->lines = newLine;
- } else if (prevLine) {
- /* add after prevLine */
- newLine->next = prevLine->next;
- prevLine->next = newLine;
- }
-
- return newLine;
-}
-/* val may be NULL */
-struct singleLine * addLine(struct singleEntry * entry,
- struct configFileInfo * cfi,
- enum lineType_e type, char * defaultIndent,
- const char * val) {
- struct singleLine * line, * prev;
- struct keywordTypes * kw;
- struct singleLine tmpl;
-
- /* NB: This function shouldn't allocate items on the heap, rather on the
- * stack since it calls addLineTmpl which will make copies.
- */
- if (type == LT_TITLE && cfi->titleBracketed) {
- /* we're doing a bracketed title (zipl) */
- tmpl.type = type;
- tmpl.numElements = 1;
- tmpl.elements = alloca(sizeof(*tmpl.elements));
- tmpl.elements[0].item = alloca(strlen(val)+3);
- sprintf(tmpl.elements[0].item, "[%s]", val);
- tmpl.elements[0].indent = "";
- val = NULL;
- } else if (type == LT_MENUENTRY) {
- char *lineend = "--class gnu-linux --class gnu --class os {";
- if (!val) {
- fprintf(stderr, "Line type LT_MENUENTRY requires a value\n");
- abort();
- }
- kw = getKeywordByType(type, cfi);
- if (!kw) {
- fprintf(stderr, "Looking up keyword for unknown type %d\n", type);
- abort();
- }
- tmpl.indent = "";
- tmpl.type = type;
- tmpl.numElements = 3;
- tmpl.elements = alloca(sizeof(*tmpl.elements) * tmpl.numElements);
- tmpl.elements[0].item = kw->key;
- tmpl.elements[0].indent = alloca(2);
- sprintf(tmpl.elements[0].indent, "%c", kw->nextChar);
- tmpl.elements[1].item = (char *)val;
- tmpl.elements[1].indent = alloca(2);
- sprintf(tmpl.elements[1].indent, "%c", kw->nextChar);
- tmpl.elements[2].item = alloca(strlen(lineend)+1);
- strcpy(tmpl.elements[2].item, lineend);
- tmpl.elements[2].indent = "";
- } else {
- kw = getKeywordByType(type, cfi);
- if (!kw) {
- fprintf(stderr, "Looking up keyword for unknown type %d\n", type);
- abort();
- }
- tmpl.type = type;
- tmpl.numElements = val ? 2 : 1;
- tmpl.elements = alloca(sizeof(*tmpl.elements) * tmpl.numElements);
- tmpl.elements[0].item = kw->key;
- tmpl.elements[0].indent = alloca(2);
- sprintf(tmpl.elements[0].indent, "%c", kw->nextChar);
if (val) {
- tmpl.elements[1].item = (char *)val;
- tmpl.elements[1].indent = "";
- }
- }
-
- /* The last non-empty line gives us the indention to us and the line
- to insert after. Note that comments are considered empty lines, which
- may not be ideal? If there are no lines or we are looking at the
- first line, we use defaultIndent (the first line is normally indented
- differently from the rest) */
- for (line = entry->lines, prev = NULL; line; line = line->next) {
- if (line->numElements) prev = line;
- /* fall back on the last line if prev isn't otherwise set */
- if (!line->next && !prev) prev = line;
- }
-
- struct singleLine *menuEntry;
- menuEntry = getLineByType(LT_MENUENTRY, entry->lines);
- if (tmpl.type == LT_ENTRY_END) {
- if (menuEntry)
- tmpl.indent = menuEntry->indent;
- else
- tmpl.indent = defaultIndent ?: "";
- } else if (tmpl.type != LT_MENUENTRY) {
- if (menuEntry)
- tmpl.indent = "\t";
- else if (prev == entry->lines)
- tmpl.indent = defaultIndent ?: "";
- else
- tmpl.indent = prev->indent;
- }
-
- return addLineTmpl(entry, &tmpl, prev, val, cfi);
-}
-
-void removeLine(struct singleEntry * entry, struct singleLine * line) {
- struct singleLine * prev;
- int i;
-
- for (i = 0; i < line->numElements; i++) {
- free(line->elements[i].item);
- free(line->elements[i].indent);
- }
- free(line->elements);
- free(line->indent);
-
- if (line == entry->lines) {
- entry->lines = line->next;
- } else {
- prev = entry->lines;
- while (prev->next != line) prev = prev->next;
- prev->next = line->next;
- }
-
- free(line);
-}
-
-static void requote(struct singleLine *tmplLine, struct configFileInfo * cfi)
-{
- struct singleLine newLine = {
- .indent = tmplLine->indent,
- .type = tmplLine->type,
- .next = tmplLine->next,
- };
- int firstQuotedItem = -1;
- int quoteLen = 0;
- int j;
- int element = 0;
- char *c;
-
- c = malloc(strlen(tmplLine->elements[0].item) + 1);
- strcpy(c, tmplLine->elements[0].item);
- insertElement(&newLine, c, element++, cfi);
- free(c);
- c = NULL;
-
- for (j = 1; j < tmplLine->numElements; j++) {
- if (firstQuotedItem == -1) {
- quoteLen += strlen(tmplLine->elements[j].item);
-
- if (isquote(tmplLine->elements[j].item[0])) {
- firstQuotedItem = j;
- quoteLen += strlen(tmplLine->elements[j].indent);
- } else {
- c = malloc(quoteLen + 1);
- strcpy(c, tmplLine->elements[j].item);
- insertElement(&newLine, c, element++, cfi);
- free(c);
- quoteLen = 0;
- }
- } else {
- int itemlen = strlen(tmplLine->elements[j].item);
- quoteLen += itemlen;
- quoteLen += strlen(tmplLine->elements[j].indent);
-
- if (isquote(tmplLine->elements[j].item[itemlen - 1])) {
- c = malloc(quoteLen + 1);
- c[0] = '\0';
- for (int i = firstQuotedItem; i < j+1; i++) {
- strcat(c, tmplLine->elements[i].item);
- strcat(c, tmplLine->elements[i].indent);
- }
- insertElement(&newLine, c, element++, cfi);
- free(c);
-
- firstQuotedItem = -1;
- quoteLen = 0;
- }
- }
- }
- while (tmplLine->numElements)
- removeElement(tmplLine, 0);
- if (tmplLine->elements)
- free(tmplLine->elements);
-
- tmplLine->numElements = newLine.numElements;
- tmplLine->elements = newLine.elements;
-}
-
-static void insertElement(struct singleLine * line,
- const char * item, int insertHere,
- struct configFileInfo * cfi)
-{
- struct keywordTypes * kw;
- char indent[2] = "";
-
- /* sanity check */
- if (insertHere > line->numElements) {
- dbgPrintf("insertElement() adjusting insertHere from %d to %d\n",
- insertHere, line->numElements);
- insertHere = line->numElements;
- }
-
- line->elements = realloc(line->elements, (line->numElements + 1) *
- sizeof(*line->elements));
- memmove(&line->elements[insertHere+1],
- &line->elements[insertHere],
- (line->numElements - insertHere) *
- sizeof(*line->elements));
- line->elements[insertHere].item = strdup(item);
-
- kw = getKeywordByType(line->type, cfi);
-
- if (line->numElements == 0) {
- indent[0] = '\0';
- } else if (insertHere == 0) {
- indent[0] = kw->nextChar;
- } else if (kw->separatorChar != '\0') {
- indent[0] = kw->separatorChar;
- } else {
- indent[0] = ' ';
- }
-
- if (insertHere > 0 && line->elements[insertHere-1].indent[0] == '\0') {
- /* move the end-of-line forward */
- line->elements[insertHere].indent =
- line->elements[insertHere-1].indent;
- line->elements[insertHere-1].indent = strdup(indent);
- } else {
- line->elements[insertHere].indent = strdup(indent);
- }
-
- line->numElements++;
-
- dbgPrintf("insertElement(%s, '%s%s', %d)\n",
- line->elements[0].item,
- line->elements[insertHere].item,
- line->elements[insertHere].indent,
- insertHere);
-}
-
-static void removeElement(struct singleLine * line, int removeHere) {
- int i;
-
- /* sanity check */
- if (removeHere >= line->numElements) return;
-
- dbgPrintf("removeElement(%s, %d:%s)\n", line->elements[0].item,
- removeHere, line->elements[removeHere].item);
-
- free(line->elements[removeHere].item);
-
- if (removeHere > 1) {
- /* previous argument gets this argument's post-indentation */
- free(line->elements[removeHere-1].indent);
- line->elements[removeHere-1].indent =
- line->elements[removeHere].indent;
- } else {
- free(line->elements[removeHere].indent);
- }
-
- /* now collapse the array, but don't bother to realloc smaller */
- for (i = removeHere; i < line->numElements - 1; i++)
- line->elements[i] = line->elements[i + 1];
-
- line->numElements--;
-}
-
-int argMatch(const char * one, const char * two) {
- char * first, * second;
- char * chptr;
-
- first = strcpy(alloca(strlen(one) + 1), one);
- second = strcpy(alloca(strlen(two) + 1), two);
-
- chptr = strchr(first, '=');
- if (chptr) *chptr = '\0';
-
- chptr = strchr(second, '=');
- if (chptr) *chptr = '\0';
-
- return strcmp(first, second);
-}
-
-int updateActualImage(struct grubConfig * cfg, const char * image,
- const char * prefix, const char * addArgs,
- const char * removeArgs, int multibootArgs) {
- struct singleEntry * entry;
- struct singleLine * line, * rootLine;
- int index = 0;
- int i, k;
- const char ** newArgs, ** oldArgs;
- const char ** arg;
- int useKernelArgs, useRoot;
- int firstElement;
- int *usedElements;
- int doreplace;
-
- if (!image) return 0;
-
- if (!addArgs) {
- newArgs = malloc(sizeof(*newArgs));
- *newArgs = NULL;
- } else {
- if (poptParseArgvString(addArgs, NULL, &newArgs)) {
- fprintf(stderr,
- _("grubby: error separating arguments '%s'\n"), addArgs);
- return 1;
- }
- }
-
- if (!removeArgs) {
- oldArgs = malloc(sizeof(*oldArgs));
- *oldArgs = NULL;
- } else {
- if (poptParseArgvString(removeArgs, NULL, &oldArgs)) {
- fprintf(stderr,
- _("grubby: error separating arguments '%s'\n"), removeArgs);
- free(newArgs);
- return 1;
- }
- }
-
-
- useKernelArgs = (getKeywordByType(LT_KERNELARGS, cfg->cfi)
- && (!multibootArgs || cfg->cfi->mbConcatArgs));
-
- useRoot = (getKeywordByType(LT_ROOT, cfg->cfi)
- && !multibootArgs);
-
- for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {
-
- if (multibootArgs && !entry->multiboot)
- continue;
-
- /* Determine where to put the args. If this config supports
- * LT_KERNELARGS, use that. Otherwise use
- * LT_HYPER/LT_KERNEL/LT_MBMODULE lines.
- */
- if (useKernelArgs) {
- line = getLineByType(LT_KERNELARGS, entry->lines);
- if (!line) {
- /* no LT_KERNELARGS, need to add it */
- line = addLine(entry, cfg->cfi, LT_KERNELARGS,
- cfg->secondaryIndent, NULL);
- }
- firstElement = 1;
-
- } else if (multibootArgs) {
- line = getLineByType(LT_HYPER, entry->lines);
- if (!line) {
- /* a multiboot entry without LT_HYPER? */
- continue;
- }
- firstElement = 2;
-
- } else {
- line = getLineByType(LT_KERNEL|LT_MBMODULE|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
- if (!line) {
- /* no LT_KERNEL or LT_MBMODULE in this entry? */
- continue;
- }
- firstElement = 2;
- }
-
- /* handle the elilo case which does:
- * append="hypervisor args -- kernel args"
- */
- if (entry->multiboot && cfg->cfi->mbConcatArgs) {
- /* this is a multiboot entry, make sure there's
- * -- on the args line
- */
- for (i = firstElement; i < line->numElements; i++) {
- if (!strcmp(line->elements[i].item, "--"))
- break;
- }
- if (i == line->numElements) {
- /* assume all existing args are kernel args,
- * prepend -- to make it official
+ /* override the inherited value with our own.
+ * This is a little weak because it only applies to elements[1]
*/
- insertElement(line, "--", firstElement, cfg->cfi);
- i = firstElement;
- }
- if (!multibootArgs) {
- /* kernel args start after the -- */
- firstElement = i + 1;
- }
- } else if (cfg->cfi->mbConcatArgs) {
- /* this is a non-multiboot entry, remove hyper args */
- for (i = firstElement; i < line->numElements; i++) {
- if (!strcmp(line->elements[i].item, "--"))
- break;
- }
- if (i < line->numElements) {
- /* remove args up to -- */
- while (strcmp(line->elements[firstElement].item, "--"))
- removeElement(line, firstElement);
- /* remove -- */
- removeElement(line, firstElement);
- }
+ if (newLine->numElements > 1)
+ removeElement(newLine, 1);
+ insertElement(newLine, val, 1, cfi);
+
+ /* but try to keep the rootspec from the template... sigh */
+ if (tmplLine->
+ type & (LT_HYPER | LT_KERNEL | LT_MBMODULE | LT_INITRD |
+ LT_KERNEL_EFI | LT_INITRD_EFI | LT_KERNEL_16 |
+ LT_INITRD_16)) {
+ char *rootspec =
+ getRootSpecifier(tmplLine->elements[1].item);
+ if (rootspec != NULL) {
+ free(newLine->elements[1].item);
+ newLine->elements[1].item =
+ sdupprintf("%s%s", rootspec, val);
+ }
+ }
+ }
+
+ dbgPrintf("addLineTmpl(%s)\n", newLine->numElements ?
+ newLine->elements[0].item : "");
+
+ if (!entry->lines) {
+ /* first one on the list */
+ entry->lines = newLine;
+ } else if (prevLine) {
+ /* add after prevLine */
+ newLine->next = prevLine->next;
+ prevLine->next = newLine;
}
- usedElements = calloc(line->numElements, sizeof(*usedElements));
+ return newLine;
+}
- for (k = 0, arg = newArgs; *arg; arg++, k++) {
+/* val may be NULL */
+struct singleLine *addLine(struct singleEntry *entry,
+ struct configFileInfo *cfi,
+ enum lineType_e type, char *defaultIndent,
+ const char *val)
+{
+ struct singleLine *line, *prev;
+ struct keywordTypes *kw;
+ struct singleLine tmpl;
- doreplace = 1;
- for (i = firstElement; i < line->numElements; i++) {
- if (multibootArgs && cfg->cfi->mbConcatArgs &&
- !strcmp(line->elements[i].item, "--"))
- {
- /* reached the end of hyper args, insert here */
- doreplace = 0;
- break;
+ /* NB: This function shouldn't allocate items on the heap, rather on
+ * the stack since it calls addLineTmpl which will make copies.
+ */
+ if (type == LT_TITLE && cfi->titleBracketed) {
+ /* we're doing a bracketed title (zipl) */
+ tmpl.type = type;
+ tmpl.numElements = 1;
+ tmpl.elements = alloca(sizeof(*tmpl.elements));
+ tmpl.elements[0].item = alloca(strlen(val) + 3);
+ sprintf(tmpl.elements[0].item, "[%s]", val);
+ tmpl.elements[0].indent = "";
+ val = NULL;
+ } else if (type == LT_MENUENTRY) {
+ char *lineend = "--class gnu-linux --class gnu --class os {";
+ if (!val) {
+ fprintf(stderr,
+ "Line type LT_MENUENTRY requires a value\n");
+ abort();
+ }
+ kw = getKeywordByType(type, cfi);
+ if (!kw) {
+ fprintf(stderr,
+ "Looking up keyword for unknown type %d\n",
+ type);
+ abort();
+ }
+ tmpl.indent = "";
+ tmpl.type = type;
+ tmpl.numElements = 3;
+ tmpl.elements =
+ alloca(sizeof(*tmpl.elements) * tmpl.numElements);
+ tmpl.elements[0].item = kw->key;
+ tmpl.elements[0].indent = alloca(2);
+ sprintf(tmpl.elements[0].indent, "%c", kw->nextChar);
+ tmpl.elements[1].item = (char *)val;
+ tmpl.elements[1].indent = alloca(2);
+ sprintf(tmpl.elements[1].indent, "%c", kw->nextChar);
+ tmpl.elements[2].item = alloca(strlen(lineend) + 1);
+ strcpy(tmpl.elements[2].item, lineend);
+ tmpl.elements[2].indent = "";
+ } else {
+ kw = getKeywordByType(type, cfi);
+ if (!kw) {
+ fprintf(stderr,
+ "Looking up keyword for unknown type %d\n",
+ type);
+ abort();
+ }
+ tmpl.type = type;
+ tmpl.numElements = val ? 2 : 1;
+ tmpl.elements =
+ alloca(sizeof(*tmpl.elements) * tmpl.numElements);
+ tmpl.elements[0].item = kw->key;
+ tmpl.elements[0].indent = alloca(2);
+ sprintf(tmpl.elements[0].indent, "%c", kw->nextChar);
+ if (val) {
+ tmpl.elements[1].item = (char *)val;
+ tmpl.elements[1].indent = "";
}
- if (usedElements[i])
- continue;
- if (!argMatch(line->elements[i].item, *arg)) {
- usedElements[i]=1;
- break;
- }
- }
-
- if (i < line->numElements && doreplace) {
- /* direct replacement */
+ }
+
+ /* The last non-empty line gives us the indention to us and the line
+ * to insert after. Note that comments are considered empty lines,
+ * which may not be ideal? If there are no lines or we are looking at
+ * the first line, we use defaultIndent (the first line is normally
+ * indented differently from the rest) */
+ for (line = entry->lines, prev = NULL; line; line = line->next) {
+ if (line->numElements)
+ prev = line;
+ /* fall back on the last line if prev isn't otherwise set */
+ if (!line->next && !prev)
+ prev = line;
+ }
+
+ struct singleLine *menuEntry;
+ menuEntry = getLineByType(LT_MENUENTRY, entry->lines);
+ if (tmpl.type == LT_ENTRY_END) {
+ if (menuEntry)
+ tmpl.indent = menuEntry->indent;
+ else
+ tmpl.indent = defaultIndent ? : "";
+ } else if (tmpl.type != LT_MENUENTRY) {
+ if (menuEntry)
+ tmpl.indent = "\t";
+ else if (prev == entry->lines)
+ tmpl.indent = defaultIndent ? : "";
+ else
+ tmpl.indent = prev->indent;
+ }
+
+ return addLineTmpl(entry, &tmpl, prev, val, cfi);
+}
+
+void removeLine(struct singleEntry *entry, struct singleLine *line)
+{
+ struct singleLine *prev;
+ int i;
+
+ for (i = 0; i < line->numElements; i++) {
free(line->elements[i].item);
- line->elements[i].item = strdup(*arg);
-
- } else if (useRoot && !strncmp(*arg, "root=/dev/", 10)) {
- /* root= replacement */
- rootLine = getLineByType(LT_ROOT, entry->lines);
- if (rootLine) {
- free(rootLine->elements[1].item);
- rootLine->elements[1].item = strdup(*arg + 5);
+ free(line->elements[i].indent);
+ }
+ free(line->elements);
+ free(line->indent);
+
+ if (line == entry->lines) {
+ entry->lines = line->next;
+ } else {
+ prev = entry->lines;
+ while (prev->next != line)
+ prev = prev->next;
+ prev->next = line->next;
+ }
+
+ free(line);
+}
+
+static void requote(struct singleLine *tmplLine, struct configFileInfo *cfi)
+{
+ struct singleLine newLine = {
+ .indent = tmplLine->indent,
+ .type = tmplLine->type,
+ .next = tmplLine->next,
+ };
+ int firstQuotedItem = -1;
+ int quoteLen = 0;
+ int j;
+ int element = 0;
+ char *c;
+
+ c = malloc(strlen(tmplLine->elements[0].item) + 1);
+ strcpy(c, tmplLine->elements[0].item);
+ insertElement(&newLine, c, element++, cfi);
+ free(c);
+ c = NULL;
+
+ for (j = 1; j < tmplLine->numElements; j++) {
+ if (firstQuotedItem == -1) {
+ quoteLen += strlen(tmplLine->elements[j].item);
+
+ if (isquote(tmplLine->elements[j].item[0])) {
+ firstQuotedItem = j;
+ quoteLen +=
+ strlen(tmplLine->elements[j].indent);
+ } else {
+ c = malloc(quoteLen + 1);
+ strcpy(c, tmplLine->elements[j].item);
+ insertElement(&newLine, c, element++, cfi);
+ free(c);
+ quoteLen = 0;
+ }
} else {
- rootLine = addLine(entry, cfg->cfi, LT_ROOT,
- cfg->secondaryIndent, *arg + 5);
- }
- }
-
- else {
- /* insert/append */
- insertElement(line, *arg, i, cfg->cfi);
- usedElements = realloc(usedElements, line->numElements *
- sizeof(*usedElements));
- memmove(&usedElements[i + 1], &usedElements[i],
- line->numElements - i - 1);
- usedElements[i] = 1;
-
- /* if we updated a root= here even though there is a
- LT_ROOT available we need to remove the LT_ROOT entry
- (this will happen if we switch from a device to a label) */
- if (useRoot && !strncmp(*arg, "root=", 5)) {
- rootLine = getLineByType(LT_ROOT, entry->lines);
- if (rootLine)
- removeLine(entry, rootLine);
- }
- }
- }
-
- free(usedElements);
-
- for (arg = oldArgs; *arg; arg++) {
- for (i = firstElement; i < line->numElements; i++) {
- if (multibootArgs && cfg->cfi->mbConcatArgs &&
- !strcmp(line->elements[i].item, "--"))
- /* reached the end of hyper args, stop here */
- break;
- if (!argMatch(line->elements[i].item, *arg)) {
- removeElement(line, i);
- break;
- }
- }
- /* handle removing LT_ROOT line too */
- if (useRoot && !strncmp(*arg, "root=", 5)) {
- rootLine = getLineByType(LT_ROOT, entry->lines);
- if (rootLine)
- removeLine(entry, rootLine);
- }
- }
-
- if (line->numElements == 1) {
- /* don't need the line at all (note it has to be a
- LT_KERNELARGS for this to happen */
- removeLine(entry, line);
- }
- }
-
- free(newArgs);
- free(oldArgs);
-
- return 0;
+ int itemlen = strlen(tmplLine->elements[j].item);
+ quoteLen += itemlen;
+ quoteLen += strlen(tmplLine->elements[j].indent);
+
+ if (isquote(tmplLine->elements[j].item[itemlen - 1])) {
+ c = malloc(quoteLen + 1);
+ c[0] = '\0';
+ for (int i = firstQuotedItem; i < j + 1; i++) {
+ strcat(c, tmplLine->elements[i].item);
+ strcat(c, tmplLine->elements[i].indent);
+ }
+ insertElement(&newLine, c, element++, cfi);
+ free(c);
+
+ firstQuotedItem = -1;
+ quoteLen = 0;
+ }
+ }
+ }
+ while (tmplLine->numElements)
+ removeElement(tmplLine, 0);
+ if (tmplLine->elements)
+ free(tmplLine->elements);
+
+ tmplLine->numElements = newLine.numElements;
+ tmplLine->elements = newLine.elements;
}
-int updateImage(struct grubConfig * cfg, const char * image,
- const char * prefix, const char * addArgs,
- const char * removeArgs,
- const char * addMBArgs, const char * removeMBArgs) {
- int rc = 0;
+static void insertElement(struct singleLine *line,
+ const char *item, int insertHere,
+ struct configFileInfo *cfi)
+{
+ struct keywordTypes *kw;
+ char indent[2] = "";
+
+ /* sanity check */
+ if (insertHere > line->numElements) {
+ dbgPrintf
+ ("insertElement() adjusting insertHere from %d to %d\n",
+ insertHere, line->numElements);
+ insertHere = line->numElements;
+ }
- if (!image) return rc;
+ line->elements = realloc(line->elements, (line->numElements + 1) *
+ sizeof(*line->elements));
+ memmove(&line->elements[insertHere + 1],
+ &line->elements[insertHere],
+ (line->numElements - insertHere) * sizeof(*line->elements));
+ line->elements[insertHere].item = strdup(item);
+
+ kw = getKeywordByType(line->type, cfi);
+
+ if (line->numElements == 0) {
+ indent[0] = '\0';
+ } else if (insertHere == 0) {
+ indent[0] = kw->nextChar;
+ } else if (kw->separatorChar != '\0') {
+ indent[0] = kw->separatorChar;
+ } else {
+ indent[0] = ' ';
+ }
- /* update the main args first... */
- if (addArgs || removeArgs)
- rc = updateActualImage(cfg, image, prefix, addArgs, removeArgs, 0);
- if (rc) return rc;
+ if (insertHere > 0 && line->elements[insertHere - 1].indent[0] == '\0') {
+ /* move the end-of-line forward */
+ line->elements[insertHere].indent =
+ line->elements[insertHere - 1].indent;
+ line->elements[insertHere - 1].indent = strdup(indent);
+ } else {
+ line->elements[insertHere].indent = strdup(indent);
+ }
- /* and now any multiboot args */
- if (addMBArgs || removeMBArgs)
- rc = updateActualImage(cfg, image, prefix, addMBArgs, removeMBArgs, 1);
- return rc;
+ line->numElements++;
+
+ dbgPrintf("insertElement(%s, '%s%s', %d)\n",
+ line->elements[0].item,
+ line->elements[insertHere].item,
+ line->elements[insertHere].indent, insertHere);
}
-int addMBInitrd(struct grubConfig * cfg, const char *newMBKernel,
- const char * image, const char * prefix, const char * initrd,
- const char * title) {
- struct singleEntry * entry;
- struct singleLine * line, * kernelLine, *endLine = NULL;
- int index = 0;
-
- if (!image) return 0;
-
- for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {
- kernelLine = getLineByType(LT_MBMODULE, entry->lines);
- if (!kernelLine) continue;
-
- /* if title is supplied, the entry's title must match it. */
- if (title) {
- char *linetitle;
-
- line = getLineByType(LT_TITLE|LT_MENUENTRY, entry->lines);
- if (!line)
- continue;
-
- linetitle = extractTitle(cfg, line);
- if (!linetitle)
- continue;
- if (strcmp(title, linetitle)) {
- free(linetitle);
- continue;
- }
- free(linetitle);
- }
-
- if (prefix) {
- int prefixLen = strlen(prefix);
- if (!strncmp(initrd, prefix, prefixLen))
- initrd += prefixLen;
- }
- endLine = getLineByType(LT_ENTRY_END, entry->lines);
- if (endLine)
- removeLine(entry, endLine);
- line = addLine(entry, cfg->cfi, preferredLineType(LT_MBMODULE,cfg->cfi),
- kernelLine->indent, initrd);
- if (!line)
- return 1;
- if (endLine) {
- line = addLine(entry, cfg->cfi, LT_ENTRY_END, "", NULL);
- if (!line)
- return 1;
+static void removeElement(struct singleLine *line, int removeHere)
+{
+ int i;
+
+ /* sanity check */
+ if (removeHere >= line->numElements)
+ return;
+
+ dbgPrintf("removeElement(%s, %d:%s)\n", line->elements[0].item,
+ removeHere, line->elements[removeHere].item);
+
+ free(line->elements[removeHere].item);
+
+ if (removeHere > 1) {
+ /* previous argument gets this argument's post-indentation */
+ free(line->elements[removeHere - 1].indent);
+ line->elements[removeHere - 1].indent =
+ line->elements[removeHere].indent;
+ } else {
+ free(line->elements[removeHere].indent);
}
- break;
- }
+ /* now collapse the array, but don't bother to realloc smaller */
+ for (i = removeHere; i < line->numElements - 1; i++)
+ line->elements[i] = line->elements[i + 1];
- return 0;
+ line->numElements--;
}
-int updateInitrd(struct grubConfig * cfg, const char * image,
- const char * prefix, const char * initrd, const char * title) {
- struct singleEntry * entry;
- struct singleLine * line, * kernelLine, *endLine = NULL;
- int index = 0;
-
- if (!image) return 0;
-
- for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {
- kernelLine = getLineByType(LT_KERNEL|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
- if (!kernelLine) continue;
-
- /* if title is supplied, the entry's title must match it. */
- if (title) {
- char *linetitle;
-
- line = getLineByType(LT_TITLE|LT_MENUENTRY, entry->lines);
- if (!line)
- continue;
-
- linetitle = extractTitle(cfg, line);
- if (!linetitle)
- continue;
- if (strcmp(title, linetitle)) {
- free(linetitle);
- continue;
- }
- free(linetitle);
- }
-
- line = getLineByType(LT_INITRD|LT_INITRD_EFI|LT_INITRD_16, entry->lines);
- if (line)
- removeLine(entry, line);
- if (prefix) {
- int prefixLen = strlen(prefix);
- if (!strncmp(initrd, prefix, prefixLen))
- initrd += prefixLen;
- }
- endLine = getLineByType(LT_ENTRY_END, entry->lines);
- if (endLine)
- removeLine(entry, endLine);
- enum lineType_e lt;
- switch(kernelLine->type) {
- case LT_KERNEL:
- lt = LT_INITRD;
- break;
- case LT_KERNEL_EFI:
- lt = LT_INITRD_EFI;
+int argMatch(const char *one, const char *two)
+{
+ char *first, *second;
+ char *chptr;
+
+ first = strcpy(alloca(strlen(one) + 1), one);
+ second = strcpy(alloca(strlen(two) + 1), two);
+
+ chptr = strchr(first, '=');
+ if (chptr)
+ *chptr = '\0';
+
+ chptr = strchr(second, '=');
+ if (chptr)
+ *chptr = '\0';
+
+ return strcmp(first, second);
+}
+
+int updateActualImage(struct grubConfig *cfg, const char *image,
+ const char *prefix, const char *addArgs,
+ const char *removeArgs, int multibootArgs)
+{
+ struct singleEntry *entry;
+ struct singleLine *line, *rootLine;
+ int index = 0;
+ int i, k;
+ const char **newArgs, **oldArgs;
+ const char **arg;
+ int useKernelArgs, useRoot;
+ int firstElement;
+ int *usedElements;
+ int doreplace;
+
+ if (!image)
+ return 0;
+
+ if (!addArgs) {
+ newArgs = malloc(sizeof(*newArgs));
+ *newArgs = NULL;
+ } else {
+ if (poptParseArgvString(addArgs, NULL, &newArgs)) {
+ fprintf(stderr,
+ _("grubby: error separating arguments '%s'\n"),
+ addArgs);
+ return 1;
+ }
+ }
+
+ if (!removeArgs) {
+ oldArgs = malloc(sizeof(*oldArgs));
+ *oldArgs = NULL;
+ } else {
+ if (poptParseArgvString(removeArgs, NULL, &oldArgs)) {
+ fprintf(stderr,
+ _("grubby: error separating arguments '%s'\n"),
+ removeArgs);
+ free(newArgs);
+ return 1;
+ }
+ }
+
+ useKernelArgs = (getKeywordByType(LT_KERNELARGS, cfg->cfi)
+ && (!multibootArgs || cfg->cfi->mbConcatArgs));
+
+ useRoot = (getKeywordByType(LT_ROOT, cfg->cfi)
+ && !multibootArgs);
+
+ for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {
+
+ if (multibootArgs && !entry->multiboot)
+ continue;
+
+ /* Determine where to put the args. If this config supports
+ * LT_KERNELARGS, use that. Otherwise use
+ * LT_HYPER/LT_KERNEL/LT_MBMODULE lines.
+ */
+ if (useKernelArgs) {
+ line = getLineByType(LT_KERNELARGS, entry->lines);
+ if (!line) {
+ /* no LT_KERNELARGS, need to add it */
+ line = addLine(entry, cfg->cfi, LT_KERNELARGS,
+ cfg->secondaryIndent, NULL);
+ }
+ firstElement = 1;
+
+ } else if (multibootArgs) {
+ line = getLineByType(LT_HYPER, entry->lines);
+ if (!line) {
+ /* a multiboot entry without LT_HYPER? */
+ continue;
+ }
+ firstElement = 2;
+
+ } else {
+ line =
+ getLineByType(LT_KERNEL | LT_MBMODULE |
+ LT_KERNEL_EFI | LT_KERNEL_16,
+ entry->lines);
+ if (!line) {
+ /* no LT_KERNEL or LT_MBMODULE in this entry? */
+ continue;
+ }
+ firstElement = 2;
+ }
+
+ /* handle the elilo case which does:
+ * append="hypervisor args -- kernel args"
+ */
+ if (entry->multiboot && cfg->cfi->mbConcatArgs) {
+ /* this is a multiboot entry, make sure there's
+ * -- on the args line
+ */
+ for (i = firstElement; i < line->numElements; i++) {
+ if (!strcmp(line->elements[i].item, "--"))
+ break;
+ }
+ if (i == line->numElements) {
+ /* assume all existing args are kernel args,
+ * prepend -- to make it official
+ */
+ insertElement(line, "--", firstElement,
+ cfg->cfi);
+ i = firstElement;
+ }
+ if (!multibootArgs) {
+ /* kernel args start after the -- */
+ firstElement = i + 1;
+ }
+ } else if (cfg->cfi->mbConcatArgs) {
+ /* this is a non-multiboot entry, remove hyper args */
+ for (i = firstElement; i < line->numElements; i++) {
+ if (!strcmp(line->elements[i].item, "--"))
+ break;
+ }
+ if (i < line->numElements) {
+ /* remove args up to -- */
+ while (strcmp
+ (line->elements[firstElement].item,
+ "--"))
+ removeElement(line, firstElement);
+ /* remove -- */
+ removeElement(line, firstElement);
+ }
+ }
+
+ usedElements = calloc(line->numElements, sizeof(*usedElements));
+
+ for (k = 0, arg = newArgs; *arg; arg++, k++) {
+
+ doreplace = 1;
+ for (i = firstElement; i < line->numElements; i++) {
+ if (multibootArgs && cfg->cfi->mbConcatArgs &&
+ !strcmp(line->elements[i].item, "--")) {
+ /* reached the end of hyper args, insert here */
+ doreplace = 0;
+ break;
+ }
+ if (usedElements[i])
+ continue;
+ if (!argMatch(line->elements[i].item, *arg)) {
+ usedElements[i] = 1;
+ break;
+ }
+ }
+
+ if (i < line->numElements && doreplace) {
+ /* direct replacement */
+ free(line->elements[i].item);
+ line->elements[i].item = strdup(*arg);
+
+ } else if (useRoot && !strncmp(*arg, "root=/dev/", 10)) {
+ /* root= replacement */
+ rootLine = getLineByType(LT_ROOT, entry->lines);
+ if (rootLine) {
+ free(rootLine->elements[1].item);
+ rootLine->elements[1].item =
+ strdup(*arg + 5);
+ } else {
+ rootLine =
+ addLine(entry, cfg->cfi, LT_ROOT,
+ cfg->secondaryIndent,
+ *arg + 5);
+ }
+ }
+
+ else {
+ /* insert/append */
+ insertElement(line, *arg, i, cfg->cfi);
+ usedElements =
+ realloc(usedElements,
+ line->numElements *
+ sizeof(*usedElements));
+ memmove(&usedElements[i + 1], &usedElements[i],
+ line->numElements - i - 1);
+ usedElements[i] = 1;
+
+ /* if we updated a root= here even though
+ * there is a LT_ROOT available we need to
+ * remove the LT_ROOT entry (this will happen
+ * if we switch from a device to a label) */
+ if (useRoot && !strncmp(*arg, "root=", 5)) {
+ rootLine =
+ getLineByType(LT_ROOT,
+ entry->lines);
+ if (rootLine)
+ removeLine(entry, rootLine);
+ }
+ }
+ }
+
+ free(usedElements);
+
+ for (arg = oldArgs; *arg; arg++) {
+ for (i = firstElement; i < line->numElements; i++) {
+ if (multibootArgs && cfg->cfi->mbConcatArgs &&
+ !strcmp(line->elements[i].item, "--"))
+ /* reached the end of hyper args, stop here */
+ break;
+ if (!argMatch(line->elements[i].item, *arg)) {
+ removeElement(line, i);
+ break;
+ }
+ }
+ /* handle removing LT_ROOT line too */
+ if (useRoot && !strncmp(*arg, "root=", 5)) {