summaryrefslogtreecommitdiffstats
path: root/atacmds.cpp
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-15 17:14:45 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-15 17:14:45 +0000
commit43e8530e93493bb978c446a2023134bdd4277e50 (patch)
treee8c0d3c0c394b17381f48fb2d288f166b4f22440 /atacmds.cpp
parentInitial commit. (diff)
downloadsmartmontools-43e8530e93493bb978c446a2023134bdd4277e50.tar.xz
smartmontools-43e8530e93493bb978c446a2023134bdd4277e50.zip
Adding upstream version 7.4.upstream/7.4upstream
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'atacmds.cpp')
-rw-r--r--atacmds.cpp2841
1 files changed, 2841 insertions, 0 deletions
diff --git a/atacmds.cpp b/atacmds.cpp
new file mode 100644
index 0000000..49f33e3
--- /dev/null
+++ b/atacmds.cpp
@@ -0,0 +1,2841 @@
+/*
+ * atacmds.cpp
+ *
+ * Home page of code is: https://www.smartmontools.org
+ *
+ * Copyright (C) 2002-11 Bruce Allen
+ * Copyright (C) 2008-21 Christian Franke
+ * Copyright (C) 1999-2000 Michael Cornwell <cornwell@acm.org>
+ * Copyright (C) 2000 Andre Hedrick <andre@linux-ide.org>
+ *
+ * SPDX-License-Identifier: GPL-2.0-or-later
+ */
+
+#include "config.h"
+#define __STDC_FORMAT_MACROS 1 // enable PRI* for C++
+
+#include <inttypes.h>
+#include <stdio.h>
+#include <string.h>
+#include <errno.h>
+#include <stdlib.h>
+#include <ctype.h>
+
+#include "atacmds.h"
+#include "knowndrives.h" // get_default_attr_defs()
+#include "utility.h"
+#include "dev_ata_cmd_set.h" // for parsed_ata_device
+
+const char * atacmds_cpp_cvsid = "$Id: atacmds.cpp 5456 2023-02-14 18:26:21Z chrfranke $"
+ ATACMDS_H_CVSID;
+
+// Print ATA debug messages?
+unsigned char ata_debugmode = 0;
+
+// Suppress serial number?
+// (also used in scsiprint.cpp)
+bool dont_print_serial_number = false;
+
+
+#define SMART_CYL_LOW 0x4F
+#define SMART_CYL_HI 0xC2
+
+// SMART RETURN STATUS yields SMART_CYL_HI,SMART_CYL_LOW to indicate drive
+// is healthy and SRET_STATUS_HI_EXCEEDED,SRET_STATUS_MID_EXCEEDED to
+// indicate that a threshold exceeded condition has been detected.
+// Those values (byte pairs) are placed in ATA register "LBA 23:8".
+#define SRET_STATUS_HI_EXCEEDED 0x2C
+#define SRET_STATUS_MID_EXCEEDED 0xF4
+
+
+// Get ID and increase flag of current pending or offline
+// uncorrectable attribute.
+unsigned char get_unc_attr_id(bool offline, const ata_vendor_attr_defs & defs,
+ bool & increase)
+{
+ unsigned char id = (!offline ? 197 : 198);
+ const ata_vendor_attr_defs::entry & def = defs[id];
+ if (def.flags & ATTRFLAG_INCREASING)
+ increase = true; // '-v 19[78],increasing' option
+ else if (def.name.empty() || (id == 198 && def.name == "Offline_Scan_UNC_SectCt"))
+ increase = false; // no or '-v 198,offlinescanuncsectorct' option
+ else
+ id = 0; // other '-v 19[78],...' option
+ return id;
+}
+
+#if 0 // TODO: never used
+// This are the meanings of the Self-test failure checkpoint byte.
+// This is in the self-test log at offset 4 bytes into the self-test
+// descriptor and in the SMART READ DATA structure at byte offset
+// 371. These codes are not well documented. The meanings returned by
+// this routine are used (at least) by Maxtor and IBM. Returns NULL if
+// not recognized. Currently the maximum length is 15 bytes.
+const char *SelfTestFailureCodeName(unsigned char which){
+
+ switch (which) {
+ case 0:
+ return "Write_Test";
+ case 1:
+ return "Servo_Basic";
+ case 2:
+ return "Servo_Random";
+ case 3:
+ return "G-list_Scan";
+ case 4:
+ return "Handling_Damage";
+ case 5:
+ return "Read_Scan";
+ default:
+ return NULL;
+ }
+}
+#endif
+
+
+// Table of raw print format names
+struct format_name_entry
+{
+ const char * name;
+ ata_attr_raw_format format;
+};
+
+const format_name_entry format_names[] = {
+ {"raw8" , RAWFMT_RAW8},
+ {"raw16" , RAWFMT_RAW16},
+ {"raw48" , RAWFMT_RAW48},
+ {"hex48" , RAWFMT_HEX48},
+ {"raw56" , RAWFMT_RAW56},
+ {"hex56" , RAWFMT_HEX56},
+ {"raw64" , RAWFMT_RAW64},
+ {"hex64" , RAWFMT_HEX64},
+ {"raw16(raw16)" , RAWFMT_RAW16_OPT_RAW16},
+ {"raw16(avg16)" , RAWFMT_RAW16_OPT_AVG16},
+ {"raw24(raw8)" , RAWFMT_RAW24_OPT_RAW8},
+ {"raw24/raw24" , RAWFMT_RAW24_DIV_RAW24},
+ {"raw24/raw32" , RAWFMT_RAW24_DIV_RAW32},
+ {"sec2hour" , RAWFMT_SEC2HOUR},
+ {"min2hour" , RAWFMT_MIN2HOUR},
+ {"halfmin2hour" , RAWFMT_HALFMIN2HOUR},
+ {"msec24hour32" , RAWFMT_MSEC24_HOUR32},
+ {"tempminmax" , RAWFMT_TEMPMINMAX},
+ {"temp10x" , RAWFMT_TEMP10X},
+};
+
+const unsigned num_format_names = sizeof(format_names)/sizeof(format_names[0]);
+
+// Table to map old to new '-v' option arguments
+const char * const map_old_vendor_opts[][2] = {
+ { "9,halfminutes" , "9,halfmin2hour,Power_On_Half_Minutes"},
+ { "9,minutes" , "9,min2hour,Power_On_Minutes"},
+ { "9,seconds" , "9,sec2hour,Power_On_Seconds"},
+ { "9,temp" , "9,tempminmax,Temperature_Celsius"},
+ {"192,emergencyretractcyclect" , "192,raw48,Emerg_Retract_Cycle_Ct"},
+ {"193,loadunload" , "193,raw24/raw24"},
+ {"194,10xCelsius" , "194,temp10x,Temperature_Celsius_x10"},
+ {"194,unknown" , "194,raw48,Unknown_Attribute"},
+ {"197,increasing" , "197,raw48+,Total_Pending_Sectors"}, // '+' sets flag
+ {"198,offlinescanuncsectorct" , "198,raw48,Offline_Scan_UNC_SectCt"}, // see also get_unc_attr_id() above
+ {"198,increasing" , "198,raw48+,Total_Offl_Uncorrectabl"}, // '+' sets flag
+ {"200,writeerrorcount" , "200,raw48,Write_Error_Count"},
+ {"201,detectedtacount" , "201,raw48,Detected_TA_Count"},
+ {"220,temp" , "220,tempminmax,Temperature_Celsius"},
+};
+
+const unsigned num_old_vendor_opts = sizeof(map_old_vendor_opts)/sizeof(map_old_vendor_opts[0]);
+
+// Parse vendor attribute display def (-v option).
+// Return false on error.
+bool parse_attribute_def(const char * opt, ata_vendor_attr_defs & defs,
+ ata_vendor_def_prior priority)
+{
+ // Map old -> new options
+ unsigned i;
+ for (i = 0; i < num_old_vendor_opts; i++) {
+ if (!strcmp(opt, map_old_vendor_opts[i][0])) {
+ opt = map_old_vendor_opts[i][1];
+ break;
+ }
+ }
+
+ // Parse option
+ int len = strlen(opt);
+ int id = 0, n1 = -1, n2 = -1;
+ char fmtname[32+1], attrname[32+1], hddssd[3+1];
+ attrname[0] = hddssd[0] = 0;
+
+ if (opt[0] == 'N') {
+ // "N,format[,name]"
+ if (!( sscanf(opt, "N,%32[^,]%n,%32[^,]%n", fmtname, &n1, attrname, &n2) >= 1
+ && (n1 == len || n2 == len)))
+ return false;
+ }
+ else {
+ // "id,format[+][,name[,HDD|SSD]]"
+ int n3 = -1;
+ if (!( sscanf(opt, "%d,%32[^,]%n,%32[^,]%n,%3[DHS]%n",
+ &id, fmtname, &n1, attrname, &n2, hddssd, &n3) >= 2
+ && 1 <= id && id <= 255
+ && ( n1 == len || n2 == len
+ // ",HDD|SSD" for DEFAULT settings only
+ || (n3 == len && priority == PRIOR_DEFAULT))))
+ return false;
+ }
+
+ unsigned flags = 0;
+ // For "-v 19[78],increasing" above
+ if (fmtname[strlen(fmtname)-1] == '+') {
+ fmtname[strlen(fmtname)-1] = 0;
+ flags = ATTRFLAG_INCREASING;
+ }
+
+ // Split "format[:byteorder]"
+ char byteorder[8+1] = "";
+ if (strchr(fmtname, ':')) {
+ if (priority == PRIOR_DEFAULT)
+ // TODO: Allow Byteorder in DEFAULT entry
+ return false;
+ n1 = n2 = -1;
+ if (!( sscanf(fmtname, "%*[^:]%n:%8[012345rvwz]%n", &n1, byteorder, &n2) >= 1
+ && n2 == (int)strlen(fmtname)))
+ return false;
+ fmtname[n1] = 0;
+ if (strchr(byteorder, 'v'))
+ flags |= (ATTRFLAG_NO_NORMVAL|ATTRFLAG_NO_WORSTVAL);
+ if (strchr(byteorder, 'w'))
+ flags |= ATTRFLAG_NO_WORSTVAL;
+ }
+
+ // Find format name
+ for (i = 0; ; i++) {
+ if (i >= num_format_names)
+ return false; // Not found
+ if (!strcmp(fmtname, format_names[i].name))
+ break;
+ }
+ ata_attr_raw_format format = format_names[i].format;
+
+ // 64-bit formats use the normalized and worst value bytes.
+ if (!*byteorder && (format == RAWFMT_RAW64 || format == RAWFMT_HEX64))
+ flags |= (ATTRFLAG_NO_NORMVAL|ATTRFLAG_NO_WORSTVAL);
+
+ // ",HDD|SSD" suffix for DEFAULT settings
+ if (hddssd[0]) {
+ if (!strcmp(hddssd, "HDD"))
+ flags |= ATTRFLAG_HDD_ONLY;
+ else if (!strcmp(hddssd, "SSD"))
+ flags |= ATTRFLAG_SSD_ONLY;
+ else
+ return false;
+ }
+
+ if (!id) {
+ // "N,format" -> set format for all entries
+ for (i = 0; i < MAX_ATTRIBUTE_NUM; i++) {
+ if (defs[i].priority >= priority)
+ continue;
+ if (attrname[0])
+ defs[i].name = attrname;
+ defs[i].priority = priority;
+ defs[i].raw_format = format;
+ defs[i].flags = flags;
+ snprintf(defs[i].byteorder, sizeof(defs[i].byteorder), "%s", byteorder);
+ }
+ }
+ else if (defs[id].priority <= priority) {
+ // "id,format[,name]"
+ if (attrname[0])
+ defs[id].name = attrname;
+ defs[id].raw_format = format;
+ defs[id].priority = priority;
+ defs[id].flags = flags;
+ snprintf(defs[id].byteorder, sizeof(defs[id].byteorder), "%s", byteorder);
+ }
+
+ return true;
+}
+
+
+// Return a multiline string containing a list of valid arguments for
+// parse_attribute_def(). The strings are preceded by tabs and followed
+// (except for the last) by newlines.
+std::string create_vendor_attribute_arg_list()
+{
+ std::string s;
+ unsigned i;
+ for (i = 0; i < num_format_names; i++)
+ s += strprintf("%s\tN,%s[:012345rvwz][,ATTR_NAME]",
+ (i>0 ? "\n" : ""), format_names[i].name);
+ for (i = 0; i < num_old_vendor_opts; i++)
+ s += strprintf("\n\t%s", map_old_vendor_opts[i][0]);
+ return s;
+}
+
+
+// Parse firmwarebug def (-F option).
+// Return false on error.
+bool parse_firmwarebug_def(const char * opt, firmwarebug_defs & firmwarebugs)
+{
+ if (!strcmp(opt, "none"))
+ firmwarebugs.set(BUG_NONE);
+ else if (!strcmp(opt, "nologdir"))
+ firmwarebugs.set(BUG_NOLOGDIR);
+ else if (!strcmp(opt, "samsung"))
+ firmwarebugs.set(BUG_SAMSUNG);
+ else if (!strcmp(opt, "samsung2"))
+ firmwarebugs.set(BUG_SAMSUNG2);
+ else if (!strcmp(opt, "samsung3"))
+ firmwarebugs.set(BUG_SAMSUNG3);
+ else if (!strcmp(opt, "xerrorlba"))
+ firmwarebugs.set(BUG_XERRORLBA);
+ else
+ return false;
+ return true;
+}
+
+// Return a string of valid argument words for parse_firmwarebug_def()
+const char * get_valid_firmwarebug_args()
+{
+ return "none, nologdir, samsung, samsung2, samsung3, xerrorlba";
+}
+
+
+// swap two bytes. Point to low address
+void swap2(char *location){
+ char tmp=*location;
+ *location=*(location+1);
+ *(location+1)=tmp;
+ return;
+}
+
+// swap four bytes. Point to low address
+void swap4(char *location){
+ char tmp=*location;
+ *location=*(location+3);
+ *(location+3)=tmp;
+ swap2(location+1);
+ return;
+}
+
+// swap eight bytes. Points to low address
+void swap8(char *location){
+ char tmp=*location;
+ *location=*(location+7);
+ *(location+7)=tmp;
+ tmp=*(location+1);
+ *(location+1)=*(location+6);
+ *(location+6)=tmp;
+ swap4(location+2);
+ return;
+}
+
+// When using the overloaded swapx() function with member of packed ATA structs,
+// it is required to pass a possibly unaligned pointer as argument.
+// Clang++ 4.0 prints -Waddress-of-packed-member warning in this case.
+// The SWAPV() macro below is a replacement which prevents the use of such pointers.
+template <typename T>
+static T get_swapx_val(T x)
+ { swapx(&x); return x; }
+
+#define SWAPV(x) ((x) = get_swapx_val(x))
+
+// Invalidate serial number and WWN and adjust checksum in IDENTIFY data
+static void invalidate_serno(ata_identify_device * id)
+{
+ unsigned char sum = 0;
+ unsigned i;
+ for (i = 0; i < sizeof(id->serial_no); i++) {
+ sum += id->serial_no[i]; sum -= id->serial_no[i] = 'X';
+ }
+ unsigned char * b = (unsigned char *)id;
+ for (i = 2*108; i < 2*112; i++) { // words108-111: WWN
+ sum += b[i]; sum -= b[i] = 0x00;
+ }
+
+ if (isbigendian())
+ SWAPV(id->words088_255[255-88]);
+ if ((id->words088_255[255-88] & 0x00ff) == 0x00a5)
+ id->words088_255[255-88] += sum << 8;
+ if (isbigendian())
+ SWAPV(id->words088_255[255-88]);
+}
+
+static const char * const commandstrings[]={
+ "SMART ENABLE",
+ "SMART DISABLE",
+ "SMART AUTOMATIC ATTRIBUTE SAVE",
+ "SMART IMMEDIATE OFFLINE",
+ "SMART AUTO OFFLINE",
+ "SMART STATUS",
+ "SMART STATUS CHECK",
+ "SMART READ ATTRIBUTE VALUES",
+ "SMART READ ATTRIBUTE THRESHOLDS",
+ "SMART READ LOG",
+ "IDENTIFY DEVICE",
+ "IDENTIFY PACKET DEVICE",
+ "CHECK POWER MODE",
+ "SMART WRITE LOG",
+ "WARNING (UNDEFINED COMMAND -- CONTACT DEVELOPERS AT " PACKAGE_BUGREPORT ")\n"
+};
+
+
+static const char * preg(const ata_register & r, char (& buf)[8])
+{
+ if (!r.is_set())
+ //return "n/a ";
+ return "....";
+ snprintf(buf, sizeof(buf), "0x%02x", r.val());
+ return buf;
+}
+
+static void print_regs(const char * prefix, const ata_in_regs & r, const char * suffix = "\n")
+{
+ char bufs[7][8];
+ pout("%s FR=%s, SC=%s, LL=%s, LM=%s, LH=%s, DEV=%s, CMD=%s%s", prefix,
+ preg(r.features, bufs[0]), preg(r.sector_count, bufs[1]), preg(r.lba_low, bufs[2]),
+ preg(r.lba_mid, bufs[3]), preg(r.lba_high, bufs[4]), preg(r.device, bufs[5]),
+ preg(r.command, bufs[6]), suffix);
+}
+
+static void print_regs(const char * prefix, const ata_out_regs & r, const char * suffix = "\n")
+{
+ char bufs[7][8];
+ pout("%sERR=%s, SC=%s, LL=%s, LM=%s, LH=%s, DEV=%s, STS=%s%s", prefix,
+ preg(r.error, bufs[0]), preg(r.sector_count, bufs[1]), preg(r.lba_low, bufs[2]),
+ preg(r.lba_mid, bufs[3]), preg(r.lba_high, bufs[4]), preg(r.device, bufs[5]),
+ preg(r.status, bufs[6]), suffix);
+}
+
+static void prettyprint(const unsigned char *p, const char *name){
+ pout("\n===== [%s] DATA START (BASE-16) =====\n", name);
+ for (int i=0; i<512; i+=16, p+=16)
+#define P(n) (' ' <= p[n] && p[n] <= '~' ? (int)p[n] : '.')
+ // print complete line to avoid slow tty output and extra lines in syslog.
+ pout("%03d-%03d: %02x %02x %02x %02x %02x %02x %02x %02x "
+ "%02x %02x %02x %02x %02x %02x %02x %02x"
+ " |%c%c%c%c%c%c%c%c%c%c%c%c%c%c%c%c|"
+ "%c",
+ i, i+16-1,
+ p[ 0], p[ 1], p[ 2], p[ 3], p[ 4], p[ 5], p[ 6], p[ 7],
+ p[ 8], p[ 9], p[10], p[11], p[12], p[13], p[14], p[15],
+ P( 0), P( 1), P( 2), P( 3), P( 4), P( 5), P( 6), P( 7),
+ P( 8), P( 9), P(10), P(11), P(12), P(13), P(14), P(15),
+ '\n');
+#undef P
+ pout("===== [%s] DATA END (512 Bytes) =====\n\n", name);
+}
+
+// This function provides the pretty-print reporting for SMART
+// commands: it implements the various -r "reporting" options for ATA
+// ioctls.
+int smartcommandhandler(ata_device * device, smart_command_set command, int select, char *data){
+ // TODO: Rework old stuff below
+ // This conditional is true for commands that return data
+ int getsdata=(command==PIDENTIFY ||
+ command==IDENTIFY ||
+ command==READ_LOG ||
+ command==READ_THRESHOLDS ||
+ command==READ_VALUES ||
+ command==CHECK_POWER_MODE);
+
+ int sendsdata=(command==WRITE_LOG);
+
+ // If reporting is enabled, say what the command will be before it's executed
+ if (ata_debugmode) {
+ // conditional is true for commands that use parameters
+ int usesparam=(command==READ_LOG ||
+ command==AUTO_OFFLINE ||
+ command==AUTOSAVE ||
+ command==IMMEDIATE_OFFLINE ||
+ command==WRITE_LOG);
+
+ pout("\nREPORT-IOCTL: Device=%s Command=%s", device->get_dev_name(), commandstrings[command]);
+ if (usesparam)
+ pout(" InputParameter=%d\n", select);
+ else
+ pout("\n");
+ }
+
+ if ((getsdata || sendsdata) && !data){
+ pout("REPORT-IOCTL: Unable to execute command %s : data destination address is NULL\n", commandstrings[command]);
+ return -1;
+ }
+
+ // The reporting is cleaner, and we will find coding bugs faster, if
+ // the commands that failed clearly return empty (zeroed) data
+ // structures
+ if (getsdata) {
+ if (command==CHECK_POWER_MODE)
+ data[0]=0;
+ else
+ memset(data, '\0', 512);
+ }
+
+
+ // if requested, pretty-print the input data structure
+ if (ata_debugmode > 1 && sendsdata)
+ //pout("REPORT-IOCTL: Device=%s Command=%s\n", device->get_dev_name(), commandstrings[command]);
+ prettyprint((unsigned char *)data, commandstrings[command]);
+
+ // now execute the command
+ int retval = -1;
+ {
+ ata_cmd_in in;
+ // Set common register values
+ switch (command) {
+ default: // SMART commands
+ in.in_regs.command = ATA_SMART_CMD;
+ in.in_regs.lba_high = SMART_CYL_HI; in.in_regs.lba_mid = SMART_CYL_LOW;
+ break;
+ case IDENTIFY: case PIDENTIFY: case CHECK_POWER_MODE: // Non SMART commands
+ break;
+ }
+ // Set specific values
+ switch (command) {
+ case IDENTIFY:
+ in.in_regs.command = ATA_IDENTIFY_DEVICE;
+ in.set_data_in(data, 1);
+ break;
+ case PIDENTIFY:
+ in.in_regs.command = ATA_IDENTIFY_PACKET_DEVICE;
+ in.set_data_in(data, 1);
+ break;
+ case CHECK_POWER_MODE:
+ in.in_regs.command = ATA_CHECK_POWER_MODE;
+ in.out_needed.sector_count = true; // Powermode returned here
+ break;
+ case READ_VALUES:
+ in.in_regs.features = ATA_SMART_READ_VALUES;
+ in.set_data_in(data, 1);
+ break;
+ case READ_THRESHOLDS:
+ in.in_regs.features = ATA_SMART_READ_THRESHOLDS;
+ in.in_regs.lba_low = 1; // TODO: CORRECT ???
+ in.set_data_in(data, 1);
+ break;
+ case READ_LOG:
+ in.in_regs.features = ATA_SMART_READ_LOG_SECTOR;
+ in.in_regs.lba_low = select;
+ in.set_data_in(data, 1);
+ break;
+ case WRITE_LOG:
+ in.in_regs.features = ATA_SMART_WRITE_LOG_SECTOR;
+ in.in_regs.lba_low = select;
+ in.set_data_out(data, 1);
+ break;
+ case ENABLE:
+ in.in_regs.features = ATA_SMART_ENABLE;
+ in.in_regs.lba_low = 1; // TODO: CORRECT ???
+ break;
+ case DISABLE:
+ in.in_regs.features = ATA_SMART_DISABLE;
+ in.in_regs.lba_low = 1; // TODO: CORRECT ???
+ break;
+ case STATUS_CHECK:
+ in.out_needed.lba_high = in.out_needed.lba_mid = true; // Status returned here
+ /* FALLTHRU */
+ case STATUS:
+ in.in_regs.features = ATA_SMART_STATUS;
+ break;
+ case AUTO_OFFLINE:
+ in.in_regs.features = ATA_SMART_AUTO_OFFLINE;
+ in.in_regs.sector_count = select; // Caution: Non-DATA command!
+ break;
+ case AUTOSAVE:
+ in.in_regs.features = ATA_SMART_AUTOSAVE;
+ in.in_regs.sector_count = select; // Caution: Non-DATA command!
+ break;
+ case IMMEDIATE_OFFLINE:
+ in.in_regs.features = ATA_SMART_IMMEDIATE_OFFLINE;
+ in.in_regs.lba_low = select;
+ break;
+ default:
+ pout("Unrecognized command %d in smartcommandhandler()\n"
+ "Please contact " PACKAGE_BUGREPORT "\n", command);
+ device->set_err(ENOSYS);
+ return -1;
+ }
+
+ if (ata_debugmode)
+ print_regs(" Input: ", in.in_regs,
+ (in.direction==ata_cmd_in::data_in ? " IN\n":
+ in.direction==ata_cmd_in::data_out ? " OUT\n":"\n"));
+
+ ata_cmd_out out;
+
+ auto start_usec = (ata_debugmode ? get_timer_usec() : -1);
+
+ bool ok = device->ata_pass_through(in, out);
+
+ if (start_usec >= 0) {
+ auto duration_usec = get_timer_usec() - start_usec;
+ if (duration_usec > 0)
+ pout(" [Duration: %.6fs]\n", duration_usec / 1000000.0);
+ }
+
+ if (ata_debugmode && out.out_regs.is_set())
+ print_regs(" Output: ", out.out_regs);
+
+ if (ok) switch (command) {
+ default:
+ retval = 0;
+ break;
+ case CHECK_POWER_MODE:
+ if (out.out_regs.sector_count.is_set()) {
+ data[0] = out.out_regs.sector_count;
+ retval = 0;
+ }
+ else {
+ pout("CHECK POWER MODE: incomplete response, ATA output registers missing\n");
+ device->set_err(ENOSYS);
+ retval = -1;
+ }
+ break;
+ case STATUS_CHECK:
+ // Cyl low and Cyl high unchanged means "Good SMART status"
+ if ((out.out_regs.lba_high == SMART_CYL_HI) &&
+ (out.out_regs.lba_mid == SMART_CYL_LOW))
+ retval = 0;
+ // These values mean "Bad SMART status"
+ else if ((out.out_regs.lba_high == SRET_STATUS_HI_EXCEEDED) &&
+ (out.out_regs.lba_mid == SRET_STATUS_MID_EXCEEDED))
+ retval = 1;
+ else if (out.out_regs.lba_mid == SMART_CYL_LOW) {
+ retval = 0;
+ if (ata_debugmode)
+ pout("SMART STATUS RETURN: half healthy response sequence, "
+ "probable SAT/USB truncation\n");
+ } else if (out.out_regs.lba_mid == SRET_STATUS_MID_EXCEEDED) {
+ retval = 1;
+ if (ata_debugmode)
+ pout("SMART STATUS RETURN: half unhealthy response sequence, "
+ "probable SAT/USB truncation\n");
+ }
+ else if (!out.out_regs.is_set()) {
+ device->set_err(ENOSYS, "Incomplete response, ATA output registers missing");
+ retval = -1;
+ }
+ else {
+ // We haven't gotten output that makes sense; print out some debugging info
+ pout("SMART Status command failed\n");
+ pout("Please get assistance from %s\n", PACKAGE_URL);
+ pout("Register values returned from SMART Status command are:\n");
+ print_regs(" ", out.out_regs);
+ device->set_err(ENOSYS, "Invalid ATA output register values");
+ retval = -1;
+ }
+ break;
+ }
+ }
+
+ // If requested, invalidate serial number before any printing is done
+ if ((command == IDENTIFY || command == PIDENTIFY) && !retval && dont_print_serial_number)
+ invalidate_serno( reinterpret_cast<ata_identify_device *>(data) );
+
+ // If reporting is enabled, say what output was produced by the command
+ if (ata_debugmode) {
+ if (retval && device->get_errno())
+ pout("REPORT-IOCTL: Device=%s Command=%s returned %d errno=%d [%s]\n",
+ device->get_dev_name(), commandstrings[command], retval,
+ device->get_errno(), device->get_errmsg());
+ else
+ pout("REPORT-IOCTL: Device=%s Command=%s returned %d\n",
+ device->get_dev_name(), commandstrings[command], retval);
+
+ // if requested, pretty-print the output data structure
+ if (ata_debugmode > 1 && getsdata) {
+ if (command==CHECK_POWER_MODE)
+ pout("Sector Count Register (BASE-16): %02x\n", (unsigned char)(*data));
+ else
+ prettyprint((unsigned char *)data, commandstrings[command]);
+ }
+ }
+
+ return retval;
+}
+
+// Get capacity and sector sizes from IDENTIFY data
+void ata_get_size_info(const ata_identify_device * id, ata_size_info & sizes)
+{
+ sizes.sectors = sizes.capacity = 0;
+ sizes.log_sector_size = sizes.phy_sector_size = 0;
+ sizes.log_sector_offset = 0;
+
+ // Return if no LBA support
+ if (!(id->words047_079[49-47] & 0x0200))
+ return;
+
+ // Determine 28-bit LBA capacity
+ unsigned lba28 = (unsigned)id->words047_079[61-47] << 16
+ | (unsigned)id->words047_079[60-47] ;
+
+ // Determine 48-bit LBA capacity if supported
+ uint64_t lba48 = 0;
+ if ((id->command_set_2 & 0xc400) == 0x4400)
+ lba48 = (uint64_t)id->words088_255[103-88] << 48
+ | (uint64_t)id->words088_255[102-88] << 32
+ | (uint64_t)id->words088_255[101-88] << 16
+ | (uint64_t)id->words088_255[100-88] ;
+
+ // Return if capacity unknown (ATAPI CD/DVD)
+ if (!(lba28 || lba48))
+ return;
+
+ // Determine sector sizes
+ sizes.log_sector_size = sizes.phy_sector_size = 512;
+
+ unsigned short word106 = id->words088_255[106-88];
+ if ((word106 & 0xc000) == 0x4000) {
+ // Long Logical/Physical Sectors (LLS/LPS) ?
+ if (word106 & 0x1000)
+ // Logical sector size is specified in 16-bit words
+ sizes.log_sector_size = sizes.phy_sector_size =
+ ((id->words088_255[118-88] << 16) | id->words088_255[117-88]) << 1;
+
+ if (word106 & 0x2000)
+ // Physical sector size is multiple of logical sector size
+ sizes.phy_sector_size <<= (word106 & 0x0f);
+
+ unsigned short word209 = id->words088_255[209-88];
+ if ((word209 & 0xc000) == 0x4000)
+ sizes.log_sector_offset = (word209 & 0x3fff) * sizes.log_sector_size;
+ }
+
+ // Some early 4KiB LLS disks (Samsung N3U-3) return bogus lba28 value
+ if (lba48 >= lba28 || (lba48 && sizes.log_sector_size > 512))
+ sizes.sectors = lba48;
+ else
+ sizes.sectors = lba28;
+
+ sizes.capacity = sizes.sectors * sizes.log_sector_size;
+}
+
+// This function computes the checksum of a single disk sector (512
+// bytes). Returns zero if checksum is OK, nonzero if the checksum is
+// incorrect. The size (512) is correct for all SMART structures.
+unsigned char checksum(const void * data)
+{
+ unsigned char sum = 0;
+ for (int i = 0; i < 512; i++)
+ sum += ((const unsigned char *)data)[i];
+ return sum;
+}
+
+// Copies n bytes (or n-1 if n is odd) from in to out, but swaps adjacents
+// bytes.
+static void swapbytes(char * out, const char * in, size_t n)
+{
+ for (size_t i = 0; i < n; i += 2) {
+ out[i] = in[i+1];
+ out[i+1] = in[i];
+ }
+}
+
+// Copies in to out, but removes leading and trailing whitespace.
+static void trim(char * out, const char * in)
+{
+ // Find the first non-space character (maybe none).
+ int first = -1;
+ int i;
+ for (i = 0; in[i]; i++)
+ if (!isspace((int)in[i])) {
+ first = i;
+ break;
+ }
+
+ if (first == -1) {
+ // There are no non-space characters.
+ out[0] = '\0';
+ return;
+ }
+
+ // Find the last non-space character.
+ for (i = strlen(in)-1; i >= first && isspace((int)in[i]); i--)
+ ;
+ int last = i;
+
+ strncpy(out, in+first, last-first+1);
+ out[last-first+1] = '\0';
+}
+
+// Convenience function for formatting strings from ata_identify_device
+void ata_format_id_string(char * out, const unsigned char * in, int n)
+{
+ char tmp[65];
+ n = n > 64 ? 64 : n;
+ swapbytes(tmp, (const char *)in, n);
+ tmp[n] = '\0';
+ trim(out, tmp);
+}
+
+// returns -1 if command fails or the device is in Sleep mode, else
+// value of Sector Count register. Sector Count result values:
+// 00h device is in Standby mode.
+// 80h device is in Idle mode.
+// FFh device is in Active mode or Idle mode.
+
+int ataCheckPowerMode(ata_device * device) {
+ unsigned char result;
+
+ if ((smartcommandhandler(device, CHECK_POWER_MODE, 0, (char *)&result)))
+ return -1;
+
+ return (int)result;
+}
+
+// Issue a no-data ATA command with optional sector count register value
+bool ata_nodata_command(ata_device * device, unsigned char command,
+ int sector_count /* = -1 */)
+{
+ ata_cmd_in in;
+ in.in_regs.command = command;
+ if (sector_count >= 0)
+ in.in_regs.sector_count = sector_count;
+
+ return device->ata_pass_through(in);
+}
+
+// Issue SET FEATURES command with optional sector count register value
+bool ata_set_features(ata_device * device, unsigned char features,
+ int sector_count /* = -1 */)
+{
+ ata_cmd_in in;
+ in.in_regs.command = ATA_SET_FEATURES;
+ in.in_regs.features = features;
+ if (sector_count >= 0)
+ in.in_regs.sector_count = sector_count;
+
+ return device->ata_pass_through(in);
+}
+
+// Reads current Device Identity info (512 bytes) into buf. Returns 0
+// if all OK. Returns -1 if no ATA Device identity can be
+// established. Returns >0 if Device is ATA Packet Device (not SMART
+// capable). The value of the integer helps identify the type of
+// Packet device, which is useful so that the user can connect the
+// formal device number with whatever object is inside their computer.
+int ata_read_identity(ata_device * device, ata_identify_device * buf, bool fix_swapped_id,
+ unsigned char * raw_buf /* = 0 */)
+{
+ // See if device responds either to IDENTIFY DEVICE or IDENTIFY
+ // PACKET DEVICE
+ bool packet = false;
+ if ((smartcommandhandler(device, IDENTIFY, 0, (char *)buf))){
+ smart_device::error_info err = device->get_err();
+ if (smartcommandhandler(device, PIDENTIFY, 0, (char *)buf)){
+ device->set_err(err);
+ return -1;
+ }
+ packet = true;
+ }
+
+ if (fix_swapped_id) {
+ // Swap ID strings
+ unsigned i;
+ for (i = 0; i < sizeof(buf->serial_no)-1; i += 2)
+ swap2((char *)(buf->serial_no+i));
+ for (i = 0; i < sizeof(buf->fw_rev)-1; i += 2)
+ swap2((char *)(buf->fw_rev+i));
+ for (i = 0; i < sizeof(buf->model)-1; i += 2)
+ swap2((char *)(buf->model+i));
+ }
+
+ // If requested, save raw data before endianness adjustments
+ if (raw_buf)
+ memcpy(raw_buf, buf, sizeof(*buf));
+
+ // If there is a checksum there, validate it
+ unsigned char * rawbyte = (unsigned char *)buf;
+ if (rawbyte[512-2] == 0xa5 && checksum(rawbyte))
+ checksumwarning("Drive Identity Structure");
+
+ // if machine is big-endian, swap byte order as needed
+ if (isbigendian()){
+ // swap various capability words that are needed
+ unsigned i;
+ for (i=0; i<33; i++)
+ swap2((char *)(buf->words047_079+i));
+ for (i=80; i<=87; i++)
+ swap2((char *)(rawbyte+2*i));
+ for (i=0; i<168; i++)
+ swap2((char *)(buf->words088_255+i));
+ }
+
+ // AT Attachment 8 - ATA/ATAPI Command Set (ATA8-ACS)
+ // T13/1699-D Revision 6a (Final Draft), September 6, 2008.
+ // Sections 7.16.7 and 7.17.6:
+ //
+ // Word 0 of IDENTIFY DEVICE data:
+ // Bit 15 = 0 : ATA device
+ //
+ // Word 0 of IDENTIFY PACKET DEVICE data:
+ // Bits 15:14 = 10b : ATAPI device
+ // Bits 15:14 = 11b : Reserved
+ // Bits 12:8 : Device type (SPC-4, e.g 0x05 = CD/DVD)
+
+ // CF+ and CompactFlash Specification Revision 4.0, May 24, 2006.
+ // Section 6.2.1.6:
+ //
+ // Word 0 of IDENTIFY DEVICE data:
+ // 848Ah = Signature for CompactFlash Storage Card
+ // 044Ah = Alternate value turns on ATA device while preserving all retired bits
+ // 0040h = Alternate value turns on ATA device while zeroing all retired bits
+
+ // Assume ATA if IDENTIFY DEVICE returns CompactFlash Signature
+ if (!packet && rawbyte[1] == 0x84 && rawbyte[0] == 0x8a)
+ return 0;
+
+ // If this is a PACKET DEVICE, return device type
+ if (rawbyte[1] & 0x80)
+ return 1+(rawbyte[1] & 0x1f);
+
+ // Not a PACKET DEVICE
+ return 0;
+}
+
+// Get World Wide Name (WWN) fields.
+// Return NAA field or -1 if WWN is unsupported.
+// Table 34 of T13/1699-D Revision 6a (ATA8-ACS), September 6, 2008.
+// (WWN was introduced in ATA/ATAPI-7 and is mandatory since ATA8-ACS Revision 3b)
+int ata_get_wwn(const ata_identify_device * id, unsigned & oui, uint64_t & unique_id)
+{
+ // Don't use word 84 to be compatible with some older ATA-7 disks
+ unsigned short word087 = id->csf_default;
+ if ((word087 & 0xc100) != 0x4100)
+ return -1; // word not valid or WWN support bit 8 not set
+
+ unsigned short word108 = id->words088_255[108-88];
+ unsigned short word109 = id->words088_255[109-88];
+ unsigned short word110 = id->words088_255[110-88];
+ unsigned short word111 = id->words088_255[111-88];
+
+ oui = ((word108 & 0x0fff) << 12) | (word109 >> 4);
+ unique_id = ((uint64_t)(word109 & 0xf) << 32)
+ | (unsigned)((word110 << 16) | word111);
+ return (word108 >> 12);
+}
+
+// Get nominal media rotation rate.
+// Returns: 0 = not reported, 1 = SSD, >1 = HDD rpm, < 0 = -(Unknown value)
+int ata_get_rotation_rate(const ata_identify_device * id)
+{
+ // Table 37 of T13/1699-D (ATA8-ACS) Revision 6a, September 6, 2008
+ // Table A.31 of T13/2161-D (ACS-3) Revision 3b, August 25, 2012
+ unsigned short word217 = id->words088_255[217-88];
+ if (word217 == 0x0000 || word217 == 0xffff)
+ return 0;
+ else if (word217 == 0x0001)
+ return 1;
+ else if (word217 > 0x0400)
+ return word217;
+ else
+ return -(int)word217;
+}
+
+// returns 1 if SMART supported, 0 if SMART unsupported, -1 if can't tell
+int ataSmartSupport(const ata_identify_device * drive)
+{
+ unsigned short word82=drive->command_set_1;
+ unsigned short word83=drive->command_set_2;
+
+ // check if words 82/83 contain valid info
+ if ((word83>>14) == 0x01)
+ // return value of SMART support bit
+ return word82 & 0x0001;
+
+ // since we can're rely on word 82, we don't know if SMART supported
+ return -1;
+}
+
+// returns 1 if SMART enabled, 0 if SMART disabled, -1 if can't tell
+int ataIsSmartEnabled(const ata_identify_device * drive)
+{
+ unsigned short word85=drive->cfs_enable_1;
+ unsigned short word87=drive->csf_default;
+
+ // check if words 85/86/87 contain valid info
+ if ((word87>>14) == 0x01)
+ // return value of SMART enabled bit
+ return word85 & 0x0001;
+
+ // Since we can't rely word85, we don't know if SMART is enabled.
+ return -1;
+}
+
+
+// Reads SMART attributes into *data
+int ataReadSmartValues(ata_device * device, struct ata_smart_values *data){
+
+ if (smartcommandhandler(device, READ_VALUES, 0, (char *)data)){
+ return -1;
+ }
+
+ // compute checksum
+ if (checksum(data))
+ checksumwarning("SMART Attribute Data Structure");
+
+ // swap endian order if needed
+ if (isbigendian()){
+ int i;
+ swap2((char *)&(data->revnumber));
+ swap2((char *)&(data->total_time_to_complete_off_line));
+ swap2((char *)&(data->smart_capability));
+ SWAPV(data->extend_test_completion_time_w);
+ for (i=0; i<NUMBER_ATA_SMART_ATTRIBUTES; i++){
+ struct ata_smart_attribute *x=data->vendor_attributes+i;
+ swap2((char *)&(x->flags));
+ }
+ }
+
+ return 0;
+}
+
+
+// This corrects some quantities that are byte reversed in the SMART
+// SELF TEST LOG
+static void fixsamsungselftestlog(ata_smart_selftestlog * data)
+{
+ // bytes 508/509 (numbered from 0) swapped (swap of self-test index
+ // with one byte of reserved.
+ swap2((char *)&(data->mostrecenttest));
+
+ // LBA low register (here called 'selftestnumber", containing
+ // information about the TYPE of the self-test) is byte swapped with
+ // Self-test execution status byte. These are bytes N, N+1 in the
+ // entries.
+ for (int i = 0; i < 21; i++)
+ swap2((char *)&(data->selftest_struct[i].selftestnumber));
+
+ return;
+}
+
+// Reads the Self Test Log (log #6)
+int ataReadSelfTestLog (ata_device * device, ata_smart_selftestlog * data,
+ firmwarebug_defs firmwarebugs)
+{
+
+ // get data from device
+ if (smartcommandhandler(device, READ_LOG, 0x06, (char *)data)){
+ return -1;
+ }
+
+ // compute its checksum, and issue a warning if needed
+ if (checksum(data))
+ checksumwarning("SMART Self-Test Log Structure");
+
+ // fix firmware bugs in self-test log
+ if (firmwarebugs.is_set(BUG_SAMSUNG))
+ fixsamsungselftestlog(data);
+
+ // swap endian order if needed
+ if (isbigendian()){
+ int i;
+ swap2((char*)&(data->revnumber));
+ for (i=0; i<21; i++){
+ struct ata_smart_selftestlog_struct *x=data->selftest_struct+i;
+ swap2((char *)&(x->timestamp));
+ swap4((char *)&(x->lbafirstfailure));
+ }
+ }
+
+ return 0;
+}
+
+// Print checksum warning for multi sector log
+static void check_multi_sector_sum(const void * data, unsigned nsectors, const char * msg)
+{
+ unsigned errs = 0;
+ for (unsigned i = 0; i < nsectors; i++) {
+ if (checksum((const unsigned char *)data + i*512))
+ errs++;
+ }
+ if (errs > 0) {
+ if (nsectors == 1)
+ checksumwarning(msg);
+ else
+ checksumwarning(strprintf("%s (%u/%u)", msg, errs, nsectors).c_str());
+ }
+}
+
+// Read SMART Extended Self-test Log
+bool ataReadExtSelfTestLog(ata_device * device, ata_smart_extselftestlog * log,
+ unsigned nsectors)
+{
+ if (!ataReadLogExt(device, 0x07, 0x00, 0, log, nsectors))
+ return false;
+
+ check_multi_sector_sum(log, nsectors, "SMART Extended Self-test Log Structure");
+
+ if (isbigendian()) {
+ for (unsigned i = 0; i < nsectors; i++) {
+ SWAPV(log[i].log_desc_index);
+ for (unsigned j = 0; j < 19; j++)
+ SWAPV(log[i].log_descs[j].timestamp);
+ }
+ }
+ return true;
+}
+
+// Write GP Log page(s)
+bool ataWriteLogExt(ata_device * device, unsigned char logaddr,
+ unsigned page, void * data, unsigned nsectors)
+{
+ ata_cmd_in in;
+ in.in_regs.command = ATA_WRITE_LOG_EXT;
+ in.set_data_out(data, nsectors);
+ in.in_regs.lba_low = logaddr;
+ in.in_regs.lba_mid_16 = page;
+ in.set_data_out(data, nsectors);
+
+ ata_cmd_out out;
+ if (!device->ata_pass_through(in, out)) { // TODO: Debug output
+ if (nsectors <= 1) {
+ pout("ATA_WRITE_LOG_EXT (addr=0x%02x, page=%u, n=%u) failed: %s\n",
+ logaddr, page, nsectors, device->get_errmsg());
+ return false;
+ }
+
+ // Recurse to retry with single sectors,
+ // multi-sector reads may not be supported by ioctl.
+ for (unsigned i = 0; i < nsectors; i++) {
+ if (!ataWriteLogExt(device, logaddr, page + i,
+ (char *)data + 512*i, 1))
+ return false;
+ }
+ }
+
+ return true;
+}
+
+// Read GP Log page(s)
+bool ataReadLogExt(ata_device * device, unsigned char logaddr,
+ unsigned char features, unsigned page,
+ void * data, unsigned nsectors)
+{
+ ata_cmd_in in;
+ in.in_regs.command = ATA_READ_LOG_EXT;
+ in.in_regs.features = features; // log specific
+ in.set_data_in_48bit(data, nsectors);
+ in.in_regs.lba_low = logaddr;
+ in.in_regs.lba_mid_16 = page;
+
+ if (!device->ata_pass_through(in)) { // TODO: Debug output
+ if (nsectors <= 1) {
+ pout("ATA_READ_LOG_EXT (addr=0x%02x:0x%02x, page=%u, n=%u) failed: %s\n",
+ logaddr, features, page, nsectors, device->get_errmsg());
+ return false;
+ }
+
+ // Recurse to retry with single sectors,
+ // multi-sector reads may not be supported by ioctl.
+ for (unsigned i = 0; i < nsectors; i++) {
+ if (!ataReadLogExt(device, logaddr,
+ features, page + i,
+ (char *)data + 512*i, 1))
+ return false;
+ }
+ }
+
+ return true;
+}
+
+// Read SMART Log page(s)
+bool ataReadSmartLog(ata_device * device, unsigned char logaddr,
+ void * data, unsigned nsectors)
+{
+ ata_cmd_in in;
+ in.in_regs.command = ATA_SMART_CMD;
+ in.in_regs.features = ATA_SMART_READ_LOG_SECTOR;
+ in.set_data_in(data, nsectors);
+ in.in_regs.lba_high = SMART_CYL_HI;
+ in.in_regs.lba_mid = SMART_CYL_LOW;
+ in.in_regs.lba_low = logaddr;
+
+ if (!device->ata_pass_through(in)) { // TODO: Debug output
+ pout("ATA_SMART_READ_LOG failed: %s\n", device->get_errmsg());
+ return false;
+ }
+ return true;
+}
+
+
+
+// Reads the SMART or GPL Log Directory (log #0)
+int ataReadLogDirectory(ata_device * device, ata_smart_log_directory * data, bool gpl)
+{
+ if (!gpl) { // SMART Log directory
+ if (smartcommandhandler(device, READ_LOG, 0x00, (char *)data))
+ return -1;
+ }
+ else { // GP Log directory
+ if (!ataReadLogExt(device, 0x00, 0x00, 0, data, 1))
+ return -1;
+ }
+
+ // swap endian order if needed
+ if (isbigendian())
+ SWAPV(data->logversion);
+
+ return 0;
+}
+
+
+// Reads the selective self-test log (log #9)
+int ataReadSelectiveSelfTestLog(ata_device * device, struct ata_selective_self_test_log *data){
+
+ // get data from device
+ if (smartcommandhandler(device, READ_LOG, 0x09, (char *)data)){
+ return -1;
+ }
+
+ // compute its checksum, and issue a warning if needed
+ if (checksum(data))
+ checksumwarning("SMART Selective Self-Test Log Structure");
+
+ // swap endian order if needed
+ if (isbigendian()){
+ int i;
+ swap2((char *)&(data->logversion));
+ for (i=0;i<5;i++){
+ swap8((char *)&(data->span[i].start));
+ swap8((char *)&(data->span[i].end));
+ }
+ swap8((char *)&(data->currentlba));
+ swap2((char *)&(data->currentspan));
+ swap2((char *)&(data->flags));
+ swap2((char *)&(data->pendingtime));
+ }
+
+ return 0;
+}
+
+// Writes the selective self-test log (log #9)
+int ataWriteSelectiveSelfTestLog(ata_device * device, ata_selective_selftest_args & args,
+ const ata_smart_values * sv, uint64_t num_sectors,
+ const ata_selective_selftest_args * prev_args)
+{
+ // Disk size must be known
+ if (!num_sectors) {
+ pout("Disk size is unknown, unable to check selective self-test spans\n");
+ return -1;
+ }
+
+ // Read log
+ struct ata_selective_self_test_log sstlog, *data=&sstlog;
+ unsigned char *ptr=(unsigned char *)data;
+ if (ataReadSelectiveSelfTestLog(device, data)) {
+ pout("SMART Read Selective Self-test Log failed: %s\n", device->get_errmsg());
+ pout("Since Read failed, will not attempt to WRITE Selective Self-test Log\n");
+ return -1;
+ }
+
+ // Set log version
+ data->logversion = 1;
+
+ // Host is NOT allowed to write selective self-test log if a selective
+ // self-test is in progress.
+ if (0<data->currentspan && data->currentspan<6 && ((sv->self_test_exec_status)>>4)==15) {
+ pout("SMART Selective or other Self-test in progress\n");
+ return -4;
+ }
+
+ // Set start/end values based on old spans for special -t select,... options
+ int i;
+ for (i = 0; i < args.num_spans; i++) {
+ int mode = args.span[i].mode;
+ uint64_t start = args.span[i].start;
+ uint64_t end = args.span[i].end;
+ if (mode == SEL_CONT) {// redo or next depending on last test status
+ switch (sv->self_test_exec_status >> 4) {
+ case 1: case 2: // Aborted/Interrupted by host
+ pout("Continue Selective Self-Test: Redo last span\n");
+ mode = SEL_REDO;
+ break;
+ default: // All others
+ pout("Continue Selective Self-Test: Start next span\n");
+ mode = SEL_NEXT;
+ break;
+ }
+ }
+
+ if ( (mode == SEL_REDO || mode == SEL_NEXT)
+ && prev_args && i < prev_args->num_spans
+ && !data->span[i].start && !data->span[i].end) {
+ // Some drives do not preserve the selective self-test log across
+ // power-cyles. If old span on drive is cleared use span provided
+ // by caller. This is used by smartd (first span only).
+ data->span[i].start = prev_args->span[i].start;
+ data->span[i].end = prev_args->span[i].end;
+ }
+
+ switch (mode) {
+ case SEL_RANGE: // -t select,START-END
+ break;
+ case SEL_REDO: // -t select,redo... => Redo current
+ start = data->span[i].start;
+ if (end > 0) { // -t select,redo+SIZE
+ end--; end += start; // [oldstart, oldstart+SIZE)
+ }
+ else // -t select,redo
+ end = data->span[i].end; // [oldstart, oldend]
+ break;
+ case SEL_NEXT: // -t select,next... => Do next
+ if (data->span[i].end == 0) {
+ start = end = 0; break; // skip empty spans
+ }
+ start = data->span[i].end + 1;
+ if (start >= num_sectors)
+ start = 0; // wrap around
+ if (end > 0) { // -t select,next+SIZE
+ end--; end += start; // (oldend, oldend+SIZE]
+ }
+ else { // -t select,next
+ uint64_t oldsize = data->span[i].end - data->span[i].start + 1;
+ end = start + oldsize - 1; // (oldend, oldend+oldsize]
+ if (end >= num_sectors) {
+ // Adjust size to allow round-robin testing without future size decrease
+ uint64_t spans = (num_sectors + oldsize-1) / oldsize;
+ uint64_t newsize = (num_sectors + spans-1) / spans;
+ uint64_t newstart = num_sectors - newsize, newend = num_sectors - 1;
+ pout("Span %d changed from %" PRIu64 "-%" PRIu64 " (%" PRIu64 " sectors)\n",
+ i, start, end, oldsize);
+ pout(" to %" PRIu64 "-%" PRIu64 " (%" PRIu64 " sectors) (%" PRIu64 " spans)\n",
+ newstart, newend, newsize, spans);
+ start = newstart; end = newend;
+ }
+ }
+ break;
+ default:
+ pout("ataWriteSelectiveSelfTestLog: Invalid mode %d\n", mode);
+ return -1;
+ }
+ // Range check
+ if (start < num_sectors && num_sectors <= end) {
+ if (end != ~(uint64_t)0) // -t select,N-max
+ pout("Size of self-test span %d decreased according to disk size\n", i);
+ end = num_sectors - 1;
+ }
+ if (!(start <= end && end < num_sectors)) {
+ pout("Invalid selective self-test span %d: %" PRIu64 "-%" PRIu64 " (%" PRIu64 " sectors)\n",
+ i, start, end, num_sectors);
+ return -1;
+ }
+ // Return the actual mode and range to caller.
+ args.span[i].mode = mode;
+ args.span[i].start = start;
+ args.span[i].end = end;
+ }
+
+ // Clear spans
+ for (i=0; i<5; i++)
+ memset(data->span+i, 0, sizeof(struct test_span));
+
+ // Set spans for testing
+ for (i = 0; i < args.num_spans; i++){
+ data->span[i].start = args.span[i].start;
+ data->span[i].end = args.span[i].end;
+ }
+
+ // host must initialize to zero before initiating selective self-test
+ data->currentlba=0;
+ data->currentspan=0;
+
+ // Perform off-line scan after selective test?
+ if (args.scan_after_select == 1)
+ // NO
+ data->flags &= ~SELECTIVE_FLAG_DOSCAN;
+ else if (args.scan_after_select == 2)
+ // YES
+ data->flags |= SELECTIVE_FLAG_DOSCAN;
+
+ // Must clear active and pending flags before writing
+ data->flags &= ~(SELECTIVE_FLAG_ACTIVE);
+ data->flags &= ~(SELECTIVE_FLAG_PENDING);
+
+ // modify pending time?
+ if (args.pending_time)
+ data->pendingtime = (unsigned short)(args.pending_time-1);
+
+ // Set checksum to zero, then compute checksum
+ data->checksum=0;
+ unsigned char cksum=0;
+ for (i=0; i<512; i++)
+ cksum+=ptr[i];
+ cksum=~cksum;
+ cksum+=1;
+ data->checksum=cksum;
+
+ // swap endian order if needed
+ if (isbigendian()){
+ swap2((char *)&(data->logversion));
+ for (int b = 0; b < 5; b++) {
+ swap8((char *)&(data->span[b].start));
+ swap8((char *)&(data->span[b].end));
+ }
+ swap8((char *)&(data->currentlba));
+ swap2((char *)&(data->currentspan));
+ swap2((char *)&(data->flags));
+ swap2((char *)&(data->pendingtime));
+ }
+
+ // write new selective self-test log
+ if (smartcommandhandler(device, WRITE_LOG, 0x09, (char *)data)){
+ pout("Write Selective Self-test Log failed: %s\n", device->get_errmsg());
+ return -3;
+ }
+
+ return 0;
+}
+
+// This corrects some quantities that are byte reversed in the SMART
+// ATA ERROR LOG.
+static void fixsamsungerrorlog(ata_smart_errorlog * data)
+{
+ // FIXED IN SAMSUNG -25 FIRMWARE???
+ // Device error count in bytes 452-3
+ swap2((char *)&(data->ata_error_count));
+
+ // FIXED IN SAMSUNG -22a FIRMWARE
+ // step through 5 error log data structures
+ for (int i = 0; i < 5; i++){
+ // step through 5 command data structures
+ for (int j = 0; j < 5; j++)
+ // Command data structure 4-byte millisec timestamp. These are
+ // bytes (N+8, N+9, N+10, N+11).
+ swap4((char *)&(data->errorlog_struct[i].commands[j].timestamp));
+ // Error data structure two-byte hour life timestamp. These are
+ // bytes (N+28, N+29).
+ swap2((char *)&(data->errorlog_struct[i].error_struct.timestamp));
+ }
+ return;
+}
+
+// NEEDED ONLY FOR SAMSUNG -22 (some) -23 AND -24?? FIRMWARE
+static void fixsamsungerrorlog2(ata_smart_errorlog * data)
+{
+ // Device error count in bytes 452-3
+ swap2((char *)&(data->ata_error_count));
+ return;
+}
+
+// Reads the Summary SMART Error Log (log #1). The Comprehensive SMART
+// Error Log is #2, and the Extended Comprehensive SMART Error log is
+// #3
+int ataReadErrorLog (ata_device * device, ata_smart_errorlog *data,
+ firmwarebug_defs firmwarebugs)
+{
+
+ // get data from device
+ if (smartcommandhandler(device, READ_LOG, 0x01, (char *)data)){
+ return -1;
+ }
+
+ // compute its checksum, and issue a warning if needed
+ if (checksum(data))
+ checksumwarning("SMART ATA Error Log Structure");
+
+ // Some disks have the byte order reversed in some SMART Summary
+ // Error log entries
+ if (firmwarebugs.is_set(BUG_SAMSUNG))
+ fixsamsungerrorlog(data);
+ else if (firmwarebugs.is_set(BUG_SAMSUNG2))
+ fixsamsungerrorlog2(data);
+
+ // swap endian order if needed
+ if (isbigendian()){
+ int i,j;
+
+ // Device error count in bytes 452-3
+ swap2((char *)&(data->ata_error_count));
+
+ // step through 5 error log data structures
+ for (i=0; i<5; i++){
+ // step through 5 command data structures
+ for (j=0; j<5; j++)
+ // Command data structure 4-byte millisec timestamp
+ swap4((char *)&(data->errorlog_struct[i].commands[j].timestamp));
+ // Error data structure life timestamp
+ swap2((char *)&(data->errorlog_struct[i].error_struct.timestamp));
+ }
+ }
+
+ return 0;
+}
+
+
+// Fix LBA byte ordering of Extended Comprehensive Error Log
+// if little endian instead of ATA register ordering is provided
+template <class T>
+static inline void fix_exterrlog_lba_cmd(T & cmd)
+{
+ T org = cmd;
+ cmd.lba_mid_register_hi = org.lba_high_register;
+ cmd.lba_low_register_hi = org.lba_mid_register_hi;
+ cmd.lba_high_register = org.lba_mid_register;
+ cmd.lba_mid_register = org.lba_low_register_hi;
+}
+
+static void fix_exterrlog_lba(ata_smart_exterrlog * log, unsigned nsectors)
+{
+ for (unsigned i = 0; i < nsectors; i++) {
+ for (int ei = 0; ei < 4; ei++) {
+ ata_smart_exterrlog_error_log & entry = log[i].error_logs[ei];
+ fix_exterrlog_lba_cmd(entry.error);
+ for (int ci = 0; ci < 5; ci++)
+ fix_exterrlog_lba_cmd(entry.commands[ci]);
+ }
+ }
+}
+
+// Read Extended Comprehensive Error Log
+bool ataReadExtErrorLog(ata_device * device, ata_smart_exterrlog * log,
+ unsigned page, unsigned nsectors, firmwarebug_defs firmwarebugs)
+{
+ if (!ataReadLogExt(device, 0x03, 0x00, page, log, nsectors))
+ return false;
+
+ check_multi_sector_sum(log, nsectors, "SMART Extended Comprehensive Error Log Structure");
+
+ if (isbigendian()) {
+ SWAPV(log->device_error_count);
+ SWAPV(log->error_log_index);
+ for (unsigned i = 0; i < nsectors; i++) {
+ for (unsigned j = 0; j < 4; j++) {
+ for (unsigned k = 0; k < 5; k++)
+ SWAPV(log[i].error_logs[j].commands[k].timestamp);
+ SWAPV(log[i].error_logs[j].error.timestamp);
+ }
+ }
+ }
+
+ if (firmwarebugs.is_set(BUG_XERRORLBA))
+ fix_exterrlog_lba(log, nsectors);
+
+ return true;
+}
+
+
+int ataReadSmartThresholds (ata_device * device, struct ata_smart_thresholds_pvt *data){
+
+ // get data from device
+ if (smartcommandhandler(device, READ_THRESHOLDS, 0, (char *)data)){
+ return -1;
+ }
+
+ // compute its checksum, and issue a warning if needed
+ if (checksum(data))
+ checksumwarning("SMART Attribute Thresholds Structure");
+
+ // swap endian order if needed
+ if (isbigendian())
+ swap2((char *)&(data->revnumber));
+
+ return 0;
+}
+
+int ataEnableSmart (ata_device * device ){
+ if (smartcommandhandler(device, ENABLE, 0, NULL)){
+ return -1;
+ }
+ return 0;
+}
+
+int ataDisableSmart (ata_device * device ){
+
+ if (smartcommandhandler(device, DISABLE, 0, NULL)){
+ return -1;
+ }
+ return 0;
+}
+
+int ataEnableAutoSave(ata_device * device){
+ if (smartcommandhandler(device, AUTOSAVE, 241, NULL)){
+ return -1;
+ }
+ return 0;
+}
+
+int ataDisableAutoSave(ata_device * device){
+
+ if (smartcommandhandler(device, AUTOSAVE, 0, NULL)){
+ return -1;
+ }
+ return 0;
+}
+
+// In *ALL* ATA standards the Enable/Disable AutoOffline command is
+// marked "OBSOLETE". It is defined in SFF-8035i Revision 2, and most
+// vendors still support it for backwards compatibility. IBM documents
+// it for some drives.
+int ataEnableAutoOffline (ata_device * device){
+
+ /* timer hard coded to 4 hours */
+ if (smartcommandhandler(device, AUTO_OFFLINE, 248, NULL)){
+ return -1;
+ }
+ return 0;
+}
+
+// Another Obsolete Command. See comments directly above, associated
+// with the corresponding Enable command.
+int ataDisableAutoOffline (ata_device * device){
+
+ if (smartcommandhandler(device, AUTO_OFFLINE, 0, NULL)){
+ return -1;
+ }
+ return 0;
+}
+
+// If SMART is enabled, supported, and working, then this call is
+// guaranteed to return 1, else zero. Note that it should return 1
+// regardless of whether the disk's SMART status is 'healthy' or
+// 'failing'.
+int ataDoesSmartWork(ata_device * device){
+ int retval=smartcommandhandler(device, STATUS, 0, NULL);
+
+ if (-1 == retval)
+ return 0;
+
+ return 1;
+}
+
+// This function uses a different interface (DRIVE_TASK) than the
+// other commands in this file.
+int ataSmartStatus2(ata_device * device){
+ return smartcommandhandler(device, STATUS_CHECK, 0, NULL);
+}
+
+// This is the way to execute ALL tests: offline, short self-test,
+// extended self test, with and without captive mode, etc.
+// TODO: Move to ataprint.cpp ?
+int ataSmartTest(ata_device * device, int testtype, bool force,
+ const ata_selective_selftest_args & selargs,
+ const ata_smart_values * sv, uint64_t num_sectors)
+{
+ char cmdmsg[128]; const char *type, *captive;
+ int cap, retval, select=0;
+
+ // Boolean, if set, says test is captive
+ cap=testtype & CAPTIVE_MASK;
+
+ // Set up strings that describe the type of test
+ if (cap)
+ captive="captive";
+ else
+ captive="off-line";
+
+ if (testtype==OFFLINE_FULL_SCAN)
+ type="off-line";
+ else if (testtype==SHORT_SELF_TEST || testtype==SHORT_CAPTIVE_SELF_TEST)
+ type="Short self-test";
+ else if (testtype==EXTEND_SELF_TEST || testtype==EXTEND_CAPTIVE_SELF_TEST)
+ type="Extended self-test";
+ else if (testtype==CONVEYANCE_SELF_TEST || testtype==CONVEYANCE_CAPTIVE_SELF_TEST)
+ type="Conveyance self-test";
+ else if ((select=(testtype==SELECTIVE_SELF_TEST || testtype==SELECTIVE_CAPTIVE_SELF_TEST)))
+ type="Selective self-test";
+ else
+ type = 0;
+
+ // Check whether another test is already running
+ if (type && (sv->self_test_exec_status >> 4) == 0xf) {
+ if (!force) {
+ pout("Can't start self-test without aborting current test (%d0%% remaining),\n"
+ "%srun 'smartctl -X' to abort test.\n",
+ sv->self_test_exec_status & 0x0f,
+ (!select ? "add '-t force' option to override, or " : ""));
+ return -1;
+ }
+ }
+ else
+ force = false;
+
+ // If doing a selective self-test, first use WRITE_LOG to write the
+ // selective self-test log.
+ ata_selective_selftest_args selargs_io = selargs; // filled with info about actual spans
+ if (select && (retval = ataWriteSelectiveSelfTestLog(device, selargs_io, sv, num_sectors))) {
+ if (retval==-4)
+ pout("Can't start selective self-test without aborting current test: use '-X' option to smartctl.\n");
+ return retval;
+ }
+
+ // Print ouf message that we are sending the command to test
+ if (testtype==ABORT_SELF_TEST)
+ snprintf(cmdmsg, sizeof(cmdmsg), "Abort SMART off-line mode self-test routine");
+ else if (!type)
+ snprintf(cmdmsg, sizeof(cmdmsg), "SMART EXECUTE OFF-LINE IMMEDIATE subcommand 0x%02x", testtype);
+ else
+ snprintf(cmdmsg, sizeof(cmdmsg), "Execute SMART %s routine immediately in %s mode", type, captive);
+ pout("Sending command: \"%s\".\n",cmdmsg);
+
+ if (select) {
+ int i;
+ pout("SPAN STARTING_LBA ENDING_LBA\n");
+ for (i = 0; i < selargs_io.num_spans; i++)
+ pout(" %d %20" PRId64 " %20" PRId64 "\n", i,
+ selargs_io.span[i].start,
+ selargs_io.span[i].end);
+ }
+
+ // Now send the command to test
+ if (smartcommandhandler(device, IMMEDIATE_OFFLINE, testtype, NULL)) {
+ if (!(cap && device->get_errno() == EIO)) {
+ pout("Command \"%s\" failed: %s\n", cmdmsg, device->get_errmsg());
+ return -1;
+ }
+ }
+
+ // Since the command succeeded, tell user
+ if (testtype==ABORT_SELF_TEST)
+ pout("Self-testing aborted!\n");
+ else {
+ pout("Drive command \"%s\" successful.\n", cmdmsg);
+ if (type)
+ pout("Testing has begun%s.\n", (force ? " (previous test aborted)" : ""));
+ }
+ return 0;
+}
+
+/* Test Time Functions */
+int TestTime(const ata_smart_values *data, int testtype)
+{
+ switch (testtype){
+ case OFFLINE_FULL_SCAN:
+ return (int) data->total_time_to_complete_off_line;
+ case SHORT_SELF_TEST:
+ case SHORT_CAPTIVE_SELF_TEST:
+ return (int) data->short_test_completion_time;
+ case EXTEND_SELF_TEST:
+ case EXTEND_CAPTIVE_SELF_TEST:
+ if (data->extend_test_completion_time_b == 0xff
+ && data->extend_test_completion_time_w != 0x0000
+ && data->extend_test_completion_time_w != 0xffff)
+ return data->extend_test_completion_time_w; // ATA-8
+ else
+ return data->extend_test_completion_time_b;
+ case CONVEYANCE_SELF_TEST:
+ case CONVEYANCE_CAPTIVE_SELF_TEST:
+ return (int) data->conveyance_test_completion_time;
+ default:
+ return 0;
+ }
+}
+
+// This function tells you both about the ATA error log and the
+// self-test error log capability (introduced in ATA-5). The bit is
+// poorly documented in the ATA/ATAPI standard. Starting with ATA-6,
+// SMART error logging is also indicated in bit 0 of DEVICE IDENTIFY
+// word 84 and 87. Top two bits must match the pattern 01. BEFORE
+// ATA-6 these top two bits still had to match the pattern 01, but the
+// remaining bits were reserved (==0).
+bool isSmartErrorLogCapable(const ata_smart_values * data, const ata_identify_device * identity)
+{
+ unsigned short word84=identity->command_set_extension;
+ unsigned short word87=identity->csf_default;
+ int isata6=identity->major_rev_num & (0x01<<6);
+ int isata7=identity->major_rev_num & (0x01<<7);
+
+ if ((isata6 || isata7) && (word84>>14) == 0x01 && (word84 & 0x01))
+ return true;
+
+ if ((isata6 || isata7) && (word87>>14) == 0x01 && (word87 & 0x01))
+ return true;
+
+ // otherwise we'll use the poorly documented capability bit
+ return !!(data->errorlog_capability & 0x01);
+}
+
+// See previous function. If the error log exists then the self-test
+// log should (must?) also exist.
+bool isSmartTestLogCapable(const ata_smart_values * data, const ata_identify_device *identity)
+{
+ unsigned short word84=identity->command_set_extension;
+ unsigned short word87=identity->csf_default;
+ int isata6=identity->major_rev_num & (0x01<<6);
+ int isata7=identity->major_rev_num & (0x01<<7);
+
+ if ((isata6 || isata7) && (word84>>14) == 0x01 && (word84 & 0x02))
+ return true;
+
+ if ((isata6 || isata7) && (word87>>14) == 0x01 && (word87 & 0x02))
+ return true;
+
+
+ // otherwise we'll use the poorly documented capability bit
+ return !!(data->errorlog_capability & 0x01);
+}
+
+
+bool isGeneralPurposeLoggingCapable(const ata_identify_device *identity)
+{
+ unsigned short word84=identity->command_set_extension;
+ unsigned short word87=identity->csf_default;
+
+ // If bit 14 of word 84 is set to one and bit 15 of word 84 is
+ // cleared to zero, the contents of word 84 contains valid support
+ // information. If not, support information is not valid in this
+ // word.
+ if ((word84>>14) == 0x01)
+ // If bit 5 of word 84 is set to one, the device supports the
+ // General Purpose Logging feature set.
+ return !!(word84 & (0x01 << 5));
+
+ // If bit 14 of word 87 is set to one and bit 15 of word 87 is
+ // cleared to zero, the contents of words (87:85) contain valid
+ // information. If not, information is not valid in these words.
+ if ((word87>>14) == 0x01)
+ // If bit 5 of word 87 is set to one, the device supports
+ // the General Purpose Logging feature set.
+ return !!(word87 & (0x01 << 5));
+
+ // not capable
+ return false;
+}
+
+// Get attribute state
+ata_attr_state ata_get_attr_state(const ata_smart_attribute & attr,
+ int attridx,
+ const ata_smart_threshold_entry * thresholds,
+ const ata_vendor_attr_defs & defs,
+ unsigned char * threshval /* = 0 */)
+{
+ if (!attr.id)
+ return ATTRSTATE_NON_EXISTING;
+
+ // Normalized values (current,worst,threshold) not valid
+ // if specified by '-v' option.
+ // (Some SSD disks uses these bytes to store raw value).
+ if (defs[attr.id].flags & ATTRFLAG_NO_NORMVAL)
+ return ATTRSTATE_NO_NORMVAL;
+
+ // Normally threshold is at same index as attribute
+ int i = attridx;
+ if (thresholds[i].id != attr.id) {
+ // Find threshold id in table
+ for (i = 0; thresholds[i].id != attr.id; ) {
+ if (++i >= NUMBER_ATA_SMART_ATTRIBUTES)
+ // Threshold id missing or thresholds cannot be read
+ return ATTRSTATE_NO_THRESHOLD;
+ }
+ }
+ unsigned char threshold = thresholds[i].threshold;
+
+ // Return threshold if requested
+ if (threshval)
+ *threshval = threshold;
+
+ // Don't report a failed attribute if its threshold is 0.
+ // ATA-3 (X3T13/2008D Revision 7b) declares 0x00 as the "always passing"
+ // threshold (Later ATA versions declare all thresholds as "obsolete").
+ // In practice, threshold value 0 is often used for usage attributes.
+ if (!threshold)
+ return ATTRSTATE_OK;
+
+ // Failed now if current value is below threshold
+ if (attr.current <= threshold)
+ return ATTRSTATE_FAILED_NOW;
+
+ // Failed in the past if worst value is below threshold
+ if (!(defs[attr.id].flags & ATTRFLAG_NO_WORSTVAL) && attr.worst <= threshold)
+ return ATTRSTATE_FAILED_PAST;
+
+ return ATTRSTATE_OK;
+}
+
+// Get attribute raw value.
+uint64_t ata_get_attr_raw_value(const ata_smart_attribute & attr,
+ const ata_vendor_attr_defs & defs)
+{
+ const ata_vendor_attr_defs::entry & def = defs[attr.id];
+ // TODO: Allow Byteorder in DEFAULT entry
+
+ // Use default byteorder if not specified
+ const char * byteorder = def.byteorder;
+ if (!*byteorder) {
+ switch (def.raw_format) {
+ case RAWFMT_RAW64:
+ case RAWFMT_HEX64:
+ byteorder = "543210wv"; break;
+ case RAWFMT_RAW56:
+ case RAWFMT_HEX56:
+ case RAWFMT_RAW24_DIV_RAW32:
+ case RAWFMT_MSEC24_HOUR32:
+ byteorder = "r543210"; break;
+ default:
+ byteorder = "543210"; break;
+ }
+ }
+
+ // Build 64-bit value from selected bytes
+ uint64_t rawvalue = 0;
+ for (int i = 0; byteorder[i]; i++) {
+ unsigned char b;
+ switch (byteorder[i]) {
+ case '0': b = attr.raw[0]; break;
+ case '1': b = attr.raw[1]; break;
+ case '2': b = attr.raw[2]; break;
+ case '3': b = attr.raw[3]; break;
+ case '4': b = attr.raw[4]; break;
+ case '5': b = attr.raw[5]; break;
+ case 'r': b = attr.reserv; break;
+ case 'v': b = attr.current; break;
+ case 'w': b = attr.worst; break;
+ default : b = 0; break;
+ }
+ rawvalue <<= 8; rawvalue |= b;
+ }
+
+ return rawvalue;
+}
+
+// Helper functions for RAWFMT_TEMPMINMAX
+static inline int check_temp_word(unsigned word)
+{
+ if (word <= 0x7f)
+ return 0x11; // >= 0, signed byte or word
+ if (word <= 0xff)
+ return 0x01; // < 0, signed byte
+ if (0xff80 <= word)
+ return 0x10; // < 0, signed word
+ return 0x00;
+}
+
+static bool check_temp_range(int t, unsigned char ut1, unsigned char ut2,
+ int & lo, int & hi)
+{
+ int t1 = (signed char)ut1, t2 = (signed char)ut2;
+ if (t1 > t2) {
+ int tx = t1; t1 = t2; t2 = tx;
+ }
+
+ if ( -60 <= t1 && t1 <= t && t <= t2 && t2 <= 120
+ && !(t1 == -1 && t2 <= 0) ) {
+ lo = t1; hi = t2;
+ return true;
+ }
+ return false;
+}
+
+// Format attribute raw value.
+std::string ata_format_attr_raw_value(const ata_smart_attribute & attr,
+ const ata_vendor_attr_defs & defs)
+{
+ // Get 48 bit or 64 bit raw value
+ uint64_t rawvalue = ata_get_attr_raw_value(attr, defs);
+
+ // Split into bytes and words
+ unsigned char raw[6];
+ raw[0] = (unsigned char) rawvalue;
+ raw[1] = (unsigned char)(rawvalue >> 8);
+ raw[2] = (unsigned char)(rawvalue >> 16);
+ raw[3] = (unsigned char)(rawvalue >> 24);
+ raw[4] = (unsigned char)(rawvalue >> 32);
+ raw[5] = (unsigned char)(rawvalue >> 40);
+ unsigned word[3];
+ word[0] = raw[0] | (raw[1] << 8);
+ word[1] = raw[2] | (raw[3] << 8);
+ word[2] = raw[4] | (raw[5] << 8);
+
+ // Get print format
+ ata_attr_raw_format format = defs[attr.id].raw_format;
+ if (format == RAWFMT_DEFAULT) {
+ // Get format from DEFAULT entry
+ format = get_default_attr_defs()[attr.id].raw_format;
+ if (format == RAWFMT_DEFAULT)
+ // Unknown Attribute
+ format = RAWFMT_RAW48;
+ }
+
+ // Print
+ std::string s;
+ switch (format) {
+ case RAWFMT_RAW8:
+ s = strprintf("%d %d %d %d %d %d",
+ raw[5], raw[4], raw[3], raw[2], raw[1], raw[0]);
+ break;
+
+ case RAWFMT_RAW16:
+ s = strprintf("%u %u %u", word[2], word[1], word[0]);
+ break;
+
+ case RAWFMT_RAW48:
+ case RAWFMT_RAW56:
+ case RAWFMT_RAW64:
+ s = strprintf("%" PRIu64, rawvalue);
+ break;
+
+ case RAWFMT_HEX48:
+ s = strprintf("0x%012" PRIx64, rawvalue);
+ break;
+
+ case RAWFMT_HEX56:
+ s = strprintf("0x%014" PRIx64, rawvalue);
+ break;
+
+ case RAWFMT_HEX64:
+ s = strprintf("0x%016" PRIx64, rawvalue);
+ break;
+
+ case RAWFMT_RAW16_OPT_RAW16:
+ s = strprintf("%u", word[0]);
+ if (word[1] || word[2])
+ s += strprintf(" (%u %u)", word[2], word[1]);
+ break;
+
+ case RAWFMT_RAW16_OPT_AVG16:
+ s = strprintf("%u", word[0]);
+ if (word[1])
+ s += strprintf(" (Average %u)", word[1]);
+ break;
+
+ case RAWFMT_RAW24_OPT_RAW8:
+ s = strprintf("%u", (unsigned)(rawvalue & 0x00ffffffULL));
+ if (raw[3] || raw[4] || raw[5])
+ s += strprintf(" (%d %d %d)", raw[5], raw[4], raw[3]);
+ break;
+
+ case RAWFMT_RAW24_DIV_RAW24:
+ s = strprintf("%u/%u",
+ (unsigned)(rawvalue >> 24), (unsigned)(rawvalue & 0x00ffffffULL));
+ break;
+
+ case RAWFMT_RAW24_DIV_RAW32:
+ s = strprintf("%u/%u",
+ (unsigned)(rawvalue >> 32), (unsigned)(rawvalue & 0xffffffffULL));
+ break;
+
+ case RAWFMT_MIN2HOUR:
+ {
+ // minutes
+ int64_t temp = word[0]+(word[1]<<16);
+ int64_t tmp1 = temp/60;
+ int64_t tmp2 = temp%60;
+ s = strprintf("%" PRIu64 "h+%02" PRIu64 "m", tmp1, tmp2);
+ if (word[2])
+ s += strprintf(" (%u)", word[2]);
+ }
+ break;
+
+ case RAWFMT_SEC2HOUR:
+ {
+ // seconds
+ int64_t hours = rawvalue/3600;
+ int64_t minutes = (rawvalue-3600*hours)/60;
+ int64_t seconds = rawvalue%60;
+ s = strprintf("%" PRIu64 "h+%02" PRIu64 "m+%02" PRIu64 "s", hours, minutes, seconds);
+ }
+ break;
+
+ case RAWFMT_HALFMIN2HOUR:
+ {
+ // 30-second counter
+ int64_t hours = rawvalue/120;
+ int64_t minutes = (rawvalue-120*hours)/2;
+ s += strprintf("%" PRIu64 "h+%02" PRIu64 "m", hours, minutes);
+ }
+ break;
+
+ case RAWFMT_MSEC24_HOUR32:
+ {
+ // hours + milliseconds
+ unsigned hours = (unsigned)(rawvalue & 0xffffffffULL);
+ unsigned milliseconds = (unsigned)(rawvalue >> 32);
+ unsigned seconds = milliseconds / 1000;
+ s = strprintf("%uh+%02um+%02u.%03us",
+ hours, seconds / 60, seconds % 60, milliseconds % 1000);
+ }
+ break;
+
+ case RAWFMT_TEMPMINMAX:
+ // Temperature
+ {
+ // Search for possible min/max values
+ // [5][4][3][2][1][0] raw[]
+ // [ 2 ] [ 1 ] [ 0 ] word[]
+ // xx HH xx LL xx TT (Hitachi/HGST)
+ // xx LL xx HH xx TT (Kingston SSDs)
+ // 00 00 HH LL xx TT (Maxtor, Samsung, Seagate, Toshiba)
+ // 00 00 00 HH LL TT (WDC)
+ // CC CC HH LL xx TT (WDC, CCCC=over temperature count)
+ // (xx = 00/ff, possibly sign extension of lower byte)
+
+ int t = (signed char)raw[0];
+ int lo = 0, hi = 0;
+
+ int tformat;
+ int ctw0 = check_temp_word(word[0]);
+ if (!word[2]) {
+ if (!word[1] && ctw0)
+ // 00 00 00 00 xx TT
+ tformat = 0;
+ else if (ctw0 && check_temp_range(t, raw[2], raw[3], lo, hi))
+ // 00 00 HL LH xx TT
+ tformat = 1;
+ else if (!raw[3] && check_temp_range(t, raw[1], raw[2], lo, hi))
+ // 00 00 00 HL LH TT
+ tformat = 2;
+ else
+ tformat = -1;
+ }
+ else if (ctw0) {
+ if ( (ctw0 & check_temp_word(word[1]) & check_temp_word(word[2])) != 0x00
+ && check_temp_range(t, raw[2], raw[4], lo, hi) )
+ // xx HL xx LH xx TT
+ tformat = 3;
+ else if ( word[2] < 0x7fff
+ && check_temp_range(t, raw[2], raw[3], lo, hi)
+ && hi >= 40 )
+ // CC CC HL LH xx TT
+ tformat = 4;
+ else
+ tformat = -2;
+ }
+ else
+ tformat = -3;
+
+ switch (tformat) {
+ case 0:
+ s = strprintf("%d", t);
+ break;
+ case 1: case 2: case 3:
+ s = strprintf("%d (Min/Max %d/%d)", t, lo, hi);
+ break;
+ case 4:
+ s = strprintf("%d (Min/Max %d/%d #%d)", t, lo, hi, word[2]);
+ break;
+ default:
+ s = strprintf("%d (%d %d %d %d %d)", raw[0], raw[5], raw[4], raw[3], raw[2], raw[1]);
+ break;
+ }
+ }
+ break;
+
+ case RAWFMT_TEMP10X:
+ // ten times temperature in Celsius
+ s = strprintf("%d.%d", word[0]/10, word[0]%10);
+ break;
+
+ default:
+ s = "?"; // Should not happen
+ break;
+ }
+
+ return s;
+}
+
+// Get attribute name
+std::string ata_get_smart_attr_name(unsigned char id, const ata_vendor_attr_defs & defs,
+ int rpm /* = 0 */)
+{
+ if (!defs[id].name.empty())
+ return defs[id].name;
+ else {
+ const ata_vendor_attr_defs::entry & def = get_default_attr_defs()[id];
+ if (def.name.empty())
+ return "Unknown_Attribute";
+ else if ((def.flags & ATTRFLAG_HDD_ONLY) && rpm == 1)
+ return "Unknown_SSD_Attribute";
+ else if ((def.flags & ATTRFLAG_SSD_ONLY) && rpm > 1)
+ return "Unknown_HDD_Attribute";
+ else
+ return def.name;
+ }
+}
+
+// Find attribute index for attribute id, -1 if not found.
+int ata_find_attr_index(unsigned char id, const ata_smart_values & smartval)
+{
+ if (!id)
+ return -1;
+ for (int i = 0; i < NUMBER_ATA_SMART_ATTRIBUTES; i++) {
+ if (smartval.vendor_attributes[i].id == id)
+ return i;
+ }
+ return -1;
+}
+
+// Return Temperature Attribute raw value selected according to possible
+// non-default interpretations. If the Attribute does not exist, return 0
+unsigned char ata_return_temperature_value(const ata_smart_values * data, const ata_vendor_attr_defs & defs)
+{
+ for (int i = 0; i < 4; i++) {
+ static const unsigned char ids[4] = {194, 190, 9, 220};
+ unsigned char id = ids[i];
+ const ata_attr_raw_format format = defs[id].raw_format;
+ if (!( ((id == 194 || id == 190) && format == RAWFMT_DEFAULT)
+ || format == RAWFMT_TEMPMINMAX || format == RAWFMT_TEMP10X))
+ continue;
+ int idx = ata_find_attr_index(id, *data);
+ if (idx < 0)
+ continue;
+ uint64_t raw = ata_get_attr_raw_value(data->vendor_attributes[idx], defs);
+ unsigned temp;
+ // ignore possible min/max values in high words
+ if (format == RAWFMT_TEMP10X) // -v N,temp10x
+ temp = ((unsigned short)raw + 5) / 10;
+ else
+ temp = (unsigned char)raw;
+ if (!(0 < temp && temp < 128))
+ continue;
+ return temp;
+ }
+ // No valid attribute found
+ return 0;
+}
+
+
+// Read SCT Status
+int ataReadSCTStatus(ata_device * device, ata_sct_status_response * sts)
+{
+ // read SCT status via SMART log 0xe0
+ memset(sts, 0, sizeof(*sts));
+ if (smartcommandhandler(device, READ_LOG, 0xe0, (char *)sts)){
+ pout("Read SCT Status failed: %s\n", device->get_errmsg());
+ return -1;
+ }
+
+ // swap endian order if needed
+ if (isbigendian()){
+ SWAPV(sts->format_version);
+ SWAPV(sts->sct_version);
+ SWAPV(sts->sct_spec);
+ SWAPV(sts->ext_status_code);
+ SWAPV(sts->action_code);
+ SWAPV(sts->function_code);
+ SWAPV(sts->over_limit_count);
+ SWAPV(sts->under_limit_count);
+ SWAPV(sts->smart_status);
+ SWAPV(sts->min_erc_time);
+ }
+
+ // Check format version
+ if (!(sts->format_version == 2 || sts->format_version == 3)) {
+ pout("Unknown SCT Status format version %u, should be 2 or 3.\n", sts->format_version);
+ return -1;
+ }
+ return 0;
+}
+
+// Read SCT Temperature History Table
+int ataReadSCTTempHist(ata_device * device, ata_sct_temperature_history_table * tmh,
+ ata_sct_status_response * sts)
+{
+ // Initial SCT status must be provided by caller
+
+ // Do nothing if other SCT command is executing
+ if (sts->ext_status_code == 0xffff) {
+ pout("Another SCT command is executing, abort Read Data Table\n"
+ "(SCT ext_status_code 0x%04x, action_code=%u, function_code=%u)\n",
+ sts->ext_status_code, sts->action_code, sts->function_code);
+ return -1;
+ }
+
+ ata_sct_data_table_command cmd; memset(&cmd, 0, sizeof(cmd));
+ // CAUTION: DO NOT CHANGE THIS VALUE (SOME ACTION CODES MAY ERASE DISK)
+ cmd.action_code = 5; // Data table command
+ cmd.function_code = 1; // Read table
+ cmd.table_id = 2; // Temperature History Table
+
+ // swap endian order if needed
+ if (isbigendian()) {
+ SWAPV(cmd.action_code);
+ SWAPV(cmd.function_code);
+ SWAPV(cmd.table_id);
+ }
+
+ // write command via SMART log page 0xe0
+ if (smartcommandhandler(device, WRITE_LOG, 0xe0, (char *)&cmd)){
+ pout("Write SCT Data Table failed: %s\n", device->get_errmsg());
+ return -1;
+ }
+
+ // read SCT data via SMART log page 0xe1
+ memset(tmh, 0, sizeof(*tmh));
+ if (smartcommandhandler(device, READ_LOG, 0xe1, (char *)tmh)){
+ pout("Read SCT Data Table failed: %s\n", device->get_errmsg());
+ return -1;
+ }
+
+ // re-read and check SCT status
+ if (ataReadSCTStatus(device, sts))
+ return -1;
+
+ if (!(sts->ext_status_code == 0 && sts->action_code == 5 && sts->function_code == 1)) {
+ pout("Unexpected SCT status 0x%04x (action_code=%u, function_code=%u)\n",
+ sts->ext_status_code, sts->action_code, sts->function_code);
+ return -1;
+ }
+
+ // swap endian order if needed
+ if (isbigendian()){
+ SWAPV(tmh->format_version);
+ SWAPV(tmh->sampling_period);
+ SWAPV(tmh->interval);
+ SWAPV(tmh->cb_index);
+ SWAPV(tmh->cb_size);
+ }
+ return 0;
+}
+
+// Common function for Get/Set SCT Feature Control:
+// Write Cache, Write Cache Reordering, etc.
+static int ataGetSetSCTFeatureControl(ata_device * device, unsigned short feature_code,
+ unsigned short state, bool persistent, bool set)
+{
+ // Check initial status
+ ata_sct_status_response sts;
+ if (ataReadSCTStatus(device, &sts))
+ return -1;
+
+ // Do nothing if other SCT command is executing
+ if (sts.ext_status_code == 0xffff) {
+ pout("Another SCT command is executing, abort Feature Control\n"
+ "(SCT ext_status_code 0x%04x, action_code=%u, function_code=%u)\n",
+ sts.ext_status_code, sts.action_code, sts.function_code);
+ return -1;
+ }
+
+ ata_sct_feature_control_command cmd; memset(&cmd, 0, sizeof(cmd));
+ // CAUTION: DO NOT CHANGE THIS VALUE (SOME ACTION CODES MAY ERASE DISK)
+ cmd.action_code = 4; // Feature Control command
+ cmd.function_code = (set ? 1 : 2); // 1=Set, 2=Get
+ cmd.feature_code = feature_code;
+ cmd.state = state;
+ cmd.option_flags = (persistent ? 0x01 : 0x00);
+
+ // swap endian order if needed
+ if (isbigendian()) {
+ SWAPV(cmd.action_code);
+ SWAPV(cmd.function_code);
+ SWAPV(cmd.feature_code);
+ SWAPV(cmd.state);
+ SWAPV(cmd.option_flags);
+ }
+
+ // write command via SMART log page 0xe0
+ // TODO: Debug output
+ ata_cmd_in in;
+ in.in_regs.command = ATA_SMART_CMD;
+ in.in_regs.lba_high = SMART_CYL_HI; in.in_regs.lba_mid = SMART_CYL_LOW;
+ in.in_regs.features = ATA_SMART_WRITE_LOG_SECTOR;
+ in.in_regs.lba_low = 0xe0;
+ in.set_data_out(&cmd, 1);
+
+ if (!set)
+ // Time limit returned in ATA registers
+ in.out_needed.sector_count = in.out_needed.lba_low = true;
+
+ ata_cmd_out out;
+ if (!device->ata_pass_through(in, out)) {
+ pout("Write SCT (%cet) Feature Control Command failed: %s\n",
+ (!set ? 'G' : 'S'), device->get_errmsg());
+ return -1;
+ }
+ state = out.out_regs.sector_count | (out.out_regs.lba_low << 8);
+
+ // re-read and check SCT status
+ if (ataReadSCTStatus(device, &sts))
+ return -1;
+
+ if (!(sts.ext_status_code == 0 && sts.action_code == 4 && sts.function_code == (set ? 1 : 2))) {
+ pout("Unexpected SCT status 0x%04x (action_code=%u, function_code=%u)\n",
+ sts.ext_status_code, sts.action_code, sts.function_code);
+ return -1;
+ }
+ return state;
+}
+
+// Get/Set Write Cache Reordering
+int ataGetSetSCTWriteCacheReordering(ata_device * device, bool enable, bool persistent, bool set)
+{
+ return ataGetSetSCTFeatureControl(device, 2 /* Enable/Disable Write Cache Reordering */,
+ (enable ? 1 : 2), persistent, set);
+}
+
+// Get/Set Write Cache (force enable, force disable,
+int ataGetSetSCTWriteCache(ata_device * device, unsigned short state, bool persistent, bool set)
+{
+ return ataGetSetSCTFeatureControl(device, 1 /* Enable/Disable Write Cache */,
+ state, persistent, set);
+}
+
+// Set SCT Temperature Logging Interval
+int ataSetSCTTempInterval(ata_device * device, unsigned interval, bool persistent)
+{
+ // Check initial status
+ ata_sct_status_response sts;
+ if (ataReadSCTStatus(device, &sts))
+ return -1;
+
+ // Do nothing if other SCT command is executing
+ if (sts.ext_status_code == 0xffff) {
+ pout("Another SCT command is executing, abort Feature Control\n"
+ "(SCT ext_status_code 0x%04x, action_code=%u, function_code=%u)\n",
+ sts.ext_status_code, sts.action_code, sts.function_code);
+ return -1;
+ }
+
+ ata_sct_feature_control_command cmd; memset(&cmd, 0, sizeof(cmd));
+ // CAUTION: DO NOT CHANGE THIS VALUE (SOME ACTION CODES MAY ERASE DISK)
+ cmd.action_code = 4; // Feature Control command
+ cmd.function_code = 1; // Set state
+ cmd.feature_code = 3; // Temperature logging interval
+ cmd.state = interval;
+ cmd.option_flags = (persistent ? 0x01 : 0x00);
+
+ // swap endian order if needed
+ if (isbigendian()) {
+ SWAPV(cmd.action_code);
+ SWAPV(cmd.function_code);
+ SWAPV(cmd.feature_code);
+ SWAPV(cmd.state);
+ SWAPV(cmd.option_flags);
+ }
+
+ // write command via SMART log page 0xe0
+ if (smartcommandhandler(device, WRITE_LOG, 0xe0, (char *)&cmd)){
+ pout("Write SCT Feature Control Command failed: %s\n", device->get_errmsg());
+ return -1;
+ }
+
+ // re-read and check SCT status
+ if (ataReadSCTStatus(device, &sts))
+ return -1;
+
+ if (!(sts.ext_status_code == 0 && sts.action_code == 4 && sts.function_code == 1)) {
+ pout("Unexpected SCT status 0x%04x (action_code=%u, function_code=%u)\n",
+ sts.ext_status_code, sts.action_code, sts.function_code);
+ return -1;
+ }
+ return 0;
+}
+
+// Get/Set SCT Error Recovery Control
+static int ataGetSetSCTErrorRecoveryControltime(ata_device * device, unsigned type,
+ bool set, unsigned short & time_limit,
+ bool power_on, bool mfg_default)
+{
+ // Check initial status
+ ata_sct_status_response sts;
+ if (ataReadSCTStatus(device, &sts))
+ return -1;
+
+ // Do nothing if other SCT command is executing
+ if (sts.ext_status_code == 0xffff) {
+ pout("Another SCT command is executing, abort Error Recovery Control\n"
+ "(SCT ext_status_code 0x%04x, action_code=%u, function_code=%u)\n",
+ sts.ext_status_code, sts.action_code, sts.function_code);
+ return -1;
+ }
+
+ ata_sct_error_recovery_control_command cmd; memset(&cmd, 0, sizeof(cmd));
+ // CAUTION: DO NOT CHANGE THIS VALUE (SOME ACTION CODES MAY ERASE DISK)
+ cmd.action_code = 3; // Error Recovery Control command
+
+ // 1=Set timer, 2=Get timer, 3=Set Power-on timer, 4=Get Power-on timer, 5=Restore mfg default
+ if (mfg_default) {
+ cmd.function_code = 5;
+ } else if (power_on) {
+ cmd.function_code = (set ? 3 : 4);
+ } else {
+ cmd.function_code = (set ? 1 : 2);
+ }
+ unsigned short saved_function_code = cmd.function_code;
+
+ cmd.selection_code = type; // 1=Read timer, 2=Write timer
+ if (set)
+ cmd.time_limit = time_limit;
+
+ // swap endian order if needed
+ if (isbigendian()) {
+ SWAPV(cmd.action_code);
+ SWAPV(cmd.function_code);
+ SWAPV(cmd.selection_code);
+ SWAPV(cmd.time_limit);
+ }
+
+ // write command via SMART log page 0xe0
+ // TODO: Debug output
+ ata_cmd_in in;
+ in.in_regs.command = ATA_SMART_CMD;
+ in.in_regs.lba_high = SMART_CYL_HI; in.in_regs.lba_mid = SMART_CYL_LOW;
+ in.in_regs.features = ATA_SMART_WRITE_LOG_SECTOR;
+ in.in_regs.lba_low = 0xe0;
+ in.set_data_out(&cmd, 1);
+
+ if (!set)
+ // Time limit returned in ATA registers
+ in.out_needed.sector_count = in.out_needed.lba_low = true;
+
+ ata_cmd_out out;
+ if (!device->ata_pass_through(in, out)) {
+ pout("Write SCT (%cet) Error Recovery Control Command failed: %s\n",
+ (!set ? 'G' : 'S'), device->get_errmsg());
+ return -1;
+ }
+
+ // re-read and check SCT status
+ if (ataReadSCTStatus(device, &sts))
+ return -1;
+
+ if (!(sts.ext_status_code == 0 && sts.action_code == 3 && sts.function_code == saved_function_code)) {
+ pout("Unexpected SCT status 0x%04x (action_code=%u, function_code=%u)\n",
+ sts.ext_status_code, sts.action_code, sts.function_code);
+ return -1;
+ }
+
+ if (!set) {
+ // Check whether registers are properly returned by ioctl()
+ if (!(out.out_regs.sector_count.is_set() && out.out_regs.lba_low.is_set())) {
+ // TODO: Output register support should be checked within each ata_pass_through()
+ // implementation before command is issued.
+ pout("SMART WRITE LOG does not return COUNT and LBA_LOW register\n");
+ return -1;
+ }
+ if ( out.out_regs.sector_count == in.in_regs.sector_count
+ && out.out_regs.lba_low == in.in_regs.lba_low ) {
+ // 0xe001 (5734.5s) - this is most likely a broken ATA pass-through implementation
+ pout("SMART WRITE LOG returns COUNT and LBA_LOW register unchanged\n");
+ return -1;
+ }
+
+ // Return value to caller
+ time_limit = out.out_regs.sector_count | (out.out_regs.lba_low << 8);
+ }
+
+ return 0;
+}
+
+// Get SCT Error Recovery Control
+int ataGetSCTErrorRecoveryControltime(ata_device * device, unsigned type, unsigned short & time_limit, bool power_on)
+{
+ return ataGetSetSCTErrorRecoveryControltime(device, type, false/*get*/, time_limit, power_on, false);
+}
+
+// Set SCT Error Recovery Control
+int ataSetSCTErrorRecoveryControltime(ata_device * device, unsigned type, unsigned short time_limit,
+ bool power_on, bool mfg_default)
+{
+ return ataGetSetSCTErrorRecoveryControltime(device, type, true/*set*/, time_limit, power_on, mfg_default);
+}
+
+
+/////////////////////////////////////////////////////////////////////////////
+// Pseudo-device to parse "smartctl -r ataioctl,2 ..." output and simulate
+// an ATA device with same behaviour
+
+namespace {
+
+class parsed_ata_device
+: public /*implements*/ ata_device_with_command_set
+{
+public:
+ parsed_ata_device(smart_interface * intf, const char * dev_name);
+
+ virtual ~parsed_ata_device();
+
+ virtual bool is_open() const;
+
+ virtual bool open();
+
+ virtual bool close();
+
+ virtual bool ata_identify_is_cached() const;
+
+protected:
+ virtual int ata_command_interface(smart_command_set command, int select, char * data);
+
+private:
+ // Table of parsed commands, return value, data
+ struct parsed_ata_command
+ {
+ smart_command_set command;
+ int select;
+ int retval, errval;
+ char * data;
+ };
+
+ enum { max_num_commands = 32 };
+ parsed_ata_command m_command_table[max_num_commands];
+
+ int m_num_commands;
+ int m_next_replay_command;
+ bool m_replay_out_of_sync;
+ bool m_ata_identify_is_cached;
+};
+
+static const char * nextline(const char * s, int & lineno)
+{
+ for (s += strcspn(s, "\r\n"); *s == '\r' || *s == '\n'; s++) {
+ if (*s == '\r' && s[1] == '\n')
+ s++;
+ lineno++;
+ }
+ return s;
+}
+
+static int name2command(const char * s)
+{
+ for (int i = 0; i < (int)(sizeof(commandstrings)/sizeof(commandstrings[0])); i++) {
+ if (!strcmp(s, commandstrings[i]))
+ return i;
+ }
+ return -1;
+}
+
+static bool matchcpy(char * dest, size_t size, const char * src,
+ const regular_expression::match_range & srcmatch)
+{
+ if (srcmatch.rm_so < 0)
+ return false;
+ size_t n = srcmatch.rm_eo - srcmatch.rm_so;
+ if (n >= size)
+ n = size-1;
+ memcpy(dest, src + srcmatch.rm_so, n);
+ dest[n] = 0;
+ return true;
+}
+
+static inline int matchtoi(const char * src, const regular_expression::match_range & srcmatch, int defval)
+{
+ if (srcmatch.rm_so < 0)
+ return defval;
+ return atoi(src + srcmatch.rm_so);
+}
+
+parsed_ata_device::parsed_ata_device(smart_interface * intf, const char * dev_name)
+: smart_device(intf, dev_name, "ata", ""),
+ m_num_commands(0),
+ m_next_replay_command(0),
+ m_replay_out_of_sync(false),
+ m_ata_identify_is_cached(false)
+{
+ memset(m_command_table, 0, sizeof(m_command_table));
+}
+
+parsed_ata_device::~parsed_ata_device()
+{
+ parsed_ata_device::close();
+}
+
+bool parsed_ata_device::is_open() const
+{
+ return (m_num_commands > 0);
+}
+
+// Parse stdin and build command table
+bool parsed_ata_device::open()
+{
+ const char * pathname = get_dev_name();
+ if (strcmp(pathname, "-"))
+ return set_err(EINVAL);
+ pathname = "<stdin>";
+ // Fill buffer
+ char buffer[64*1024];
+ int size = 0;
+ while (size < (int)sizeof(buffer)) {
+ int nr = fread(buffer, 1, sizeof(buffer), stdin);
+ if (nr <= 0)
+ break;
+ size += nr;
+ }
+ if (size <= 0)
+ return set_err(ENOENT, "%s: Unexpected EOF", pathname);
+ if (size >= (int)sizeof(buffer))
+ return set_err(EIO, "%s: Buffer overflow", pathname);
+ buffer[size] = 0;
+
+ // Regex to match output from "-r ataioctl,2"
+ static const char pattern[] = "^"
+ "(" // (1
+ "REPORT-IOCTL: DeviceF?D?=[^ ]+ Command=([A-Z ]*[A-Z])" // (2)
+ "(" // (3
+ "( InputParameter=([0-9]+))?" // (4 (5))
+ "|"
+ "( returned (-?[0-9]+)( errno=([0-9]+)[^\r\n]*)?)" // (6 (7) (8 (9)))
+ ")" // )
+ "[\r\n]" // EOL match necessary to match optional parts above
+ "|"
+ "===== \\[([A-Z ]*[A-Z])\\] DATA START " // (10)
+ "|"
+ " *(En|Dis)abled status cached by OS, " // (11)
+ ")"; // )
+
+ // Compile regex
+ const regular_expression regex(pattern);
+
+ // Parse buffer
+ const char * errmsg = 0;
+ int i = -1, state = 0, lineno = 1;
+ for (const char * line = buffer; *line; line = nextline(line, lineno)) {
+ // Match line
+ if (!(line[0] == 'R' || line[0] == '=' || line[0] == ' '))
+ continue;
+ const int nmatch = 1+11;
+ regular_expression::match_range match[nmatch];
+ if (!regex.execute(line, nmatch, match))
+ continue;
+
+ char cmdname[40];
+ if (matchcpy(cmdname, sizeof(cmdname), line, match[2])) { // "REPORT-IOCTL:... Command=%s ..."
+ int nc = name2command(cmdname);
+ if (nc < 0) {
+ errmsg = "Unknown ATA command name"; break;
+ }
+ if (match[7].rm_so < 0) { // "returned %d"
+ // Start of command
+ if (!(state == 0 || state == 2)) {
+ errmsg = "Missing REPORT-IOCTL result"; break;
+ }
+ if (++i >= max_num_commands) {
+ errmsg = "Too many ATA commands"; break;
+ }
+ m_command_table[i].command = (smart_command_set)nc;
+ m_command_table[i].select = matchtoi(line, match[5], 0); // "InputParameter=%d"
+ state = 1;
+ }
+ else {
+ // End of command
+ if (!(state == 1 && (int)m_command_table[i].command == nc)) {
+ errmsg = "Missing REPORT-IOCTL start"; break;
+ }
+ m_command_table[i].retval = matchtoi(line, match[7], -1); // "returned %d"
+ m_command_table[i].errval = matchtoi(line, match[9], 0); // "errno=%d"
+ state = 2;
+ }
+ }
+ else if (matchcpy(cmdname, sizeof(cmdname), line, match[10])) { // "===== [%s] DATA START "
+ // Start of sector hexdump
+ int nc = name2command(cmdname);
+ if (!(state == (nc == WRITE_LOG ? 1 : 2) && (int)m_command_table[i].command == nc)) {
+ errmsg = "Unexpected DATA START"; break;
+ }
+ line = nextline(line, lineno);
+ char * data = (char *)malloc(512);
+ unsigned j;
+ for (j = 0; j < 32; j++) {
+ unsigned b[16];
+ unsigned u1, u2; int n1 = -1;
+ if (!(sscanf(line, "%3u-%3u: "
+ "%2x %2x %2x %2x %2x %2x %2x %2x "
+ "%2x %2x %2x %2x %2x %2x %2x %2x%n",
+ &u1, &u2,
+ b+ 0, b+ 1, b+ 2, b+ 3, b+ 4, b+ 5, b+ 6, b+ 7,
+ b+ 8, b+ 9, b+10, b+11, b+12, b+13, b+14, b+15, &n1) == 18
+ && n1 >= 56 && u1 == j*16 && u2 == j*16+15))
+ break;
+ for (unsigned k = 0; k < 16; k++)
+ data[j*16+k] = b[k];
+ line = nextline(line, lineno);
+ }
+ if (j < 32) {
+ free(data);
+ errmsg = "Incomplete sector hex dump"; break;
+ }
+ m_command_table[i].data = data;
+ if (nc != WRITE_LOG)
+ state = 0;
+ }
+ else if (match[11].rm_so > 0) { // "(En|Dis)abled status cached by OS"
+ m_ata_identify_is_cached = true;
+ }
+ }
+
+ if (!(state == 0 || state == 2))
+ errmsg = "Missing REPORT-IOCTL result";
+
+ if (!errmsg && i < 0)
+ errmsg = "No information found";
+
+ m_num_commands = i+1;
+ m_next_replay_command = 0;
+ m_replay_out_of_sync = false;
+
+ if (errmsg) {
+ close();
+ return set_err(EIO, "%s(%d): Syntax error: %s", pathname, lineno, errmsg);
+ }
+ return true;
+}
+
+// Report warnings and free command table
+bool parsed_ata_device::close()
+{
+ if (m_replay_out_of_sync)
+ pout("REPLAY-IOCTL: Warning: commands replayed out of sync\n");
+ else if (m_next_replay_command != 0)
+ pout("REPLAY-IOCTL: Warning: %d command(s) not replayed\n", m_num_commands-m_next_replay_command);
+
+ for (int i = 0; i < m_num_commands; i++) {
+ if (m_command_table[i].data) {
+ free(m_command_table[i].data); m_command_table[i].data = 0;
+ }
+ }
+ m_num_commands = 0;
+ m_next_replay_command = 0;
+ m_replay_out_of_sync = false;
+ return true;
+}
+
+
+bool parsed_ata_device::ata_identify_is_cached() const
+{
+ return m_ata_identify_is_cached;
+}
+
+
+// Simulate ATA command from command table
+int parsed_ata_device::ata_command_interface(smart_command_set command, int select, char * data)
+{
+ // Find command, try round-robin if out of sync
+ int i = m_next_replay_command;
+ for (int j = 0; ; j++) {
+ if (j >= m_num_commands) {
+ pout("REPLAY-IOCTL: Warning: Command not found\n");
+ errno = ENOSYS;
+ return -1;
+ }
+ if (m_command_table[i].command == command && m_command_table[i].select == select)
+ break;
+ if (!m_replay_out_of_sync) {
+ m_replay_out_of_sync = true;
+ pout("REPLAY-IOCTL: Warning: Command #%d is out of sync\n", i+1);
+ }
+ if (++i >= m_num_commands)
+ i = 0;
+ }
+ m_next_replay_command = i;
+ if (++m_next_replay_command >= m_num_commands)
+ m_next_replay_command = 0;
+
+ // Return command data
+ switch (command) {
+ case IDENTIFY:
+ case PIDENTIFY:
+ case READ_VALUES:
+ case READ_THRESHOLDS:
+ case READ_LOG:
+ if (m_command_table[i].data)
+ memcpy(data, m_command_table[i].data, 512);
+ break;
+ case WRITE_LOG:
+ if (!(m_command_table[i].data && !memcmp(data, m_command_table[i].data, 512)))
+ pout("REPLAY-IOCTL: Warning: WRITE LOG data does not match\n");
+ break;
+ case CHECK_POWER_MODE:
+ data[0] = (char)0xff;
+ default:
+ break;
+ }
+
+ if (m_command_table[i].errval)
+ errno = m_command_table[i].errval;
+ return m_command_table[i].retval;
+}
+
+} // namespace
+
+ata_device * get_parsed_ata_device(smart_interface * intf, const char * dev_name)
+{
+ return new parsed_ata_device(intf, dev_name);
+}