From: Robert Millan <rmh@aybabtu.com>
Date: Wed, 27 Jan 2010 01:49:11 +0000 (+0000)
Subject: 2010-01-27  Robert Millan  <rmh.grub@aybabtu.com>
X-Git-Tag: 1.98~111
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=af75a9f19a81c2c5fada0c44779a491fae388e96;p=thirdparty%2Fgrub.git

2010-01-27  Robert Millan  <rmh.grub@aybabtu.com>

	* util/grub-fstest.c (fstest): Rewrite allocation, fixing a few
	memleak conditions.
---

diff --git a/ChangeLog b/ChangeLog
index e3988f150..c4930acb8 100644
--- a/ChangeLog
+++ b/ChangeLog
@@ -1,3 +1,8 @@
+2010-01-27  Robert Millan  <rmh.grub@aybabtu.com>
+
+	* util/grub-fstest.c (fstest): Rewrite allocation, fixing a few
+	memleak conditions.
+
 2010-01-27  Carles Pina i Estany  <carles@pina.cat>
 
 	* util/lvm.c: New macro LVM_DEV_MAPPER_STRING.
diff --git a/util/grub-fstest.c b/util/grub-fstest.c
index bf30286a4..c03c43451 100644
--- a/util/grub-fstest.c
+++ b/util/grub-fstest.c
@@ -280,27 +280,29 @@ fstest (char **images, int num_disks, int cmd, int n, char **args)
 {
   char *host_file;
   char *loop_name;
-  char *argv[3] = { "-p" };
+  char *argv[3];
   int i;
 
+  argv[0] = "-p";
+
   for (i = 0; i < num_disks; i++)
     {
       loop_name = grub_xasprintf ("loop%d", i);
-      host_file = grub_xasprintf ("(host)%s", images[i]);
+      if (!loop_name)
+	grub_util_error (grub_errmsg);
 
-      if (!loop_name || !host_file)
-	{
-	  grub_free (loop_name);
-	  grub_free (host_file);
-	  grub_util_error (grub_errmsg);
-	  return;
-	}
+      host_file = grub_xasprintf ("(host)%s", images[i]);
+      if (!host_file)
+	grub_util_error (grub_errmsg);
 
       argv[1] = loop_name;
       argv[2] = host_file;
 
       if (execute_command ("loopback", 3, argv))
         grub_util_error ("loopback command fails");
+
+      grub_free (loop_name);
+      grub_free (host_file);
     }
 
   grub_lvm_fini ();
@@ -336,19 +338,16 @@ fstest (char **images, int num_disks, int cmd, int n, char **args)
 
   for (i = 0; i < num_disks; i++)
     {
-      grub_free (loop_name);
       loop_name = grub_xasprintf ("loop%d", i);
       if (!loop_name)
-	{
-	  grub_free (host_file);
-	  grub_util_error (grub_errmsg);
-	  return;
-	}
+	grub_util_error (grub_errmsg);
+
+      argv[1] = loop_name;
+
       execute_command ("loopback", 2, argv);
-    }
 
-  grub_free (loop_name);
-  grub_free (host_file);
+      grub_free (loop_name);
+    }
 }
 
 static struct option options[] = {