12 typedef struct sProperty
18 typedef struct sDevice
29 int parse_property(char *str, int size, Property *prop)
34 for(i=0; (equals<0 && i<size); ++i)
41 prop->name = malloc(equals+1);
42 strncpy(prop->name, str, equals);
43 prop->name[equals] = 0;
45 prop->value = malloc(size-equals);
46 strncpy(prop->value, str+equals+1, size-equals-1);
47 prop->value[size-equals-1] = 0;
52 Property *get_device_properties(char *node)
63 printf("Running udevadm info -q property -n \"%s\"\n", node);
68 execl("/sbin/udevadm", "udevadm", "info", "-q", "property", "-n", node, NULL);
77 Property *props = NULL;
83 buf = (char *)malloc(bufsize);
95 len = read(pipe_fd[0], buf+pos, bufsize-pos);
104 for(i=0; (newline<0 && i<pos); ++i)
113 buf = (char *)realloc(buf, bufsize);
117 if(parse_property(buf, newline, &prop)==0)
119 props = (Property *)realloc(props, (n_props+2)*sizeof(Property));
120 props[n_props] = prop;
123 memmove(buf, buf+newline+1, pos-newline-1);
134 props[n_props].name = NULL;
135 props[n_props].value = NULL;
138 waitpid(pid, NULL, 0);
152 char *get_property_value(Property *props, char *name)
155 for(i=0; props[i].name; ++i)
156 if(strcmp(props[i].name, name)==0)
157 return props[i].value;
161 int match_property_value(Property *props, char *name, char *value)
163 char *v = get_property_value(props, name);
166 return strcmp(v, value)==0;
169 void free_properties(Property *props)
174 for(i=0; props[i].name; ++i)
177 free(props[i].value);
182 char **get_mounted_devices(void)
186 char **mounted = NULL;
189 mtab = setmntent("/etc/mtab", "r");
193 while((me = getmntent(mtab)))
195 mounted = (char **)realloc(mounted, (n_mounted+2)*sizeof(char *));
196 mounted[n_mounted] = strdup(me->mnt_fsname);
201 mounted[n_mounted] = NULL;
206 int is_mounted(char **mounted, char *devname)
209 for(i=0; mounted[i]; ++i)
210 if(!strcmp(devname, mounted[i]))
215 void free_mounted_devices(char **mounted)
220 for(i=0; mounted[i]; ++i)
225 int is_removable(char *devpath)
232 len = snprintf(fnbuf, sizeof(fnbuf), "/sys%s", devpath);
233 if(len+10>=(int)sizeof(fnbuf))
236 for(ptr=fnbuf+len; (ptr>fnbuf && *ptr!='/'); --ptr) ;
237 strcpy(ptr, "/removable");
238 fd = open(fnbuf, O_RDONLY);
247 printf(" Removable\n");
251 printf(" Not removable\n");
257 int check_buses(char *devpath, char **buses)
263 len = snprintf(fnbuf, sizeof(fnbuf), "/sys%s", devpath);
264 if(len+10>=(int)sizeof(fnbuf))
267 for(ptr=fnbuf+len; ptr>fnbuf+12; --ptr)
271 strcpy(ptr, "/subsystem");
272 len = readlink(fnbuf, linkbuf, sizeof(linkbuf)-1);
279 for(; (len>0 && linkbuf[len-1]!='/'); --len) ;
281 printf(" Subsystem of %s is %s\n", fnbuf, linkbuf+len);
282 for(i=0; buses[i]; ++i)
283 if(strcmp(linkbuf+len, buses[i])==0)
291 int can_mount(Property *props)
293 static char *removable_buses[] = { "usb", "firewire", 0 };
297 if(!match_property_value(props, "DEVTYPE", "partition"))
300 devpath = get_property_value(props, "DEVPATH");
301 if(is_removable(devpath))
304 for(i=0; removable_buses[i]; ++i)
305 if(match_property_value(props, "ID_BUS", removable_buses[i]))
308 return check_buses(devpath, removable_buses);
311 char **get_device_nodes(char *dirname)
320 char **checked = NULL;
324 dir = opendir(dirname);
328 while((de = readdir(dir)))
333 if(de->d_name[0]=='.' && (de->d_name[1]==0 || (de->d_name[1]=='.' && de->d_name[2]==0)))
336 snprintf(fnbuf, sizeof(fnbuf), "%s/%s", dirname, de->d_name);
340 if(S_ISLNK(st.st_mode))
343 len = readlink(fnbuf, linkbuf, sizeof(linkbuf)-1);
353 for(i=0; (!duplicate && i<n_checked); ++i)
354 if(strcmp(node, checked[i])==0)
360 printf("Device %s is a duplicate\n", fnbuf);
364 checked = (char **)realloc(checked, (n_checked+1)*sizeof(char *));
365 checked[n_checked] = strdup(node);
368 nodes = (char **)realloc(nodes, (n_nodes+2)*sizeof(char *));
369 nodes[n_nodes] = strdup(fnbuf);
376 for(i=0; i<n_checked; ++i)
382 nodes[n_nodes] = NULL;
387 Device *get_devices(void)
390 Device *devices = NULL;
392 char **mounted = NULL;
395 nodes = get_device_nodes("/dev/disk/by-id");
396 mounted = get_mounted_devices();
398 for(i=0; nodes[i]; ++i)
401 printf("Examining device %s\n", nodes[i]);
403 Property *props = get_device_properties(nodes[i]);
407 printf(" No properties\n");
414 for(j=0; props[j].name; ++j)
415 printf(" %s = %s\n", props[j].name, props[j].value);
429 printf(" Using device\n");
431 devname = get_property_value(props, "DEVNAME");
433 label = get_property_value(props, "ID_FS_LABEL");
435 label = get_property_value(props, "ID_FS_UUID");
441 for(ptr=label; *ptr; ++ptr)
446 vendor = get_property_value(props, "ID_VENDOR");
447 model = get_property_value(props, "ID_MODEL");
449 pos = snprintf(buf, sizeof(buf), "%s", label);
451 pos += snprintf(buf+pos, sizeof(buf)-pos, " (%s %s)", vendor, model);
455 devices = (Device *)realloc(devices, (n_devices+2)*sizeof(Device));
456 devices[n_devices].node = nodes[i];
457 devices[n_devices].label = strdup(label);
458 devices[n_devices].description = strdup(buf);
459 devices[n_devices].mounted = is_mounted(mounted, devname);
460 devices[n_devices].time = st.st_mtime;
465 free_properties(props);
469 free_mounted_devices(mounted);
473 devices[n_devices].node = NULL;
474 devices[n_devices].label = NULL;
475 devices[n_devices].description = NULL;
481 void free_devices(Device *devices)
486 for(i=0; devices[i].node; ++i)
488 free(devices[i].node);
489 free(devices[i].label);
490 free(devices[i].description);
495 void row_activated(GtkTreeView *list, GtkTreePath *path, GtkTreeViewColumn *column, gpointer user_data)
499 int umount = *(int *)user_data;
501 model = gtk_tree_view_get_model(list);
503 if(gtk_tree_model_get_iter(model, &iter, path))
509 gtk_tree_model_get(model, &iter, 1, &device, -1);
519 printf("Running pumount %s\n", device->node);
521 printf("Running pmount %s %s\n", device->node, device->label);
529 execl("/usr/bin/pumount", "pumount", device->node, NULL);
531 execl("/usr/bin/pmount", "pmount", device->node, device->label, NULL);
546 len = read(pipe_fd[0], buf+pos, sizeof(buf)-pos-1);
554 waitpid(pid, &status, 0);
555 if(!WIFEXITED(status) || WEXITSTATUS(status))
559 dialog = gtk_message_dialog_new(NULL, 0, GTK_MESSAGE_ERROR, GTK_BUTTONS_OK, "%s", buf);
560 g_signal_connect(dialog, "response", >k_main_quit, NULL);
561 gtk_widget_show_all(dialog);
574 int main(int argc, char **argv)
580 GtkTreeSelection *selection;
589 gtk_init(&argc, &argv);
591 while((opt = getopt(argc, argv, "vu"))!=-1) switch(opt)
601 window = gtk_window_new(GTK_WINDOW_TOPLEVEL);
602 gtk_container_set_border_width(GTK_CONTAINER(window), 5);
603 g_signal_connect(window, "destroy", G_CALLBACK(>k_main_quit), NULL);
605 viewport = gtk_viewport_new(NULL, NULL);
606 gtk_viewport_set_shadow_type(GTK_VIEWPORT(viewport), GTK_SHADOW_IN);
607 gtk_container_add(GTK_CONTAINER(window), viewport);
609 list = gtk_tree_view_new();
610 gtk_container_add(GTK_CONTAINER(viewport), list);
611 g_signal_connect(list, "row-activated", G_CALLBACK(&row_activated), &umount);
613 store = gtk_list_store_new(2, G_TYPE_STRING, G_TYPE_POINTER);
614 gtk_tree_view_set_model(GTK_TREE_VIEW(list), GTK_TREE_MODEL(store));
615 gtk_tree_view_insert_column_with_attributes(GTK_TREE_VIEW(list),
616 -1, "Device", gtk_cell_renderer_text_new(), "text", 0, NULL);
618 selection = gtk_tree_view_get_selection(GTK_TREE_VIEW(list));
620 devices = get_devices();
625 for(i=0; devices[i].node; ++i)
626 if(!devices[i].mounted==!umount)
628 gtk_list_store_append(store, &iter);
629 gtk_list_store_set(store, &iter, 0, devices[i].description, 1, &devices[i], -1);
630 if(devices[i].time>latest)
632 latest = devices[i].time;
633 gtk_tree_selection_select_iter(selection, &iter);
642 gtk_widget_show_all(window);
647 dialog = gtk_message_dialog_new(NULL, 0, GTK_MESSAGE_WARNING, GTK_BUTTONS_OK,
648 "No devices to %s", (umount ? "unmount" : "mount"));
649 g_signal_connect(dialog, "response", G_CALLBACK(>k_main_quit), NULL);
650 gtk_widget_show_all(dialog);
655 free_devices(devices);