]> code.delx.au - refind/blobdiff - refind/config.c
TianoCore build support; new use_graphics_for refind.conf token
[refind] / refind / config.c
index ebc7d083e4f872b09e0594f997284a2a66e3e586..eb5040c5b59085db4b5eb61f1983d635f5c451ea 100644 (file)
 #include "menu.h"
 #include "config.h"
 #include "screen.h"
-#include "refit_call_wrapper.h"
+#include "../include/refit_call_wrapper.h"
 
 // constants
 
 #define CONFIG_FILE_NAME         L"refind.conf"
-#define LINUX_OPTIONS_FILENAMES  L"refind_linux.conf,refind-linux.conf,linux.conf"
+#define LINUX_OPTIONS_FILENAMES  L"refind_linux.conf,refind-linux.conf"
 #define MAXCONFIGFILESIZE        (128*1024)
 
 #define ENCODING_ISO8859_1  (0)
@@ -138,18 +138,18 @@ static CHAR16 *ReadLine(REFIT_FILE *File)
 {
     CHAR16  *Line, *q;
     UINTN   LineLength;
-    
+
     if (File->Buffer == NULL)
         return NULL;
-    
+
     if (File->Encoding == ENCODING_ISO8859_1 || File->Encoding == ENCODING_UTF8) {
-        
+
         CHAR8 *p, *LineStart, *LineEnd;
-        
+
         p = File->Current8Ptr;
         if (p >= File->End8Ptr)
             return NULL;
-        
+
         LineStart = p;
         for (; p < File->End8Ptr; p++)
             if (*p == 13 || *p == 10)
@@ -159,12 +159,12 @@ static CHAR16 *ReadLine(REFIT_FILE *File)
             if (*p != 13 && *p != 10)
                 break;
         File->Current8Ptr = p;
-        
+
         LineLength = (UINTN)(LineEnd - LineStart) + 1;
         Line = AllocatePool(LineLength * sizeof(CHAR16));
         if (Line == NULL)
             return NULL;
-        
+
         q = Line;
         if (File->Encoding == ENCODING_ISO8859_1) {
             for (p = LineStart; p < LineEnd; )
@@ -267,40 +267,39 @@ VOID FreeTokenLine(IN OUT CHAR16 ***TokenList, IN OUT UINTN *TokenCount)
     FreeList((VOID ***)TokenList, TokenCount);
 }
 
-//
 // handle a parameter with a single integer argument
-//
-
 static VOID HandleInt(IN CHAR16 **TokenList, IN UINTN TokenCount, OUT UINTN *Value)
 {
-    if (TokenCount < 2) {
-        return;
-    }
-    if (TokenCount > 2) {
-        return;
-    }
-    *Value = Atoi(TokenList[1]);
+    if (TokenCount == 2)
+       *Value = Atoi(TokenList[1]);
 }
 
-//
 // handle a parameter with a single string argument
-//
+static VOID HandleString(IN CHAR16 **TokenList, IN UINTN TokenCount, OUT CHAR16 **Target) {
+   if (TokenCount == 2) {
+      if (*Target != NULL)
+         FreePool(*Target);
+      *Target = StrDuplicate(TokenList[1]);
+   } // if
+} // static VOID HandleString()
 
-static VOID HandleString(IN CHAR16 **TokenList, IN UINTN TokenCount, OUT CHAR16 **Value)
-{
-    if (TokenCount < 2) {
-        return;
-    }
-    if (TokenCount > 2) {
-        return;
-    }
-    *Value = StrDuplicate(TokenList[1]);
-}
+// Handle a parameter with a series of string arguments, to be added to a comma-delimited
+// list. Passes each token through the CleanUpPathNameSlashes() function to ensure
+// consistency in subsequent comparisons of filenames.
+static VOID HandleStrings(IN CHAR16 **TokenList, IN UINTN TokenCount, OUT CHAR16 **Target) {
+   UINTN i;
 
-//
-// read config file
-//
+   if (*Target != NULL) {
+      FreePool(*Target);
+      *Target = NULL;
+   } // if
+   for (i = 1; i < TokenCount; i++) {
+      CleanUpPathNameSlashes(TokenList[i]);
+      MergeStrings(Target, TokenList[i], L',');
+   }
+} // static VOID HandleStrings()
 
+// read config file
 VOID ReadConfig(VOID)
 {
     EFI_STATUS      Status;
@@ -326,11 +325,7 @@ VOID ReadConfig(VOID)
         if (StriCmp(TokenList[0], L"timeout") == 0) {
             HandleInt(TokenList, TokenCount, &(GlobalConfig.Timeout));
 
-        // Note: I'm using "disable" as equivalent to "hideui" for the moment (as of rEFInd 0.2.4)
-        // because I've folded two options into one and removed some values, so I want to catch
-        // existing configurations as much as possible. The "disable" equivalency to "hideui" will
-        // be removed sooner or later, leaving only "hideui".
-        } else if ((StriCmp(TokenList[0], L"hideui") == 0) || (StriCmp(TokenList[0], L"disable") == 0)) {
+        } else if (StriCmp(TokenList[0], L"hideui") == 0) {
             for (i = 1; i < TokenCount; i++) {
                 FlagName = TokenList[i];
                 if (StriCmp(FlagName, L"banner") == 0) {
@@ -350,10 +345,8 @@ VOID ReadConfig(VOID)
                 }
             }
 
-        } else if ((StriCmp(TokenList[0], L"icons_dir") == 0) && (TokenCount == 2)) {
-           if (GlobalConfig.IconsDir != NULL)
-              FreePool(GlobalConfig.IconsDir);
-           GlobalConfig.IconsDir = StrDuplicate(TokenList[1]);
+        } else if (StriCmp(TokenList[0], L"icons_dir") == 0) {
+           HandleString(TokenList, TokenCount, &(GlobalConfig.IconsDir));
 
         } else if (StriCmp(TokenList[0], L"scanfor") == 0) {
            for (i = 0; i < NUM_SCAN_OPTIONS; i++) {
@@ -364,20 +357,13 @@ VOID ReadConfig(VOID)
            }
 
         } else if (StriCmp(TokenList[0], L"also_scan_dirs") == 0) {
-           if (GlobalConfig.AlsoScan != NULL) {
-              FreePool(GlobalConfig.AlsoScan);
-              GlobalConfig.AlsoScan = NULL;
-           } // if
-           for (i = 1; i < TokenCount; i++)
-              MergeStrings(&GlobalConfig.AlsoScan, TokenList[i], L',');
+            HandleStrings(TokenList, TokenCount, &(GlobalConfig.AlsoScan));
+
+        } else if ((StriCmp(TokenList[0], L"don't_scan_dirs") == 0) || (StriCmp(TokenList[0], L"dont_scan_dirs") == 0)) {
+            HandleStrings(TokenList, TokenCount, &(GlobalConfig.DontScan));
 
         } else if (StriCmp(TokenList[0], L"scan_driver_dirs") == 0) {
-           if (GlobalConfig.DriverDirs != NULL) {
-              FreePool(GlobalConfig.DriverDirs);
-              GlobalConfig.DriverDirs = NULL;
-           } // if
-           for (i = 1; i < TokenCount; i++)
-              MergeStrings(&GlobalConfig.DriverDirs, TokenList[i], L',');
+            HandleStrings(TokenList, TokenCount, &(GlobalConfig.DriverDirs));
 
         } else if (StriCmp(TokenList[0], L"showtools") == 0) {
             SetMem(GlobalConfig.ShowTools, NUM_TOOLS * sizeof(UINTN), 0);
@@ -396,21 +382,21 @@ VOID ReadConfig(VOID)
                 } else if (StriCmp(FlagName, L"shutdown") == 0) {
                    GlobalConfig.ShowTools[i - 1] = TAG_SHUTDOWN;
                 } else {
-                    Print(L" unknown showtools flag: '%s'\n", FlagName);
+                   Print(L" unknown showtools flag: '%s'\n", FlagName);
                 }
             } // showtools options
 
         } else if (StriCmp(TokenList[0], L"banner") == 0) {
-            HandleString(TokenList, TokenCount, &(GlobalConfig.BannerFileName));
+           HandleString(TokenList, TokenCount, &(GlobalConfig.BannerFileName));
 
         } else if (StriCmp(TokenList[0], L"selection_small") == 0) {
-            HandleString(TokenList, TokenCount, &(GlobalConfig.SelectionSmallFileName));
+           HandleString(TokenList, TokenCount, &(GlobalConfig.SelectionSmallFileName));
 
         } else if (StriCmp(TokenList[0], L"selection_big") == 0) {
-            HandleString(TokenList, TokenCount, &(GlobalConfig.SelectionBigFileName));
+           HandleString(TokenList, TokenCount, &(GlobalConfig.SelectionBigFileName));
 
         } else if (StriCmp(TokenList[0], L"default_selection") == 0) {
-            HandleString(TokenList, TokenCount, &(GlobalConfig.DefaultSelection));
+           HandleString(TokenList, TokenCount, &(GlobalConfig.DefaultSelection));
 
         } else if (StriCmp(TokenList[0], L"textonly") == 0) {
             GlobalConfig.TextOnly = TRUE;
@@ -419,11 +405,27 @@ VOID ReadConfig(VOID)
            GlobalConfig.RequestedScreenWidth = Atoi(TokenList[1]);
            GlobalConfig.RequestedScreenHeight = Atoi(TokenList[2]);
 
+        } else if (StriCmp(TokenList[0], L"use_graphics_for") == 0) {
+           GlobalConfig.GraphicsFor = 0;
+           for (i = 1; i < TokenCount; i++) {
+              if (StriCmp(TokenList[i], L"osx")) {
+                 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_OSX;
+              } else if (StriCmp(TokenList[i], L"linux")) {
+                 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_LINUX;
+              } else if (StriCmp(TokenList[i], L"elilo")) {
+                 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_ELILO;
+              } else if (StriCmp(TokenList[i], L"grub")) {
+                 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_GRUB;
+              } else if (StriCmp(TokenList[i], L"windows")) {
+                 GlobalConfig.GraphicsFor |= GRAPHICS_FOR_WINDOWS;
+              }
+           } // for (graphics_on tokens)
+
         } else if (StriCmp(TokenList[0], L"scan_all_linux_kernels") == 0) {
            GlobalConfig.ScanAllLinux = TRUE;
 
-        } else if ((StriCmp(TokenList[0], L"max_tags") == 0) && (TokenCount > 1)) {
-           GlobalConfig.MaxTags = Atoi(TokenList[1]);
+        } else if (StriCmp(TokenList[0], L"max_tags") == 0) {
+           HandleInt(TokenList, TokenCount, &(GlobalConfig.MaxTags));
         }
 
         FreeTokenLine(&TokenList, &TokenCount);
@@ -447,11 +449,13 @@ static VOID AddSubmenu(LOADER_ENTRY *Entry, REFIT_FILE *File, REFIT_VOLUME *Volu
    SubEntry->me.Title        = StrDuplicate(Title);
 
    while (((TokenCount = ReadTokenLine(File, &TokenList)) > 0) && (StriCmp(TokenList[0], L"}") != 0)) {
+
       if ((StriCmp(TokenList[0], L"loader") == 0) && (TokenCount > 1)) { // set the boot loader filename
          if (SubEntry->LoaderPath != NULL)
             FreePool(SubEntry->LoaderPath);
          SubEntry->LoaderPath = StrDuplicate(TokenList[1]);
          SubEntry->DevicePath = FileDevicePath(Volume->DeviceHandle, SubEntry->LoaderPath);
+
       } else if (StriCmp(TokenList[0], L"initrd") == 0) {
          if (SubEntry->InitrdPath != NULL)
             FreePool(SubEntry->InitrdPath);
@@ -459,6 +463,7 @@ static VOID AddSubmenu(LOADER_ENTRY *Entry, REFIT_FILE *File, REFIT_VOLUME *Volu
          if (TokenCount > 1) {
             SubEntry->InitrdPath = StrDuplicate(TokenList[1]);
          }
+
       } else if (StriCmp(TokenList[0], L"options") == 0) {
          if (SubEntry->LoadOptions != NULL)
             FreePool(SubEntry->LoadOptions);
@@ -466,15 +471,20 @@ static VOID AddSubmenu(LOADER_ENTRY *Entry, REFIT_FILE *File, REFIT_VOLUME *Volu
          if (TokenCount > 1) {
             SubEntry->LoadOptions = StrDuplicate(TokenList[1]);
          } // if/else
+
       } else if ((StriCmp(TokenList[0], L"add_options") == 0) && (TokenCount > 1)) {
          MergeStrings(&SubEntry->LoadOptions, TokenList[1], L' ');
+
       } else if ((StriCmp(TokenList[0], L"graphics") == 0) && (TokenCount > 1)) {
          SubEntry->UseGraphicsMode = (StriCmp(TokenList[1], L"on") == 0);
+
       } else if (StriCmp(TokenList[0], L"disabled") == 0) {
          SubEntry->Enabled = FALSE;
       } // ief/elseif
+
       FreeTokenLine(&TokenList, &TokenCount);
    } // while()
+
    if (SubEntry->InitrdPath != NULL) {
       MergeStrings(&SubEntry->LoadOptions, L"initrd=", L' ');
       MergeStrings(&SubEntry->LoadOptions, SubEntry->InitrdPath, 0);
@@ -536,7 +546,8 @@ static LOADER_ENTRY * AddStanzaEntries(REFIT_FILE *File, REFIT_VOLUME *Volume, C
       return NULL;
 
    Entry->Title           = StrDuplicate(Title);
-   Entry->me.Title        = PoolPrint(L"Boot %s from %s", (Title != NULL) ? Title : L"Unknown", CurrentVolume->VolName);
+   Entry->me.Title        = AllocateZeroPool(256 * sizeof(CHAR16));
+   SPrint(Entry->me.Title, 255, L"Boot %s from %s", (Title != NULL) ? Title : L"Unknown", CurrentVolume->VolName);
    Entry->me.Row          = 0;
    Entry->me.BadgeImage   = CurrentVolume->VolBadgeImage;
    Entry->VolName         = CurrentVolume->VolName;
@@ -554,7 +565,8 @@ static LOADER_ENTRY * AddStanzaEntries(REFIT_FILE *File, REFIT_VOLUME *Volume, C
       } else if ((StriCmp(TokenList[0], L"volume") == 0) && (TokenCount > 1)) {
          if (FindVolume(&CurrentVolume, TokenList[1])) {
             FreePool(Entry->me.Title);
-            Entry->me.Title        = PoolPrint(L"Boot %s from %s", (Title != NULL) ? Title : L"Unknown", CurrentVolume->VolName);
+            Entry->me.Title        = AllocateZeroPool(256 * sizeof(CHAR16));
+            SPrint(Entry->me.Title, 255, L"Boot %s from %s", (Title != NULL) ? Title : L"Unknown", CurrentVolume->VolName);
             Entry->me.BadgeImage   = CurrentVolume->VolBadgeImage;
             Entry->VolName         = CurrentVolume->VolName;
          } // if match found