Better error checking and reporting

diff --git a/main.c b/main.c
index 10bf929ed3a5ee496c83dd704b73502bacbed04e..13e669625ba6a3f62834e2d240ba19c4a7659851 100644 (file)
--- a/main.c
+++ b/main.c
@@ -101,7 +101,7 @@ Property *get_device_properties(char *node)
                        {
                                /*printf("Got property '%s' = '%s'\n", prop.name, prop.value);*/
 
-                               props = (Property *)realloc(props, (n_props*2)*sizeof(Property));
+                               props = (Property *)realloc(props, (n_props+2)*sizeof(Property));
                                props[n_props] = prop;
                                ++n_props;
 
@@ -168,6 +168,9 @@ Device *get_devices(void)
        int n_devices = 0;
        
        dir = opendir("/dev/disk/by-id");
+       if(!dir)
+               return NULL;
+
        while((de = readdir(dir)))
        {
                if(de->d_name[0]=='.' && (de->d_name[1]==0 || (de->d_name[1]=='.' && de->d_name[2]==0)))
@@ -217,8 +220,13 @@ Device *get_devices(void)
                free_properties(props);
        }
 
-       devices[n_devices].node = NULL;
-       devices[n_devices].label = NULL;
+       closedir(dir);
+
+       if(devices)
+       {
+               devices[n_devices].node = NULL;
+               devices[n_devices].label = NULL;
+       }
 
        return devices;
 }
@@ -247,25 +255,60 @@ void row_activated(GtkTreeView *list, GtkTreePath *path, GtkTreeViewColumn *colu
        {
                Device *device;
                int pid;
+               int pipe_fd[2];
 
                gtk_tree_model_get(model, &iter, 1, &device, -1);
 
+               pipe(pipe_fd);
+
                pid = fork();
                if(pid==0)
                {
+                       close(pipe_fd[0]);
+                       dup2(pipe_fd[1], 1);
+                       dup2(pipe_fd[1], 2);
                        execl("/usr/bin/pmount", "pmount", device->node, device->label, NULL);
                        _exit(1);
                }
                else if(pid>0)
                {
-                       waitpid(pid, NULL, 0);
+                       char buf[1024];
+                       int pos = 0;
+                       int status;
+
+                       close(pipe_fd[1]);
+
+                       while(1)
+                       {
+                               int len;
+
+                               len = read(pipe_fd[0], buf+pos, sizeof(buf)-pos-1);
+                               if(len<=0)
+                                       break;
+                               pos += len;
+                       }
+
+                       buf[pos] = 0;
+
+                       waitpid(pid, &status, 0);
+                       if(!WIFEXITED(status) || WEXITSTATUS(status))
+                       {
+                               GtkWidget *dialog;
+
+                               dialog = gtk_message_dialog_new(NULL, 0, GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, "%s", buf);
+                               g_signal_connect(dialog, "response", &gtk_main_quit, NULL);
+                               gtk_widget_show_all(dialog);
+                       }
+                       else
+                               gtk_main_quit();
+               }
+               else
+               {
                }
        }
 
        (void)column;
        (void)user_data;
-
-       gtk_main_quit();
 }
 
 int main(int argc, char **argv)
@@ -302,19 +345,31 @@ int main(int argc, char **argv)
        selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(list));
 
        devices = get_devices();
-       latest = 0;
-       for(i=0; devices[i].node; ++i)
+       if(devices)
        {
-               gtk_list_store_append(store, &iter);
-               gtk_list_store_set(store, &iter, 0, devices[i].description, 1, &devices[i], -1);
-               if(devices[i].time>latest)
+               latest = 0;
+               for(i=0; devices[i].node; ++i)
                {
-                       latest = devices[i].time;
-                       gtk_tree_selection_select_iter(selection, &iter);
+                       gtk_list_store_append(store, &iter);
+                       gtk_list_store_set(store, &iter, 0, devices[i].description, 1, &devices[i], -1);
+                       if(devices[i].time>latest)
+                       {
+                               latest = devices[i].time;
+                               gtk_tree_selection_select_iter(selection, &iter);
+                       }
                }
+
+               gtk_widget_show_all(window);
+       }
+       else
+       {
+               GtkWidget *dialog;
+
+               dialog = gtk_message_dialog_new(NULL, 0, GTK_MESSAGE_WARNING, GTK_BUTTONS_OK, "No devices found");
+               g_signal_connect(dialog, "response", &gtk_main_quit, NULL);
+               gtk_widget_show_all(dialog);
        }
 
-       gtk_widget_show_all(window);
        gtk_main();
 
        free_devices(devices);