#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/select.h>
+#include <sys/inotify.h>
#include <gtk/gtk.h>
#include <gdk/gdkkeysyms.h>
typedef struct sGuiContext
{
int manager;
+ Device *devices;
GtkWidget *list;
GtkWidget *button;
char *post_mount_command;
+ int inotify_fd;
+ int dev_wd;
+ GIOChannel *inotify_channel;
+ int refresh_pending;
} GuiContext;
int verbosity = 0;
return 0;
}
+/**
+Refreshes both the internal devices array and the GUI list.
+*/
+int refresh_devices(GuiContext *context, int umount)
+{
+ GtkListStore *store;
+ GtkTreeSelection *selection;
+ int n_listed;
+ time_t latest;
+ int i;
+ GtkTreeIter iter;
+
+ store = GTK_LIST_STORE(gtk_tree_view_get_model(GTK_TREE_VIEW(context->list)));
+ gtk_list_store_clear(store);
+
+ free_devices(context->devices);
+
+ context->devices = get_devices();
+ if(!context->devices)
+ return 0;
+
+ selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(context->list));
+
+ /* Populate the list with devices in appropriate state. */
+ n_listed = 0;
+ latest = 0;
+ for(i=0; context->devices[i].node; ++i)
+ {
+ Device *dev;
+
+ dev = &context->devices[i];
+ if(umount<0 || !dev->mount_point==!umount)
+ {
+ gtk_list_store_append(store, &iter);
+ gtk_list_store_set(store, &iter,
+ 0, dev->description,
+ 1, dev,
+ 2, !!dev->mount_point,
+ 3, dev->mount_point,
+ -1);
+ if(dev->time>latest)
+ {
+ /* Pre-select the device that appeared on the system most recently. */
+ latest = dev->time;
+ gtk_tree_selection_select_iter(selection, &iter);
+ }
+
+ ++n_listed;
+ }
+ }
+
+ return n_listed;
+}
+
+/**
+Handles an automatic refresh of the device list in response to inotify events.
+*/
+gboolean refresh_devices_idle(gpointer data)
+{
+ GuiContext *context = (GuiContext *)data;
+
+ refresh_devices(context, -1);
+ context->refresh_pending = 0;
+
+ return FALSE;
+}
+
/**
Callback for selection in the device list changing. Updates the action button
label according to device status.
return FALSE;
}
+/**
+Callback for inotify events.
+*/
+gboolean inotify_event_available(GIOChannel *source, GIOCondition condition, gpointer user_data)
+{
+ GuiContext *context = (GuiContext *)user_data;
+ int fd;
+ char eventbuf[sizeof(struct inotify_event)+NAME_MAX+1];
+ int len;
+
+ fd = g_io_channel_unix_get_fd(source);
+ len = read(fd, eventbuf, sizeof(eventbuf));
+ if(len>=(int)sizeof(struct inotify_event))
+ {
+ if(!context->refresh_pending)
+ {
+ g_timeout_add(500, &refresh_devices_idle, context);
+ context->refresh_pending = 1;
+ }
+ }
+
+ (void)condition;
+
+ return TRUE;
+}
+
void show_help(void)
{
printf("pmount-gui\n"
GtkListStore *store;
GtkTreeSelection *selection;
GtkCellRenderer *mounted_toggle;
- GtkTreeIter iter;
- Device *devices;
- int i;
- time_t latest;
int opt;
int umount = 0;
int n_listed;
context.manager = 0;
+ context.devices = NULL;
context.post_mount_command = NULL;
+ context.inotify_fd = -1;
+ context.dev_wd = -1;
+ context.inotify_channel = NULL;
+ context.refresh_pending = 0;
gtk_init(&argc, &argv);
gtk_tree_view_column_add_attribute(mounted_column, mount_point_renderer, "text", 3);
gtk_tree_view_insert_column(GTK_TREE_VIEW(context.list), mounted_column, -1);
+
+ context.inotify_fd = inotify_init();
+ if(context.inotify_fd>=0)
+ {
+ context.dev_wd = inotify_add_watch(context.inotify_fd, "/dev/disk/by-id", IN_CREATE|IN_DELETE);
+
+ context.inotify_channel = g_io_channel_unix_new(context.inotify_fd);
+ g_io_add_watch(context.inotify_channel, G_IO_IN, &inotify_event_available, &context);
+ }
+ else
+ printf("Warning: Unable to initialize inotify\n");
+
+ umount = -1;
}
selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(context.list));
g_signal_connect(context.button, "clicked", G_CALLBACK(&button_clicked), &context);
gtk_box_pack_start(GTK_BOX(box), context.button, FALSE, TRUE, 0);
- devices = get_devices();
- n_listed = 0;
- if(devices)
- {
- /* Populate the list with devices in appropriate state. */
- latest = 0;
- for(i=0; devices[i].node; ++i)
- if(!devices[i].mount_point==!umount || context.manager)
- {
- gtk_list_store_append(store, &iter);
- gtk_list_store_set(store, &iter,
- 0, devices[i].description,
- 1, &devices[i],
- 2, !!devices[i].mount_point,
- 3, devices[i].mount_point,
- -1);
- if(devices[i].time>latest)
- {
- /* Pre-select the device that appeared on the system most recently. */
- latest = devices[i].time;
- gtk_tree_selection_select_iter(selection, &iter);
- }
-
- ++n_listed;
- }
-
- }
+ n_listed = refresh_devices(&context, umount);
if(n_listed || context.manager)
gtk_widget_show_all(window);
gtk_main();
- free_devices(devices);
+ free_devices(context.devices);
+ if(context.inotify_fd>=0)
+ close(context.inotify_fd);
return 0;
}