Magellan Linux

Diff of /trunk/grubby/grubby.c

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

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

Legend:
Removed from v.2977  
changed lines
  Added in v.3135