dcavalca / rpms / grub2

Forked from rpms/grub2 3 years ago
Clone
0dc71c
From b5adb83344dfeb6ff142dc41bd2a82d29029c1c2 Mon Sep 17 00:00:00 2001
0dc71c
From: Peter Jones <pjones@redhat.com>
0dc71c
Date: Fri, 3 Oct 2014 11:08:03 -0400
0dc71c
Subject: [PATCH] Try $prefix if $fw_path doesn't work.
0dc71c
0dc71c
Related: rhbz#1148652
0dc71c
0dc71c
Signed-off-by: Peter Jones <pjones@redhat.com>
0dc71c
---
0dc71c
 grub-core/kern/ieee1275/init.c |  30 ++++++-----
0dc71c
 grub-core/net/net.c            |   2 +-
0dc71c
 grub-core/normal/main.c        | 118 ++++++++++++++++++++---------------------
0dc71c
 3 files changed, 75 insertions(+), 75 deletions(-)
0dc71c
0dc71c
diff --git a/grub-core/kern/ieee1275/init.c b/grub-core/kern/ieee1275/init.c
0dc71c
index 8191f8c..8ca4bf7 100644
0dc71c
--- a/grub-core/kern/ieee1275/init.c
0dc71c
+++ b/grub-core/kern/ieee1275/init.c
0dc71c
@@ -131,23 +131,25 @@ grub_machine_get_bootlocation (char **device, char **path)
0dc71c
       grub_free (canon);
0dc71c
     }
0dc71c
   else
0dc71c
-    *device = grub_ieee1275_encode_devname (bootpath);
0dc71c
-  grub_free (type);
0dc71c
-
0dc71c
-  filename = grub_ieee1275_get_filename (bootpath);
0dc71c
-  if (filename)
0dc71c
     {
0dc71c
-      char *lastslash = grub_strrchr (filename, '\\');
0dc71c
-
0dc71c
-      /* Truncate at last directory.  */
0dc71c
-      if (lastslash)
0dc71c
+      filename = grub_ieee1275_get_filename (bootpath);
0dc71c
+      if (filename)
0dc71c
         {
0dc71c
-	  *lastslash = '\0';
0dc71c
-	  grub_translate_ieee1275_path (filename);
0dc71c
-
0dc71c
-	  *path = filename;
0dc71c
-	}
0dc71c
+          char *lastslash = grub_strrchr (filename, '\\');
0dc71c
+
0dc71c
+          /* Truncate at last directory.  */
0dc71c
+          if (lastslash)
0dc71c
+            {
0dc71c
+              *lastslash = '\0';
0dc71c
+              grub_translate_ieee1275_path (filename);
0dc71c
+
0dc71c
+              *path = filename;
0dc71c
+            }
0dc71c
+        }
0dc71c
+      *device = grub_ieee1275_encode_devname (bootpath);
0dc71c
     }
0dc71c
+
0dc71c
+  grub_free (type);
0dc71c
   grub_free (bootpath);
0dc71c
 }
0dc71c
 
0dc71c
diff --git a/grub-core/net/net.c b/grub-core/net/net.c
0dc71c
index 578e057..10bfed3 100644
0dc71c
--- a/grub-core/net/net.c
0dc71c
+++ b/grub-core/net/net.c
0dc71c
@@ -1817,7 +1817,7 @@ grub_net_search_configfile (char *config)
0dc71c
   /* Remove the remaining minus sign at the end. */
0dc71c
   config[config_len] = '\0';
0dc71c
 
0dc71c
-  return GRUB_ERR_NONE;
0dc71c
+  return GRUB_ERR_FILE_NOT_FOUND;
0dc71c
 }
0dc71c
 
0dc71c
 static struct grub_preboot *fini_hnd;
0dc71c
diff --git a/grub-core/normal/main.c b/grub-core/normal/main.c
0dc71c
index 4190985..aa14499 100644
0dc71c
--- a/grub-core/normal/main.c
0dc71c
+++ b/grub-core/normal/main.c
0dc71c
@@ -331,74 +331,72 @@ grub_enter_normal_mode (const char *config)
0dc71c
   grub_boot_time ("Exiting normal mode");
0dc71c
 }
0dc71c
 
