#include <stdlib.h>
#include <stdio.h>
#include <string.h>
+#include <unistd.h>
+#include <fcntl.h>
#include <dirent.h>
#include <mntent.h>
#include <sys/stat.h>
}
else if(pid>0)
{
- char buf[256];
+ char *buf;
+ int bufsize;
int pos = 0;
int eof = 0;
Property *props = NULL;
close(pipe_fd[1]);
+ bufsize = 256;
+ buf = (char *)malloc(bufsize);
+
while(1)
{
int newline;
{
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)
newline = i;
if(newline<0)
- break;
+ {
+ if(eof)
+ break;
+ bufsize *= 2;
+ buf = (char *)realloc(buf, bufsize);
+ continue;
+ }
if(parse_property(buf, newline, &prop)==0)
{
break;
}
+ free(buf);
+
if(props)
{
props[n_props].name = NULL;
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<n_checked); ++i)
+ if(strcmp(node, checked[i])==0)
+ duplicate = 1;
+ }
+ if(duplicate)
+ {
+ if(verbosity>=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<n_checked; ++i)
+ free(checked[i]);
+ free(checked);
+ }
+
+ if(nodes)
+ nodes[n_nodes] = NULL;
+
+ return nodes;
+}
+
+Device *get_devices(void)
+{
+ char **nodes = NULL;
+ Device *devices = NULL;
+ int n_devices = 0;
+ char **mounted = NULL;
+ int i;
+
+ nodes = get_device_nodes("/dev/disk/by-id");
+ mounted = get_mounted_devices();
+
+ for(i=0; nodes[i]; ++i)
+ {
if(verbosity>=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)
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;
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)
{
free(devices[i].node);
free(devices[i].label);
+ free(devices[i].description);
}
free(devices);
}
pid = fork();
if(pid==0)
{
- if(verbosity>=2)
+ if(verbosity>=1)
{
if(umount)
printf("Running pumount %s\n", device->node);