This enables to capture snapshot of controller information via device
coredump machanism.

The nvme device coredump creates the following coredump files.

- regs: NVMe controller registers (00h to 4Fh)
- sq<qid>: Submission queue
- cq<qid>: Completion queue
- telemetry-ctrl-log: Telemetry controller-initiated log (if available)
- data: Empty

The reason for an empty 'data' file is to provide a uniform way to notify
the device coredump is no longer needed by writing the 'data' file.

Since all existing drivers using the device coredump provide a 'data' file
if the nvme device coredump doesn't provide it, the userspace programs need
to know which driver provides what coredump file.

Cc: Johannes Berg <[email protected]>
Cc: Keith Busch <[email protected]>
Cc: Jens Axboe <[email protected]>
Cc: Christoph Hellwig <[email protected]>
Cc: Sagi Grimberg <[email protected]>
Cc: Minwoo Im <[email protected]>
Signed-off-by: Akinobu Mita <[email protected]>
---
* v2
- Exclude the doorbell registers from register dump.
- Save controller registers in a binary format instead of a text format.
- Create an empty 'data' file in the device coredump.
- Save telemetry controller-initiated log if available

 drivers/nvme/host/Kconfig |   1 +
 drivers/nvme/host/core.c  |   1 +
 drivers/nvme/host/pci.c   | 425 ++++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 427 insertions(+)

diff --git a/drivers/nvme/host/Kconfig b/drivers/nvme/host/Kconfig
index 0f345e2..c3a06af 100644
--- a/drivers/nvme/host/Kconfig
+++ b/drivers/nvme/host/Kconfig
@@ -5,6 +5,7 @@ config BLK_DEV_NVME
        tristate "NVM Express block device"
        depends on PCI && BLOCK
        select NVME_CORE
+       select WANT_DEV_COREDUMP
        ---help---
          The NVM Express driver is for solid state drives directly
          connected to the PCI or PCI Express bus.  If you know you
diff --git a/drivers/nvme/host/core.c b/drivers/nvme/host/core.c
index 42f09d6..8d297c7 100644
--- a/drivers/nvme/host/core.c
+++ b/drivers/nvme/host/core.c
@@ -2457,6 +2457,7 @@ int nvme_get_log(struct nvme_ctrl *ctrl, u32 nsid, u8 
log_page, u8 lsp,
 
        return nvme_submit_sync_cmd(ctrl->admin_q, &c, log, size);
 }
