busybox/util-linux/volume_id/get_devname.c

400 lines
8.1 KiB
C
Raw Normal View History

/* vi: set sw=4 ts=4: */
/*
* Support functions for mounting devices by label/uuid
*
* Copyright (C) 2006 by Jason Schoon <floydpink@gmail.com>
* Some portions cribbed from e2fsprogs, util-linux, dosfstools
*
* Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
*/
#include "volume_id_internal.h"
#define BLKGETSIZE64 _IOR(0x12,114,size_t)
#define PROC_PARTITIONS "/proc/partitions"
#define PROC_CDROMS "/proc/sys/dev/cdrom/info"
#define DEVLABELDIR "/dev"
#define SYS_BLOCK "/sys/block"
static struct uuidCache_s {
struct uuidCache_s *next;
char uuid[16];
char *device;
char *label;
int major, minor;
} *uuidCache;
/* for now, only ext2, ext3 and xfs are supported */
static int
get_label_uuid(const char *device, char **label, char **uuid, int iso_only)
{
int rv = 1;
uint64_t size;
struct volume_id *vid;
vid = volume_id_open_node(device);
if (ioctl(vid->fd, BLKGETSIZE64, &size) != 0) {
size = 0;
}
#if ENABLE_FEATURE_VOLUMEID_ISO9660
if (iso_only ?
volume_id_probe_iso9660(vid, 0) != 0 :
volume_id_probe_all(vid, 0, size) != 0) {
goto ret;
}
#else
if (volume_id_probe_all(vid, 0, size) != 0) {
goto ret;
}
#endif
if (vid->label[0] != '\0') {
*label = xstrndup(vid->label, sizeof(vid->label));
*uuid = xstrndup(vid->uuid, sizeof(vid->uuid));
printf("Found label %s on %s (uuid:%s)\n", *label, device, *uuid);
rv = 0;
}
ret:
free_volume_id(vid);
return rv;
}
static void
uuidcache_addentry(char * device, int major, int minor, char *label, char *uuid)
{
struct uuidCache_s *last;
if (!uuidCache) {
last = uuidCache = xzalloc(sizeof(*uuidCache));
} else {
for (last = uuidCache; last->next; last = last->next)
continue;
last->next = xzalloc(sizeof(*uuidCache));
last = last->next;
}
/*last->next = NULL; - xzalloc did it*/
last->label = label;
last->device = device;
last->major = major;
last->minor = minor;
memcpy(last->uuid, uuid, sizeof(last->uuid));
}
static void
uuidcache_check_device(const char *device_name, int ma, int mi, int iso_only)
{
char device[110];
char *uuid = NULL, *label = NULL;
char *ptr;
char *deviceDir = NULL;
int mustRemove = 0;
int mustRemoveDir = 0;
int i;
sprintf(device, "%s/%s", DEVLABELDIR, device_name);
if (access(device, F_OK)) {
ptr = device;
i = 0;
while (*ptr)
if (*ptr++ == '/')
i++;
if (i > 2) {
deviceDir = alloca(strlen(device) + 1);
strcpy(deviceDir, device);
ptr = deviceDir + (strlen(device) - 1);
while (*ptr != '/')
*ptr-- = '\0';
if (mkdir(deviceDir, 0644)) {
printf("mkdir: cannot create directory %s: %d\n", deviceDir, errno);
} else {
mustRemoveDir = 1;
}
}
mknod(device, S_IFBLK | 0600, makedev(ma, mi));
mustRemove = 1;
}
if (!get_label_uuid(device, &label, &uuid, iso_only))
uuidcache_addentry(strdup(device), ma, mi,
label, uuid);
if (mustRemove) unlink(device);
if (mustRemoveDir) rmdir(deviceDir);
}
static void
uuidcache_init_partitions(void)
{
char line[100];
int ma, mi;
unsigned long long sz;
FILE *procpt;
int firstPass;
int handleOnFirst;
char *chptr;
procpt = xfopen(PROC_PARTITIONS, "r");
/*
# cat /proc/partitions
major minor #blocks name
8 0 293036184 sda
8 1 6835626 sda1
8 2 1 sda2
8 5 979933 sda5
8 6 15623181 sda6
8 7 97659103 sda7
8 8 171935631 sda8
*/
for (firstPass = 1; firstPass >= 0; firstPass--) {
fseek(procpt, 0, SEEK_SET);
while (fgets(line, sizeof(line), procpt)) {
/* The original version of this code used sscanf, but
diet's sscanf is quite limited */
chptr = line;
if (*chptr != ' ') continue;
chptr++;
ma = bb_strtou(chptr, &chptr, 0);
if (ma < 0) continue;
chptr = skip_whitespace(chptr);
mi = bb_strtou(chptr, &chptr, 0);
if (mi < 0) continue;
chptr = skip_whitespace(chptr);
sz = bb_strtoull(chptr, &chptr, 0);
if ((long long)sz == -1LL) continue;
chptr = skip_whitespace(chptr);
/* skip extended partitions (heuristic: size 1) */
if (sz == 1)
continue;
*strchrnul(chptr, '\n') = '\0';
/* now chptr => device name */
if (!chptr[0])
continue;
/* look only at md devices on first pass */
handleOnFirst = (chptr[0] == 'm' && chptr[1] == 'd');
if (firstPass != handleOnFirst)
continue;
/* heuristic: partition name ends in a digit */
if (isdigit(chptr[strlen(chptr) - 1])) {
uuidcache_check_device(chptr, ma, mi, 0);
}
}
}
fclose(procpt);
}
static int
dev_get_major_minor(char *device_name, int *major, int *minor)
{
char * dev_path;
int fd;
char dev[7];
char *major_ptr, *minor_ptr;
dev_path = alloca(strlen(SYS_BLOCK) + strlen(device_name) + 6);
sprintf(dev_path, "%s/%s/dev", SYS_BLOCK, device_name);
fd = open(dev_path, O_RDONLY);
if (fd < 0) return 1;
full_read(fd, dev, sizeof(dev));
close(fd);
major_ptr = dev;
minor_ptr = strchr(dev, ':');
if (!minor_ptr) return 1;
*minor_ptr++ = '\0';
*major = strtol(major_ptr, NULL, 10);
*minor = strtol(minor_ptr, NULL, 10);
return 0;
}
static void
uuidcache_init_cdroms(void)
{
char line[100];
int ma, mi;
FILE *proccd;
proccd = fopen(PROC_CDROMS, "r");
if (!proccd) {
static smallint warn = 0;
if (!warn) {
warn = 1;
bb_error_msg("mount: could not open %s, so UUID and LABEL "
"conversion cannot be done for CD-Roms.",
PROC_CDROMS);
}
return;
}
while (fgets(line, sizeof(line), proccd)) {
const char *drive_name_string = "drive name:\t\t";
if (!strncmp(line, drive_name_string, strlen(drive_name_string))) {
char *device_name;
device_name = strtok(line + strlen(drive_name_string), "\t\n");
while (device_name) {
dev_get_major_minor(device_name, &ma, &mi);
uuidcache_check_device(device_name, ma, mi, 1);
device_name = strtok(NULL, "\t\n");
}
break;
}
}
fclose(proccd);
}
static void
uuidcache_init(void)
{
if (uuidCache)
return;
uuidcache_init_partitions();
uuidcache_init_cdroms();
}
#define UUID 1
#define VOL 2
#ifdef UNUSED
static char *
get_spec_by_x(int n, const char *t, int * majorPtr, int * minorPtr)
{
struct uuidCache_s *uc;
uuidcache_init();
uc = uuidCache;
while(uc) {
switch (n) {
case UUID:
if (!memcmp(t, uc->uuid, sizeof(uc->uuid))) {
*majorPtr = uc->major;
*minorPtr = uc->minor;
return uc->device;
}
break;
case VOL:
if (!strcmp(t, uc->label)) {
*majorPtr = uc->major;
*minorPtr = uc->minor;
return uc->device;
}
break;
}
uc = uc->next;
}
return NULL;
}
static unsigned char
fromhex(char c)
{
if (isdigit(c))
return (c - '0');
if (islower(c))
return (c - 'a' + 10);
return (c - 'A' + 10);
}
static char *
get_spec_by_uuid(const char *s, int * major, int * minor)
{
unsigned char uuid[16];
int i;
if (strlen(s) != 36 ||
s[8] != '-' || s[13] != '-' || s[18] != '-' || s[23] != '-')
goto bad_uuid;
for (i=0; i<16; i++) {
if (*s == '-') s++;
if (!isxdigit(s[0]) || !isxdigit(s[1]))
goto bad_uuid;
uuid[i] = ((fromhex(s[0])<<4) | fromhex(s[1]));
s += 2;
}
return get_spec_by_x(UUID, (char *)uuid, major, minor);
bad_uuid:
fprintf(stderr, _("mount: bad UUID"));
return 0;
}
static char *
get_spec_by_volume_label(const char *s, int *major, int *minor)
{
return get_spec_by_x(VOL, s, major, minor);
}
static int display_uuid_cache(void)
{
struct uuidCache_s *u;
size_t i;
uuidcache_init();
u = uuidCache;
while (u) {
printf("%s %s ", u->device, u->label);
for (i = 0; i < sizeof(u->uuid); i++) {
if (i == 4 || i == 6 || i == 8 || i == 10)
printf("-");
printf("%x", u->uuid[i] & 0xff);
}
printf("\n");
u = u->next;
}
return 0;
}
#endif // UNUSED
/* Used by mount and findfs */
char *get_devname_from_label(const char *spec)
{
struct uuidCache_s *uc;
int spec_len = strlen(spec);
uuidcache_init();
uc = uuidCache;
while (uc) {
if (uc->label && !strncmp(spec, uc->label, spec_len)) {
return xstrdup(uc->device);
}
uc = uc->next;
}
return NULL;
}
char *get_devname_from_uuid(const char *spec)
{
struct uuidCache_s *uc;
uuidcache_init();
uc = uuidCache;
while (uc) {
if (!memcmp(spec, uc->uuid, sizeof(uc->uuid))) {
return xstrdup(uc->device);
}
uc = uc->next;
}
return NULL;
}