[gnome-disk-utility] Don't assume only GduDrive presentables are shown in GduDiskSelectionWidget

[David Zeuthen <davidz src gnome org>]

commit 3d1c6dfaf27d7d235164f481389249e403c4fe00
Author: David Zeuthen <davidz redhat com>
Date:   Mon Dec 14 11:43:05 2009 -0500

    Don't assume only GduDrive presentables are shown in GduDiskSelectionWidget
    
    Signed-off-by: David Zeuthen <davidz redhat com>

 src/gdu-gtk/gdu-disk-selection-widget.c |   16 ++++++++++------
 1 files changed, 10 insertions(+), 6 deletions(-)
---
diff --git a/src/gdu-gtk/gdu-disk-selection-widget.c b/src/gdu-gtk/gdu-disk-selection-widget.c
index 559f232..a6fff09 100644
--- a/src/gdu-gtk/gdu-disk-selection-widget.c
+++ b/src/gdu-gtk/gdu-disk-selection-widget.c
@@ -771,13 +771,17 @@ model_visible_func (GtkTreeModel  *model,
                 guint64 largest_segment;
                 gboolean whole_disk_is_uninitialized;
 
-                if (gdu_drive_has_unallocated_space (GDU_DRIVE (p),
-                                                     &whole_disk_is_uninitialized,
-                                                     &largest_segment,
-                                                     NULL)) {
-                        if (largest_segment >= widget->priv->component_size) {
-                                ret = TRUE;
+                if (GDU_IS_DRIVE (p)) {
+                        if (gdu_drive_has_unallocated_space (GDU_DRIVE (p),
+                                                             &whole_disk_is_uninitialized,
+                                                             &largest_segment,
+                                                             NULL)) {
+                                if (largest_segment >= widget->priv->component_size) {
+                                        ret = TRUE;
+                                }
                         }
+                } else {
+                        ret = TRUE;
                 }
                 g_object_unref (p);
         }