Magellan Linux

  • Back to homepage
    • /[magellan-source]/trunk/grubby/grubby.c

    Diff of /trunk/grubby/grubby.c

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

    revision 3014 by niro, Tue Jun 27 14:34:11 2017 UTC revision 3029 by niro, Tue Jun 27 14:46:56 2017 UTC
    # Line 132  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    
    # Line 677  struct grubConfig { Line 681  struct grubConfig {
    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;
    # Line 1296  static struct grubConfig *readConfig(con Line 1302  static struct grubConfig *readConfig(con
    1302   cfg->theLines = NULL;   cfg->theLines = NULL;
    1303   cfg->entries = NULL;   cfg->entries = NULL;
    1304   cfg->fallbackImage = 0;   cfg->fallbackImage = 0;
    1305     cfg->isModified = 0;
    1306    
    1307   /* copy everything we have */   /* copy everything we have */
    1308   while (*head) {   while (*head) {
    # Line 1615  static struct grubConfig *readConfig(con Line 1622  static struct grubConfig *readConfig(con
    1622   *end == ' ' || *end == '\t'))   *end == ' ' || *end == '\t'))
    1623   end++;   end++;
    1624   if (*end)   if (*end)
    1625   cfg->defaultImage = -1;   cfg->defaultImage = NO_DEFAULT_ENTRY;
    1626   } else if (defaultLine->numElements == 3) {   } else if (defaultLine->numElements == 3) {
    1627   char *value = defaultLine->elements[2].item;   char *value = defaultLine->elements[2].item;
    1628   while (*value && (*value == '"' ||   while (*value && (*value == '"' ||
    # Line 1628  static struct grubConfig *readConfig(con Line 1635  static struct grubConfig *readConfig(con
    1635   *end == ' ' || *end == '\t'))   *end == ' ' || *end == '\t'))
    1636   end++;   end++;
    1637   if (*end)   if (*end)
    1638   cfg->defaultImage = -1;   cfg->defaultImage = NO_DEFAULT_ENTRY;
    1639   }   }
    1640   } else if (cfi->defaultSupportSaved &&   } else if (cfi->defaultSupportSaved &&
    1641     !strncmp(defaultLine->elements[1].item, "saved",     !strncmp(defaultLine->elements[1].item, "saved",
    # Line 1638  static struct grubConfig *readConfig(con Line 1645  static struct grubConfig *readConfig(con
    1645   cfg->defaultImage =   cfg->defaultImage =
    1646      strtol(defaultLine->elements[1].item, &end, 10);      strtol(defaultLine->elements[1].item, &end, 10);
    1647   if (*end)   if (*end)
    1648   cfg->defaultImage = -1;   cfg->defaultImage = NO_DEFAULT_ENTRY;
    1649   } else if (defaultLine->numElements >= 2) {   } else if (defaultLine->numElements >= 2) {
    1650   int i = 0;   int i = 0;
    1651   while ((entry = findEntryByIndex(cfg, i))) {   while ((entry = findEntryByIndex(cfg, i))) {
    # Line 1666  static struct grubConfig *readConfig(con Line 1673  static struct grubConfig *readConfig(con
    1673   if (entry) {   if (entry) {
    1674   cfg->defaultImage = i;   cfg->defaultImage = i;
    1675   } else {   } else {
    1676   cfg->defaultImage = -1;   cfg->defaultImage = NO_DEFAULT_ENTRY;
    1677   }   }
    1678   }   }
    1679   } else if (cfg->cfi->defaultIsSaved && cfg->cfi->getEnv) {   } else if (cfg->cfi->defaultIsSaved && cfg->cfi->getEnv) {
    # Line 1683  static struct grubConfig *readConfig(con Line 1690  static struct grubConfig *readConfig(con
    1690   cfg->defaultImage = index;   cfg->defaultImage = index;
    1691   }   }
    1692   } else {   } else {
    1693   cfg->defaultImage = 0;   cfg->defaultImage = FIRST_ENTRY_INDEX;
    1694   }   }
    1695    
    1696   return cfg;   return cfg;
    # Line 1703  static void writeDefault(FILE * out, cha Line 1710  static void writeDefault(FILE * out, cha
    1710   fprintf(out, "%sdefault%ssaved\n", indent, separator);   fprintf(out, "%sdefault%ssaved\n", indent, separator);
    1711   else if (cfg->cfi->defaultIsSaved) {   else if (cfg->cfi->defaultIsSaved) {
    1712   fprintf(out, "%sset default=\"${saved_entry}\"\n", indent);   fprintf(out, "%sset default=\"${saved_entry}\"\n", indent);
    1713   if (cfg->defaultImage >= 0 && cfg->cfi->setEnv) {   if (cfg->defaultImage >= FIRST_ENTRY_INDEX && cfg->cfi->setEnv) {
    1714   char *title;   char *title;
    1715   entry = findEntryByIndex(cfg, cfg->defaultImage);   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);   line = getLineByType(LT_MENUENTRY, entry->lines);
    1730   if (!line)   if (!line)
    1731   line = getLineByType(LT_TITLE, entry->lines);   line = getLineByType(LT_TITLE, entry->lines);
    # Line 1716  static void writeDefault(FILE * out, cha Line 1736  static void writeDefault(FILE * out, cha
    1736   "saved_entry", title);   "saved_entry", title);
    1737   }   }
    1738   }   }
    1739   } else if (cfg->defaultImage > -1) {   } else if (cfg->defaultImage >= FIRST_ENTRY_INDEX) {
    1740   if (cfg->cfi->defaultIsIndex) {   if (cfg->cfi->defaultIsIndex) {
    1741   if (cfg->cfi->defaultIsVariable) {   if (cfg->cfi->defaultIsVariable) {
    1742   fprintf(out, "%sset default=\"%d\"\n", indent,   fprintf(out, "%sset default=\"%d\"\n", indent,
    # Line 2420  struct singleEntry *findTemplate(struct Line 2440  struct singleEntry *findTemplate(struct
    2440   }   }
    2441   }   }
    2442   }   }
    2443   } else if (cfg->defaultImage > -1) {   } else if (cfg->defaultImage >= FIRST_ENTRY_INDEX) {
    2444   entry = findEntryByIndex(cfg, cfg->defaultImage);   entry = findEntryByIndex(cfg, cfg->defaultImage);
    2445   if (entry && suitableImage(entry, prefix, skipRemoved, flags)) {   if (entry && suitableImage(entry, prefix, skipRemoved, flags)) {
    2446   if (indexPtr)   if (indexPtr)
    # Line 2432  struct singleEntry *findTemplate(struct Line 2452  struct singleEntry *findTemplate(struct
    2452   index = 0;   index = 0;
    2453   while ((entry = findEntryByIndex(cfg, index))) {   while ((entry = findEntryByIndex(cfg, index))) {
    2454   if (suitableImage(entry, prefix, skipRemoved, flags)) {   if (suitableImage(entry, prefix, skipRemoved, flags)) {
    2455   int j;   int j, unmodifiedIndex;
    2456   for (j = 0; j < index; j++) {  
    2457     unmodifiedIndex = index;
    2458    
    2459     for (j = 0; j < unmodifiedIndex; j++) {
    2460   entry2 = findEntryByIndex(cfg, j);   entry2 = findEntryByIndex(cfg, j);
    2461   if (entry2->skip)   if (entry2->skip)
    2462   index--;   index--;
    # Line 2494  void markRemovedImage(struct grubConfig Line 2517  void markRemovedImage(struct grubConfig
    2517   entry->skip = 1;   entry->skip = 1;
    2518  }  }
    2519    
    2520  void setDefaultImage(struct grubConfig *config, int hasNew,  void setDefaultImage(struct grubConfig *config, int isAddingBootEntry,
    2521       const char *defaultKernelPath, int newIsDefault,       const char *defaultKernelPath, int newBootEntryIsDefault,
    2522       const char *prefix, int flags, int index)       const char *prefix, int flags,
    2523  {       int newDefaultBootEntryIndex, int newBootEntryIndex)
    2524   struct singleEntry *entry, *entry2, *newDefault;  {
    2525   int i, j;   struct singleEntry *bootEntry, *newDefault;
    2526     int indexToVerify, firstKernelEntryIndex, currentLookupIndex;
    2527   if (newIsDefault) {  
    2528   config->defaultImage = 0;          /* initialize */
    2529            currentLookupIndex = FIRST_ENTRY_INDEX;
    2530    
    2531     /* handle the two cases where the user explictly picks the default
    2532     * boot entry index as it would exist post-modification */
    2533    
    2534     /* Case 1: user chose to make the latest boot entry the default */
    2535     if (newBootEntryIsDefault) {
    2536     config->defaultImage = newBootEntryIndex;
    2537   return;   return;
    2538   } else if ((index >= 0) && config->cfi->defaultIsIndex) {   }
    2539   if (findEntryByIndex(config, index))  
    2540   config->defaultImage = index;   /* Case 2: user picked an arbitrary index as the default boot entry */
    2541   else   if (newDefaultBootEntryIndex >= FIRST_ENTRY_INDEX
    2542   config->defaultImage = -1;      && config->cfi->defaultIsIndex) {
    2543   return;   indexToVerify = newDefaultBootEntryIndex;
    2544   } else if (defaultKernelPath) {  
    2545   i = 0;   /* user chose to make latest boot entry the default */
    2546   if (findEntryByPath(config, defaultKernelPath, prefix, &i)) {   if (newDefaultBootEntryIndex == newBootEntryIndex) {
    2547   config->defaultImage = i;   config->defaultImage = newBootEntryIndex;
      } else {  
      config->defaultImage = -1;  
    2548   return;   return;
    2549   }   }
      }  
    2550    
    2551   /* defaultImage now points to what we'd like to use, but before any   /* the user picks the default index based on the
    2552   * order changes */   * order of the bootloader configuration after
    2553   if ((config->defaultImage == DEFAULT_SAVED) ||   * modification; ensure we are checking for the
    2554      (config->defaultImage == DEFAULT_SAVED_GRUB2))   * existence of the correct entry */
    2555   /* default is set to saved, we don't want to change it */   if (newBootEntryIndex < newDefaultBootEntryIndex) {
    2556     if (!config->isModified)
    2557     indexToVerify--;
    2558     }
    2559    
    2560     /* verify the user selected index will exist */
    2561     if (findEntryByIndex(config, indexToVerify)) {
    2562     config->defaultImage = newDefaultBootEntryIndex;
    2563     } else {
    2564     config->defaultImage = NO_DEFAULT_ENTRY;
    2565     }
    2566    
    2567   return;   return;
    2568     }
    2569    
    2570   if (config->defaultImage > -1)   /* handle cases where the index value may shift */
      entry = findEntryByIndex(config, config->defaultImage);  
      else  
      entry = NULL;  
    2571    
    2572   if (entry && !entry->skip) {   /* check validity of existing default or first-entry-found
    2573   /* we can preserve the default */     selection */
    2574   if (hasNew)   if (defaultKernelPath) {
    2575   config->defaultImage++;                  /* we must initialize this */
    2576                    firstKernelEntryIndex = 0;
    2577     /* user requested first-entry-found */
    2578     if (!findEntryByPath(config, defaultKernelPath,
    2579         prefix, &firstKernelEntryIndex)) {
    2580     /* don't change default if can't find match */
    2581     config->defaultImage = NO_DEFAULT_ENTRY;
    2582     return;
    2583     }
    2584    
    2585     config->defaultImage = firstKernelEntryIndex;
    2586    
    2587     /* this is where we start looking for decrement later */
    2588     currentLookupIndex = config->defaultImage;
    2589    
    2590   /* count the number of entries erased before this one */   if (isAddingBootEntry && !config->isModified &&
    2591   for (j = 0; j < config->defaultImage; j++) {      (newBootEntryIndex < config->defaultImage)) {
    2592   entry2 = findEntryByIndex(config, j);   /* increment because new entry added before default */
    2593   if (entry2->skip)   config->defaultImage++;
      config->defaultImage--;  
    2594   }   }
      } else if (hasNew) {  
      config->defaultImage = 0;  
    2595   } else {   } else {
    2596   /* Either we just erased the default (or the default line was                  /* check to see if the default is stored in the environment */
    2597   * bad to begin with) and didn't put a new one in. We'll use                  if (config->defaultImage < FIRST_ENTRY_INDEX) {
    2598   * the first valid image. */                      if (config->defaultImage == DEFAULT_SAVED || config->defaultImage == DEFAULT_SAVED_GRUB2)
    2599                        {
    2600                            if (config->cfi->defaultIsSaved) {
    2601                                if (config->cfi->getEnv) {
    2602                                    char *defaultTitle = config->cfi->getEnv(config->cfi, "saved_entry");
    2603    
    2604                                    if (defaultTitle) {
    2605                                        if (isnumber(defaultTitle)) {
    2606                                            currentLookupIndex = atoi(defaultTitle);
    2607                                        } else {
    2608                                            findEntryByTitle(config, defaultTitle, &currentLookupIndex);
    2609                                        }
    2610                                        /* set the default Image to an actual index */
    2611                                        config->defaultImage = currentLookupIndex;
    2612                                    }
    2613                                }
    2614                             }
    2615                        }
    2616                    } else {
    2617                            /* use pre-existing default entry from the file*/
    2618                            currentLookupIndex = config->defaultImage;
    2619                    }
    2620    
    2621     if (isAddingBootEntry
    2622        && (newBootEntryIndex <= config->defaultImage)) {
    2623     config->defaultImage++;
    2624    
    2625     if (config->isModified) {
    2626     currentLookupIndex++;
    2627     }
    2628     }
    2629     }
    2630    
    2631     /* sanity check - is this entry index valid? */
    2632     bootEntry = findEntryByIndex(config, currentLookupIndex);
    2633    
    2634     if ((bootEntry && bootEntry->skip) || !bootEntry) {
    2635     /* entry is to be skipped or is invalid */
    2636     if (isAddingBootEntry) {
    2637     config->defaultImage = newBootEntryIndex;
    2638     return;
    2639     }
    2640   newDefault =   newDefault =
    2641      findTemplate(config, prefix, &config->defaultImage, 1,      findTemplate(config, prefix, &config->defaultImage, 1,
    2642   flags);   flags);
    2643   if (!newDefault)   if (!newDefault) {
    2644   config->defaultImage = -1;   config->defaultImage = NO_DEFAULT_ENTRY;
    2645     }
    2646    
    2647     return;
    2648     }
    2649    
    2650     currentLookupIndex--;
    2651    
    2652     /* decrement index by the total number of entries deleted */
    2653    
    2654     for (indexToVerify = currentLookupIndex;
    2655         indexToVerify >= FIRST_ENTRY_INDEX; indexToVerify--) {
    2656    
    2657     bootEntry = findEntryByIndex(config, indexToVerify);
    2658    
    2659     if (bootEntry && bootEntry->skip) {
    2660     config->defaultImage--;
    2661     }
    2662   }   }
    2663  }  }
    2664    
    # Line 2582  void setFallbackImage(struct grubConfig Line 2687  void setFallbackImage(struct grubConfig
    2687   }   }
    2688  }  }
    2689    
    2690  void displayEntry(struct singleEntry *entry, const char *prefix, int index)  void displayEntry(struct grubConfig *config, struct singleEntry *entry, const char *prefix, int index)
    2691  {  {
    2692   struct singleLine *line;   struct singleLine *line;
    2693   char *root = NULL;   char *root = NULL;
    # Line 2678  void displayEntry(struct singleEntry *en Line 2783  void displayEntry(struct singleEntry *en
    2783    
    2784   line = getLineByType(LT_TITLE, entry->lines);   line = getLineByType(LT_TITLE, entry->lines);
    2785   if (line) {   if (line) {
    2786   printf("title=%s\n", line->elements[1].item);                  char *entryTitle;
    2787                    /* if we can extractTitle, then it's a zipl config and
    2788                     * if not then we go ahead with what's existed prior */
    2789                    entryTitle = extractTitle(config, line);
    2790                    if (!entryTitle) {
    2791                        entryTitle=line->elements[1].item;
    2792                    }
    2793     printf("title=%s\n", entryTitle);
    2794   } else {   } else {
    2795   char *title;   char *title;
    2796   line = getLineByType(LT_MENUENTRY, entry->lines);   line = getLineByType(LT_MENUENTRY, entry->lines);
    # Line 3094  int displayInfo(struct grubConfig *confi Line 3206  int displayInfo(struct grubConfig *confi
    3206   printf("lba\n");   printf("lba\n");
    3207   }   }
    3208    
    3209   displayEntry(entry, prefix, i);   displayEntry(config, entry, prefix, i);
    3210    
    3211   i++;   i++;
    3212   while ((entry = findEntryByPath(config, kernel, prefix, &i))) {   while ((entry = findEntryByPath(config, kernel, prefix, &i))) {
    3213   displayEntry(entry, prefix, i);   displayEntry(config, entry, prefix, i);
    3214   i++;   i++;
    3215   }   }
    3216    
    # Line 4725  int addNewKernel(struct grubConfig *conf Line 4837  int addNewKernel(struct grubConfig *conf
    4837   }   }
    4838    
    4839   if (updateImage(config, indexs, prefix, newKernelArgs, NULL,   if (updateImage(config, indexs, prefix, newKernelArgs, NULL,
    4840   newMBKernelArgs, NULL))   newMBKernelArgs, NULL)) {
    4841     config->isModified = 1;
    4842   return 1;   return 1;
    4843     }
    4844    
    4845   return 0;   return 0;
    4846  }  }
    # Line 5178  int main(int argc, const char **argv) Line 5292  int main(int argc, const char **argv)
    5292   struct singleEntry *entry;   struct singleEntry *entry;
    5293   char *rootspec;   char *rootspec;
    5294    
    5295   if (config->defaultImage == -1)   if (config->defaultImage == NO_DEFAULT_ENTRY)
    5296   return 0;   return 0;
    5297   if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&   if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&
    5298      cfi->defaultIsSaved)      cfi->defaultIsSaved)
    5299   config->defaultImage = 0;   config->defaultImage = FIRST_ENTRY_INDEX;
    5300   entry = findEntryByIndex(config, config->defaultImage);   entry = findEntryByIndex(config, config->defaultImage);
    5301   if (!entry)   if (!entry)
    5302   return 0;   return 0;
    # Line 5205  int main(int argc, const char **argv) Line 5319  int main(int argc, const char **argv)
    5319   struct singleLine *line;   struct singleLine *line;
    5320   struct singleEntry *entry;   struct singleEntry *entry;
    5321    
    5322   if (config->defaultImage == -1)   if (config->defaultImage == NO_DEFAULT_ENTRY)
    5323   return 0;   return 0;
    5324   if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&   if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&
    5325      cfi->defaultIsSaved)      cfi->defaultIsSaved)
    5326   config->defaultImage = 0;   config->defaultImage = FIRST_ENTRY_INDEX;
    5327   entry = findEntryByIndex(config, config->defaultImage);   entry = findEntryByIndex(config, config->defaultImage);
    5328   if (!entry)   if (!entry)
    5329   return 0;   return 0;
    # Line 5239  int main(int argc, const char **argv) Line 5353  int main(int argc, const char **argv)
    5353   return 0;   return 0;
    5354    
    5355   } else if (displayDefaultIndex) {   } else if (displayDefaultIndex) {
    5356   if (config->defaultImage == -1)   if (config->defaultImage == NO_DEFAULT_ENTRY)
    5357   return 0;   return 0;
    5358   if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&   if (config->defaultImage == DEFAULT_SAVED_GRUB2 &&
    5359      cfi->defaultIsSaved)      cfi->defaultIsSaved)
    5360   config->defaultImage = 0;   config->defaultImage = FIRST_ENTRY_INDEX;
    5361   printf("%i\n", config->defaultImage);   printf("%i\n", config->defaultImage);
    5362   return 0;   return 0;
    5363    
    # Line 5259  int main(int argc, const char **argv) Line 5373  int main(int argc, const char **argv)
    5373   markRemovedImage(config, removeKernelPath, bootPrefix);   markRemovedImage(config, removeKernelPath, bootPrefix);
    5374   markRemovedImage(config, removeMBKernel, bootPrefix);   markRemovedImage(config, removeMBKernel, bootPrefix);
    5375   setDefaultImage(config, newKernelPath != NULL, defaultKernel,   setDefaultImage(config, newKernelPath != NULL, defaultKernel,
    5376   makeDefault, bootPrefix, flags, defaultIndex);   makeDefault, bootPrefix, flags, defaultIndex,
    5377     newIndex);
    5378   setFallbackImage(config, newKernelPath != NULL);   setFallbackImage(config, newKernelPath != NULL);
    5379   if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs,   if (updateImage(config, updateKernelPath, bootPrefix, newKernelArgs,
    5380   removeArgs, newMBKernelArgs, removeMBKernelArgs))   removeArgs, newMBKernelArgs, removeMBKernelArgs))
    Legend:
    Removed from v.3014  
    changed lines
      Added in v.3029