0dc71c
+static grub_err_t
0dc71c
+grub_try_normal (const char *variable)
0dc71c
+{
0dc71c
+    char *config;
0dc71c
+    const char *prefix;
0dc71c
+    grub_err_t err = GRUB_ERR_FILE_NOT_FOUND;
0dc71c
+
0dc71c
+    prefix = grub_env_get (variable);
0dc71c
+    if (!prefix)
0dc71c
+      return GRUB_ERR_FILE_NOT_FOUND;
0dc71c
+
0dc71c
+    if (grub_strncmp (prefix + 1, "tftp", sizeof ("tftp") - 1) == 0)
0dc71c
+      {
0dc71c
+	grub_size_t config_len;
0dc71c
+	config_len = grub_strlen (prefix) +
0dc71c
+	  sizeof ("/grub.cfg-XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX");
0dc71c
+	config = grub_malloc (config_len);
0dc71c
+
0dc71c
+	if (! config)
0dc71c
+	  return GRUB_ERR_FILE_NOT_FOUND;
0dc71c
+
0dc71c
+	grub_snprintf (config, config_len, "%s/grub.cfg", prefix);
0dc71c
+	err = grub_net_search_configfile (config);
0dc71c
+      }
0dc71c
+
0dc71c
+    if (err != GRUB_ERR_NONE)
0dc71c
+      {
0dc71c
+	config = grub_xasprintf ("%s/grub.cfg", prefix);
0dc71c
+	if (config)
0dc71c
+	  {
0dc71c
+	    grub_file_t file;
0dc71c
+	    file = grub_file_open (config);
0dc71c
+	    if (file)
0dc71c
+	      {
0dc71c
+		grub_file_close (file);
0dc71c
+		err = GRUB_ERR_NONE;
0dc71c
+	      }
0dc71c
+	  }
0dc71c
+      }
0dc71c
+
0dc71c
+    if (err == GRUB_ERR_NONE)
0dc71c
+      grub_enter_normal_mode (config);
0dc71c
+
0dc71c
+    grub_errno = 0;
0dc71c
+    grub_free (config);
0dc71c
+    return err;
0dc71c
+}
0dc71c
+
0dc71c
 /* Enter normal mode from rescue mode.  */
0dc71c
 static grub_err_t
0dc71c
 grub_cmd_normal (struct grub_command *cmd __attribute__ ((unused)),
0dc71c
 		 int argc, char *argv[])
0dc71c
 {
0dc71c
-  if (argc == 0)
0dc71c
+  if (argc)
0dc71c
+    grub_enter_normal_mode (argv[0]);
0dc71c
+  else
0dc71c
     {
0dc71c
-      /* Guess the config filename. It is necessary to make CONFIG static,
0dc71c
-	 so that it won't get broken by longjmp.  */
0dc71c
-      char *config;
0dc71c
-      const char *prefix;
0dc71c
-
0dc71c
-      prefix = grub_env_get ("fw_path");
0dc71c
-      if (! prefix)
0dc71c
-	      prefix = grub_env_get ("prefix");
0dc71c
-
0dc71c
-      if (prefix)
0dc71c
-	{
0dc71c
-	  if (grub_strncmp (prefix + 1, "tftp", sizeof ("tftp") - 1) == 0)
0dc71c
-	    {
0dc71c
-	      grub_size_t config_len;
0dc71c
-	      config_len = grub_strlen (prefix) +
0dc71c
-		sizeof ("/grub.cfg-XXXXXXXX-XXXX-XXXX-XXXX-XXXXXXXXXXXX");
0dc71c
-	      config = grub_malloc (config_len);
0dc71c
-
0dc71c
-	      if (! config)
0dc71c
-		goto quit;
0dc71c
-
0dc71c
-	      grub_snprintf (config, config_len, "%s/grub.cfg", prefix);
0dc71c
-
0dc71c
-	      grub_net_search_configfile (config);
0dc71c
-
0dc71c
-	      grub_enter_normal_mode (config);
0dc71c
-	      grub_free (config);
0dc71c
-	      config = NULL;
0dc71c
-	    }
0dc71c
-
0dc71c
-	  if (!config)
0dc71c
-	    {
0dc71c
-	      config = grub_xasprintf ("%s/grub.cfg", prefix);
0dc71c
-	      if (config)
0dc71c
-		{
0dc71c
-		  grub_file_t file;
0dc71c
-
0dc71c
-		  file = grub_file_open (config);
0dc71c
-		  if (file)
0dc71c
-		    {
0dc71c
-		      grub_file_close (file);
0dc71c
-		      grub_enter_normal_mode (config);
0dc71c
-		    }
0dc71c
-		  else
0dc71c
-		    {
0dc71c
-		      /*  Ignore all errors.  */
0dc71c
-		      grub_errno = 0;
0dc71c
-		    }
0dc71c
-		  grub_free (config);
0dc71c
-		}
0dc71c
-	    }
0dc71c
-	}
0dc71c
-      else
0dc71c
-	{
0dc71c
-	  grub_enter_normal_mode (0);
0dc71c
-	}
0dc71c
+      /* Guess the config filename. */
0dc71c
+      grub_err_t err;
0dc71c
+      err = grub_try_normal ("fw_path");
0dc71c
+      if (err == GRUB_ERR_FILE_NOT_FOUND)
0dc71c
+        err = grub_try_normal ("prefix");
0dc71c
+      if (err == GRUB_ERR_FILE_NOT_FOUND)
0dc71c
+        grub_enter_normal_mode (0);
0dc71c
     }
0dc71c
-  else
0dc71c
-    grub_enter_normal_mode (argv[0]);
0dc71c
 
0dc71c
-quit:
0dc71c
   return 0;
0dc71c
 }
0dc71c
 
0dc71c
-- 
0dc71c
1.9.3
0dc71c