#include <fcntl.h>
#include <dirent.h>
#include <mntent.h>
+#include <errno.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/select.h>
typedef struct sDevice
{
char *node;
+ char *devname;
char *label;
char *description;
- int mounted;
+ char *mount_point;
time_t time;
} Device;
int verbosity = 0;
+char *post_mount_command = NULL;
/**
Parses a string of the form name=value and places the components in a Property
}
/**
-Reads device names from an fstab/mtab file.
+Returns an array of user-mountable devices listed in fstab.
*/
-char **get_mount_entries(char *filename, int (*predicate)(struct mntent *))
+char **get_fstab_devices(void)
{
FILE *file;
struct mntent *me;
char **devices = NULL;
int n_devices = 0;
- file = setmntent(filename, "r");
+ file = setmntent("/etc/fstab", "r");
if(!file)
return NULL;
while((me = getmntent(file)))
- if(!predicate || predicate(me))
+ if(hasmntopt(me, "user")!=NULL)
{
devices = (char **)realloc(devices, (n_devices+2)*sizeof(char *));
devices[n_devices] = strdup(me->mnt_fsname);
return devices;
}
-/**
-Returns an array of all currently mounted devices.
-*/
-char **get_mounted_devices(void)
-{
- return get_mount_entries("/etc/mtab", NULL);
-}
-
-/**
-Checks if an fstab entry has the user option set.
-*/
-int is_user_mountable(struct mntent *me)
-{
- return hasmntopt(me, "user")!=NULL;
-}
-
-/**
-Returns an array of user-mountable devices listed in fstab.
-*/
-char **get_fstab_devices(void)
-{
- return get_mount_entries("/etc/fstab", &is_user_mountable);
-}
-
/**
Checks if an array of strings contains the specified string.
*/
return nodes;
}
-/** Returns an array of all mountable devices. */
+/**
+Reads the list of mounted devices from /etc/mtab and records the mount points.
+*/
+void check_mounts(Device *devices)
+{
+ FILE *file;
+ struct mntent *me;
+
+ file = setmntent("/etc/mtab", "r");
+ if(!file)
+ return;
+
+ while((me = getmntent(file)))
+ {
+ int i;
+
+ for(i=0; devices[i].node; ++i)
+ if(!strcmp(devices[i].devname, me->mnt_fsname))
+ {
+ devices[i].mount_point = strdup(me->mnt_dir);
+
+ if(verbosity>=1)
+ printf("Device %s is mounted on %s\n", devices[i].node, devices[i].mount_point);
+ }
+ }
+
+ endmntent(file);
+}
+
+/**
+Returns an array of all mountable devices.
+*/
Device *get_devices(void)
{
char **nodes = NULL;
int i;
nodes = get_device_nodes("/dev/disk/by-id");
- mounted = get_mounted_devices();
fstab = get_fstab_devices();
for(i=0; nodes[i]; ++i)
/* Reserve space for a sentinel entry. */
devices = (Device *)realloc(devices, (n_devices+2)*sizeof(Device));
devices[n_devices].node = nodes[i];
+ devices[n_devices].devname = strdup(devname);
devices[n_devices].label = strdup(label);
devices[n_devices].description = strdup(buf);
- devices[n_devices].mounted = is_in_array(mounted, devname);
+ devices[n_devices].mount_point = NULL;
devices[n_devices].time = st.st_mtime;
++n_devices;
}
{
/* Terminate the array with NULL pointers. */
devices[n_devices].node = NULL;
+ devices[n_devices].devname = NULL;
devices[n_devices].label = NULL;
devices[n_devices].description = NULL;
+ devices[n_devices].mount_point = NULL;
+
+ check_mounts(devices);
}
return devices;
for(i=0; devices[i].node; ++i)
{
free(devices[i].node);
+ free(devices[i].devname);
free(devices[i].label);
free(devices[i].description);
+ if(devices[i].mount_point)
+ free(devices[i].mount_point);
}
free(devices);
}
/**
-Callback for activating a row in the device list. Mounts or unmounts the
-device depending on operating mode.
+Mounts a device if it was not mounted, or unmounts if it was.
*/
-void row_activated(GtkTreeView *list, GtkTreePath *path, GtkTreeViewColumn *column, gpointer user_data)
+int toggle_device(Device *device, char *out_buf, int out_size)
{
- GtkTreeModel *model;
- GtkTreeIter iter;
- int umount = *(int *)user_data;
- char buf[1024];
+ int umount = !!device->mount_point;
+ char mount_point[1024];
int pos = 0;
int status = 0;
fd_set fds;
struct timeval timeout;
- Device *device;
int pid;
int pipe_fd[2];
+ char suffix = 0;
- model = gtk_tree_view_get_model(list);
+ out_buf[0] = 0;
- if(!gtk_tree_model_get_iter(model, &iter, path))
- return;
+ /* Find a mount point that does not exist yet. */
+ while(1)
+ {
+ int len;
- gtk_tree_model_get(model, &iter, 1, &device, -1);
+ len = snprintf(mount_point, sizeof(mount_point), "/media/%s", device->label);
+ if(len+2>=(int)sizeof(mount_point))
+ return -1;
+
+ if(suffix)
+ len += snprintf(mount_point+len, sizeof(mount_point)-len, "_%c", suffix);
+
+ if(access(mount_point, F_OK)<0 && errno==ENOENT)
+ break;
+
+ if(suffix==0)
+ suffix = '1';
+ else if(suffix<'9')
+ ++suffix;
+ else
+ return -1;
+ }
pipe(pipe_fd);
if(umount)
printf("Running pumount %s\n", device->node);
else
- printf("Running pmount %s %s\n", device->node, device->label);
+ printf("Running pmount %s %s\n", device->node, mount_point+7);
}
close(pipe_fd[0]);
if(umount)
execl("/usr/bin/pumount", "pumount", device->node, NULL);
else
- execl("/usr/bin/pmount", "pmount", device->node, device->label, NULL);
+ execl("/usr/bin/pmount", "pmount", device->node, mount_point+7, NULL);
_exit(1);
}
else if(pid<0)
- return;
+ return -1;
/* Parent process */
{
int len;
- len = read(pipe_fd[0], buf+pos, sizeof(buf)-pos-1);
+ len = read(pipe_fd[0], out_buf+pos, out_size-pos-1);
if(len<=0)
break;
pos += len;
}
}
+ close(pipe_fd[0]);
if(pid)
waitpid(pid, &status, 0);
- buf[pos] = 0;
+ out_buf[pos] = 0;
if(verbosity>=1)
{
}
if(!WIFEXITED(status) || WEXITSTATUS(status))
+ return -1;
+
+ if(umount)
+ {
+ free(device->mount_point);
+ device->mount_point = NULL;
+ }
+ else
+ device->mount_point = strdup(mount_point);
+
+ return 0;
+}
+
+/**
+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)
+{
+ GtkTreeModel *model;
+ GtkTreeIter iter;
+ Device *device;
+ int ret;
+ char output[1024];
+ int pid;
+
+ model = gtk_tree_view_get_model(list);
+
+ if(!gtk_tree_model_get_iter(model, &iter, path))
+ return;
+
+ gtk_tree_model_get(model, &iter, 1, &device, -1);
+ ret = toggle_device(device, output, sizeof(output));
+ if(ret)
{
GtkWidget *dialog;
/* 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", buf);
+ dialog = gtk_message_dialog_new(NULL, 0, GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, "%s", output);
g_signal_connect(dialog, "response", >k_main_quit, NULL);
gtk_widget_show_all(dialog);
return;
gtk_main_quit();
+ if(post_mount_command && device->mount_point)
+ {
+ if(verbosity>=1)
+ printf("Running %s in %s\n", post_mount_command, device->mount_point);
+
+ pid = fork();
+ if(pid==0)
+ {
+ chdir(device->mount_point);
+ execlp(post_mount_command, post_mount_command, NULL);
+ _exit(1);
+ }
+ }
+
(void)column;
+ (void)user_data;
}
/**
return FALSE;
}
+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]\n\n"
+ "Options:\n"
+ " -v Increase verbosity\n"
+ " -u Unmount a device (default is mount)\n"
+ " -r Run a command after mounting\n"
+ " -h Display this help\n");
+}
+
int main(int argc, char **argv)
{
GtkWidget *window;
gtk_init(&argc, &argv);
- while((opt = getopt(argc, argv, "vu"))!=-1) switch(opt)
+ while((opt = getopt(argc, argv, "vur:h"))!=-1) switch(opt)
{
case 'v':
++verbosity;
case 'u':
umount = 1;
break;
+ case 'r':
+ post_mount_command = optarg;
+ break;
+ case 'h':
+ show_help();
+ return 0;
}
window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
list = gtk_tree_view_new();
gtk_container_add(GTK_CONTAINER(viewport), list);
- g_signal_connect(list, "row-activated", G_CALLBACK(&row_activated), &umount);
+ g_signal_connect(list, "row-activated", G_CALLBACK(&row_activated), NULL);
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));
/* Populate the list with devices in appropriate state. */
latest = 0;
for(i=0; devices[i].node; ++i)
- if(!devices[i].mounted==!umount)
+ 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);
projects / pmount-gui.git / blobdiff