add ata_id to read serial numbers from ATA drives

Signed-off-by: Kay Sievers <kay.sievers@suse.de>

1 files changed, 150 insertions, 0 deletions

diff --git a/extras/ata_id/ata_id.c b/extras/ata_id/ata_id.c
new file mode 100644
index 0000000000..e26cb6bd2d
--- /dev/null
+++ b/extras/ata_id/ata_id.c
@@ -0,0 +1,150 @@
+/*
+ * ata_id - reads product/serial number from ATA drives
+ *
+ * Copyright (C) 2005 Kay Sievers <kay.sievers@vrfy.org>
+ *
+ *	This library is free software; you can redistribute it and/or
+ *	modify it under the terms of the GNU Lesser General Public
+ *	License as published by the Free Software Foundation; either
+ *	version 2.1 of the License, or (at your option) any later version.
+ *
+ *	This library is distributed in the hope that it will be useful,
+ *	but WITHOUT ANY WARRANTY; without even the implied warranty of
+ *	MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ *	Lesser General Public License for more details.
+ *
+ *	You should have received a copy of the GNU Lesser General Public
+ *	License along with this library; if not, write to the Free Software
+ *	Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
+ */
+
+#ifndef _GNU_SOURCE
+#define _GNU_SOURCE 1
+#endif
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <ctype.h>
+#include <errno.h>
+#include <sys/ioctl.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <linux/types.h>
+#include <linux/hdreg.h>
+
+#include "../../logging.h"
+#include "../../udev_utils.h"
+
+#ifdef USE_LOG
+void log_message(int priority, const char *format, ...)
+{
+	va_list args;
+	static int udev_log = -1;
+
+	if (udev_log == -1) {
+		const char *value;
+
+		value = getenv("UDEV_LOG");
+		if (value)
+			udev_log = log_priority(value);
+		else
+			udev_log = LOG_ERR;
+	}
+
+	if (priority > udev_log)
+		return;
+
+	va_start(args, format);
+	vsyslog(priority, format, args);
+	va_end(args);
+}
+#endif
+
+static void set_str(char *to, const unsigned char *from, int count)
+{
+	int i, j;
+	int len;
+
+	len = strnlen(from, count);
+	while (isspace(from[len-1]))
+		len--;
+
+	i = 0;
+	while (isspace(from[i]) && (i < len))
+		i++;
+
+	j = 0;
+	while (i < len) {
+		switch(from[i]) {
+		case '/':
+		case ' ':
+			to[j++] = '_';
+		default:
+			to[j++] = from[i];
+		}
+		i++;
+	}
+	to[j] = '\0';
+}
+
+int main(int argc, char *argv[])
+{
+	struct hd_driveid id;
+	char model[41];
+	char serial[21];
+	char revision[9];
+	const char *node = NULL;
+	int i;
+	int export = 0;
+	int fd;
+	int rc = 0;
+
+	for (i = 1 ; i < argc; i++) {
+		char *arg = argv[i];
+
+		if (strcmp(arg, "--export") == 0) {
+			export = 1;
+		} else
+			node = arg;
+	}
+	if (!node) {
+		err("no node specified");
+		rc = 1;
+		goto exit;
+	}
+
+	fd = open(node, O_RDONLY);
+	if (fd < 0)
+		if (errno == ENOMEDIUM)
+			fd = open(node, O_RDONLY|O_NONBLOCK);
+	if (fd < 0) {
+		err("unable to open '%s'", node);
+		rc = 1;
+		goto exit;
+	}
+
+	if (ioctl(fd, HDIO_GET_IDENTITY, &id)) {
+		err("HDIO_GET_IDENTITY failed for '%s'", node);
+		rc = 3;
+		goto close;
+	}
+
+	set_str(model, id.model, 40);
+	set_str(serial, id.serial_no, 20);
+	set_str(revision, id.fw_rev, 8);
+
+	if (export) {
+		printf("ID_MODEL=%s\n", model);
+		printf("ID_SERIAL=%s\n", serial);
+		printf("ID_REVISION=%s\n", revision);
+	} else
+		printf("%s_%s\n", model, serial);
+
+close:
+	close(fd);
+exit:
+	logging_close();
+	return rc;
+}