#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/select.h>
+ #include <sys/inotify.h>
#include <gtk/gtk.h>
#include <gdk/gdkkeysyms.h>
time_t time;
} Device;
+ typedef struct sGuiContext
+ {
+ int manager;
+ int autohide;
+ Device *devices;
+ GtkWidget *window;
+ GtkWidget *list;
+ GtkWidget *button;
+ char *post_mount_command;
+ int inotify_fd;
+ int dev_wd;
+ GIOChannel *inotify_channel;
+ int refresh_pending;
+ } GuiContext;
+
int verbosity = 0;
- char *post_mount_command = NULL;
+char **pmount_argv = NULL;
/**
Parses a string of the form name=value and places the components in a Property
if(pid==0)
{
/* Child process */
+
+ /* Complete construction of pmount call */
+ int last;
+
+ pmount_argv[0] = "pmount";
+
+ last = 0;
+ while(pmount_argv[++last]);
+
+ pmount_argv[last] = device->node;
+ pmount_argv[last+1] = mount_point+7;
+ pmount_argv[last+2] = NULL;
+
if(verbosity>=1)
{
if(umount)
printf("Running pumount %s\n", device->node);
else
- printf("Running pmount %s %s\n", device->node, mount_point+7);
+ {
+ int i = 0;
+ printf("Running pmount");
+ while(pmount_argv[++i])
+ printf(" %s", pmount_argv[i]);
+ printf("\n");
+ }
}
close(pipe_fd[0]);
if(umount)
execl("/usr/bin/pumount", "pumount", device->node, NULL);
else
- execl("/usr/bin/pmount", "pmount", device->node, mount_point+7, NULL);
+ execvp("/usr/bin/pmount", pmount_argv);
_exit(1);
}
else if(pid<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;
+ int n_listed;
+
+ n_listed = refresh_devices(context, -1);
+ context->refresh_pending = 0;
+
+ if(context->autohide)
+ {
+ if(n_listed)
+ gtk_widget_show_all(context->window);
+ else
+ gtk_widget_hide(context->window);
+ }
+
+ return FALSE;
+ }
+
+ /**
+ Callback for selection in the device list changing. Updates the action button
+ label according to device status.
+ */
+ void selection_changed(GtkTreeSelection *selection, gpointer user_data)
+ {
+ GuiContext *context = (GuiContext *)user_data;
+ GtkTreeIter iter;
+ GtkTreeModel *model;
+ Device *device;
+
+ if(!gtk_tree_selection_get_selected(selection, &model, &iter))
+ return;
+
+ gtk_tree_model_get(model, &iter, 1, &device, -1);
+ gtk_button_set_label(GTK_BUTTON(context->button), (device->mount_point ? "Unmount" : "Mount"));
+ }
+
/**
Callback for activating a row in the device list. Mounts or unmounts the
device depending on operating mode.
*/
void row_activated(GtkTreeView *list, GtkTreePath *path, GtkTreeViewColumn *column, gpointer user_data)
{
+ GuiContext *context = (GuiContext *)user_data;
GtkTreeModel *model;
GtkTreeIter iter;
Device *device;
/* Pmount terminated with nonzero status or a signal. Display an
error to the user. */
dialog = gtk_message_dialog_new(NULL, 0, GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, "%s", output);
- g_signal_connect(dialog, "response", >k_main_quit, NULL);
+ if(context->manager)
+ g_signal_connect(dialog, "response", G_CALLBACK(>k_widget_destroy), dialog);
+ else
+ g_signal_connect(dialog, "response", G_CALLBACK(>k_main_quit), NULL);
gtk_widget_show_all(dialog);
return;
}
- gtk_main_quit();
+ if(context->manager)
+ {
+ gtk_list_store_set(GTK_LIST_STORE(model), &iter,
+ 2, !!device->mount_point,
+ 3, device->mount_point,
+ -1);
+
+ gtk_button_set_label(GTK_BUTTON(context->button), (device->mount_point ? "Unmount" : "Mount"));
+ }
+ else
+ gtk_main_quit();
- if(post_mount_command && device->mount_point)
+ if(context->post_mount_command && device->mount_point)
{
if(verbosity>=1)
- printf("Running %s in %s\n", post_mount_command, device->mount_point);
+ printf("Running %s in %s\n", context->post_mount_command, device->mount_point);
pid = fork();
if(pid==0)
{
chdir(device->mount_point);
- execlp(post_mount_command, post_mount_command, NULL);
+ execlp(context->post_mount_command, context->post_mount_command, NULL);
_exit(1);
}
}
*/
void button_clicked(GtkButton *button, gpointer user_data)
{
- GtkWidget *list = (GtkWidget *)user_data;
+ GuiContext *context = (GuiContext *)user_data;
GtkTreeSelection *selection;
GtkTreeIter iter;
GtkTreeModel *model;
GtkTreePath *path;
+ GtkTreeViewColumn *column;
- selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(list));
+ selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(context->list));
gtk_tree_selection_get_selected(selection, &model, &iter);
path = gtk_tree_model_get_path(model, &iter);
- gtk_tree_view_row_activated(GTK_TREE_VIEW(list), path, gtk_tree_view_get_column(GTK_TREE_VIEW(list), 0));
+ column = gtk_tree_view_get_column(GTK_TREE_VIEW(context->list), 0);
+ gtk_tree_view_row_activated(GTK_TREE_VIEW(context->list), path, column);
gtk_tree_path_free(path);
(void)button;
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"
"Copyright (c) 2011-2015 Mikko Rasa, Mikkosoft Productions\n\n"
- "Usage: pmount-gui [-v] [-u] [-r <command>] [-h] [-- <pmount options>]\n\n"
- "Usage: pmount-gui [-v] [-u] [-r <command>] [-m|-M] [-h]\n\n"
++ "Usage: pmount-gui [-v] [-u] [-r <command>] [-m|-M] [-h] [-- <pmount options>]\n\n"
"Options:\n"
" -v Increase verbosity\n"
" -u Unmount a device (default is mount)\n"
" -r Run a command after mounting\n"
+ " -m Start a persistent mount manager\n"
+ " -M Like -m, but hide the window if there are no devices\n"
" -h Display this help\n");
}
int main(int argc, char **argv)
{
- GtkWidget *window;
+ GuiContext context;
GtkWidget *box;
GtkWidget *viewport;
- GtkWidget *list;
GtkListStore *store;
GtkTreeSelection *selection;
- GtkWidget *button;
- GtkTreeIter iter;
- Device *devices;
- int i;
- time_t latest;
+ GtkCellRenderer *mounted_toggle;
int opt;
int umount = 0;
int n_listed;
+ context.manager = 0;
+ context.autohide = 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);
- while((opt = getopt(argc, argv, "vur:h"))!=-1) switch(opt)
+ while((opt = getopt(argc, argv, "vur:mMh"))!=-1) switch(opt)
{
case 'v':
++verbosity;
umount = 1;
break;
case 'r':
- post_mount_command = optarg;
+ context.post_mount_command = optarg;
+ break;
+ case 'm':
+ context.manager = 1;
+ context.autohide = 0;
+ break;
+ case 'M':
+ context.manager = 1;
+ context.autohide = 1;
break;
case 'h':
show_help();
return 0;
}
- window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
- gtk_container_set_border_width(GTK_CONTAINER(window), 5);
- g_signal_connect(window, "destroy", G_CALLBACK(>k_main_quit), NULL);
- g_signal_connect(window, "key-press-event", G_CALLBACK(&key_press), NULL);
+ /* argc - optind equals to number of options after "--" */
+ /* One empty element at the beginning is for program name for execvp */
+ /* Three empty elements at the end are required options for pmount + NULL */
+
+ pmount_argv = malloc(sizeof(char *)*(argc-optind+4));
+
+ for(i=0; i<(argc-optind); ++i)
+ pmount_argv[i+1] = argv[optind+i];
+ pmount_argv[argc-optind+1] = NULL;
+
+ context.window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
+ gtk_container_set_border_width(GTK_CONTAINER(context.window), 5);
+ g_signal_connect(context.window, "destroy", G_CALLBACK(>k_main_quit), NULL);
box = gtk_vbox_new(FALSE, 5);
- gtk_container_add(GTK_CONTAINER(window), box);
+ gtk_container_add(GTK_CONTAINER(context.window), box);
viewport = gtk_viewport_new(NULL, NULL);
gtk_viewport_set_shadow_type(GTK_VIEWPORT(viewport), GTK_SHADOW_IN);
gtk_box_pack_start(GTK_BOX(box), viewport, TRUE, TRUE, 0);
- list = gtk_tree_view_new();
- gtk_container_add(GTK_CONTAINER(viewport), list);
- g_signal_connect(list, "row-activated", G_CALLBACK(&row_activated), NULL);
+ context.list = gtk_tree_view_new();
+ gtk_container_add(GTK_CONTAINER(viewport), context.list);
+ g_signal_connect(context.list, "row-activated", G_CALLBACK(&row_activated), &context);
- store = gtk_list_store_new(2, G_TYPE_STRING, G_TYPE_POINTER);
- gtk_tree_view_set_model(GTK_TREE_VIEW(list), GTK_TREE_MODEL(store));
- gtk_tree_view_insert_column_with_attributes(GTK_TREE_VIEW(list),
+ store = gtk_list_store_new(4, G_TYPE_STRING, G_TYPE_POINTER, G_TYPE_BOOLEAN, G_TYPE_STRING);
+ gtk_tree_view_set_model(GTK_TREE_VIEW(context.list), GTK_TREE_MODEL(store));
+ gtk_tree_view_insert_column_with_attributes(GTK_TREE_VIEW(context.list),
-1, "Device", gtk_cell_renderer_text_new(), "text", 0, NULL);
- selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(list));
+ if(context.manager)
+ {
+ GtkTreeViewColumn *mounted_column;
+ GtkCellRenderer *mount_point_renderer;
- button = gtk_button_new_with_label(umount ? "Unmount" : "Mount");
- g_signal_connect(button, "clicked", G_CALLBACK(&button_clicked), list);
- gtk_box_pack_start(GTK_BOX(box), button, FALSE, TRUE, 0);
+ mounted_column = gtk_tree_view_column_new();
+ gtk_tree_view_column_set_title(mounted_column, "Mounted");
- 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)
- {
- 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)
- {
- /* Pre-select the device that appeared on the system most recently. */
- latest = devices[i].time;
- gtk_tree_selection_select_iter(selection, &iter);
- }
+ mounted_toggle = gtk_cell_renderer_toggle_new();
+ gtk_tree_view_column_pack_start(mounted_column, mounted_toggle, FALSE);
+ gtk_tree_view_column_add_attribute(mounted_column, mounted_toggle, "active", 2);
- ++n_listed;
- }
+ mount_point_renderer = gtk_cell_renderer_text_new();
+ gtk_tree_view_column_pack_start(mounted_column, mount_point_renderer, TRUE);
+ 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);
- if(n_listed)
- gtk_widget_show_all(window);
+ 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;
+ }
else
+ g_signal_connect(context.window, "key-press-event", G_CALLBACK(&key_press), NULL);
+
+ selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(context.list));
+ g_signal_connect(selection, "changed", G_CALLBACK(&selection_changed), &context);
+
+ context.button = gtk_button_new_with_label(umount ? "Unmount" : "Mount");
+ g_signal_connect(context.button, "clicked", G_CALLBACK(&button_clicked), &context);
+ gtk_box_pack_start(GTK_BOX(box), context.button, FALSE, TRUE, 0);
+
+ n_listed = refresh_devices(&context, umount);
+
+ if(n_listed || (context.manager && !context.autohide))
+ gtk_widget_show_all(context.window);
+ else if(!context.manager)
{
GtkWidget *dialog;
gtk_main();
- free_devices(devices);
+ free_devices(context.devices);
+ if(context.inotify_fd>=0)
+ close(context.inotify_fd);
+ free(pmount_argv);
+
return 0;
}