X-Git-Url: https://git.wh0rd.org/?p=dump.git;a=blobdiff_plain;f=compat%2Flib%2Fbylabel.c;h=5a40a70e6eb5af34dfbe3d388a99904a38c80df2;hp=c91457d3d232d7b47ff0209dc2a7c0531c79d4c6;hb=dba36a3e9b1fb9e7c207ab8b5f7cb7f502867780;hpb=1099d48c9c09059aade8605beddfa4ef98f575aa

diff --git a/compat/lib/bylabel.c b/compat/lib/bylabel.c
index c91457d..5a40a70 100644
--- a/compat/lib/bylabel.c
+++ b/compat/lib/bylabel.c
@@ -11,6 +11,9 @@
  * - Ported to dump/restore
  */
 
+#include <config.h>
+#include <compatlfs.h>
+#include <sys/types.h>
 #include <stdio.h>
 #include <sys/param.h>
 #include <string.h>
@@ -18,18 +21,26 @@
 #include <fcntl.h>
 #include <unistd.h>
 #include <malloc.h>
+#include <sys/cdefs.h>
 #include "bylabel.h"
 
+#ifndef HAVE_BLKID
+
 #define PROC_PARTITIONS "/proc/partitions"
 #define DEVLABELDIR	"/dev"
 
-#define EXT2_SUPER_MAGIC 0xEF53
+#define EXT2_SUPER_OFFSET	1024
+#define EXT2_SUPER_SIZE		sizeof(struct ext2_super_block)
+#define EXT2_SUPER_MAGIC	0xEF53
+
+#define VOLNAMSZ	16
+
 struct ext2_super_block {
 	unsigned char	s_dummy1[56];
 	unsigned char	s_magic[2];
 	unsigned char	s_dummy2[46];
 	unsigned char	s_uuid[16];
-	unsigned char	s_volume_name[16];
+	unsigned char	s_volume_name[VOLNAMSZ];
 };
 #define ext2magic(s)	((unsigned int) s.s_magic[0] + (((unsigned int) s.s_magic[1]) << 8))
 
@@ -51,13 +62,13 @@ get_label_uuid(const char *device, char **label, char *uuid) {
 	int fd;
 	struct ext2_super_block e2sb;
 
-	fd = open(device, O_RDONLY);
+	fd = OPEN(device, O_RDONLY);
 	if (fd < 0)
 		return 1;
 
-	if (lseek(fd, 1024, SEEK_SET) != 1024
-	    || read(fd, (char *) &e2sb, sizeof(e2sb)) != sizeof(e2sb)
-	    || (ext2magic(e2sb) != EXT2_SUPER_MAGIC)) {
+	if (LSEEK(fd, EXT2_SUPER_OFFSET, SEEK_SET) != EXT2_SUPER_OFFSET ||
+	    read(fd, (char *) &e2sb, EXT2_SUPER_SIZE) != EXT2_SUPER_SIZE ||
+	    ext2magic(e2sb) != EXT2_SUPER_MAGIC) {
 		close(fd);
 		return 1;
 	}
@@ -66,7 +77,9 @@ get_label_uuid(const char *device, char **label, char *uuid) {
 
 	/* superblock is ext2 - now what is its label? */
 	memcpy(uuid, e2sb.s_uuid, sizeof(e2sb.s_uuid));
-	*label = strdup(e2sb.s_volume_name);
+	*label = malloc(VOLNAMSZ + 1);
+	strncpy(*label, e2sb.s_volume_name, VOLNAMSZ);
+	(*label)[VOLNAMSZ] = 0;
 
 	return 0;
 }
@@ -229,3 +242,20 @@ get_device_name(const char * item) {
 
 	return rc;
 }
+
+const char *
+get_device_label(const char * spec) {
+	struct uuidCache_s *uc;
+
+	uuidcache_init();
+	uc = uuidCache;
+
+	while(uc) {
+		if (!strcmp(spec, uc->device))
+			return uc->label[0] == '\0' ? NULL : strdup(uc->label);
+		uc = uc->next;
+	}
+	return NULL;
+}
+
+#endif /* !HAVE_BLKID */