Magellan Linux

Diff of /trunk/grubby/grubby.c

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

revision 1854 by niro, Mon Jul 2 13:12:32 2012 UTC revision 2718 by niro, Wed Jul 16 11:01:36 2014 UTC
# Line 36  Line 36 
36  #include <signal.h>  #include <signal.h>
37  #include <blkid/blkid.h>  #include <blkid/blkid.h>
38    
39    #include "log.h"
40    
41  #ifndef DEBUG  #ifndef DEBUG
42  #define DEBUG 0  #define DEBUG 0
43  #endif  #endif
# Line 56  int debug = 0; /* Currently just for tem Line 58  int debug = 0; /* Currently just for tem
58  #define NOOP_OPCODE 0x90  #define NOOP_OPCODE 0x90
59  #define JMP_SHORT_OPCODE 0xeb  #define JMP_SHORT_OPCODE 0xeb
60    
61    int isEfi = 0;
62    
63    #if defined(__aarch64__)
64    #define isEfiOnly 1
65    #else
66    #define isEfiOnly 0
67    #endif
68    
69    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;
# Line 82  enum lineType_e { Line 94  enum lineType_e {
94      LT_MENUENTRY    = 1 << 17,      LT_MENUENTRY    = 1 << 17,
95      LT_ENTRY_END    = 1 << 18,      LT_ENTRY_END    = 1 << 18,
96      LT_SET_VARIABLE = 1 << 19,      LT_SET_VARIABLE = 1 << 19,
97      LT_UNKNOWN      = 1 << 20,      LT_KERNEL_EFI   = 1 << 20,
98        LT_INITRD_EFI   = 1 << 21,
99        LT_KERNEL_16    = 1 << 22,
100        LT_INITRD_16    = 1 << 23,
101        LT_DEVTREE      = 1 << 24,
102        LT_UNKNOWN      = 1 << 25,
103  };  };
104    
105  struct singleLine {  struct singleLine {
# Line 111  struct singleEntry { Line 128  struct singleEntry {
128  #define NEED_ARGS    (1 << 3)  #define NEED_ARGS    (1 << 3)
129  #define NEED_MB      (1 << 4)  #define NEED_MB      (1 << 4)
130  #define NEED_END     (1 << 5)  #define NEED_END     (1 << 5)
131    #define NEED_DEVTREE (1 << 6)
132    
133  #define MAIN_DEFAULT    (1 << 0)  #define MAIN_DEFAULT    (1 << 0)
134  #define DEFAULT_SAVED       -2  #define DEFAULT_SAVED       -2
# Line 128  struct configFileInfo; Line 146  struct configFileInfo;
146  typedef const char *(*findConfigFunc)(struct configFileInfo *);  typedef const char *(*findConfigFunc)(struct configFileInfo *);
147  typedef const int (*writeLineFunc)(struct configFileInfo *,  typedef const int (*writeLineFunc)(struct configFileInfo *,
148   struct singleLine *line);   struct singleLine *line);
149    typedef char *(*getEnvFunc)(struct configFileInfo *, char *name);
150    typedef int (*setEnvFunc)(struct configFileInfo *, char *name, char *value);
151    
152  struct configFileInfo {  struct configFileInfo {
153      char * defaultConfig;      char * defaultConfig;
154      findConfigFunc findConfig;      findConfigFunc findConfig;
155      writeLineFunc writeLine;      writeLineFunc writeLine;
156        getEnvFunc getEnv;
157        setEnvFunc setEnv;
158      struct keywordTypes * keywords;      struct keywordTypes * keywords;
159        int caseInsensitive;
160      int defaultIsIndex;      int defaultIsIndex;
161      int defaultIsVariable;      int defaultIsVariable;
162      int defaultSupportSaved;      int defaultSupportSaved;
163        int defaultIsSaved;
164        int defaultIsUnquoted;
165      enum lineType_e entryStart;      enum lineType_e entryStart;
166      enum lineType_e entryEnd;      enum lineType_e entryEnd;
167      int needsBootPrefix;      int needsBootPrefix;
# Line 148  struct configFileInfo { Line 173  struct configFileInfo {
173      int mbInitRdIsModule;      int mbInitRdIsModule;
174      int mbConcatArgs;      int mbConcatArgs;
175      int mbAllowExtraInitRds;      int mbAllowExtraInitRds;
176        char *envFile;
177  };  };
178    
179  struct keywordTypes grubKeywords[] = {  struct keywordTypes grubKeywords[] = {
# Line 167  const char *grubFindConfig(struct config Line 193  const char *grubFindConfig(struct config
193   "/boot/grub/grub.conf",   "/boot/grub/grub.conf",
194   "/boot/grub/menu.lst",   "/boot/grub/menu.lst",
195   "/etc/grub.conf",   "/etc/grub.conf",
196     "/boot/grub2/grub.cfg",
197     "/boot/grub2-efi/grub.cfg",
198   NULL   NULL
199      };      };
200      static int i = -1;      static int i = -1;
# Line 205  struct keywordTypes grub2Keywords[] = { Line 233  struct keywordTypes grub2Keywords[] = {
233      { "default",    LT_DEFAULT,     ' ' },      { "default",    LT_DEFAULT,     ' ' },
234      { "fallback",   LT_FALLBACK,    ' ' },      { "fallback",   LT_FALLBACK,    ' ' },
235      { "linux",      LT_KERNEL,      ' ' },      { "linux",      LT_KERNEL,      ' ' },
236        { "linuxefi",   LT_KERNEL_EFI,  ' ' },
237        { "linux16",    LT_KERNEL_16,   ' ' },
238      { "initrd",     LT_INITRD,      ' ', ' ' },      { "initrd",     LT_INITRD,      ' ', ' ' },
239        { "initrdefi",  LT_INITRD_EFI,  ' ', ' ' },
240        { "initrd16",   LT_INITRD_16,   ' ', ' ' },
241      { "module",     LT_MBMODULE,    ' ' },      { "module",     LT_MBMODULE,    ' ' },
242      { "kernel",     LT_HYPER,       ' ' },      { "kernel",     LT_HYPER,       ' ' },
243        { "devicetree", LT_DEVTREE,  ' ' },
244      { NULL, 0, 0 },      { NULL, 0, 0 },
245  };  };
246    
# Line 219  const char *grub2FindConfig(struct confi Line 252  const char *grub2FindConfig(struct confi
252      };      };
253      static int i = -1;      static int i = -1;
254      static const char *grub_cfg = "/boot/grub/grub.cfg";      static const char *grub_cfg = "/boot/grub/grub.cfg";
255        int rc = -1;
256    
257      if (i == -1) {      if (i == -1) {
258   for (i = 0; configFiles[i] != NULL; i++) {   for (i = 0; configFiles[i] != NULL; i++) {
259      dbgPrintf("Checking \"%s\": ", configFiles[i]);      dbgPrintf("Checking \"%s\": ", configFiles[i]);
260      if (!access(configFiles[i], R_OK)) {      if ((rc = access(configFiles[i], R_OK))) {
261     if (errno == EACCES) {
262        printf("Unable to access bootloader configuration file "
263           "\"%s\": %m\n", configFiles[i]);
264        exit(1);
265     }
266     continue;
267        } else {
268   dbgPrintf("found\n");   dbgPrintf("found\n");
269   return configFiles[i];   return configFiles[i];
270      }      }
# Line 242  const char *grub2FindConfig(struct confi Line 283  const char *grub2FindConfig(struct confi
283      return configFiles[i];      return configFiles[i];
284  }  }
285    
286    /* kind of hacky.  It'll give the first 1024 bytes, ish. */
287    static char *grub2GetEnv(struct configFileInfo *info, char *name)
288    {
289        static char buf[1025];
290        char *s = NULL;
291        char *ret = NULL;
292        char *envFile = info->envFile ? info->envFile : "/boot/grub/grubenv";
293        int rc = asprintf(&s, "grub-editenv %s list | grep '^%s='", envFile, name);
294    
295        if (rc < 0)
296     return NULL;
297    
298        FILE *f = popen(s, "r");
299        if (!f)
300     goto out;
301    
302        memset(buf, '\0', sizeof (buf));
303        ret = fgets(buf, 1024, f);
304        pclose(f);
305    
306        if (ret) {
307     ret += strlen(name) + 1;
308     ret[strlen(ret) - 1] = '\0';
309        }
310        dbgPrintf("grub2GetEnv(%s): %s\n", name, ret);
311    out:
312        free(s);
313        return ret;
314    }
315    
316    static int sPopCount(const char *s, const char *c)
317    {
318        int ret = 0;
319        if (!s)
320     return -1;
321        for (int i = 0; s[i] != '\0'; i++)
322     for (int j = 0; c[j] != '\0'; j++)
323        if (s[i] == c[j])
324     ret++;
325        return ret;
326    }
327    
328    static char *shellEscape(const char *s)
329    {
330        int l = strlen(s) + sPopCount(s, "'") * 2;
331    
332        char *ret = calloc(l+1, sizeof (*ret));
333        if (!ret)
334     return NULL;
335        for (int i = 0, j = 0; s[i] != '\0'; i++, j++) {
336     if (s[i] == '\'')
337        ret[j++] = '\\';
338     ret[j] = s[i];
339        }
340        return ret;
341    }
342    
343    static void unquote(char *s)
344    {
345        int l = strlen(s);
346    
347        if ((s[l-1] == '\'' && s[0] == '\'') || (s[l-1] == '"' && s[0] == '"')) {
348     memmove(s, s+1, l-2);
349     s[l-2] = '\0';
350        }
351    }
352    
353    static int grub2SetEnv(struct configFileInfo *info, char *name, char *value)
354    {
355        char *s = NULL;
356        int rc = 0;
357        char *envFile = info->envFile ? info->envFile : "/boot/grub/grubenv";
358    
359        unquote(value);
360        value = shellEscape(value);
361        if (!value)
362        return -1;
363    
364        rc = asprintf(&s, "grub-editenv %s set '%s=%s'", envFile, name, value);
365        free(value);
366        if (rc <0)
367     return -1;
368    
369        dbgPrintf("grub2SetEnv(%s): %s\n", name, s);
370        rc = system(s);
371        free(s);
372        return rc;
373    }
374    
375    /* this is a gigantic hack to avoid clobbering grub2 variables... */
376    static int is_special_grub2_variable(const char *name)
377    {
378        if (!strcmp(name,"\"${next_entry}\""))
379     return 1;
380        if (!strcmp(name,"\"${prev_saved_entry}\""))
381     return 1;
382        return 0;
383    }
384    
385  int sizeOfSingleLine(struct singleLine * line) {  int sizeOfSingleLine(struct singleLine * line) {
386    int count = 0;    int count = 0;
387    
# Line 271  static int isquote(char q) Line 411  static int isquote(char q)
411      return 0;      return 0;
412  }  }
413    
414    static int iskernel(enum lineType_e type) {
415        return (type == LT_KERNEL || type == LT_KERNEL_EFI || type == LT_KERNEL_16);
416    }
417    
418    static int isinitrd(enum lineType_e type) {
419        return (type == LT_INITRD || type == LT_INITRD_EFI || type == LT_INITRD_16);
420    }
421    
422  char *grub2ExtractTitle(struct singleLine * line) {  char *grub2ExtractTitle(struct singleLine * line) {
423      char * current;      char * current;
424      char * current_indent;      char * current_indent;
# Line 328  char *grub2ExtractTitle(struct singleLin Line 476  char *grub2ExtractTitle(struct singleLin
476    
477  struct configFileInfo grub2ConfigType = {  struct configFileInfo grub2ConfigType = {
478      .findConfig = grub2FindConfig,      .findConfig = grub2FindConfig,
479        .getEnv = grub2GetEnv,
480        .setEnv = grub2SetEnv,
481      .keywords = grub2Keywords,      .keywords = grub2Keywords,
482      .defaultIsIndex = 1,      .defaultIsIndex = 1,
483      .defaultSupportSaved = 1,      .defaultSupportSaved = 1,
# Line 482  struct configFileInfo ziplConfigType = { Line 632  struct configFileInfo ziplConfigType = {
632  struct configFileInfo extlinuxConfigType = {  struct configFileInfo extlinuxConfigType = {
633      .defaultConfig = "/boot/extlinux/extlinux.conf",      .defaultConfig = "/boot/extlinux/extlinux.conf",
634      .keywords = extlinuxKeywords,      .keywords = extlinuxKeywords,
635        .caseInsensitive = 1,
636      .entryStart = LT_TITLE,      .entryStart = LT_TITLE,
637      .needsBootPrefix = 1,      .needsBootPrefix = 1,
638      .maxTitleLength = 255,      .maxTitleLength = 255,
639      .mbAllowExtraInitRds = 1,      .mbAllowExtraInitRds = 1,
640        .defaultIsUnquoted = 1,
641  };  };
642    
643  struct grubConfig {  struct grubConfig {
# Line 506  struct singleEntry * findEntryByIndex(st Line 658  struct singleEntry * findEntryByIndex(st
658  struct singleEntry * findEntryByPath(struct grubConfig * cfg,  struct singleEntry * findEntryByPath(struct grubConfig * cfg,
659       const char * path, const char * prefix,       const char * path, const char * prefix,
660       int * index);       int * index);
661    struct singleEntry * findEntryByTitle(struct grubConfig * cfg, char *title,
662          int * index);
663  static int readFile(int fd, char ** bufPtr);  static int readFile(int fd, char ** bufPtr);
664  static void lineInit(struct singleLine * line);  static void lineInit(struct singleLine * line);
665  struct singleLine * lineDup(struct singleLine * line);  struct singleLine * lineDup(struct singleLine * line);
# Line 570  static char * sdupprintf(const char *for Line 724  static char * sdupprintf(const char *for
724      return buf;      return buf;
725  }  }
726    
727    static enum lineType_e preferredLineType(enum lineType_e type,
728     struct configFileInfo *cfi) {
729        if (isEfi && cfi == &grub2ConfigType) {
730     switch (type) {
731     case LT_KERNEL:
732        return isEfiOnly ? LT_KERNEL : LT_KERNEL_EFI;
733     case LT_INITRD:
734        return isEfiOnly ? LT_INITRD : LT_INITRD_EFI;
735     default:
736        return type;
737     }
738    #if defined(__i386__) || defined(__x86_64__)
739        } else if (cfi == &grub2ConfigType) {
740     switch (type) {
741     case LT_KERNEL:
742        return LT_KERNEL_16;
743     case LT_INITRD:
744        return LT_INITRD_16;
745     default:
746        return type;
747     }
748    #endif
749        }
750        return type;
751    }
752    
753  static struct keywordTypes * getKeywordByType(enum lineType_e type,  static struct keywordTypes * getKeywordByType(enum lineType_e type,
754        struct configFileInfo * cfi) {        struct configFileInfo * cfi) {
755      for (struct keywordTypes *kw = cfi->keywords; kw->key; kw++) {      for (struct keywordTypes *kw = cfi->keywords; kw->key; kw++) {
# Line 603  static char * getuuidbydev(char *device) Line 783  static char * getuuidbydev(char *device)
783  static enum lineType_e getTypeByKeyword(char * keyword,  static enum lineType_e getTypeByKeyword(char * keyword,
784   struct configFileInfo * cfi) {   struct configFileInfo * cfi) {
785      for (struct keywordTypes *kw = cfi->keywords; kw->key; kw++) {      for (struct keywordTypes *kw = cfi->keywords; kw->key; kw++) {
786   if (!strcmp(keyword, kw->key))   if (cfi->caseInsensitive) {
787      return kw->type;      if (!strcasecmp(keyword, kw->key))
788                    return kw->type;
789     } else {
790        if (!strcmp(keyword, kw->key))
791            return kw->type;
792     }
793      }      }
794      return LT_UNKNOWN;      return LT_UNKNOWN;
795  }  }
# Line 643  static int isEntryStart(struct singleLin Line 828  static int isEntryStart(struct singleLin
828  /* extract the title from within brackets (for zipl) */  /* extract the title from within brackets (for zipl) */
829  static char * extractTitle(struct singleLine * line) {  static char * extractTitle(struct singleLine * line) {
830      /* bracketed title... let's extract it (leaks a byte) */      /* bracketed title... let's extract it (leaks a byte) */
831      char * title;      char * title = NULL;
832      title = strdup(line->elements[0].item);      if (line->type == LT_TITLE) {
833      title++;   title = strdup(line->elements[0].item);
834      *(title + strlen(title) - 1) = '\0';   title++;
835     *(title + strlen(title) - 1) = '\0';
836        } else if (line->type == LT_MENUENTRY)
837     title = strdup(line->elements[1].item);
838        else
839     return NULL;
840      return title;      return title;
841  }  }
842    
# Line 904  static int getNextLine(char ** bufPtr, s Line 1094  static int getNextLine(char ** bufPtr, s
1094      return 0;      return 0;
1095  }  }
1096    
1097    static int isnumber(const char *s)
1098    {
1099        int i;
1100        for (i = 0; s[i] != '\0'; i++)
1101     if (s[i] < '0' || s[i] > '9')
1102        return 0;
1103        return i;
1104    }
1105    
1106  static struct grubConfig * readConfig(const char * inName,  static struct grubConfig * readConfig(const char * inName,
1107        struct configFileInfo * cfi) {        struct configFileInfo * cfi) {
1108      int in;      int in;
# Line 918  static struct grubConfig * readConfig(co Line 1117  static struct grubConfig * readConfig(co
1117      int len;      int len;
1118      char * buf;      char * buf;
1119    
1120      if (!strcmp(inName, "-")) {      if (inName == NULL) {
1121            printf("Could not find bootloader configuration\n");
1122            exit(1);
1123        } else if (!strcmp(inName, "-")) {
1124   in = 0;   in = 0;
1125      } else {      } else {
1126   if ((in = open(inName, O_RDONLY)) < 0) {   if ((in = open(inName, O_RDONLY)) < 0) {
# Line 985  static struct grubConfig * readConfig(co Line 1187  static struct grubConfig * readConfig(co
1187      dbgPrintf("\n");      dbgPrintf("\n");
1188      struct keywordTypes *kwType = getKeywordByType(LT_DEFAULT, cfi);      struct keywordTypes *kwType = getKeywordByType(LT_DEFAULT, cfi);
1189      if (kwType && line->numElements == 3 &&      if (kwType && line->numElements == 3 &&
1190      !strcmp(line->elements[1].item, kwType->key)) {      !strcmp(line->elements[1].item, kwType->key) &&
1191        !is_special_grub2_variable(line->elements[2].item)) {
1192   dbgPrintf("Line sets default config\n");   dbgPrintf("Line sets default config\n");
1193   cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;   cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;
1194   defaultLine = line;   defaultLine = line;
1195      }      }
  } else if (line->type == LT_DEFAULT && line->numElements == 2) {  
     cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;  
     defaultLine = line;  
1196    
1197          } else if (line->type == LT_KERNEL) {          } else if (iskernel(line->type)) {
1198      /* if by some freak chance this is multiboot and the "module"      /* if by some freak chance this is multiboot and the "module"
1199       * lines came earlier in the template, make sure to use LT_HYPER       * lines came earlier in the template, make sure to use LT_HYPER
1200       * instead of LT_KERNEL now       * instead of LT_KERNEL now
1201       */       */
1202      if (entry->multiboot)      if (entry && entry->multiboot)
1203   line->type = LT_HYPER;   line->type = LT_HYPER;
1204    
1205          } else if (line->type == LT_MBMODULE) {          } else if (line->type == LT_MBMODULE) {
# Line 1011  static struct grubConfig * readConfig(co Line 1211  static struct grubConfig * readConfig(co
1211      for (struct singleLine *l = entry->lines; l; l = l->next) {      for (struct singleLine *l = entry->lines; l; l = l->next) {
1212   if (l->type == LT_HYPER)   if (l->type == LT_HYPER)
1213      break;      break;
1214   else if (l->type == LT_KERNEL) {   else if (iskernel(l->type)) {
1215      l->type = LT_HYPER;      l->type = LT_HYPER;
1216      break;      break;
1217   }   }
# Line 1025  static struct grubConfig * readConfig(co Line 1225  static struct grubConfig * readConfig(co
1225      cfg->fallbackImage = strtol(line->elements[1].item, &end, 10);      cfg->fallbackImage = strtol(line->elements[1].item, &end, 10);
1226      if (*end) cfg->fallbackImage = -1;      if (*end) cfg->fallbackImage = -1;
1227    
1228   } else if (line->type == LT_TITLE && line->numElements > 1) {   } else if ((line->type == LT_DEFAULT && cfi->defaultIsUnquoted) ||
1229      /* make the title a single argument (undoing our parsing) */                  (line->type == LT_TITLE && line->numElements > 1)) {
1230        /* make the title/default a single argument (undoing our parsing) */
1231      len = 0;      len = 0;
1232      for (int i = 1; i < line->numElements; i++) {      for (int i = 1; i < line->numElements; i++) {
1233   len += strlen(line->elements[i].item);   len += strlen(line->elements[i].item);
# Line 1133  static struct grubConfig * readConfig(co Line 1334  static struct grubConfig * readConfig(co
1334      }      }
1335   }   }
1336    
1337     if (line->type == LT_DEFAULT && line->numElements == 2) {
1338        cfg->flags &= ~GRUB_CONFIG_NO_DEFAULT;
1339        defaultLine = line;
1340     }
1341    
1342   /* If we find a generic config option which should live at the   /* If we find a generic config option which should live at the
1343     top of the file, move it there. Old versions of grubby were     top of the file, move it there. Old versions of grubby were
1344     probably responsible for putting new images in the wrong     probably responsible for putting new images in the wrong
# Line 1190  static struct grubConfig * readConfig(co Line 1396  static struct grubConfig * readConfig(co
1396          if (defaultLine->numElements > 2 &&          if (defaultLine->numElements > 2 &&
1397      cfi->defaultSupportSaved &&      cfi->defaultSupportSaved &&
1398      !strncmp(defaultLine->elements[2].item,"\"${saved_entry}\"", 16)) {      !strncmp(defaultLine->elements[2].item,"\"${saved_entry}\"", 16)) {
1399      cfg->defaultImage = DEFAULT_SAVED_GRUB2;   cfg->cfi->defaultIsSaved = 1;
1400     cfg->defaultImage = DEFAULT_SAVED_GRUB2;
1401     if (cfg->cfi->getEnv) {
1402        char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry");
1403        if (defTitle) {
1404     int index = 0;
1405     if (isnumber(defTitle)) {
1406        index = atoi(defTitle);
1407        entry = findEntryByIndex(cfg, index);
1408     } else {
1409        entry = findEntryByTitle(cfg, defTitle, &index);
1410     }
1411     if (entry)
1412        cfg->defaultImage = index;
1413        }
1414     }
1415   } else if (cfi->defaultIsVariable) {   } else if (cfi->defaultIsVariable) {
1416      char *value = defaultLine->elements[2].item;      char *value = defaultLine->elements[2].item;
1417      while (*value && (*value == '"' || *value == '\'' ||      while (*value && (*value == '"' || *value == '\'' ||
# Line 1231  static struct grubConfig * readConfig(co Line 1452  static struct grubConfig * readConfig(co
1452          cfg->defaultImage = -1;          cfg->defaultImage = -1;
1453      }      }
1454   }   }
1455        } else if (cfg->cfi->defaultIsSaved && cfg->cfi->getEnv) {
1456     char *defTitle = cfi->getEnv(cfg->cfi, "saved_entry");
1457     if (defTitle) {
1458        int index = 0;
1459        if (isnumber(defTitle)) {
1460     index = atoi(defTitle);
1461     entry = findEntryByIndex(cfg, index);
1462        } else {
1463     entry = findEntryByTitle(cfg, defTitle, &index);
1464        }
1465        if (entry)
1466     cfg->defaultImage = index;
1467     }
1468      } else {      } else {
1469          cfg->defaultImage = 0;          cfg->defaultImage = 0;
1470      }      }
# Line 1248  static void writeDefault(FILE * out, cha Line 1482  static void writeDefault(FILE * out, cha
1482    
1483      if (cfg->defaultImage == DEFAULT_SAVED)      if (cfg->defaultImage == DEFAULT_SAVED)
1484   fprintf(out, "%sdefault%ssaved\n", indent, separator);   fprintf(out, "%sdefault%ssaved\n", indent, separator);
1485      else if (cfg->defaultImage == DEFAULT_SAVED_GRUB2)      else if (cfg->cfi->defaultIsSaved) {
1486   fprintf(out, "%sset default=\"${saved_entry}\"\n", indent);   fprintf(out, "%sset default=\"${saved_entry}\"\n", indent);
1487      else if (cfg->defaultImage > -1) {   if (cfg->defaultImage >= 0 && cfg->cfi->setEnv) {
1488        char *title;
1489        entry = findEntryByIndex(cfg, cfg->defaultImage);
1490        line = getLineByType(LT_MENUENTRY, entry->lines);
1491        if (!line)
1492     line = getLineByType(LT_TITLE, entry->lines);
1493        if (line) {
1494     title = extractTitle(line);
1495     if (title)
1496        cfg->cfi->setEnv(cfg->cfi, "saved_entry", title);
1497        }
1498     }
1499        } else if (cfg->defaultImage > -1) {
1500   if (cfg->cfi->defaultIsIndex) {   if (cfg->cfi->defaultIsIndex) {
1501      if (cfg->cfi->defaultIsVariable) {      if (cfg->cfi->defaultIsVariable) {
1502          fprintf(out, "%sset default=\"%d\"\n", indent,          fprintf(out, "%sset default=\"%d\"\n", indent,
# Line 1310  static int writeConfig(struct grubConfig Line 1556  static int writeConfig(struct grubConfig
1556    
1557      /* most likely the symlink is relative, so change our      /* most likely the symlink is relative, so change our
1558         directory to the dir of the symlink */         directory to the dir of the symlink */
1559              rc = chdir(dirname(outName));      char *dir = strdupa(outName);
1560        rc = chdir(dirname(dir));
1561      do {      do {
1562   buf = alloca(len + 1);   buf = alloca(len + 1);
1563   rc = readlink(basename(outName), buf, len);   rc = readlink(basename(outName), buf, len);
# Line 1352  static int writeConfig(struct grubConfig Line 1599  static int writeConfig(struct grubConfig
1599      while (line) {      while (line) {
1600          if (line->type == LT_SET_VARIABLE && defaultKw &&          if (line->type == LT_SET_VARIABLE && defaultKw &&
1601   line->numElements == 3 &&   line->numElements == 3 &&
1602   !strcmp(line->elements[1].item, defaultKw->key)) {   !strcmp(line->elements[1].item, defaultKw->key) &&
1603     !is_special_grub2_variable(line->elements[2].item)) {
1604      writeDefault(out, line->indent, line->elements[0].indent, cfg);      writeDefault(out, line->indent, line->elements[0].indent, cfg);
1605      needs &= ~MAIN_DEFAULT;      needs &= ~MAIN_DEFAULT;
1606   } else if (line->type == LT_DEFAULT) {   } else if (line->type == LT_DEFAULT) {
# Line 1497  static char *findDiskForRoot() Line 1745  static char *findDiskForRoot()
1745      return NULL;      return NULL;
1746  }  }
1747    
1748  void printEntry(struct singleEntry * entry) {  void printEntry(struct singleEntry * entry, FILE *f) {
1749      int i;      int i;
1750      struct singleLine * line;      struct singleLine * line;
1751    
1752      for (line = entry->lines; line; line = line->next) {      for (line = entry->lines; line; line = line->next) {
1753   fprintf(stderr, "DBG: %s", line->indent);   log_message(f, "DBG: %s", line->indent);
1754   for (i = 0; i < line->numElements; i++) {   for (i = 0; i < line->numElements; i++) {
1755      /* Need to handle this, because we strip the quotes from      /* Need to handle this, because we strip the quotes from
1756       * menuentry when read it. */       * menuentry when read it. */
1757      if (line->type == LT_MENUENTRY && i == 1) {      if (line->type == LT_MENUENTRY && i == 1) {
1758   if(!isquote(*line->elements[i].item))   if(!isquote(*line->elements[i].item))
1759      fprintf(stderr, "\'%s\'", line->elements[i].item);      log_message(f, "\'%s\'", line->elements[i].item);
1760   else   else
1761      fprintf(stderr, "%s", line->elements[i].item);      log_message(f, "%s", line->elements[i].item);
1762   fprintf(stderr, "%s", line->elements[i].indent);   log_message(f, "%s", line->elements[i].indent);
1763    
1764   continue;   continue;
1765      }      }
1766            
1767      fprintf(stderr, "%s%s",      log_message(f, "%s%s",
1768      line->elements[i].item, line->elements[i].indent);      line->elements[i].item, line->elements[i].indent);
1769   }   }
1770   fprintf(stderr, "\n");   log_message(f, "\n");
1771      }      }
1772  }  }
1773    
1774  void notSuitablePrintf(struct singleEntry * entry, const char *fmt, ...)  void notSuitablePrintf(struct singleEntry * entry, int okay, const char *fmt, ...)
1775  {  {
1776      va_list argp;      static int once;
1777        va_list argp, argq;
1778    
1779        va_start(argp, fmt);
1780    
1781        va_copy(argq, argp);
1782        if (!once) {
1783     log_time(NULL);
1784     log_message(NULL, "command line: %s\n", saved_command_line);
1785        }
1786        log_message(NULL, "DBG: Image entry %s: ", okay ? "succeeded" : "failed");
1787        log_vmessage(NULL, fmt, argq);
1788    
1789        printEntry(entry, NULL);
1790        va_end(argq);
1791    
1792      if (!debug)      if (!debug) {
1793     once = 1;
1794         va_end(argp);
1795   return;   return;
1796        }
1797    
1798      va_start(argp, fmt);      if (okay) {
1799     va_end(argp);
1800     return;
1801        }
1802    
1803        if (!once)
1804     log_message(stderr, "DBG: command line: %s\n", saved_command_line);
1805        once = 1;
1806      fprintf(stderr, "DBG: Image entry failed: ");      fprintf(stderr, "DBG: Image entry failed: ");
1807      vfprintf(stderr, fmt, argp);      vfprintf(stderr, fmt, argp);
1808      printEntry(entry);      printEntry(entry, stderr);
1809      va_end(argp);      va_end(argp);
1810  }  }
1811    
# Line 1560  int suitableImage(struct singleEntry * e Line 1832  int suitableImage(struct singleEntry * e
1832      char * rootdev;      char * rootdev;
1833    
1834      if (skipRemoved && entry->skip) {      if (skipRemoved && entry->skip) {
1835   notSuitablePrintf(entry, "marked to skip\n");   notSuitablePrintf(entry, 0, "marked to skip\n");
1836   return 0;   return 0;
1837      }      }
1838    
1839      line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines);      line = getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
1840      if (!line) {      if (!line) {
1841   notSuitablePrintf(entry, "no line found\n");   notSuitablePrintf(entry, 0, "no line found\n");
1842   return 0;   return 0;
1843      }      }
1844      if (line->numElements < 2) {      if (line->numElements < 2) {
1845   notSuitablePrintf(entry, "line has only %d elements\n",   notSuitablePrintf(entry, 0, "line has only %d elements\n",
1846      line->numElements);      line->numElements);
1847   return 0;   return 0;
1848      }      }
1849    
1850      if (flags & GRUBBY_BADIMAGE_OKAY) return 1;      if (flags & GRUBBY_BADIMAGE_OKAY) {
1851        notSuitablePrintf(entry, 1, "\n");
1852        return 1;
1853        }
1854    
1855      fullName = alloca(strlen(bootPrefix) +      fullName = alloca(strlen(bootPrefix) +
1856        strlen(line->elements[1].item) + 1);        strlen(line->elements[1].item) + 1);
# Line 1586  int suitableImage(struct singleEntry * e Line 1861  int suitableImage(struct singleEntry * e
1861      sprintf(fullName, "%s%s%s", bootPrefix, hasslash ? "" : "/",      sprintf(fullName, "%s%s%s", bootPrefix, hasslash ? "" : "/",
1862              line->elements[1].item + rootspec_offset);              line->elements[1].item + rootspec_offset);
1863      if (access(fullName, R_OK)) {      if (access(fullName, R_OK)) {
1864   notSuitablePrintf(entry, "access to %s failed\n", fullName);   notSuitablePrintf(entry, 0, "access to %s failed\n", fullName);
1865   return 0;   return 0;
1866      }      }
1867      for (i = 2; i < line->numElements; i++)      for (i = 2; i < line->numElements; i++)
# Line 1607  int suitableImage(struct singleEntry * e Line 1882  int suitableImage(struct singleEntry * e
1882    
1883              /* failed to find one */              /* failed to find one */
1884              if (!line) {              if (!line) {
1885   notSuitablePrintf(entry, "no line found\n");   notSuitablePrintf(entry, 0, "no line found\n");
1886   return 0;   return 0;
1887              }              }
1888    
# Line 1616  int suitableImage(struct singleEntry * e Line 1891  int suitableImage(struct singleEntry * e
1891      if (i < line->numElements)      if (i < line->numElements)
1892          dev = line->elements[i].item + 5;          dev = line->elements[i].item + 5;
1893      else {      else {
1894   notSuitablePrintf(entry, "no root= entry found\n");   notSuitablePrintf(entry, 0, "no root= entry found\n");
1895   /* it failed too...  can't find root= */   /* it failed too...  can't find root= */
1896          return 0;          return 0;
1897              }              }
# Line 1625  int suitableImage(struct singleEntry * e Line 1900  int suitableImage(struct singleEntry * e
1900    
1901      dev = getpathbyspec(dev);      dev = getpathbyspec(dev);
1902      if (!getpathbyspec(dev)) {      if (!getpathbyspec(dev)) {
1903          notSuitablePrintf(entry, "can't find blkid entry for %s\n", dev);          notSuitablePrintf(entry, 0, "can't find blkid entry for %s\n", dev);
1904          return 0;          return 0;
1905      } else      } else
1906   dev = getpathbyspec(dev);   dev = getpathbyspec(dev);
1907    
1908      rootdev = findDiskForRoot();      rootdev = findDiskForRoot();
1909      if (!rootdev) {      if (!rootdev) {
1910          notSuitablePrintf(entry, "can't find root device\n");          notSuitablePrintf(entry, 0, "can't find root device\n");
1911   return 0;   return 0;
1912      }      }
1913    
1914      if (!getuuidbydev(rootdev) || !getuuidbydev(dev)) {      if (!getuuidbydev(rootdev) || !getuuidbydev(dev)) {
1915          notSuitablePrintf(entry, "uuid missing: rootdev %s, dev %s\n",          notSuitablePrintf(entry, 0, "uuid missing: rootdev %s, dev %s\n",
1916   getuuidbydev(rootdev), getuuidbydev(dev));   getuuidbydev(rootdev), getuuidbydev(dev));
1917          free(rootdev);          free(rootdev);
1918          return 0;          return 0;
1919      }      }
1920    
1921      if (strcmp(getuuidbydev(rootdev), getuuidbydev(dev))) {      if (strcmp(getuuidbydev(rootdev), getuuidbydev(dev))) {
1922          notSuitablePrintf(entry, "uuid mismatch: rootdev %s, dev %s\n",          notSuitablePrintf(entry, 0, "uuid mismatch: rootdev %s, dev %s\n",
1923   getuuidbydev(rootdev), getuuidbydev(dev));   getuuidbydev(rootdev), getuuidbydev(dev));
1924   free(rootdev);   free(rootdev);
1925          return 0;          return 0;
1926      }      }
1927    
1928      free(rootdev);      free(rootdev);
1929        notSuitablePrintf(entry, 1, "\n");
1930    
1931      return 1;      return 1;
1932  }  }
# Line 1684  struct singleEntry * findEntryByPath(str Line 1960  struct singleEntry * findEntryByPath(str
1960   }   }
1961    
1962   indexVars[i + 1] = -1;   indexVars[i + 1] = -1;
1963    
1964   i = 0;   i = 0;
1965   if (index) {   if (index) {
1966      while (i < *index) i++;      while (i < *index) {
1967      if (indexVars[i] == -1) return NULL;   i++;
1968     if (indexVars[i] == -1) return NULL;
1969        }
1970   }   }
1971    
1972   entry = findEntryByIndex(config, indexVars[i]);   entry = findEntryByIndex(config, indexVars[i]);
1973   if (!entry) return NULL;   if (!entry) return NULL;
1974    
1975   line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines);   line = getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
1976   if (!line) return NULL;   if (!line) return NULL;
1977    
1978   if (index) *index = indexVars[i];   if (index) *index = indexVars[i];
# Line 1743  struct singleEntry * findEntryByPath(str Line 2021  struct singleEntry * findEntryByPath(str
2021    
2022      /* check all the lines matching checkType */      /* check all the lines matching checkType */
2023      for (line = entry->lines; line; line = line->next) {      for (line = entry->lines; line; line = line->next) {
2024   line = getLineByType(entry->multiboot && checkType == LT_KERNEL ?   enum lineType_e ct = checkType;
2025       LT_KERNEL|LT_MBMODULE|LT_HYPER :   if (entry->multiboot && checkType == LT_KERNEL)
2026       checkType, line);      ct = LT_KERNEL|LT_KERNEL_EFI|LT_MBMODULE|LT_HYPER|LT_KERNEL_16;
2027   if (!line) break;  /* not found in this entry */   else if (checkType & LT_KERNEL)
2028        ct = checkType | LT_KERNEL_EFI | LT_KERNEL_16;
2029     line = getLineByType(ct, line);
2030     if (!line)
2031        break;  /* not found in this entry */
2032    
2033   if (line && line->type != LT_MENUENTRY &&   if (line && line->type != LT_MENUENTRY &&
2034   line->numElements >= 2) {   line->numElements >= 2) {
# Line 1765  struct singleEntry * findEntryByPath(str Line 2047  struct singleEntry * findEntryByPath(str
2047       * non-Linux boot entries (could find netbsd etc, though, which is       * non-Linux boot entries (could find netbsd etc, though, which is
2048       * unfortunate)       * unfortunate)
2049       */       */
2050      if (line && getLineByType(LT_KERNEL|LT_HYPER, entry->lines))      if (line && getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines))
2051   break; /* found 'im! */   break; /* found 'im! */
2052   }   }
2053    
# Line 1775  struct singleEntry * findEntryByPath(str Line 2057  struct singleEntry * findEntryByPath(str
2057      return entry;      return entry;
2058  }  }
2059    
2060    struct singleEntry * findEntryByTitle(struct grubConfig * cfg, char *title,
2061          int * index) {
2062        struct singleEntry * entry;
2063        struct singleLine * line;
2064        int i;
2065        char * newtitle;
2066    
2067        for (i = 0, entry = cfg->entries; entry; entry = entry->next, i++) {
2068     if (index && i < *index)
2069        continue;
2070     line = getLineByType(LT_TITLE, entry->lines);
2071     if (!line)
2072        line = getLineByType(LT_MENUENTRY, entry->lines);
2073     if (!line)
2074        continue;
2075     newtitle = grub2ExtractTitle(line);
2076     if (!newtitle)
2077        continue;
2078     if (!strcmp(title, newtitle))
2079        break;
2080        }
2081    
2082        if (!entry)
2083     return NULL;
2084    
2085        if (index)
2086     *index = i;
2087        return entry;
2088    }
2089    
2090  struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index) {  struct singleEntry * findEntryByIndex(struct grubConfig * cfg, int index) {
2091      struct singleEntry * entry;      struct singleEntry * entry;
2092    
# Line 1797  struct singleEntry * findTemplate(struct Line 2109  struct singleEntry * findTemplate(struct
2109      struct singleEntry * entry, * entry2;      struct singleEntry * entry, * entry2;
2110      int index;      int index;
2111    
2112      if (cfg->defaultImage > -1) {      if (cfg->cfi->defaultIsSaved) {
2113     if (cfg->cfi->getEnv) {
2114        char *defTitle = cfg->cfi->getEnv(cfg->cfi, "saved_entry");
2115        if (defTitle) {
2116     int index = 0;
2117     if (isnumber(defTitle)) {
2118        index = atoi(defTitle);
2119        entry = findEntryByIndex(cfg, index);
2120     } else {
2121        entry = findEntryByTitle(cfg, defTitle, &index);
2122     }
2123     if (entry)
2124        cfg->defaultImage = index;
2125        }
2126     }
2127        } else if (cfg->defaultImage > -1) {
2128   entry = findEntryByIndex(cfg, cfg->defaultImage);   entry = findEntryByIndex(cfg, cfg->defaultImage);
2129   if (entry && suitableImage(entry, prefix, skipRemoved, flags)) {   if (entry && suitableImage(entry, prefix, skipRemoved, flags)) {
2130      if (indexPtr) *indexPtr = cfg->defaultImage;      if (indexPtr) *indexPtr = cfg->defaultImage;
# Line 1867  void markRemovedImage(struct grubConfig Line 2194  void markRemovedImage(struct grubConfig
2194    
2195  void setDefaultImage(struct grubConfig * config, int hasNew,  void setDefaultImage(struct grubConfig * config, int hasNew,
2196       const char * defaultKernelPath, int newIsDefault,       const char * defaultKernelPath, int newIsDefault,
2197       const char * prefix, int flags) {       const char * prefix, int flags, int index) {
2198      struct singleEntry * entry, * entry2, * newDefault;      struct singleEntry * entry, * entry2, * newDefault;
2199      int i, j;      int i, j;
2200    
2201      if (newIsDefault) {      if (newIsDefault) {
2202   config->defaultImage = 0;   config->defaultImage = 0;
2203   return;   return;
2204        } else if ((index >= 0) && config->cfi->defaultIsIndex) {
2205     if (findEntryByIndex(config, index))
2206        config->defaultImage = index;
2207     else
2208        config->defaultImage = -1;
2209     return;
2210      } else if (defaultKernelPath) {      } else if (defaultKernelPath) {
2211   i = 0;   i = 0;
2212   if (findEntryByPath(config, defaultKernelPath, prefix, &i)) {   if (findEntryByPath(config, defaultKernelPath, prefix, &i)) {
# Line 1945  void displayEntry(struct singleEntry * e Line 2278  void displayEntry(struct singleEntry * e
2278      struct singleLine * line;      struct singleLine * line;
2279      char * root = NULL;      char * root = NULL;
2280      int i;      int i;
2281        int j;
2282    
2283      printf("index=%d\n", index);      printf("index=%d\n", index);
2284    
2285      line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines);      line = getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
2286      if (!line) {      if (!line) {
2287          printf("non linux entry\n");          printf("non linux entry\n");
2288          return;          return;
# Line 2013  void displayEntry(struct singleEntry * e Line 2347  void displayEntry(struct singleEntry * e
2347   printf("root=%s\n", s);   printf("root=%s\n", s);
2348      }      }
2349    
2350      line = getLineByType(LT_INITRD, entry->lines);      line = getLineByType(LT_INITRD|LT_INITRD_EFI|LT_INITRD_16, entry->lines);
2351    
2352      if (line && line->numElements >= 2) {      if (line && line->numElements >= 2) {
2353   if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))   if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))
# Line 2036  void displayEntry(struct singleEntry * e Line 2370  void displayEntry(struct singleEntry * e
2370   if (title)   if (title)
2371      printf("title=%s\n", title);      printf("title=%s\n", title);
2372      }      }
2373    
2374        for (j = 0, line = entry->lines; line; line = line->next) {
2375     if ((line->type & LT_MBMODULE) && line->numElements >= 2) {
2376        if (!strncmp(prefix, line->elements[1].item, strlen(prefix)))
2377     printf("mbmodule%d=", j);
2378        else
2379     printf("mbmodule%d=%s", j, prefix);
2380    
2381        for (i = 1; i < line->numElements; i++)
2382     printf("%s%s", line->elements[i].item, line->elements[i].indent);
2383        printf("\n");
2384        j++;
2385     }
2386        }
2387  }  }
2388    
2389  int isSuseSystem(void) {  int isSuseSystem(void) {
# Line 2414  struct singleLine * addLineTmpl(struct s Line 2762  struct singleLine * addLineTmpl(struct s
2762  {  {
2763      struct singleLine * newLine = lineDup(tmplLine);      struct singleLine * newLine = lineDup(tmplLine);
2764    
2765        if (isEfi && cfi == &grub2ConfigType) {
2766     enum lineType_e old = newLine->type;
2767     newLine->type = preferredLineType(newLine->type, cfi);
2768     if (old != newLine->type)
2769        newLine->elements[0].item = getKeyByType(newLine->type, cfi);
2770        }
2771    
2772      if (val) {      if (val) {
2773   /* override the inherited value with our own.   /* override the inherited value with our own.
2774   * This is a little weak because it only applies to elements[1]   * This is a little weak because it only applies to elements[1]
# Line 2423  struct singleLine * addLineTmpl(struct s Line 2778  struct singleLine * addLineTmpl(struct s
2778   insertElement(newLine, val, 1, cfi);   insertElement(newLine, val, 1, cfi);
2779    
2780   /* but try to keep the rootspec from the template... sigh */   /* but try to keep the rootspec from the template... sigh */
2781   if (tmplLine->type & (LT_HYPER|LT_KERNEL|LT_MBMODULE|LT_INITRD)) {   if (tmplLine->type & (LT_HYPER|LT_KERNEL|LT_MBMODULE|LT_INITRD|LT_KERNEL_EFI|LT_INITRD_EFI|LT_KERNEL_16|LT_INITRD_16)) {
2782      char * rootspec = getRootSpecifier(tmplLine->elements[1].item);      char * rootspec = getRootSpecifier(tmplLine->elements[1].item);
2783      if (rootspec != NULL) {      if (rootspec != NULL) {
2784   free(newLine->elements[1].item);   free(newLine->elements[1].item);
# Line 2460  struct singleLine *  addLine(struct sing Line 2815  struct singleLine *  addLine(struct sing
2815      /* NB: This function shouldn't allocate items on the heap, rather on the      /* NB: This function shouldn't allocate items on the heap, rather on the
2816       * stack since it calls addLineTmpl which will make copies.       * stack since it calls addLineTmpl which will make copies.
2817       */       */
   
2818      if (type == LT_TITLE && cfi->titleBracketed) {      if (type == LT_TITLE && cfi->titleBracketed) {
2819   /* we're doing a bracketed title (zipl) */   /* we're doing a bracketed title (zipl) */
2820   tmpl.type = type;   tmpl.type = type;
# Line 2794  int updateActualImage(struct grubConfig Line 3148  int updateActualImage(struct grubConfig
3148      firstElement = 2;      firstElement = 2;
3149    
3150   } else {   } else {
3151      line = getLineByType(LT_KERNEL|LT_MBMODULE, entry->lines);      line = getLineByType(LT_KERNEL|LT_MBMODULE|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
3152      if (!line) {      if (!line) {
3153   /* no LT_KERNEL or LT_MBMODULE in this entry? */   /* no LT_KERNEL or LT_MBMODULE in this entry? */
3154   continue;   continue;
# Line 2950  int updateImage(struct grubConfig * cfg, Line 3304  int updateImage(struct grubConfig * cfg,
3304      return rc;      return rc;
3305  }  }
3306    
3307    int addMBInitrd(struct grubConfig * cfg, const char *newMBKernel,
3308     const char * image, const char * prefix, const char * initrd) {
3309        struct singleEntry * entry;
3310        struct singleLine * line, * kernelLine, *endLine = NULL;
3311        int index = 0;
3312    
3313        if (!image) return 0;
3314    
3315        for (; (entry = findEntryByPath(cfg, newMBKernel, prefix, &index)); index++) {
3316            kernelLine = getLineByType(LT_MBMODULE, entry->lines);
3317            if (!kernelLine) continue;
3318    
3319            if (prefix) {
3320                int prefixLen = strlen(prefix);
3321                if (!strncmp(initrd, prefix, prefixLen))
3322                    initrd += prefixLen;
3323            }
3324     endLine = getLineByType(LT_ENTRY_END, entry->lines);
3325     if (endLine)
3326        removeLine(entry, endLine);
3327            line = addLine(entry, cfg->cfi, preferredLineType(LT_MBMODULE,cfg->cfi),
3328     kernelLine->indent, initrd);
3329            if (!line)
3330        return 1;
3331     if (endLine) {
3332        line = addLine(entry, cfg->cfi, LT_ENTRY_END, "", NULL);
3333                if (!line)
3334     return 1;
3335     }
3336    
3337            break;
3338        }
3339    
3340        return 0;
3341    }
3342    
3343  int updateInitrd(struct grubConfig * cfg, const char * image,  int updateInitrd(struct grubConfig * cfg, const char * image,
3344                   const char * prefix, const char * initrd) {                   const char * prefix, const char * initrd) {
3345      struct singleEntry * entry;      struct singleEntry * entry;
# Line 2959  int updateInitrd(struct grubConfig * cfg Line 3349  int updateInitrd(struct grubConfig * cfg
3349      if (!image) return 0;      if (!image) return 0;
3350    
3351      for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {      for (; (entry = findEntryByPath(cfg, image, prefix, &index)); index++) {
3352          kernelLine = getLineByType(LT_KERNEL, entry->lines);          kernelLine = getLineByType(LT_KERNEL|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
3353          if (!kernelLine) continue;          if (!kernelLine) continue;
3354    
3355          line = getLineByType(LT_INITRD, entry->lines);          line = getLineByType(LT_INITRD|LT_INITRD_EFI|LT_INITRD_16, entry->lines);
3356          if (line)          if (line)
3357              removeLine(entry, line);              removeLine(entry, line);
3358          if (prefix) {          if (prefix) {
# Line 2973  int updateInitrd(struct grubConfig * cfg Line 3363  int updateInitrd(struct grubConfig * cfg
3363   endLine = getLineByType(LT_ENTRY_END, entry->lines);   endLine = getLineByType(LT_ENTRY_END, entry->lines);
3364   if (endLine)   if (endLine)
3365      removeLine(entry, endLine);      removeLine(entry, endLine);
3366          line = addLine(entry, cfg->cfi, LT_INITRD, kernelLine->indent, initrd);   enum lineType_e lt;
3367     switch(kernelLine->type) {
3368        case LT_KERNEL:
3369            lt = LT_INITRD;
3370     break;
3371        case LT_KERNEL_EFI:
3372            lt = LT_INITRD_EFI;
3373     break;
3374        case LT_KERNEL_16:
3375            lt = LT_INITRD_16;
3376     break;
3377        default:
3378            lt = preferredLineType(LT_INITRD, cfg->cfi);
3379     }
3380            line = addLine(entry, cfg->cfi, lt, kernelLine->indent, initrd);
3381          if (!line)          if (!line)
3382      return 1;      return 1;
3383   if (endLine) {   if (endLine) {
# Line 3332  int addNewKernel(struct grubConfig * con Line 3736  int addNewKernel(struct grubConfig * con
3736   const char * newKernelPath, const char * newKernelTitle,   const char * newKernelPath, const char * newKernelTitle,
3737   const char * newKernelArgs, const char * newKernelInitrd,   const char * newKernelArgs, const char * newKernelInitrd,
3738   const char ** extraInitrds, int extraInitrdCount,   const char ** extraInitrds, int extraInitrdCount,
3739                   const char * newMBKernel, const char * newMBKernelArgs) {                   const char * newMBKernel, const char * newMBKernelArgs,
3740     const char * newDevTreePath) {
3741      struct singleEntry * new;      struct singleEntry * new;
3742      struct singleLine * newLine = NULL, * tmplLine = NULL, * masterLine = NULL;      struct singleLine * newLine = NULL, * tmplLine = NULL, * masterLine = NULL;
3743      int needs;      int needs;
# Line 3373  int addNewKernel(struct grubConfig * con Line 3778  int addNewKernel(struct grubConfig * con
3778          needs |= NEED_MB;          needs |= NEED_MB;
3779          new->multiboot = 1;          new->multiboot = 1;
3780      }      }
3781        if (newDevTreePath && getKeywordByType(LT_DEVTREE, config->cfi))
3782     needs |= NEED_DEVTREE;
3783    
3784      if (template) {      if (template) {
3785   for (masterLine = template->lines;   for (masterLine = template->lines;
# Line 3386  int addNewKernel(struct grubConfig * con Line 3793  int addNewKernel(struct grubConfig * con
3793      while (*chptr && isspace(*chptr)) chptr++;      while (*chptr && isspace(*chptr)) chptr++;
3794      if (*chptr == '#') continue;      if (*chptr == '#') continue;
3795    
3796      if (tmplLine->type == LT_KERNEL &&      if (iskernel(tmplLine->type) && tmplLine->numElements >= 2) {
     tmplLine->numElements >= 2) {  
3797   if (!template->multiboot && (needs & NEED_MB)) {   if (!template->multiboot && (needs & NEED_MB)) {
3798      /* it's not a multiboot template and this is the kernel      /* it's not a multiboot template and this is the kernel
3799       * line.  Try to be intelligent about inserting the       * line.  Try to be intelligent about inserting the
# Line 3464  int addNewKernel(struct grubConfig * con Line 3870  int addNewKernel(struct grubConfig * con
3870      /* template is multi but new is not,      /* template is multi but new is not,
3871       * insert the kernel in the first module slot       * insert the kernel in the first module slot
3872       */       */
3873      tmplLine->type = LT_KERNEL;      tmplLine->type = preferredLineType(LT_KERNEL, config->cfi);
3874      free(tmplLine->elements[0].item);      free(tmplLine->elements[0].item);
3875      tmplLine->elements[0].item =      tmplLine->elements[0].item =
3876   strdup(getKeywordByType(LT_KERNEL, config->cfi)->key);   strdup(getKeywordByType(tmplLine->type,
3877     config->cfi)->key);
3878      newLine = addLineTmpl(new, tmplLine, newLine,      newLine = addLineTmpl(new, tmplLine, newLine,
3879    newKernelPath + strlen(prefix), config->cfi);    newKernelPath + strlen(prefix),
3880      config->cfi);
3881      needs &= ~NEED_KERNEL;      needs &= ~NEED_KERNEL;
3882   } else if (needs & NEED_INITRD) {   } else if (needs & NEED_INITRD) {
3883      char *initrdVal;      char *initrdVal;
3884      /* template is multi but new is not,      /* template is multi but new is not,
3885       * insert the initrd in the second module slot       * insert the initrd in the second module slot
3886       */       */
3887      tmplLine->type = LT_INITRD;      tmplLine->type = preferredLineType(LT_INITRD, config->cfi);
3888      free(tmplLine->elements[0].item);      free(tmplLine->elements[0].item);
3889      tmplLine->elements[0].item =      tmplLine->elements[0].item =
3890   strdup(getKeywordByType(LT_INITRD, config->cfi)->key);   strdup(getKeywordByType(tmplLine->type,
3891     config->cfi)->key);
3892      initrdVal = getInitrdVal(config, prefix, tmplLine, newKernelInitrd, extraInitrds, extraInitrdCount);      initrdVal = getInitrdVal(config, prefix, tmplLine, newKernelInitrd, extraInitrds, extraInitrdCount);
3893      newLine = addLineTmpl(new, tmplLine, newLine, initrdVal, config->cfi);      newLine = addLineTmpl(new, tmplLine, newLine, initrdVal, config->cfi);
3894      free(initrdVal);      free(initrdVal);
3895      needs &= ~NEED_INITRD;      needs &= ~NEED_INITRD;
3896   }   }
3897    
3898      } else if (tmplLine->type == LT_INITRD &&      } else if (isinitrd(tmplLine->type) && tmplLine->numElements >= 2) {
        tmplLine->numElements >= 2) {  
3899   if (needs & NEED_INITRD &&   if (needs & NEED_INITRD &&
3900      new->multiboot && !template->multiboot &&      new->multiboot && !template->multiboot &&
3901      config->cfi->mbInitRdIsModule) {      config->cfi->mbInitRdIsModule) {
# Line 3541  int addNewKernel(struct grubConfig * con Line 3949  int addNewKernel(struct grubConfig * con
3949      static const char *prefix = "'Loading ";      static const char *prefix = "'Loading ";
3950      if (tmplLine->numElements > 1 &&      if (tmplLine->numElements > 1 &&
3951      strstr(tmplLine->elements[1].item, prefix) &&      strstr(tmplLine->elements[1].item, prefix) &&
3952      masterLine->next && masterLine->next->type == LT_KERNEL) {      masterLine->next &&
3953        iskernel(masterLine->next->type)) {
3954   char *newTitle = malloc(strlen(prefix) +   char *newTitle = malloc(strlen(prefix) +
3955   strlen(newKernelTitle) + 2);   strlen(newKernelTitle) + 2);
3956    
# Line 3556  int addNewKernel(struct grubConfig * con Line 3965  int addNewKernel(struct grubConfig * con
3965   newLine = addLineTmpl(new, tmplLine, newLine, NULL,   newLine = addLineTmpl(new, tmplLine, newLine, NULL,
3966   config->cfi);   config->cfi);
3967      }      }
3968        } else if (tmplLine->type == LT_DEVTREE &&
3969           tmplLine->numElements == 2 && newDevTreePath) {
3970            newLine = addLineTmpl(new, tmplLine, newLine,
3971          newDevTreePath + strlen(prefix),
3972          config->cfi);
3973     needs &= ~NEED_DEVTREE;
3974        } else if (tmplLine->type == LT_ENTRY_END && needs & NEED_DEVTREE) {
3975     const char *ndtp = newDevTreePath;
3976     if (!strncmp(newDevTreePath, prefix, strlen(prefix)))
3977        ndtp += strlen(prefix);
3978     newLine = addLine(new, config->cfi, LT_DEVTREE,
3979      config->secondaryIndent,
3980      ndtp);
3981     needs &= ~NEED_DEVTREE;
3982     newLine = addLineTmpl(new, tmplLine, newLine, NULL, config->cfi);
3983      } else {      } else {
3984   /* pass through other lines from the template */   /* pass through other lines from the template */
3985   newLine = addLineTmpl(new, tmplLine, newLine, NULL, config->cfi);   newLine = addLineTmpl(new, tmplLine, newLine, NULL, config->cfi);
# Line 3568  int addNewKernel(struct grubConfig * con Line 3992  int addNewKernel(struct grubConfig * con
3992   */   */
3993   switch (config->cfi->entryStart) {   switch (config->cfi->entryStart) {
3994      case LT_KERNEL:      case LT_KERNEL:
3995        case LT_KERNEL_EFI:
3996        case LT_KERNEL_16:
3997   if (new->multiboot && config->cfi->mbHyperFirst) {   if (new->multiboot && config->cfi->mbHyperFirst) {
3998      /* fall through to LT_HYPER */      /* fall through to LT_HYPER */
3999   } else {   } else {
4000      newLine = addLine(new, config->cfi, LT_KERNEL,      newLine = addLine(new, config->cfi,
4001              preferredLineType(LT_KERNEL, config->cfi),
4002        config->primaryIndent,        config->primaryIndent,
4003        newKernelPath + strlen(prefix));        newKernelPath + strlen(prefix));
4004      needs &= ~NEED_KERNEL;      needs &= ~NEED_KERNEL;
# Line 3628  int addNewKernel(struct grubConfig * con Line 4055  int addNewKernel(struct grubConfig * con
4055   }   }
4056      }      }
4057    
4058        struct singleLine *endLine = NULL;
4059        endLine = getLineByType(LT_ENTRY_END, new->lines);
4060        if (endLine) {
4061        removeLine(new, endLine);
4062        needs |= NEED_END;
4063        }
4064    
4065      /* add the remainder of the lines, i.e. those that either      /* add the remainder of the lines, i.e. those that either
4066       * 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,
4067       * all the lines following the entryStart.       * all the lines following the entryStart.
# Line 3647  int addNewKernel(struct grubConfig * con Line 4081  int addNewKernel(struct grubConfig * con
4081      if (needs & NEED_KERNEL) {      if (needs & NEED_KERNEL) {
4082   newLine = addLine(new, config->cfi,   newLine = addLine(new, config->cfi,
4083    (new->multiboot && getKeywordByType(LT_MBMODULE,    (new->multiboot && getKeywordByType(LT_MBMODULE,
4084        config->cfi)) ?        config->cfi))
4085    LT_MBMODULE : LT_KERNEL,     ? LT_MBMODULE
4086     : preferredLineType(LT_KERNEL, config->cfi),
4087    config->secondaryIndent,    config->secondaryIndent,
4088    newKernelPath + strlen(prefix));    newKernelPath + strlen(prefix));
4089   needs &= ~NEED_KERNEL;   needs &= ~NEED_KERNEL;
# Line 3664  int addNewKernel(struct grubConfig * con Line 4099  int addNewKernel(struct grubConfig * con
4099   initrdVal = getInitrdVal(config, prefix, NULL, newKernelInitrd, extraInitrds, extraInitrdCount);   initrdVal = getInitrdVal(config, prefix, NULL, newKernelInitrd, extraInitrds, extraInitrdCount);
4100   newLine = addLine(new, config->cfi,   newLine = addLine(new, config->cfi,
4101    (new->multiboot && getKeywordByType(LT_MBMODULE,    (new->multiboot && getKeywordByType(LT_MBMODULE,
4102        config->cfi)) ?        config->cfi))
4103    LT_MBMODULE : LT_INITRD,     ? LT_MBMODULE
4104       : preferredLineType(LT_INITRD, config->cfi),
4105    config->secondaryIndent,    config->secondaryIndent,
4106    initrdVal);    initrdVal);
4107   free(initrdVal);   free(initrdVal);
4108   needs &= ~NEED_INITRD;   needs &= ~NEED_INITRD;
4109      }      }
4110        if (needs & NEED_DEVTREE) {
4111     newLine = addLine(new, config->cfi, LT_DEVTREE,
4112      config->secondaryIndent,
4113      newDevTreePath);
4114     needs &= ~NEED_DEVTREE;
4115        }
4116    
4117        /* NEEDS_END must be last on bootloaders that need it... */
4118      if (needs & NEED_END) {      if (needs & NEED_END) {
4119   newLine = addLine(new, config->cfi, LT_ENTRY_END,   newLine = addLine(new, config->cfi, LT_ENTRY_END,
4120   config->secondaryIndent, NULL);   config->secondaryIndent, NULL);
# Line 3697  static void traceback(int signum) Line 4141  static void traceback(int signum)
4141      memset(array, '\0', sizeof (array));      memset(array, '\0', sizeof (array));
4142      size = backtrace(array, 40);      size = backtrace(array, 40);
4143    
4144      fprintf(stderr, "grubby recieved SIGSEGV!  Backtrace (%ld):\n",      fprintf(stderr, "grubby received SIGSEGV!  Backtrace (%ld):\n",
4145              (unsigned long)size);              (unsigned long)size);
4146      backtrace_symbols_fd(array, size, STDERR_FILENO);      backtrace_symbols_fd(array, size, STDERR_FILENO);
4147      exit(1);      exit(1);
# Line 3722  int main(int argc, const char ** argv) { Line 4166  int main(int argc, const char ** argv) {
4166      char * newKernelArgs = NULL;      char * newKernelArgs = NULL;
4167      char * newKernelInitrd = NULL;      char * newKernelInitrd = NULL;
4168      char * newKernelTitle = NULL;      char * newKernelTitle = NULL;
4169      char * newKernelVersion = NULL;      char * newDevTreePath = NULL;
4170      char * newMBKernel = NULL;      char * newMBKernel = NULL;
4171      char * newMBKernelArgs = NULL;      char * newMBKernelArgs = NULL;
4172      char * removeMBKernelArgs = NULL;      char * removeMBKernelArgs = NULL;
# Line 3732  int main(int argc, const char ** argv) { Line 4176  int main(int argc, const char ** argv) {
4176      char * removeArgs = NULL;      char * removeArgs = NULL;
4177      char * kernelInfo = NULL;      char * kernelInfo = NULL;
4178      char * extraInitrds[MAX_EXTRA_INITRDS] = { NULL };      char * extraInitrds[MAX_EXTRA_INITRDS] = { NULL };
4179        char * envPath = NULL;
4180      const char * chptr = NULL;      const char * chptr = NULL;
4181      struct configFileInfo * cfi = NULL;      struct configFileInfo * cfi = NULL;
4182      struct grubConfig * config;      struct grubConfig * config;
# Line 3740  int main(int argc, const char ** argv) { Line 4185  int main(int argc, const char ** argv) {
4185      int displayDefault = 0;      int displayDefault = 0;
4186      int displayDefaultIndex = 0;      int displayDefaultIndex = 0;
4187      int displayDefaultTitle = 0;      int displayDefaultTitle = 0;
4188        int defaultIndex = -1;
4189      struct poptOption options[] = {      struct poptOption options[] = {
4190   { "add-kernel", 0, POPT_ARG_STRING, &newKernelPath, 0,   { "add-kernel", 0, POPT_ARG_STRING, &newKernelPath, 0,
4191      _("add an entry for the specified kernel"), _("kernel-path") },      _("add an entry for the specified kernel"), _("kernel-path") },
# Line 3778  int main(int argc, const char ** argv) { Line 4224  int main(int argc, const char ** argv) {
4224      _("display the index of the default kernel") },      _("display the index of the default kernel") },
4225   { "default-title", 0, 0, &displayDefaultTitle, 0,   { "default-title", 0, 0, &displayDefaultTitle, 0,
4226      _("display the title of the default kernel") },      _("display the title of the default kernel") },
4227     { "devtree", 0, POPT_ARG_STRING, &newDevTreePath, 0,
4228        _("device tree file for new stanza"), _("dtb-path") },
4229   { "elilo", 0, POPT_ARG_NONE, &configureELilo, 0,   { "elilo", 0, POPT_ARG_NONE, &configureELilo, 0,
4230      _("configure elilo bootloader") },      _("configure elilo bootloader") },
4231     { "efi", 0, POPT_ARG_NONE, &isEfi, 0,
4232        _("force grub2 stanzas to use efi") },
4233     { "env", 0, POPT_ARG_STRING, &envPath, 0,
4234        _("path for environment data"),
4235        _("path") },
4236   { "extlinux", 0, POPT_ARG_NONE, &configureExtLinux, 0,   { "extlinux", 0, POPT_ARG_NONE, &configureExtLinux, 0,
4237      _("configure extlinux bootloader (from syslinux)") },      _("configure extlinux bootloader (from syslinux)") },
4238   { "grub", 0, POPT_ARG_NONE, &configureGrub, 0,   { "grub", 0, POPT_ARG_NONE, &configureGrub, 0,
# Line 3792  int main(int argc, const char ** argv) { Line 4245  int main(int argc, const char ** argv) {
4245   { "initrd", 0, POPT_ARG_STRING, &newKernelInitrd, 0,   { "initrd", 0, POPT_ARG_STRING, &newKernelInitrd, 0,
4246      _("initrd image for the new kernel"), _("initrd-path") },      _("initrd image for the new kernel"), _("initrd-path") },
4247   { "extra-initrd", 'i', POPT_ARG_STRING, NULL, 'i',   { "extra-initrd", 'i', POPT_ARG_STRING, NULL, 'i',
4248      _("auxilliary initrd image for things other than the new kernel"), _("initrd-path") },      _("auxiliary initrd image for things other than the new kernel"), _("initrd-path") },
4249   { "lilo", 0, POPT_ARG_NONE, &configureLilo, 0,   { "lilo", 0, POPT_ARG_NONE, &configureLilo, 0,
4250      _("configure lilo bootloader") },      _("configure lilo bootloader") },
4251   { "make-default", 0, 0, &makeDefault, 0,   { "make-default", 0, 0, &makeDefault, 0,
# Line 3812  int main(int argc, const char ** argv) { Line 4265  int main(int argc, const char ** argv) {
4265   { "set-default", 0, POPT_ARG_STRING, &defaultKernel, 0,   { "set-default", 0, POPT_ARG_STRING, &defaultKernel, 0,
4266      _("make the first entry referencing the specified kernel "      _("make the first entry referencing the specified kernel "
4267        "the default"), _("kernel-path") },        "the default"), _("kernel-path") },
4268     { "set-default-index", 0, POPT_ARG_INT, &defaultIndex, 0,
4269        _("make the given entry index the default entry"),
4270        _("entry-index") },
4271   { "silo", 0, POPT_ARG_NONE, &configureSilo, 0,   { "silo", 0, POPT_ARG_NONE, &configureSilo, 0,
4272      _("configure silo bootloader") },      _("configure silo bootloader") },
4273   { "title", 0, POPT_ARG_STRING, &newKernelTitle, 0,   { "title", 0, POPT_ARG_STRING, &newKernelTitle, 0,
# Line 3833  int main(int argc, const char ** argv) { Line 4289  int main(int argc, const char ** argv) {
4289    
4290      signal(SIGSEGV, traceback);      signal(SIGSEGV, traceback);
4291    
4292        int i = 0;
4293        for (int j = 1; j < argc; j++)
4294     i += strlen(argv[j]) + 1;
4295        saved_command_line = malloc(i);
4296        if (!saved_command_line) {
4297     fprintf(stderr, "grubby: %m\n");
4298     exit(1);
4299        }
4300        saved_command_line[0] = '\0';
4301        for (int j = 1; j < argc; j++) {
4302     strcat(saved_command_line, argv[j]);
4303     strncat(saved_command_line, j == argc -1 ? "" : " ", 1);
4304        }
4305    
4306      optCon = poptGetContext("grubby", argc, argv, options, 0);      optCon = poptGetContext("grubby", argc, argv, options, 0);
4307      poptReadDefaultConfig(optCon, 1);      poptReadDefaultConfig(optCon, 1);
4308    
# Line 3876  int main(int argc, const char ** argv) { Line 4346  int main(int argc, const char ** argv) {
4346   return 1;   return 1;
4347      } else if (configureGrub2) {      } else if (configureGrub2) {
4348   cfi = &grub2ConfigType;   cfi = &grub2ConfigType;
4349     if (envPath)
4350        cfi->envFile = envPath;
4351      } else if (configureLilo) {      } else if (configureLilo) {
4352   cfi = &liloConfigType;   cfi = &liloConfigType;
4353      } else if (configureGrub) {      } else if (configureGrub) {
# Line 3919  int main(int argc, const char ** argv) { Line 4391  int main(int argc, const char ** argv) {
4391      grubConfig = cfi->defaultConfig;      grubConfig = cfi->defaultConfig;
4392      }      }
4393    
4394      if (bootloaderProbe && (displayDefault || kernelInfo || newKernelVersion ||      if (bootloaderProbe && (displayDefault || kernelInfo ||
4395    newKernelPath || removeKernelPath || makeDefault ||      newKernelPath || removeKernelPath || makeDefault ||
4396    defaultKernel || displayDefaultIndex || displayDefaultTitle)) {      defaultKernel || displayDefaultIndex || displayDefaultTitle ||
4397        (defaultIndex >= 0))) {
4398   fprintf(stderr, _("grubby: --bootloader-probe may not be used with "   fprintf(stderr, _("grubby: --bootloader-probe may not be used with "
4399    "specified option"));    "specified option"));
4400   return 1;   return 1;
4401      }      }
4402    
4403      if ((displayDefault || kernelInfo) && (newKernelVersion || newKernelPath ||      if ((displayDefault || kernelInfo) && (newKernelPath ||
4404     removeKernelPath)) {     removeKernelPath)) {
4405   fprintf(stderr, _("grubby: --default-kernel and --info may not "   fprintf(stderr, _("grubby: --default-kernel and --info may not "
4406    "be used when adding or removing kernels\n"));    "be used when adding or removing kernels\n"));
# Line 3963  int main(int argc, const char ** argv) { Line 4436  int main(int argc, const char ** argv) {
4436   makeDefault = 1;   makeDefault = 1;
4437   defaultKernel = NULL;   defaultKernel = NULL;
4438      }      }
4439        else if (defaultKernel && (defaultIndex >= 0)) {
4440     fprintf(stderr, _("grubby: --set-default and --set-default-index "
4441      "may not be used together\n"));
4442     return 1;
4443        }
4444    
4445      if (grubConfig && !strcmp(grubConfig, "-") && !outputFile) {      if (grubConfig && !strcmp(grubConfig, "-") && !outputFile) {
4446   fprintf(stderr, _("grubby: output file must be specified if stdin "   fprintf(stderr, _("grubby: output file must be specified if stdin "
# Line 3971  int main(int argc, const char ** argv) { Line 4449  int main(int argc, const char ** argv) {
4449      }      }
4450    
4451      if (!removeKernelPath && !newKernelPath && !displayDefault && !defaultKernel      if (!removeKernelPath && !newKernelPath && !displayDefault && !defaultKernel
4452   && !kernelInfo && !bootloaderProbe && !updateKernelPath   && !kernelInfo && !bootloaderProbe && !updateKernelPath
4453          && !removeMBKernel && !displayDefaultIndex && !displayDefaultTitle) {   && !removeMBKernel && !displayDefaultIndex && !displayDefaultTitle
4454     && (defaultIndex == -1)) {
4455   fprintf(stderr, _("grubby: no action specified\n"));   fprintf(stderr, _("grubby: no action specified\n"));
4456   return 1;   return 1;
4457      }      }
# Line 4052  int main(int argc, const char ** argv) { Line 4531  int main(int argc, const char ** argv) {
4531      if (!yconfig)      if (!yconfig)
4532   yrc = 1;   yrc = 1;
4533      else      else
4534        yrc = checkForYaboot(lconfig);   yrc = checkForYaboot(yconfig);
4535   }   }
4536    
4537   if (lrc == 1 || grc == 1 || gr2c == 1 || extrc == 1 || yrc == 1 ||   if (lrc == 1 || grc == 1 || gr2c == 1 || extrc == 1 || yrc == 1 ||
4538      erc == 1)   erc == 1)
4539      return 1;      return 1;
4540    
4541   if (lrc == 2) printf("lilo\n");   if (lrc == 2) printf("lilo\n");
# Line 4069  int main(int argc, const char ** argv) { Line 4548  int main(int argc, const char ** argv) {
4548   return 0;   return 0;
4549      }      }
4550    
4551        if (grubConfig == NULL) {
4552     printf("Could not find bootloader configuration file.\n");
4553     exit(1);
4554        }
4555    
4556      config = readConfig(grubConfig, cfi);      config = readConfig(grubConfig, cfi);
4557      if (!config) return 1;      if (!config) return 1;
4558    
# Line 4078  int main(int argc, const char ** argv) { Line 4562  int main(int argc, const char ** argv) {
4562          char * rootspec;          char * rootspec;
4563    
4564   if (config->defaultImage == -1) return 0;   if (config->defaultImage == -1) return 0;
4565     if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&
4566     cfi->defaultIsSaved)
4567        config->defaultImage = 0;
4568   entry = findEntryByIndex(config, config->defaultImage);   entry = findEntryByIndex(config, config->defaultImage);
4569   if (!entry) return 0;   if (!entry) return 0;
4570   if (!suitableImage(entry, bootPrefix, 0, flags)) return 0;   if (!suitableImage(entry, bootPrefix, 0, flags)) return 0;
4571    
4572   line = getLineByType(LT_KERNEL|LT_HYPER, entry->lines);   line = getLineByType(LT_KERNEL|LT_HYPER|LT_KERNEL_EFI|LT_KERNEL_16, entry->lines);
4573   if (!line) return 0;   if (!line) return 0;
4574    
4575          rootspec = getRootSpecifier(line->elements[1].item);          rootspec = getRootSpecifier(line->elements[1].item);
# Line 4096  int main(int argc, const char ** argv) { Line 4583  int main(int argc, const char ** argv) {
4583   struct singleEntry * entry;   struct singleEntry * entry;
4584    
4585   if (config->defaultImage == -1) return 0;   if (config->defaultImage == -1) return 0;
4586     if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&
4587     cfi->defaultIsSaved)
4588        config->defaultImage = 0;
4589   entry = findEntryByIndex(config, config->defaultImage);   entry = findEntryByIndex(config, config->defaultImage);
4590   if (!entry) return 0;   if (!entry) return 0;
4591    
# Line 4118  int main(int argc, const char ** argv) { Line 4608  int main(int argc, const char ** argv) {
4608    
4609      } else if (displayDefaultIndex) {      } else if (displayDefaultIndex) {
4610          if (config->defaultImage == -1) return 0;          if (config->defaultImage == -1) return 0;
4611     if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&
4612     cfi->defaultIsSaved)
4613        config->defaultImage = 0;
4614          printf("%i\n", config->defaultImage);          printf("%i\n", config->defaultImage);
4615            return 0;
4616    
4617      } else if (kernelInfo)      } else if (kernelInfo)
4618   return displayInfo(config, kernelInfo, bootPrefix);   return displayInfo(config, kernelInfo, bootPrefix);
# Line 4131  int main(int argc, const char ** argv) { Line 4625  int main(int argc, const char ** argv) {
4625      markRemovedImage(config, removeKernelPath, bootPrefix);      markRemovedImage(config, removeKernelPath, bootPrefix);
4626      markRemovedImage(config, removeMBKernel, bootPrefix);      markRemovedImage(config, removeMBKernel, bootPrefix);
4627      setDefaultImage(config, newKernelPath != NULL, defaultKernel, makeDefault,      setDefaultImage(config, newKernelPath != NULL, defaultKernel, makeDefault,
4628      bootPrefix, flags);      bootPrefix, flags, defaultIndex);
4629      setFallbackImage(config, newKernelPath != NULL);      setFallbackImage(config, newKernelPath != NULL);
4630      if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs,      if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs,
4631                      removeArgs, newMBKernelArgs, removeMBKernelArgs)) return 1;                      removeArgs, newMBKernelArgs, removeMBKernelArgs)) return 1;
4632      if (updateKernelPath && newKernelInitrd) {      if (updateKernelPath && newKernelInitrd) {
4633              if (updateInitrd(config, updateKernelPath, bootPrefix,      if (newMBKernel) {
4634                               newKernelInitrd)) return 1;      if (addMBInitrd(config, newMBKernel, updateKernelPath,
4635     bootPrefix, newKernelInitrd))
4636        return 1;
4637        } else {
4638        if (updateInitrd(config, updateKernelPath, bootPrefix,
4639     newKernelInitrd))
4640     return 1;
4641        }
4642      }      }
4643      if (addNewKernel(config, template, bootPrefix, newKernelPath,      if (addNewKernel(config, template, bootPrefix, newKernelPath,
4644                       newKernelTitle, newKernelArgs, newKernelInitrd,                       newKernelTitle, newKernelArgs, newKernelInitrd,
4645                       (const char **)extraInitrds, extraInitrdCount,                       (const char **)extraInitrds, extraInitrdCount,
4646                       newMBKernel, newMBKernelArgs)) return 1;                       newMBKernel, newMBKernelArgs, newDevTreePath)) return 1;
4647            
4648    
4649      if (numEntries(config) == 0) {      if (numEntries(config) == 0) {

Legend:
Removed from v.1854  
changed lines
  Added in v.2718