1
0
mirror of https://github.com/systemd/systemd.git synced 2024-11-05 23:51:28 +03:00
systemd/extras/ata_id/ata_id.c
Kay Sievers 670e470543 add ata_id to read serial numbers from ATA drives
Signed-off-by: Kay Sievers <kay.sievers@suse.de>
2005-06-25 23:54:28 +02:00

151 lines
2.9 KiB
C

/*
* 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;
}