X-Git-Url: http://git.tdb.fi/?p=pmount-gui.git;a=blobdiff_plain;f=main.c;h=f62b7b0b0b6f28f7791352c75eb5d0cdee8cfefc;hp=910983fb94498a7211fdc9f4921944294a4f9a48;hb=890ae36a86ec7297a8e2ec4ac85ea44482b3938c;hpb=9547dca153610b13cab14725c8d16ccc134ec188 diff --git a/main.c b/main.c index 910983f..f62b7b0 100644 --- a/main.c +++ b/main.c @@ -1,6 +1,8 @@ #include #include #include +#include +#include #include #include #include @@ -68,7 +70,8 @@ Property *get_device_properties(char *node) } else if(pid>0) { - char buf[256]; + char *buf; + int bufsize; int pos = 0; int eof = 0; Property *props = NULL; @@ -76,6 +79,9 @@ Property *get_device_properties(char *node) close(pipe_fd[1]); + bufsize = 256; + buf = (char *)malloc(bufsize); + while(1) { int newline; @@ -86,7 +92,7 @@ Property *get_device_properties(char *node) { int len; - len = read(pipe_fd[0], buf+pos, sizeof(buf)-pos); + len = read(pipe_fd[0], buf+pos, bufsize-pos); if(len==0) eof = 1; else if(len==-1) @@ -100,7 +106,13 @@ Property *get_device_properties(char *node) newline = i; if(newline<0) - break; + { + if(eof) + break; + bufsize *= 2; + buf = (char *)realloc(buf, bufsize); + continue; + } if(parse_property(buf, newline, &prop)==0) { @@ -115,6 +127,8 @@ Property *get_device_properties(char *node) break; } + free(buf); + if(props) { props[n_props].name = NULL; @@ -208,31 +222,185 @@ void free_mounted_devices(char **mounted) free(mounted); } -Device *get_devices(void) +int is_removable(char *devpath) +{ + char fnbuf[256]; + int len; + char *ptr; + int fd; + + len = snprintf(fnbuf, sizeof(fnbuf), "/sys%s", devpath); + if(len+10>=(int)sizeof(fnbuf)) + return 0; + + for(ptr=fnbuf+len; (ptr>fnbuf && *ptr!='/'); --ptr) ; + strcpy(ptr, "/removable"); + fd = open(fnbuf, O_RDONLY); + if(fd!=-1) + { + char c; + read(fd, &c, 1); + close(fd); + if(c=='1') + { + if(verbosity>=2) + printf(" Removable\n"); + return 1; + } + if(verbosity>=2) + printf(" Not removable\n"); + } + + return 0; +} + +int check_buses(char *devpath, char **buses) +{ + char fnbuf[256]; + char *ptr; + int len; + + len = snprintf(fnbuf, sizeof(fnbuf), "/sys%s", devpath); + if(len+10>=(int)sizeof(fnbuf)) + return 0; + + for(ptr=fnbuf+len; ptr>fnbuf+12; --ptr) + if(*ptr=='/') + { + char linkbuf[256]; + strcpy(ptr, "/subsystem"); + len = readlink(fnbuf, linkbuf, sizeof(linkbuf)-1); + *ptr = 0; + + if(len!=-1) + { + int i; + linkbuf[len] = 0; + for(; (len>0 && linkbuf[len-1]!='/'); --len) ; + if(verbosity>=2) + printf(" Subsystem of %s is %s\n", fnbuf, linkbuf+len); + for(i=0; buses[i]; ++i) + if(strcmp(linkbuf+len, buses[i])==0) + return 1; + } + } + + return 0; +} + +int can_mount(Property *props) +{ + static char *removable_buses[] = { "usb", "firewire", 0 }; + char *devpath; + int i; + + if(!match_property_value(props, "DEVTYPE", "partition")) + return 0; + + devpath = get_property_value(props, "DEVPATH"); + if(is_removable(devpath)) + return 1; + + for(i=0; removable_buses[i]; ++i) + if(match_property_value(props, "ID_BUS", removable_buses[i])) + return 1; + + return check_buses(devpath, removable_buses); +} + +char **get_device_nodes(char *dirname) { DIR *dir; struct dirent *de; char fnbuf[256]; - Device *devices = NULL; - int n_devices = 0; - char **mounted = NULL; - - dir = opendir("/dev/disk/by-id"); + char linkbuf[256]; + struct stat st; + char **nodes = NULL; + int n_nodes = 0; + char **checked = NULL; + int n_checked = 0; + int i; + + dir = opendir(dirname); if(!dir) return NULL; - mounted = get_mounted_devices(); - while((de = readdir(dir))) { + char *node; + int duplicate = 0; + if(de->d_name[0]=='.' && (de->d_name[1]==0 || (de->d_name[1]=='.' && de->d_name[2]==0))) continue; - snprintf(fnbuf, sizeof(fnbuf), "/dev/disk/by-id/%s", de->d_name); + snprintf(fnbuf, sizeof(fnbuf), "%s/%s", dirname, de->d_name); + + node = fnbuf; + lstat(fnbuf, &st); + if(S_ISLNK(st.st_mode)) + { + int len; + len = readlink(fnbuf, linkbuf, sizeof(linkbuf)-1); + if(len!=-1) + { + linkbuf[len] = 0; + node = linkbuf; + } + } + + if(checked) + { + for(i=0; (!duplicate && i=2) + printf("Device %s is a duplicate\n", fnbuf); + continue; + } + + checked = (char **)realloc(checked, (n_checked+1)*sizeof(char *)); + checked[n_checked] = strdup(node); + ++n_checked; + + nodes = (char **)realloc(nodes, (n_nodes+2)*sizeof(char *)); + nodes[n_nodes] = strdup(fnbuf); + ++n_nodes; + } + + closedir(dir); + if(checked) + { + for(i=0; i=1) - printf("Examining device %s\n", fnbuf); + printf("Examining device %s\n", nodes[i]); - Property *props = get_device_properties(fnbuf); + Property *props = get_device_properties(nodes[i]); if(!props) { if(verbosity>=2) @@ -242,12 +410,12 @@ Device *get_devices(void) if(verbosity>=2) { - int i; - for(i=0; props[i].name; ++i) - printf(" %s = %s\n", props[i].name, props[i].value); + int j; + for(j=0; props[j].name; ++j) + printf(" %s = %s\n", props[j].name, props[j].value); } - if(match_property_value(props, "ID_BUS", "usb") && match_property_value(props, "DEVTYPE", "partition")) + if(can_mount(props)) { char *devname; char *label; @@ -282,20 +450,22 @@ Device *get_devices(void) if(vendor && model) pos += snprintf(buf+pos, sizeof(buf)-pos, " (%s %s)", vendor, model); - stat(fnbuf, &st); + stat(nodes[i], &st); devices = (Device *)realloc(devices, (n_devices+2)*sizeof(Device)); - devices[n_devices].node = strdup(fnbuf); + devices[n_devices].node = nodes[i]; devices[n_devices].label = strdup(label); devices[n_devices].description = strdup(buf); devices[n_devices].mounted = is_mounted(mounted, devname); devices[n_devices].time = st.st_mtime; ++n_devices; } + else + free(nodes[i]); free_properties(props); } - closedir(dir); + free(nodes); free_mounted_devices(mounted); if(devices) @@ -317,6 +487,7 @@ void free_devices(Device *devices) { free(devices[i].node); free(devices[i].label); + free(devices[i].description); } free(devices); } @@ -342,7 +513,7 @@ void row_activated(GtkTreeView *list, GtkTreePath *path, GtkTreeViewColumn *colu pid = fork(); if(pid==0) { - if(verbosity>=2) + if(verbosity>=1) { if(umount) printf("Running pumount %s\n", device->node);