+EXPORT_SYMBOL_GPL(nvme_get_log);
 
 static int nvme_get_effects_log(struct nvme_ctrl *ctrl)
 {
diff --git a/drivers/nvme/host/pci.c b/drivers/nvme/host/pci.c
index a90cf5d..4684a86 100644
--- a/drivers/nvme/host/pci.c
+++ b/drivers/nvme/host/pci.c
@@ -9,6 +9,7 @@
 #include <linux/blkdev.h>
 #include <linux/blk-mq.h>
 #include <linux/blk-mq-pci.h>
+#include <linux/devcoredump.h>
 #include <linux/dmi.h>
 #include <linux/init.h>
 #include <linux/interrupt.h>
@@ -131,6 +132,9 @@ struct nvme_dev {
        dma_addr_t host_mem_descs_dma;
        struct nvme_host_mem_buf_desc *host_mem_descs;
        void **host_mem_desc_bufs;
+
+       struct dev_coredumpm_bulk_data *dumps;
+       int num_dumps;
 };
 
 static int io_queue_depth_set(const char *val, const struct kernel_param *kp)
@@ -2867,6 +2871,426 @@ static int nvme_resume(struct device *dev)
 
 static SIMPLE_DEV_PM_OPS(nvme_dev_pm_ops, nvme_suspend, nvme_resume);
 
+#ifdef CONFIG_DEV_COREDUMP
+
+static ssize_t nvme_coredump_read(char *buffer, loff_t offset, size_t count,
+                                 void *data, size_t datalen)
+{
+       return memory_read_from_buffer(buffer, count, &offset, data, datalen);
+}
+
+static void nvme_coredump_free(void *data)
+{
+       kvfree(data);
+}
+
+static int nvme_coredump_empty(struct dev_coredumpm_bulk_data *data)
+{
+       data->name = kstrdup("data", GFP_KERNEL);
+       if (!data->name)
+               return -ENOMEM;
+
+       data->data = NULL;
+       data->datalen = 0;
+       data->read = nvme_coredump_read;
+       data->free = nvme_coredump_free;
+
+       return 0;
+}
+
+static int nvme_coredump_regs(struct dev_coredumpm_bulk_data *data,
+                             struct nvme_ctrl *ctrl)
+{
+       const int reg_size = 0x50; /* 00h to 4Fh */
+
+       data->name = kstrdup("regs", GFP_KERNEL);
+       if (!data->name)
+               return -ENOMEM;
+
+       data->data = kvzalloc(reg_size, GFP_KERNEL);
+       if (!data->data) {
+               kfree(data->name);
+               return -ENOMEM;
+       }
+       memcpy_fromio(data->data, to_nvme_dev(ctrl)->bar, reg_size);
+
+       data->datalen = reg_size;
+       data->read = nvme_coredump_read;
+       data->free = nvme_coredump_free;
+
+       return 0;
+}
+
+static void *kvmemdup(const void *src, size_t len, gfp_t gfp)
+{
+       void *p;
+
+       p = kvmalloc(len, gfp);
+       if (p)
+               memcpy(p, src, len);
+
+       return p;
+}
+
+static int nvme_coredump_queues(struct dev_coredumpm_bulk_data *bulk_data,
+                               struct nvme_ctrl *ctrl)
+{
+       int i;
+
+       for (i = 0; i < ctrl->queue_count; i++) {
+               struct dev_coredumpm_bulk_data *data = &bulk_data[2 * i];
+               struct nvme_queue *nvmeq = &to_nvme_dev(ctrl)->queues[i];
+
+               data[0].name = kasprintf(GFP_KERNEL, "sq%d", i);
+               data[0].data = kvmemdup(nvmeq->sq_cmds,
+                                       SQ_SIZE(nvmeq->q_depth), GFP_KERNEL);
+               data[0].datalen = SQ_SIZE(nvmeq->q_depth);
+               data[0].read = nvme_coredump_read;
+               data[0].free = nvme_coredump_free;
+
+               data[1].name = kasprintf(GFP_KERNEL, "cq%d", i);
+               data[1].data = kvmemdup((void *)nvmeq->cqes,
+                                       CQ_SIZE(nvmeq->q_depth), GFP_KERNEL);
+               data[1].datalen = CQ_SIZE(nvmeq->q_depth);
+               data[1].read = nvme_coredump_read;
+               data[1].free = nvme_coredump_free;
+
+               if (!data[0].name || !data[1].name ||
+                   !data[0].data || !data[1].data)
+                       goto free;
+       }
+
+       return 0;
+free:
+       for (; i >= 0; i--) {
+               struct dev_coredumpm_bulk_data *data = &bulk_data[2 * i];
+
+               kfree(data[0].name);
+               kfree(data[1].name);
+               kvfree(data[0].data);
+               kvfree(data[1].data);
+       }
+
+       return -ENOMEM;
+}
+
+static struct
+dev_coredumpm_bulk_data *nvme_coredump_alloc(struct nvme_dev *dev, int n)
+{
+       struct dev_coredumpm_bulk_data *data;
+
+       data = krealloc(dev->dumps, sizeof(*data) * (dev->num_dumps + n),
+                       GFP_KERNEL | __GFP_ZERO);
+       if (!data)
+               return NULL;
+
+       dev->dumps = data;
+       data += dev->num_dumps;
+       dev->num_dumps += n;
+
+       return data;
+}
+
+static void __nvme_coredump_discard(struct nvme_dev *dev, bool free_data)
+{
+       int i;
+
+       for (i = 0; i < dev->num_dumps; i++) {
+               kfree(dev->dumps[i].name);
+               if (free_data)
+                       dev->dumps[i].free(dev->dumps[i].data);
+       }
+
+       kfree(dev->dumps);
+       dev->dumps = NULL;
+       dev->num_dumps = 0;
+}
+
+static void nvme_coredump_discard(struct nvme_dev *dev)
+{
+       __nvme_coredump_discard(dev, true);
+}
+
+static void nvme_coredump_clear(struct nvme_dev *dev)
+{
+       __nvme_coredump_discard(dev, false);
+}
+
+static int nvme_coredump_prologue(struct nvme_dev *dev)
+{
+       struct nvme_ctrl *ctrl = &dev->ctrl;
+       struct dev_coredumpm_bulk_data *bulk_data;
+       int ret;
+
+       if (WARN_ON(dev->dumps))
+               nvme_coredump_discard(dev);
+
+       bulk_data = nvme_coredump_alloc(dev, 2 + 2 * ctrl->queue_count);
+       if (!bulk_data)
+               return -ENOMEM;
+
+       ret = nvme_coredump_empty(&bulk_data[0]);
+       if (ret)
+               goto free_bulk_data;
+
+       ret = nvme_coredump_regs(&bulk_data[1], ctrl);
+       if (ret)
+               goto free_bulk_data;
+
+       ret = nvme_coredump_queues(&bulk_data[2], ctrl);
+       if (ret)
+               goto free_bulk_data;
+
+       return 0;
+
+free_bulk_data:
+       nvme_coredump_discard(dev);
+
+       return -ENOMEM;
+}
+
+static ssize_t nvme_coredump_read_sgtable(char *buffer, loff_t offset,
+                                         size_t buf_len, void *data,
+                                         size_t data_len)
+{
+       struct sg_table *table = data;
+
+       if (data_len <= offset)
+               return 0;
+
+       if (offset + buf_len > data_len)
+               buf_len = data_len - offset;
+
+       return sg_pcopy_to_buffer(table->sgl, sg_nents(table->sgl), buffer,
+                                 buf_len, offset);
+}
+
+static void nvme_coredump_free_sgtable(void *data)
+{
+       struct sg_table *table = data;
+       struct scatterlist *sg;
+       int n = sg_nents(table->sgl);
+       int i;
+
+       for_each_sg(table->sgl, sg, n, i) {
+               struct page *page = sg_page(sg);
+
+               if (page)
+                       __free_page(page);
+
+       }
+
+       kfree(table);
+}
+
+static struct sg_table *nvme_coredump_alloc_sgtable(size_t bytes)
+{
+       struct sg_table *table;
+       struct scatterlist *sg;
+       int n = DIV_ROUND_UP(bytes, PAGE_SIZE);
+       int i;
+
+       table = kzalloc(sizeof(*table), GFP_KERNEL);
+       if (!table)
+               return NULL;
+
+       if (sg_alloc_table(table, n, GFP_KERNEL))
+               goto free_table;
+
+       for_each_sg(table->sgl, sg, n, i) {
+               struct page *page = alloc_page(GFP_KERNEL);
+               size_t size = min_t(int, bytes, PAGE_SIZE);
+
+               if (!page)
+                       goto free_page;
+
+               sg_set_page(sg, page, size, 0);
+               bytes -= size;
+       }
+
+       return table;
+free_page:
+       for_each_sg(table->sgl, sg, n, i) {
+               struct page *page = sg_page(sg);
+
+               if (page)
+                       __free_page(page);
+
+       }
+free_table:
+       kfree(table);
+
+       return NULL;
+}
+
+static int nvme_get_telemetry_log_blocks(struct nvme_ctrl *ctrl, void *buf,
+                                        size_t bytes, loff_t offset)
+{
+       const size_t chunk_size = ctrl->max_hw_sectors * ctrl->page_size;
+       loff_t pos = 0;
+
+       while (pos < bytes) {
+               size_t size = min_t(size_t, bytes - pos, chunk_size);
+               int ret;
+
+               ret = nvme_get_log(ctrl, NVME_NSID_ALL, NVME_LOG_TELEMETRY_CTRL,
+                                  0, buf + pos, size, offset + pos);
+               if (ret)
+                       return ret;
+
+               pos += size;
+       }
+
+       return 0;
+}
+
+static int nvme_get_telemetry_log(struct nvme_ctrl *ctrl,
+                                 struct sg_table *table, size_t bytes)
+{
+       int n = sg_nents(table->sgl);
+       struct scatterlist *sg;
+       size_t offset = 0;
+       int i;
+
+       for_each_sg(table->sgl, sg, n, i) {
+               struct page *page = sg_page(sg);
+               size_t size = min_t(int, bytes - offset, sg->length);
+               int ret;
+
+               ret = nvme_get_telemetry_log_blocks(ctrl, page_address(page),
+                                                   size, offset);
+               if (ret)
+                       return ret;
+
+               offset += size;
+       }
+
+       return 0;
+}
+
+static int nvme_coredump_telemetry_log(struct dev_coredumpm_bulk_data *data,
+                                      struct nvme_ctrl *ctrl)
+{
+       struct nvme_telemetry_log_page_hdr *page_hdr;
+       struct sg_table *table;
+       int log_size;
+       int ret;
+       u8 ctrldgn;
+
+       if (!(ctrl->lpa & NVME_CTRL_LPA_TELEMETRY_LOG))
+               return -EINVAL;
+       if (!(ctrl->lpa & NVME_CTRL_LPA_EXTENDED_DATA))
+               return -EINVAL;
+
+       page_hdr = kzalloc(sizeof(*page_hdr), GFP_KERNEL);
+       if (!page_hdr)
+               return -ENOMEM;
+
+       ret = nvme_get_log(ctrl, NVME_NSID_ALL, NVME_LOG_TELEMETRY_CTRL, 0,
+                          page_hdr, sizeof(*page_hdr), 0);
+       if (ret)
+               goto free_page_hdr;
+
+       if (!page_hdr->ctrlavail) {
+               ret = -EINVAL;
+               goto free_page_hdr;
+       }
+
+       log_size = (le16_to_cpu(page_hdr->dalb3) + 1) * 512;
+
+       table = nvme_coredump_alloc_sgtable(log_size);
+       if (!table) {
+               ret = -ENOMEM;
+               goto free_page_hdr;
+       }
+
+       ret = nvme_get_telemetry_log(ctrl, table, log_size);
+       if (ret)
+               goto free_table;
+
+       sg_pcopy_to_buffer(table->sgl, sg_nents(table->sgl), &ctrldgn,
+                          sizeof(ctrldgn),
+                          offsetof(typeof(*page_hdr), ctrldgn));
+
+       ret = nvme_get_log(ctrl, NVME_NSID_ALL, NVME_LOG_TELEMETRY_CTRL, 0,
+                          page_hdr, sizeof(*page_hdr), 0);
+       if (ret)
+               goto free_table;
+
+       if (page_hdr->ctrldgn != ctrldgn) {
+               ret = -EINVAL;
+               goto free_table;
+       }
+
+       data->name = kstrdup("telemetry-ctrl-log", GFP_KERNEL);
+       if (!data->name) {
+               ret = -ENOMEM;
+               goto free_table;
+       }
+
+       data->data = table;
+       data->datalen = log_size;
+       data->read = nvme_coredump_read_sgtable;
+       data->free = nvme_coredump_free_sgtable;
+
+       kfree(page_hdr);
+
+       return 0;
+free_table:
+       nvme_coredump_free_sgtable(table);
+free_page_hdr:
+       kfree(page_hdr);
+
+       return ret;
+}
+
+static void nvme_coredump_epilogue(struct nvme_dev *dev)
+{
+       struct dev_coredumpm_bulk_data *bulk_data;
+
+       if (!dev->dumps)
+               return;
+
+       bulk_data = nvme_coredump_alloc(dev, 1);
+       if (bulk_data) {
+               if (nvme_coredump_telemetry_log(bulk_data, &dev->ctrl))
+                       dev->num_dumps--;
+       }
+
+       dev_coredumpm_bulk(dev->dev, THIS_MODULE, GFP_KERNEL,
+                          dev->dumps, dev->num_dumps);
+       nvme_coredump_clear(dev);
+}
+
+static void nvme_coredump(struct device *dev)
+{
+       struct nvme_dev *ndev = dev_get_drvdata(dev);
+
+       mutex_lock(&ndev->shutdown_lock);
+
+       nvme_coredump_prologue(ndev);
+       nvme_coredump_epilogue(ndev);
+
+       mutex_unlock(&ndev->shutdown_lock);
+}
+
+#else
+
+static int nvme_coredump_prologue(struct nvme_dev *dev)
+{
+       return 0;
+}
+
+static void nvme_coredump_epilogue(struct nvme_dev *dev)
+{
+}
+
+static void nvme_coredump(struct device *dev)
+{
+}
+
+#endif /* CONFIG_DEV_COREDUMP */
+
 static pci_ers_result_t nvme_error_detected(struct pci_dev *pdev,
                                                pci_channel_state_t state)
 {
@@ -2972,6 +3396,7 @@ static struct pci_driver nvme_driver = {
        .shutdown       = nvme_shutdown,
        .driver         = {
                .pm     = &nvme_dev_pm_ops,
+               .coredump = nvme_coredump,
        },
        .sriov_configure = pci_sriov_configure_simple,
        .err_handler    = &nvme_err_handler,
-- 
2.7.4

Reply via email to