From ed628d30b05b4f5884f1d2029633cf155bf5e571 Mon Sep 17 00:00:00 2001
From: "Richard W.M. Jones" <rjones@redhat.com>
Date: Thu, 24 Nov 2011 14:06:31 +0000
Subject: [PATCH] virt-alignment-scan: Add checks for when API calls return
 NULL.

In particular, virt-alignment-scan -a /dev/null would segfault
because of an error returned by parted.
(cherry picked from commit 7b192f5a672197ad67ac23501e420c674aa2f3a9)
---
 align/scan.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/align/scan.c b/align/scan.c
index 529add5..7b4631a 100644
--- a/align/scan.c
+++ b/align/scan.c
@@ -228,9 +228,13 @@ scan (void)
   struct guestfs_partition_list *parts;
 
   devices = guestfs_list_devices (g);
+  if (devices == NULL)
+    exit (EXIT_FAILURE);
 
   for (i = 0; devices[i] != NULL; ++i) {
     parts = guestfs_part_list (g, devices[i]);
+    if (parts == NULL)
+      exit (EXIT_FAILURE);
 
     /* Canonicalize the name of the device for printing. */
     if (STRPREFIX (devices[i], "/dev/") &&
-- 
1.8.3.1