/* * ata_id - reads product/serial number from ATA drives * * Copyright (C) 2005-2008 Kay Sievers <kay.sievers@vrfy.org> * Copyright (C) 2009 Lennart Poettering <lennart@poettering.net> * Copyright (C) 2009-2010 David Zeuthen <zeuthen@gmail.com> * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation, either version 2 of the License, or * (at your option) any later version. * * This program 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 General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ #include <stdio.h> #include <stdlib.h> #include <stdint.h> #include <unistd.h> #include <fcntl.h> #include <ctype.h> #include <string.h> #include <errno.h> #include <getopt.h> #include <scsi/scsi.h> #include <scsi/sg.h> #include <scsi/scsi_ioctl.h> #include <sys/ioctl.h> #include <sys/types.h> #include <sys/stat.h> #include <linux/types.h> #include <linux/hdreg.h> #include <linux/fs.h> #include <linux/cdrom.h> #include <linux/bsg.h> #include <arpa/inet.h> #include "libudev.h" #include "libudev-private.h" #include "log.h" #define COMMAND_TIMEOUT_MSEC (30 * 1000) static int disk_scsi_inquiry_command(int fd, void *buf, size_t buf_len) { struct sg_io_v4 io_v4; uint8_t cdb[6]; uint8_t sense[32]; int ret; /* * INQUIRY, see SPC-4 section 6.4 */ memset(cdb, 0, sizeof(cdb)); cdb[0] = 0x12; /* OPERATION CODE: INQUIRY */ cdb[3] = (buf_len >> 8); /* ALLOCATION LENGTH */ cdb[4] = (buf_len & 0xff); memset(sense, 0, sizeof(sense)); memset(&io_v4, 0, sizeof(struct sg_io_v4)); io_v4.guard = 'Q'; io_v4.protocol = BSG_PROTOCOL_SCSI; io_v4.subprotocol = BSG_SUB_PROTOCOL_SCSI_CMD; io_v4.request_len = sizeof (cdb); io_v4.request = (uintptr_t) cdb; io_v4.max_response_len = sizeof (sense); io_v4.response = (uintptr_t) sense; io_v4.din_xfer_len = buf_len; io_v4.din_xferp = (uintptr_t) buf; io_v4.timeout = COMMAND_TIMEOUT_MSEC; ret = ioctl(fd, SG_IO, &io_v4); if (ret != 0) { /* could be that the driver doesn't do version 4, try version 3 */ if (errno == EINVAL) { struct sg_io_hdr io_hdr; memset(&io_hdr, 0, sizeof(struct sg_io_hdr)); io_hdr.interface_id = 'S'; io_hdr.cmdp = (unsigned char*) cdb; io_hdr.cmd_len = sizeof (cdb); io_hdr.dxferp = buf; io_hdr.dxfer_len = buf_len; io_hdr.sbp = sense; io_hdr.mx_sb_len = sizeof (sense); io_hdr.dxfer_direction = SG_DXFER_FROM_DEV; io_hdr.timeout = COMMAND_TIMEOUT_MSEC; ret = ioctl(fd, SG_IO, &io_hdr); if (ret != 0) goto out; /* even if the ioctl succeeds, we need to check the return value */ if (!(io_hdr.status == 0 && io_hdr.host_status == 0 && io_hdr.driver_status == 0)) { errno = EIO; ret = -1; goto out; } } else { goto out; } } /* even if the ioctl succeeds, we need to check the return value */ if (!(io_v4.device_status == 0 && io_v4.transport_status == 0 && io_v4.driver_status == 0)) { errno = EIO; ret = -1; goto out; } out: return ret; } static int disk_identify_command(int fd, void *buf, size_t buf_len) { struct sg_io_v4 io_v4; uint8_t cdb[12]; uint8_t sense[32]; uint8_t *desc = sense+8; int ret; /* * ATA Pass-Through 12 byte command, as described in * * T10 04-262r8 ATA Command Pass-Through * * from http://www.t10.org/ftp/t10/document.04/04-262r8.pdf */ memset(cdb, 0, sizeof(cdb)); cdb[0] = 0xa1; /* OPERATION CODE: 12 byte pass through */ cdb[1] = 4 << 1; /* PROTOCOL: PIO Data-in */ cdb[2] = 0x2e; /* OFF_LINE=0, CK_COND=1, T_DIR=1, BYT_BLOK=1, T_LENGTH=2 */ cdb[3] = 0; /* FEATURES */ cdb[4] = 1; /* SECTORS */ cdb[5] = 0; /* LBA LOW */ cdb[6] = 0; /* LBA MID */ cdb[7] = 0; /* LBA HIGH */ cdb[8] = 0 & 0x4F; /* SELECT */ cdb[9] = 0xEC; /* Command: ATA IDENTIFY DEVICE */; memset(sense, 0, sizeof(sense)); memset(&io_v4, 0, sizeof(struct sg_io_v4)); io_v4.guard = 'Q'; io_v4.protocol = BSG_PROTOCOL_SCSI; io_v4.subprotocol = BSG_SUB_PROTOCOL_SCSI_CMD; io_v4.request_len = sizeof (cdb); io_v4.request = (uintptr_t) cdb; io_v4.max_response_len = sizeof (sense); io_v4.response = (uintptr_t) sense; io_v4.din_xfer_len = buf_len; io_v4.din_xferp = (uintptr_t) buf; io_v4.timeout = COMMAND_TIMEOUT_MSEC; ret = ioctl(fd, SG_IO, &io_v4); if (ret != 0) { /* could be that the driver doesn't do version 4, try version 3 */ if (errno == EINVAL) { struct sg_io_hdr io_hdr; memset(&io_hdr, 0, sizeof(struct sg_io_hdr)); io_hdr.interface_id = 'S'; io_hdr.cmdp = (unsigned char*) cdb; io_hdr.cmd_len = sizeof (cdb); io_hdr.dxferp = buf; io_hdr.dxfer_len = buf_len; io_hdr.sbp = sense; io_hdr.mx_sb_len = sizeof (sense); io_hdr.dxfer_direction = SG_DXFER_FROM_DEV; io_hdr.timeout = COMMAND_TIMEOUT_MSEC; ret = ioctl(fd, SG_IO, &io_hdr); if (ret != 0) goto out; } else { goto out; } } if (!(sense[0] == 0x72 && desc[0] == 0x9 && desc[1] == 0x0c)) { errno = EIO; ret = -1; goto out; } out: return ret; } static int disk_identify_packet_device_command(int fd, void *buf, size_t buf_len) { struct sg_io_v4 io_v4; uint8_t cdb[16]; uint8_t sense[32]; uint8_t *desc = sense+8; int ret; /* * ATA Pass-Through 16 byte command, as described in * * T10 04-262r8 ATA Command Pass-Through * * from http://www.t10.org/ftp/t10/document.04/04-262r8.pdf */ memset(cdb, 0, sizeof(cdb)); cdb[0] = 0x85; /* OPERATION CODE: 16 byte pass through */ cdb[1] = 4 << 1; /* PROTOCOL: PIO Data-in */ cdb[2] = 0x2e; /* OFF_LINE=0, CK_COND=1, T_DIR=1, BYT_BLOK=1, T_LENGTH=2 */ cdb[3] = 0; /* FEATURES */ cdb[4] = 0; /* FEATURES */ cdb[5] = 0; /* SECTORS */ cdb[6] = 1; /* SECTORS */ cdb[7] = 0; /* LBA LOW */ cdb[8] = 0; /* LBA LOW */ cdb[9] = 0; /* LBA MID */ cdb[10] = 0; /* LBA MID */ cdb[11] = 0; /* LBA HIGH */ cdb[12] = 0; /* LBA HIGH */ cdb[13] = 0; /* DEVICE */ cdb[14] = 0xA1; /* Command: ATA IDENTIFY PACKET DEVICE */; cdb[15] = 0; /* CONTROL */ memset(sense, 0, sizeof(sense)); memset(&io_v4, 0, sizeof(struct sg_io_v4)); io_v4.guard = 'Q'; io_v4.protocol = BSG_PROTOCOL_SCSI; io_v4.subprotocol = BSG_SUB_PROTOCOL_SCSI_CMD; io_v4.request_len = sizeof (cdb); io_v4.request = (uintptr_t) cdb; io_v4.max_response_len = sizeof (sense); io_v4.response = (uintptr_t) sense; io_v4.din_xfer_len = buf_len; io_v4.din_xferp = (uintptr_t) buf; io_v4.timeout = COMMAND_TIMEOUT_MSEC; ret = ioctl(fd, SG_IO, &io_v4); if (ret != 0) { /* could be that the driver doesn't do version 4, try version 3 */ if (errno == EINVAL) { struct sg_io_hdr io_hdr; memset(&io_hdr, 0, sizeof(struct sg_io_hdr)); io_hdr.interface_id = 'S'; io_hdr.cmdp = (unsigned char*) cdb; io_hdr.cmd_len = sizeof (cdb); io_hdr.dxferp = buf; io_hdr.dxfer_len = buf_len; io_hdr.sbp = sense; io_hdr.mx_sb_len = sizeof (sense); io_hdr.dxfer_direction = SG_DXFER_FROM_DEV; io_hdr.timeout = COMMAND_TIMEOUT_MSEC; ret = ioctl(fd, SG_IO, &io_hdr); if (ret != 0) goto out; } else { goto out; } } if (!(sense[0] == 0x72 && desc[0] == 0x9 && desc[1] == 0x0c)) { errno = EIO; ret = -1; goto out; } out: return ret; } /** * disk_identify_get_string: * @identify: A block of IDENTIFY data * @offset_words: Offset of the string to get, in words. * @dest: Destination buffer for the string. * @dest_len: Length of destination buffer, in bytes. * * Copies the ATA string from @identify located at @offset_words into @dest. */ static void disk_identify_get_string(uint8_t identify[512], unsigned int offset_words, char *dest, size_t dest_len) { unsigned int c1; unsigned int c2; while (dest_len > 0) { c1 = identify[offset_words * 2 + 1]; c2 = identify[offset_words * 2]; *dest = c1; dest++; *dest = c2; dest++; offset_words++; dest_len -= 2; } } static void disk_identify_fixup_string(uint8_t identify[512], unsigned int offset_words, size_t len) { disk_identify_get_string(identify, offset_words, (char *) identify + offset_words * 2, len); } static void disk_identify_fixup_uint16 (uint8_t identify[512], unsigned int offset_words) { uint16_t *p; p = (uint16_t *) identify; p[offset_words] = le16toh (p[offset_words]); } /** * disk_identify: * @udev: The libudev context. * @fd: File descriptor for the block device. * @out_identify: Return location for IDENTIFY data. * @out_is_packet_device: Return location for whether returned data is from a IDENTIFY PACKET DEVICE. * * Sends the IDENTIFY DEVICE or IDENTIFY PACKET DEVICE command to the * device represented by @fd. If successful, then the result will be * copied into @out_identify and @out_is_packet_device. * * This routine is based on code from libatasmart, Copyright 2008 * Lennart Poettering, LGPL v2.1. * * Returns: 0 if the data was successfully obtained, otherwise * non-zero with errno set. */ static int disk_identify(struct udev *udev, int fd, uint8_t out_identify[512], int *out_is_packet_device) { int ret; uint8_t inquiry_buf[36]; int peripheral_device_type; int all_nul_bytes; int n; int is_packet_device; /* init results */ memset(out_identify, '\0', 512); is_packet_device = 0; /* If we were to use ATA PASS_THROUGH (12) on an ATAPI device * we could accidentally blank media. This is because MMC's BLANK * command has the same op-code (0x61). * * To prevent this from happening we bail out if the device * isn't a Direct Access Block Device, e.g. SCSI type 0x00 * (CD/DVD devices are type 0x05). So we send a SCSI INQUIRY * command first... libata is handling this via its SCSI * emulation layer. * * This also ensures that we're actually dealing with a device * that understands SCSI commands. * * (Yes, it is a bit perverse that we're tunneling the ATA * command through SCSI and relying on the ATA driver * emulating SCSI well-enough...) * * (See commit 160b069c25690bfb0c785994c7c3710289179107 for * the original bug-fix and see http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=556635 * for the original bug-report.) */ ret = disk_scsi_inquiry_command (fd, inquiry_buf, sizeof (inquiry_buf)); if (ret != 0) goto out; /* SPC-4, section 6.4.2: Standard INQUIRY data */ peripheral_device_type = inquiry_buf[0] & 0x1f; if (peripheral_device_type == 0x05) { is_packet_device = 1; ret = disk_identify_packet_device_command(fd, out_identify, 512); goto check_nul_bytes; } if (peripheral_device_type != 0x00) { ret = -1; errno = EIO; goto out; } /* OK, now issue the IDENTIFY DEVICE command */ ret = disk_identify_command(fd, out_identify, 512); if (ret != 0) goto out; check_nul_bytes: /* Check if IDENTIFY data is all NUL bytes - if so, bail */ all_nul_bytes = 1; for (n = 0; n < 512; n++) { if (out_identify[n] != '\0') { all_nul_bytes = 0; break; } } if (all_nul_bytes) { ret = -1; errno = EIO; goto out; } out: if (out_is_packet_device != NULL) *out_is_packet_device = is_packet_device; return ret; } static void log_fn(struct udev *udev, int priority, const char *file, int line, const char *fn, const char *format, va_list args) { vsyslog(priority, format, args); } int main(int argc, char *argv[]) { struct udev *udev; struct hd_driveid id; uint8_t identify[512]; uint16_t *identify_words; char model[41]; char model_enc[256]; char serial[21]; char revision[9]; const char *node = NULL; int export = 0; int fd; uint16_t word; int rc = 0; int is_packet_device = 0; static const struct option options[] = { { "export", no_argument, NULL, 'x' }, { "help", no_argument, NULL, 'h' }, {} }; udev = udev_new(); if (udev == NULL) goto exit; log_open(); udev_set_log_fn(udev, log_fn); while (1) { int option; option = getopt_long(argc, argv, "xh", options, NULL); if (option == -1) break; switch (option) { case 'x': export = 1; break; case 'h': printf("Usage: ata_id [--export] [--help] <device>\n" " --export print values as environment keys\n" " --help print this help text\n\n"); goto exit; } } node = argv[optind]; if (node == NULL) { log_error("no node specified\n"); rc = 1; goto exit; } fd = open(node, O_RDONLY|O_NONBLOCK); if (fd < 0) { log_error("unable to open '%s'\n", node); rc = 1; goto exit; } if (disk_identify(udev, fd, identify, &is_packet_device) == 0) { /* * fix up only the fields from the IDENTIFY data that we are going to * use and copy it into the hd_driveid struct for convenience */ disk_identify_fixup_string(identify, 10, 20); /* serial */ disk_identify_fixup_string(identify, 23, 6); /* fwrev */ disk_identify_fixup_string(identify, 27, 40); /* model */ disk_identify_fixup_uint16(identify, 0); /* configuration */ disk_identify_fixup_uint16(identify, 75); /* queue depth */ disk_identify_fixup_uint16(identify, 75); /* SATA capabilities */ disk_identify_fixup_uint16(identify, 82); /* command set supported */ disk_identify_fixup_uint16(identify, 83); /* command set supported */ disk_identify_fixup_uint16(identify, 84); /* command set supported */ disk_identify_fixup_uint16(identify, 85); /* command set supported */ disk_identify_fixup_uint16(identify, 86); /* command set supported */ disk_identify_fixup_uint16(identify, 87); /* command set supported */ disk_identify_fixup_uint16(identify, 89); /* time required for SECURITY ERASE UNIT */ disk_identify_fixup_uint16(identify, 90); /* time required for enhanced SECURITY ERASE UNIT */ disk_identify_fixup_uint16(identify, 91); /* current APM values */ disk_identify_fixup_uint16(identify, 94); /* current AAM value */ disk_identify_fixup_uint16(identify, 128); /* device lock function */ disk_identify_fixup_uint16(identify, 217); /* nominal media rotation rate */ memcpy(&id, identify, sizeof id); } else { /* If this fails, then try HDIO_GET_IDENTITY */ if (ioctl(fd, HDIO_GET_IDENTITY, &id) != 0) { log_info("HDIO_GET_IDENTITY failed for '%s': %m\n", node); rc = 2; goto close; } } identify_words = (uint16_t *) identify; memcpy (model, id.model, 40); model[40] = '\0'; udev_util_encode_string(model, model_enc, sizeof(model_enc)); util_replace_whitespace((char *) id.model, model, 40); util_replace_chars(model, NULL); util_replace_whitespace((char *) id.serial_no, serial, 20); util_replace_chars(serial, NULL); util_replace_whitespace((char *) id.fw_rev, revision, 8); util_replace_chars(revision, NULL); if (export) { /* Set this to convey the disk speaks the ATA protocol */ printf("ID_ATA=1\n"); if ((id.config >> 8) & 0x80) { /* This is an ATAPI device */ switch ((id.config >> 8) & 0x1f) { case 0: printf("ID_TYPE=cd\n"); break; case 1: printf("ID_TYPE=tape\n"); break; case 5: printf("ID_TYPE=cd\n"); break; case 7: printf("ID_TYPE=optical\n"); break; default: printf("ID_TYPE=generic\n"); break; } } else { printf("ID_TYPE=disk\n"); } printf("ID_BUS=ata\n"); printf("ID_MODEL=%s\n", model); printf("ID_MODEL_ENC=%s\n", model_enc); printf("ID_REVISION=%s\n", revision); if (serial[0] != '\0') { printf("ID_SERIAL=%s_%s\n", model, serial); printf("ID_SERIAL_SHORT=%s\n", serial); } else { printf("ID_SERIAL=%s\n", model); } if (id.command_set_1 & (1<<5)) { printf ("ID_ATA_WRITE_CACHE=1\n"); printf ("ID_ATA_WRITE_CACHE_ENABLED=%d\n", (id.cfs_enable_1 & (1<<5)) ? 1 : 0); } if (id.command_set_1 & (1<<10)) { printf("ID_ATA_FEATURE_SET_HPA=1\n"); printf("ID_ATA_FEATURE_SET_HPA_ENABLED=%d\n", (id.cfs_enable_1 & (1<<10)) ? 1 : 0); /* * TODO: use the READ NATIVE MAX ADDRESS command to get the native max address * so it is easy to check whether the protected area is in use. */ } if (id.command_set_1 & (1<<3)) { printf("ID_ATA_FEATURE_SET_PM=1\n"); printf("ID_ATA_FEATURE_SET_PM_ENABLED=%d\n", (id.cfs_enable_1 & (1<<3)) ? 1 : 0); } if (id.command_set_1 & (1<<1)) { printf("ID_ATA_FEATURE_SET_SECURITY=1\n"); printf("ID_ATA_FEATURE_SET_SECURITY_ENABLED=%d\n", (id.cfs_enable_1 & (1<<1)) ? 1 : 0); printf("ID_ATA_FEATURE_SET_SECURITY_ERASE_UNIT_MIN=%d\n", id.trseuc * 2); if ((id.cfs_enable_1 & (1<<1))) /* enabled */ { if (id.dlf & (1<<8)) printf("ID_ATA_FEATURE_SET_SECURITY_LEVEL=maximum\n"); else printf("ID_ATA_FEATURE_SET_SECURITY_LEVEL=high\n"); } if (id.dlf & (1<<5)) printf("ID_ATA_FEATURE_SET_SECURITY_ENHANCED_ERASE_UNIT_MIN=%d\n", id.trsEuc * 2); if (id.dlf & (1<<4)) printf("ID_ATA_FEATURE_SET_SECURITY_EXPIRE=1\n"); if (id.dlf & (1<<3)) printf("ID_ATA_FEATURE_SET_SECURITY_FROZEN=1\n"); if (id.dlf & (1<<2)) printf("ID_ATA_FEATURE_SET_SECURITY_LOCKED=1\n"); } if (id.command_set_1 & (1<<0)) { printf("ID_ATA_FEATURE_SET_SMART=1\n"); printf("ID_ATA_FEATURE_SET_SMART_ENABLED=%d\n", (id.cfs_enable_1 & (1<<0)) ? 1 : 0); } if (id.command_set_2 & (1<<9)) { printf("ID_ATA_FEATURE_SET_AAM=1\n"); printf("ID_ATA_FEATURE_SET_AAM_ENABLED=%d\n", (id.cfs_enable_2 & (1<<9)) ? 1 : 0); printf("ID_ATA_FEATURE_SET_AAM_VENDOR_RECOMMENDED_VALUE=%d\n", id.acoustic >> 8); printf("ID_ATA_FEATURE_SET_AAM_CURRENT_VALUE=%d\n", id.acoustic & 0xff); } if (id.command_set_2 & (1<<5)) { printf("ID_ATA_FEATURE_SET_PUIS=1\n"); printf("ID_ATA_FEATURE_SET_PUIS_ENABLED=%d\n", (id.cfs_enable_2 & (1<<5)) ? 1 : 0); } if (id.command_set_2 & (1<<3)) { printf("ID_ATA_FEATURE_SET_APM=1\n"); printf("ID_ATA_FEATURE_SET_APM_ENABLED=%d\n", (id.cfs_enable_2 & (1<<3)) ? 1 : 0); if ((id.cfs_enable_2 & (1<<3))) printf("ID_ATA_FEATURE_SET_APM_CURRENT_VALUE=%d\n", id.CurAPMvalues & 0xff); } if (id.command_set_2 & (1<<0)) printf("ID_ATA_DOWNLOAD_MICROCODE=1\n"); /* * Word 76 indicates the capabilities of a SATA device. A PATA device shall set * word 76 to 0000h or FFFFh. If word 76 is set to 0000h or FFFFh, then * the device does not claim compliance with the Serial ATA specification and words * 76 through 79 are not valid and shall be ignored. */ word = *((uint16_t *) identify + 76); if (word != 0x0000 && word != 0xffff) { printf("ID_ATA_SATA=1\n"); /* * If bit 2 of word 76 is set to one, then the device supports the Gen2 * signaling rate of 3.0 Gb/s (see SATA 2.6). * * If bit 1 of word 76 is set to one, then the device supports the Gen1 * signaling rate of 1.5 Gb/s (see SATA 2.6). */ if (word & (1<<2)) printf("ID_ATA_SATA_SIGNAL_RATE_GEN2=1\n"); if (word & (1<<1)) printf("ID_ATA_SATA_SIGNAL_RATE_GEN1=1\n"); } /* Word 217 indicates the nominal media rotation rate of the device */ word = *((uint16_t *) identify + 217); if (word != 0x0000) { if (word == 0x0001) { printf ("ID_ATA_ROTATION_RATE_RPM=0\n"); /* non-rotating e.g. SSD */ } else if (word >= 0x0401 && word <= 0xfffe) { printf ("ID_ATA_ROTATION_RATE_RPM=%d\n", word); } } /* * Words 108-111 contain a mandatory World Wide Name (WWN) in the NAA IEEE Registered identifier * format. Word 108 bits (15:12) shall contain 5h, indicating that the naming authority is IEEE. * All other values are reserved. */ word = *((uint16_t *) identify + 108); if ((word & 0xf000) == 0x5000) { uint64_t wwwn; wwwn = *((uint16_t *) identify + 108); wwwn <<= 16; wwwn |= *((uint16_t *) identify + 109); wwwn <<= 16; wwwn |= *((uint16_t *) identify + 110); wwwn <<= 16; wwwn |= *((uint16_t *) identify + 111); printf("ID_WWN=0x%llx\n", (unsigned long long int) wwwn); /* ATA devices have no vendor extension */ printf("ID_WWN_WITH_EXTENSION=0x%llx\n", (unsigned long long int) wwwn); } /* from Linux's include/linux/ata.h */ if (identify_words[0] == 0x848a || identify_words[0] == 0x844a) { printf("ID_ATA_CFA=1\n"); } else { if ((identify_words[83] & 0xc004) == 0x4004) { printf("ID_ATA_CFA=1\n"); } } } else { if (serial[0] != '\0') printf("%s_%s\n", model, serial); else printf("%s\n", model); } close: close(fd); exit: udev_unref(udev); log_close(); return rc; }