]> git.proxmox.com Git - mirror_smartmontools-debian.git/blobdiff - nvmecmds.cpp
Imported Upstream version 6.5+svn4324
[mirror_smartmontools-debian.git] / nvmecmds.cpp
diff --git a/nvmecmds.cpp b/nvmecmds.cpp
new file mode 100644 (file)
index 0000000..b3f6a0f
--- /dev/null
@@ -0,0 +1,248 @@
+/*
+ * nvmecmds.cpp
+ *
+ * Home page of code is: http://www.smartmontools.org
+ *
+ * Copyright (C) 2016 Christian Franke
+ *
+ * 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, or (at your option)
+ * any later version.
+ *
+ * You should have received a copy of the GNU General Public License
+ * (for example COPYING); If not, see <http://www.gnu.org/licenses/>.
+ *
+ */
+
+#include "config.h"
+#include "nvmecmds.h"
+
+const char * nvmecmds_cvsid = "$Id: nvmecmds.cpp 4313 2016-05-01 16:17:53Z chrfranke $"
+  NVMECMDS_H_CVSID;
+
+#include "dev_interface.h"
+#include "atacmds.h" // swapx(), ASSERT_*(), dont_print_serial_number
+#include "scsicmds.h" // dStrHex()
+#include "utility.h"
+
+using namespace smartmontools;
+
+// Check nvme_* struct sizes
+ASSERT_SIZEOF_STRUCT(nvme_id_ctrl, 4096);
+ASSERT_SIZEOF_STRUCT(nvme_id_ns, 4096);
+ASSERT_SIZEOF_STRUCT(nvme_error_log_page, 64);
+ASSERT_SIZEOF_STRUCT(nvme_smart_log, 512);
+
+
+// Print NVMe debug messages?
+unsigned char nvme_debugmode = 0;
+
+// Dump up to 4096 bytes, do not dump trailing zero bytes.
+// TODO: Handle this by new unified function in utility.cpp
+static void debug_hex_dump(const void * data, unsigned size)
+{
+  const unsigned char * p = (const unsigned char *)data;
+  const unsigned limit = 4096; // sizeof(nvme_id_ctrl)
+  unsigned sz = (size <= limit ? size : limit);
+
+  while (sz > 0x10 && !p[sz-1])
+    sz--;
+  if (sz < size) {
+    if (sz & 0x0f)
+      sz = (sz & ~0x0f) + 0x10;
+    sz += 0x10;
+    if (sz > size)
+      sz = size;
+  }
+
+  dStrHex(p, sz, 0);
+  if (sz < size)
+    pout(" ...\n");
+}
+
+// Call NVMe pass-through and print debug info if requested.
+static bool nvme_pass_through(nvme_device * device, const nvme_cmd_in & in,
+  nvme_cmd_out & out)
+{
+  int64_t start_usec = -1;
+
+  if (nvme_debugmode) {
+    pout(" [NVMe call: opcode=0x%02x, size=0x%04x, nsid=0x%08x, cdw10=0x%08x",
+      in.opcode, in.size, in.nsid, in.cdw10);
+    if (in.cdw11 || in.cdw12 || in.cdw13 || in.cdw14 || in.cdw15)
+      pout(",\n  cdw1x=0x%08x, 0x%08x, 0x%08x, 0x%08x, 0x%08x",
+       in.cdw11, in.cdw12, in.cdw13, in.cdw14, in.cdw15);
+    pout("]\n");
+
+    start_usec = smi()->get_timer_usec();
+  }
+
+  bool ok = device->nvme_pass_through(in, out);
+
+  if (   dont_print_serial_number && ok
+      && in.opcode == nvme_admin_identify && in.cdw10 == 0x01) {
+        // Invalidate serial number
+        nvme_id_ctrl & id_ctrl = *reinterpret_cast<nvme_id_ctrl *>(in.buffer);
+        memset(id_ctrl.sn, 'X', sizeof(id_ctrl.sn));
+  }
+
+  if (nvme_debugmode) {
+    if (start_usec >= 0) {
+      int64_t duration_usec = smi()->get_timer_usec() - start_usec;
+      if (duration_usec >= 500)
+        pout("  [Duration: %.3fs]\n", duration_usec / 1000000.0);
+    }
+
+    if (!ok) {
+      pout(" [NVMe call failed: ");
+      if (out.status_valid)
+        pout("NVMe Status=0x%04x", out.status);
+      else
+        pout("%s", device->get_errmsg());
+    }
+    else {
+      pout(" [NVMe call succeeded: result=0x%08x", out.result);
+      if (nvme_debugmode > 1 && in.direction() == nvme_cmd_in::data_in) {
+        pout("\n");
+        debug_hex_dump(in.buffer, in.size);
+        pout(" ");
+      }
+    }
+    pout("]\n");
+  }
+
+  return ok;
+}
+
+// Call NVMe pass-through and print debug info if requested.
+// Version without output parameters.
+static bool nvme_pass_through(nvme_device * device, const nvme_cmd_in & in)
+{
+  nvme_cmd_out out;
+  return nvme_pass_through(device, in, out);
+}
+
+// Read NVMe identify info with controller/namespace field CNS.
+static bool nvme_read_identify(nvme_device * device, unsigned nsid,
+  unsigned char cns, void * data, unsigned size)
+{
+  memset(data, 0, size);
+  nvme_cmd_in in;
+  in.set_data_in(nvme_admin_identify, data, size);
+  in.nsid = nsid;
+  in.cdw10 = cns;
+
+  return nvme_pass_through(device, in);
+}
+
+// Read NVMe Identify Controller data structure.
+bool nvme_read_id_ctrl(nvme_device * device, nvme_id_ctrl & id_ctrl)
+{
+  if (!nvme_read_identify(device, 0, 0x01, &id_ctrl, sizeof(id_ctrl)))
+    return false;
+
+  if (isbigendian()) {
+    swapx(&id_ctrl.vid);
+    swapx(&id_ctrl.ssvid);
+    swapx(&id_ctrl.cntlid);
+    swapx(&id_ctrl.oacs);
+    swapx(&id_ctrl.wctemp);
+    swapx(&id_ctrl.cctemp);
+    swapx(&id_ctrl.mtfa);
+    swapx(&id_ctrl.hmpre);
+    swapx(&id_ctrl.hmmin);
+    swapx(&id_ctrl.rpmbs);
+    swapx(&id_ctrl.nn);
+    swapx(&id_ctrl.oncs);
+    swapx(&id_ctrl.fuses);
+    swapx(&id_ctrl.awun);
+    swapx(&id_ctrl.awupf);
+    swapx(&id_ctrl.acwu);
+    swapx(&id_ctrl.sgls);
+    for (int i = 0; i < 32; i++) {
+      swapx(&id_ctrl.psd[i].max_power);
+      swapx(&id_ctrl.psd[i].entry_lat);
+      swapx(&id_ctrl.psd[i].exit_lat);
+      swapx(&id_ctrl.psd[i].idle_power);
+      swapx(&id_ctrl.psd[i].active_power);
+    }
+  }
+
+  return true;
+}
+
+// Read NVMe Identify Namespace data structure for namespace NSID.
+bool nvme_read_id_ns(nvme_device * device, unsigned nsid, nvme_id_ns & id_ns)
+{
+  if (!nvme_read_identify(device, nsid, 0x00, &id_ns, sizeof(id_ns)))
+    return false;
+
+  if (isbigendian()) {
+    swapx(&id_ns.nsze);
+    swapx(&id_ns.ncap);
+    swapx(&id_ns.nuse);
+    swapx(&id_ns.nawun);
+    swapx(&id_ns.nawupf);
+    swapx(&id_ns.nacwu);
+    swapx(&id_ns.nabsn);
+    swapx(&id_ns.nabo);
+    swapx(&id_ns.nabspf);
+    for (int i = 0; i < 16; i++)
+      swapx(&id_ns.lbaf[i].ms);
+  }
+
+  return true;
+}
+
+// Read NVMe log page with identifier LID.
+bool nvme_read_log_page(nvme_device * device, unsigned char lid, void * data, unsigned size)
+{
+  if (!(4 <= size && size <= 0x4000 && (size % 4) == 0))
+    throw std::logic_error("nvme_read_log_page(): invalid size");
+
+  memset(data, 0, size);
+  nvme_cmd_in in;
+  in.set_data_in(nvme_admin_get_log_page, data, size);
+  in.nsid = device->get_nsid();
+  in.cdw10 = lid | (((size / 4) - 1) << 16);
+
+  return nvme_pass_through(device, in);
+}
+
+// Read NVMe Error Information Log.
+bool nvme_read_error_log(nvme_device * device, nvme_error_log_page * error_log, unsigned num_entries)
+{
+  if (!nvme_read_log_page(device, 0x01, error_log, num_entries * sizeof(*error_log)))
+    return false;
+
+  if (isbigendian()) {
+    for (unsigned i = 0; i < num_entries; i++) {
+      swapx(&error_log[i].error_count);
+      swapx(&error_log[i].sqid);
+      swapx(&error_log[i].cmdid);
+      swapx(&error_log[i].status_field);
+      swapx(&error_log[i].parm_error_location);
+      swapx(&error_log[i].lba);
+      swapx(&error_log[i].nsid);
+    }
+  }
+
+  return true;
+}
+
+// Read NVMe SMART/Health Information log.
+bool nvme_read_smart_log(nvme_device * device, nvme_smart_log & smart_log)
+{
+  if (!nvme_read_log_page(device, 0x02, &smart_log, sizeof(smart_log)))
+    return false;
+
+  if (isbigendian()) {
+    swapx(&smart_log.warning_temp_time);
+    swapx(&smart_log.critical_comp_time);
+    for (int i = 0; i < 8; i++)
+      swapx(&smart_log.temp_sensor[i]);
+  }
+
+  return true;
+}