Quelle  admin-cmd.c   Sprache: C

// SPDX-License-Identifier: GPL-2.0
/*
 * NVMe admin command implementation.
 * Copyright (c) 2015-2016 HGST, a Western Digital Company.
 */
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
#include <linux/module.h>
#include <linux/rculist.h>
#include <linux/part_stat.h>

#include <generated/utsrelease.h>
#include <linux/unaligned.h>
#include "nvmet.h"

static void nvmet_execute_delete_sq(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 u16 sqid = le16_to_cpu(req->cmd->delete_queue.qid);
 u16 status;

 if (!nvmet_is_pci_ctrl(ctrl)) {
  status = nvmet_report_invalid_opcode(req);
  goto complete;
 }

 if (!sqid) {
  status = NVME_SC_QID_INVALID | NVME_STATUS_DNR;
  goto complete;
 }

 status = nvmet_check_sqid(ctrl, sqid, false);
 if (status != NVME_SC_SUCCESS)
  goto complete;

 status = ctrl->ops->delete_sq(ctrl, sqid);

complete:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_create_sq(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvme_command *cmd = req->cmd;
 u16 sqid = le16_to_cpu(cmd->create_sq.sqid);
 u16 cqid = le16_to_cpu(cmd->create_sq.cqid);
 u16 sq_flags = le16_to_cpu(cmd->create_sq.sq_flags);
 u16 qsize = le16_to_cpu(cmd->create_sq.qsize);
 u64 prp1 = le64_to_cpu(cmd->create_sq.prp1);
 u16 status;

 if (!nvmet_is_pci_ctrl(ctrl)) {
  status = nvmet_report_invalid_opcode(req);
  goto complete;
 }

 if (!sqid) {
  status = NVME_SC_QID_INVALID | NVME_STATUS_DNR;
  goto complete;
 }

 status = nvmet_check_sqid(ctrl, sqid, true);
 if (status != NVME_SC_SUCCESS)
  goto complete;

 status = nvmet_check_io_cqid(ctrl, cqid, false);
 if (status != NVME_SC_SUCCESS) {
  pr_err("SQ %u: Invalid CQID %u\n", sqid, cqid);
  goto complete;
 }

 if (!qsize || qsize > NVME_CAP_MQES(ctrl->cap)) {
  status = NVME_SC_QUEUE_SIZE | NVME_STATUS_DNR;
  goto complete;
 }

 status = ctrl->ops->create_sq(ctrl, sqid, cqid, sq_flags, qsize, prp1);

complete:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_delete_cq(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 u16 cqid = le16_to_cpu(req->cmd->delete_queue.qid);
 u16 status;

 if (!nvmet_is_pci_ctrl(ctrl)) {
  status = nvmet_report_invalid_opcode(req);
  goto complete;
 }

 status = nvmet_check_io_cqid(ctrl, cqid, false);
 if (status != NVME_SC_SUCCESS)
  goto complete;

 if (!ctrl->cqs[cqid] || nvmet_cq_in_use(ctrl->cqs[cqid])) {
  /* Some SQs are still using this CQ */
  status = NVME_SC_QID_INVALID | NVME_STATUS_DNR;
  goto complete;
 }

 status = ctrl->ops->delete_cq(ctrl, cqid);

complete:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_create_cq(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvme_command *cmd = req->cmd;
 u16 cqid = le16_to_cpu(cmd->create_cq.cqid);
 u16 cq_flags = le16_to_cpu(cmd->create_cq.cq_flags);
 u16 qsize = le16_to_cpu(cmd->create_cq.qsize);
 u16 irq_vector = le16_to_cpu(cmd->create_cq.irq_vector);
 u64 prp1 = le64_to_cpu(cmd->create_cq.prp1);
 u16 status;

 if (!nvmet_is_pci_ctrl(ctrl)) {
  status = nvmet_report_invalid_opcode(req);
  goto complete;
 }

 status = nvmet_check_io_cqid(ctrl, cqid, true);
 if (status != NVME_SC_SUCCESS)
  goto complete;

 if (!qsize || qsize > NVME_CAP_MQES(ctrl->cap)) {
  status = NVME_SC_QUEUE_SIZE | NVME_STATUS_DNR;
  goto complete;
 }

 status = ctrl->ops->create_cq(ctrl, cqid, cq_flags, qsize,
          prp1, irq_vector);

complete:
 nvmet_req_complete(req, status);
}

u32 nvmet_get_log_page_len(struct nvme_command *cmd)
{
 u32 len = le16_to_cpu(cmd->get_log_page.numdu);

 len <<= 16;
 len += le16_to_cpu(cmd->get_log_page.numdl);
 /* NUMD is a 0's based value */
 len += 1;
 len *= sizeof(u32);

 return len;
}

static u32 nvmet_feat_data_len(struct nvmet_req *req, u32 cdw10)
{
 switch (cdw10 & 0xff) {
 case NVME_FEAT_HOST_ID:
  return sizeof(req->sq->ctrl->hostid);
 default:
  return 0;
 }
}

u64 nvmet_get_log_page_offset(struct nvme_command *cmd)
{
 return le64_to_cpu(cmd->get_log_page.lpo);
}

static void nvmet_execute_get_log_page_noop(struct nvmet_req *req)
{
 nvmet_req_complete(req, nvmet_zero_sgl(req, 0, req->transfer_len));
}

static void nvmet_execute_get_log_page_error(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 unsigned long flags;
 off_t offset = 0;
 u64 slot;
 u64 i;

 spin_lock_irqsave(&ctrl->error_lock, flags);
 slot = ctrl->err_counter % NVMET_ERROR_LOG_SLOTS;

 for (i = 0; i < NVMET_ERROR_LOG_SLOTS; i++) {
  if (nvmet_copy_to_sgl(req, offset, &ctrl->slots[slot],
    sizeof(struct nvme_error_slot)))
   break;

  if (slot == 0)
   slot = NVMET_ERROR_LOG_SLOTS - 1;
  else
   slot--;
  offset += sizeof(struct nvme_error_slot);
 }
 spin_unlock_irqrestore(&ctrl->error_lock, flags);
 nvmet_req_complete(req, 0);
}

static void nvmet_execute_get_supported_log_pages(struct nvmet_req *req)
{
 struct nvme_supported_log *logs;
 u16 status;

 logs = kzalloc(sizeof(*logs), GFP_KERNEL);
 if (!logs) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 logs->lids[NVME_LOG_SUPPORTED] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_ERROR] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_SMART] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_FW_SLOT] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_CHANGED_NS] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_CMD_EFFECTS] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_ENDURANCE_GROUP] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_ANA] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_FEATURES] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_RMI] = cpu_to_le32(NVME_LIDS_LSUPP);
 logs->lids[NVME_LOG_RESERVATION] = cpu_to_le32(NVME_LIDS_LSUPP);

 status = nvmet_copy_to_sgl(req, 0, logs, sizeof(*logs));
 kfree(logs);
out:
 nvmet_req_complete(req, status);
}

static u16 nvmet_get_smart_log_nsid(struct nvmet_req *req,
  struct nvme_smart_log *slog)
{
 u64 host_reads, host_writes, data_units_read, data_units_written;
 u16 status;

 status = nvmet_req_find_ns(req);
 if (status)
  return status;

 /* we don't have the right data for file backed ns */
 if (!req->ns->bdev)
  return NVME_SC_SUCCESS;

 host_reads = part_stat_read(req->ns->bdev, ios[READ]);
 data_units_read =
  DIV_ROUND_UP(part_stat_read(req->ns->bdev, sectors[READ]), 1000);
 host_writes = part_stat_read(req->ns->bdev, ios[WRITE]);
 data_units_written =
  DIV_ROUND_UP(part_stat_read(req->ns->bdev, sectors[WRITE]), 1000);

 put_unaligned_le64(host_reads, &slog->host_reads[0]);
 put_unaligned_le64(data_units_read, &slog->data_units_read[0]);
 put_unaligned_le64(host_writes, &slog->host_writes[0]);
 put_unaligned_le64(data_units_written, &slog->data_units_written[0]);

 return NVME_SC_SUCCESS;
}

static u16 nvmet_get_smart_log_all(struct nvmet_req *req,
  struct nvme_smart_log *slog)
{
 u64 host_reads = 0, host_writes = 0;
 u64 data_units_read = 0, data_units_written = 0;
 struct nvmet_ns *ns;
 struct nvmet_ctrl *ctrl;
 unsigned long idx;

 ctrl = req->sq->ctrl;
 nvmet_for_each_enabled_ns(&ctrl->subsys->namespaces, idx, ns) {
  /* we don't have the right data for file backed ns */
  if (!ns->bdev)
   continue;
  host_reads += part_stat_read(ns->bdev, ios[READ]);
  data_units_read += DIV_ROUND_UP(
   part_stat_read(ns->bdev, sectors[READ]), 1000);
  host_writes += part_stat_read(ns->bdev, ios[WRITE]);
  data_units_written += DIV_ROUND_UP(
   part_stat_read(ns->bdev, sectors[WRITE]), 1000);
 }

 put_unaligned_le64(host_reads, &slog->host_reads[0]);
 put_unaligned_le64(data_units_read, &slog->data_units_read[0]);
 put_unaligned_le64(host_writes, &slog->host_writes[0]);
 put_unaligned_le64(data_units_written, &slog->data_units_written[0]);

 return NVME_SC_SUCCESS;
}

static void nvmet_execute_get_log_page_rmi(struct nvmet_req *req)
{
 struct nvme_rotational_media_log *log;
 struct gendisk *disk;
 u16 status;

 req->cmd->common.nsid = cpu_to_le32(le16_to_cpu(
         req->cmd->get_log_page.lsi));
 status = nvmet_req_find_ns(req);
 if (status)
  goto out;

 if (!req->ns->bdev || bdev_nonrot(req->ns->bdev)) {
  status = NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
  goto out;
 }

 if (req->transfer_len != sizeof(*log)) {
  status = NVME_SC_SGL_INVALID_DATA | NVME_STATUS_DNR;
  goto out;
 }

 log = kzalloc(sizeof(*log), GFP_KERNEL);
 if (!log)
  goto out;

 log->endgid = req->cmd->get_log_page.lsi;
 disk = req->ns->bdev->bd_disk;
 if (disk && disk->ia_ranges)
  log->numa = cpu_to_le16(disk->ia_ranges->nr_ia_ranges);
 else
  log->numa = cpu_to_le16(1);

 status = nvmet_copy_to_sgl(req, 0, log, sizeof(*log));
 kfree(log);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_get_log_page_smart(struct nvmet_req *req)
{
 struct nvme_smart_log *log;
 u16 status = NVME_SC_INTERNAL;
 unsigned long flags;

 if (req->transfer_len != sizeof(*log))
  goto out;

 log = kzalloc(sizeof(*log), GFP_KERNEL);
 if (!log)
  goto out;

 if (req->cmd->get_log_page.nsid == cpu_to_le32(NVME_NSID_ALL))
  status = nvmet_get_smart_log_all(req, log);
 else
  status = nvmet_get_smart_log_nsid(req, log);
 if (status)
  goto out_free_log;

 spin_lock_irqsave(&req->sq->ctrl->error_lock, flags);
 put_unaligned_le64(req->sq->ctrl->err_counter,
   &log->num_err_log_entries);
 spin_unlock_irqrestore(&req->sq->ctrl->error_lock, flags);

 status = nvmet_copy_to_sgl(req, 0, log, sizeof(*log));
out_free_log:
 kfree(log);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_get_cmd_effects_admin(struct nvmet_ctrl *ctrl,
     struct nvme_effects_log *log)
{
 /* For a PCI target controller, advertize support for the . */
 if (nvmet_is_pci_ctrl(ctrl)) {
  log->acs[nvme_admin_delete_sq] =
  log->acs[nvme_admin_create_sq] =
  log->acs[nvme_admin_delete_cq] =
  log->acs[nvme_admin_create_cq] =
   cpu_to_le32(NVME_CMD_EFFECTS_CSUPP);
 }

 log->acs[nvme_admin_get_log_page] =
 log->acs[nvme_admin_identify] =
 log->acs[nvme_admin_abort_cmd] =
 log->acs[nvme_admin_set_features] =
 log->acs[nvme_admin_get_features] =
 log->acs[nvme_admin_async_event] =
 log->acs[nvme_admin_keep_alive] =
  cpu_to_le32(NVME_CMD_EFFECTS_CSUPP);
}

static void nvmet_get_cmd_effects_nvm(struct nvme_effects_log *log)
{
 log->iocs[nvme_cmd_read] =
 log->iocs[nvme_cmd_flush] =
 log->iocs[nvme_cmd_dsm] =
 log->iocs[nvme_cmd_resv_acquire] =
 log->iocs[nvme_cmd_resv_register] =
 log->iocs[nvme_cmd_resv_release] =
 log->iocs[nvme_cmd_resv_report] =
  cpu_to_le32(NVME_CMD_EFFECTS_CSUPP);
 log->iocs[nvme_cmd_write] =
 log->iocs[nvme_cmd_write_zeroes] =
  cpu_to_le32(NVME_CMD_EFFECTS_CSUPP | NVME_CMD_EFFECTS_LBCC);
}

static void nvmet_get_cmd_effects_zns(struct nvme_effects_log *log)
{
 log->iocs[nvme_cmd_zone_append] =
 log->iocs[nvme_cmd_zone_mgmt_send] =
  cpu_to_le32(NVME_CMD_EFFECTS_CSUPP | NVME_CMD_EFFECTS_LBCC);
 log->iocs[nvme_cmd_zone_mgmt_recv] =
  cpu_to_le32(NVME_CMD_EFFECTS_CSUPP);
}

static void nvmet_execute_get_log_cmd_effects_ns(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvme_effects_log *log;
 u16 status = NVME_SC_SUCCESS;

 log = kzalloc(sizeof(*log), GFP_KERNEL);
 if (!log) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 switch (req->cmd->get_log_page.csi) {
 case NVME_CSI_NVM:
  nvmet_get_cmd_effects_admin(ctrl, log);
  nvmet_get_cmd_effects_nvm(log);
  break;
 case NVME_CSI_ZNS:
  if (!IS_ENABLED(CONFIG_BLK_DEV_ZONED)) {
   status = NVME_SC_INVALID_IO_CMD_SET;
   goto free;
  }
  nvmet_get_cmd_effects_admin(ctrl, log);
  nvmet_get_cmd_effects_nvm(log);
  nvmet_get_cmd_effects_zns(log);
  break;
 default:
  status = NVME_SC_INVALID_LOG_PAGE;
  goto free;
 }

 status = nvmet_copy_to_sgl(req, 0, log, sizeof(*log));
free:
 kfree(log);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_get_log_changed_ns(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 u16 status = NVME_SC_INTERNAL;
 size_t len;

 if (req->transfer_len != NVME_MAX_CHANGED_NAMESPACES * sizeof(__le32))
  goto out;

 mutex_lock(&ctrl->lock);
 if (ctrl->nr_changed_ns == U32_MAX)
  len = sizeof(__le32);
 else
  len = ctrl->nr_changed_ns * sizeof(__le32);
 status = nvmet_copy_to_sgl(req, 0, ctrl->changed_ns_list, len);
 if (!status)
  status = nvmet_zero_sgl(req, len, req->transfer_len - len);
 ctrl->nr_changed_ns = 0;
 nvmet_clear_aen_bit(req, NVME_AEN_BIT_NS_ATTR);
 mutex_unlock(&ctrl->lock);
out:
 nvmet_req_complete(req, status);
}

static u32 nvmet_format_ana_group(struct nvmet_req *req, u32 grpid,
  struct nvme_ana_group_desc *desc)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvmet_ns *ns;
 unsigned long idx;
 u32 count = 0;

 if (!(req->cmd->get_log_page.lsp & NVME_ANA_LOG_RGO)) {
  nvmet_for_each_enabled_ns(&ctrl->subsys->namespaces, idx, ns) {
   if (ns->anagrpid == grpid)
    desc->nsids[count++] = cpu_to_le32(ns->nsid);
  }
 }

 desc->grpid = cpu_to_le32(grpid);
 desc->nnsids = cpu_to_le32(count);
 desc->chgcnt = cpu_to_le64(nvmet_ana_chgcnt);
 desc->state = req->port->ana_state[grpid];
 memset(desc->rsvd17, 0, sizeof(desc->rsvd17));
 return struct_size(desc, nsids, count);
}

static void nvmet_execute_get_log_page_endgrp(struct nvmet_req *req)
{
 u64 host_reads, host_writes, data_units_read, data_units_written;
 struct nvme_endurance_group_log *log;
 u16 status;

 /*
 * The target driver emulates each endurance group as its own
 * namespace, reusing the nsid as the endurance group identifier.
 */
 req->cmd->common.nsid = cpu_to_le32(le16_to_cpu(
         req->cmd->get_log_page.lsi));
 status = nvmet_req_find_ns(req);
 if (status)
  goto out;

 log = kzalloc(sizeof(*log), GFP_KERNEL);
 if (!log) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 if (!req->ns->bdev)
  goto copy;

 host_reads = part_stat_read(req->ns->bdev, ios[READ]);
 data_units_read =
  DIV_ROUND_UP(part_stat_read(req->ns->bdev, sectors[READ]), 1000);
 host_writes = part_stat_read(req->ns->bdev, ios[WRITE]);
 data_units_written =
  DIV_ROUND_UP(part_stat_read(req->ns->bdev, sectors[WRITE]), 1000);

 put_unaligned_le64(host_reads, &log->hrc[0]);
 put_unaligned_le64(data_units_read, &log->dur[0]);
 put_unaligned_le64(host_writes, &log->hwc[0]);
 put_unaligned_le64(data_units_written, &log->duw[0]);
copy:
 status = nvmet_copy_to_sgl(req, 0, log, sizeof(*log));
 kfree(log);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_get_log_page_ana(struct nvmet_req *req)
{
 struct nvme_ana_rsp_hdr hdr = { 0, };
 struct nvme_ana_group_desc *desc;
 size_t offset = sizeof(struct nvme_ana_rsp_hdr); /* start beyond hdr */
 size_t len;
 u32 grpid;
 u16 ngrps = 0;
 u16 status;

 status = NVME_SC_INTERNAL;
 desc = kmalloc(struct_size(desc, nsids, NVMET_MAX_NAMESPACES),
         GFP_KERNEL);
 if (!desc)
  goto out;

 down_read(&nvmet_ana_sem);
 for (grpid = 1; grpid <= NVMET_MAX_ANAGRPS; grpid++) {
  if (!nvmet_ana_group_enabled[grpid])
   continue;
  len = nvmet_format_ana_group(req, grpid, desc);
  status = nvmet_copy_to_sgl(req, offset, desc, len);
  if (status)
   break;
  offset += len;
  ngrps++;
 }
 for ( ; grpid <= NVMET_MAX_ANAGRPS; grpid++) {
  if (nvmet_ana_group_enabled[grpid])
   ngrps++;
 }

 hdr.chgcnt = cpu_to_le64(nvmet_ana_chgcnt);
 hdr.ngrps = cpu_to_le16(ngrps);
 nvmet_clear_aen_bit(req, NVME_AEN_BIT_ANA_CHANGE);
 up_read(&nvmet_ana_sem);

 kfree(desc);

 /* copy the header last once we know the number of groups */
 status = nvmet_copy_to_sgl(req, 0, &hdr, sizeof(hdr));
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_get_log_page_features(struct nvmet_req *req)
{
 struct nvme_supported_features_log *features;
 u16 status;

 features = kzalloc(sizeof(*features), GFP_KERNEL);
 if (!features) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 features->fis[NVME_FEAT_NUM_QUEUES] =
  cpu_to_le32(NVME_FIS_FSUPP | NVME_FIS_CSCPE);
 features->fis[NVME_FEAT_KATO] =
  cpu_to_le32(NVME_FIS_FSUPP | NVME_FIS_CSCPE);
 features->fis[NVME_FEAT_ASYNC_EVENT] =
  cpu_to_le32(NVME_FIS_FSUPP | NVME_FIS_CSCPE);
 features->fis[NVME_FEAT_HOST_ID] =
  cpu_to_le32(NVME_FIS_FSUPP | NVME_FIS_CSCPE);
 features->fis[NVME_FEAT_WRITE_PROTECT] =
  cpu_to_le32(NVME_FIS_FSUPP | NVME_FIS_NSCPE);
 features->fis[NVME_FEAT_RESV_MASK] =
  cpu_to_le32(NVME_FIS_FSUPP | NVME_FIS_NSCPE);

 status = nvmet_copy_to_sgl(req, 0, features, sizeof(*features));
 kfree(features);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_get_log_page(struct nvmet_req *req)
{
 if (!nvmet_check_transfer_len(req, nvmet_get_log_page_len(req->cmd)))
  return;

 switch (req->cmd->get_log_page.lid) {
 case NVME_LOG_SUPPORTED:
  return nvmet_execute_get_supported_log_pages(req);
 case NVME_LOG_ERROR:
  return nvmet_execute_get_log_page_error(req);
 case NVME_LOG_SMART:
  return nvmet_execute_get_log_page_smart(req);
 case NVME_LOG_FW_SLOT:
  /*
 * We only support a single firmware slot which always is
 * active, so we can zero out the whole firmware slot log and
 * still claim to fully implement this mandatory log page.
 */
  return nvmet_execute_get_log_page_noop(req);
 case NVME_LOG_CHANGED_NS:
  return nvmet_execute_get_log_changed_ns(req);
 case NVME_LOG_CMD_EFFECTS:
  return nvmet_execute_get_log_cmd_effects_ns(req);
 case NVME_LOG_ENDURANCE_GROUP:
  return nvmet_execute_get_log_page_endgrp(req);
 case NVME_LOG_ANA:
  return nvmet_execute_get_log_page_ana(req);
 case NVME_LOG_FEATURES:
  return nvmet_execute_get_log_page_features(req);
 case NVME_LOG_RMI:
  return nvmet_execute_get_log_page_rmi(req);
 case NVME_LOG_RESERVATION:
  return nvmet_execute_get_log_page_resv(req);
 }
 pr_debug("unhandled lid %d on qid %d\n",
        req->cmd->get_log_page.lid, req->sq->qid);
 req->error_loc = offsetof(struct nvme_get_log_page_command, lid);
 nvmet_req_complete(req, NVME_SC_INVALID_FIELD | NVME_STATUS_DNR);
}

static void nvmet_execute_identify_ctrl(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvmet_subsys *subsys = ctrl->subsys;
 struct nvme_id_ctrl *id;
 u32 cmd_capsule_size, ctratt;
 u16 status = 0;

 if (!subsys->subsys_discovered) {
  mutex_lock(&subsys->lock);
  subsys->subsys_discovered = true;
  mutex_unlock(&subsys->lock);
 }

 id = kzalloc(sizeof(*id), GFP_KERNEL);
 if (!id) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 id->vid = cpu_to_le16(subsys->vendor_id);
 id->ssvid = cpu_to_le16(subsys->subsys_vendor_id);

 memcpy(id->sn, ctrl->subsys->serial, NVMET_SN_MAX_SIZE);
 memcpy_and_pad(id->mn, sizeof(id->mn), subsys->model_number,
         strlen(subsys->model_number), ' ');
 memcpy_and_pad(id->fr, sizeof(id->fr),
         subsys->firmware_rev, strlen(subsys->firmware_rev), ' ');

 put_unaligned_le24(subsys->ieee_oui, id->ieee);

 id->rab = 6;

 if (nvmet_is_disc_subsys(ctrl->subsys))
  id->cntrltype = NVME_CTRL_DISC;
 else
  id->cntrltype = NVME_CTRL_IO;

 /* we support multiple ports, multiples hosts and ANA: */
 id->cmic = NVME_CTRL_CMIC_MULTI_PORT | NVME_CTRL_CMIC_MULTI_CTRL |
  NVME_CTRL_CMIC_ANA;

 /* Limit MDTS according to transport capability */
 if (ctrl->ops->get_mdts)
  id->mdts = ctrl->ops->get_mdts(ctrl);
 else
  id->mdts = 0;

 id->cntlid = cpu_to_le16(ctrl->cntlid);
 id->ver = cpu_to_le32(ctrl->subsys->ver);

 /* XXX: figure out what to do about RTD3R/RTD3 */
 id->oaes = cpu_to_le32(NVMET_AEN_CFG_OPTIONAL);
 ctratt = NVME_CTRL_ATTR_HID_128_BIT | NVME_CTRL_ATTR_TBKAS;
 if (nvmet_is_pci_ctrl(ctrl))
  ctratt |= NVME_CTRL_ATTR_RHII;
 id->ctratt = cpu_to_le32(ctratt);

 id->oacs = 0;

 /*
 * We don't really have a practical limit on the number of abort
 * comands.  But we don't do anything useful for abort either, so
 * no point in allowing more abort commands than the spec requires.
 */
 id->acl = 3;

 id->aerl = NVMET_ASYNC_EVENTS - 1;

 /* first slot is read-only, only one slot supported */
 id->frmw = (1 << 0) | (1 << 1);
 id->lpa = (1 << 0) | (1 << 1) | (1 << 2);
 id->elpe = NVMET_ERROR_LOG_SLOTS - 1;
 id->npss = 0;

 /* We support keep-alive timeout in granularity of seconds */
 id->kas = cpu_to_le16(NVMET_KAS);

 id->sqes = (0x6 << 4) | 0x6;
 id->cqes = (0x4 << 4) | 0x4;

 /* no enforcement soft-limit for maxcmd - pick arbitrary high value */
 id->maxcmd = cpu_to_le16(NVMET_MAX_CMD(ctrl));

 id->nn = cpu_to_le32(NVMET_MAX_NAMESPACES);
 id->mnan = cpu_to_le32(NVMET_MAX_NAMESPACES);
 id->oncs = cpu_to_le16(NVME_CTRL_ONCS_DSM |
   NVME_CTRL_ONCS_WRITE_ZEROES |
   NVME_CTRL_ONCS_RESERVATIONS);

 /* XXX: don't report vwc if the underlying device is write through */
 id->vwc = NVME_CTRL_VWC_PRESENT;

 /*
 * We can't support atomic writes bigger than a LBA without support
 * from the backend device.
 */
 id->awun = 0;
 id->awupf = 0;

 /* we always support SGLs */
 id->sgls = cpu_to_le32(NVME_CTRL_SGLS_BYTE_ALIGNED);
 if (ctrl->ops->flags & NVMF_KEYED_SGLS)
  id->sgls |= cpu_to_le32(NVME_CTRL_SGLS_KSDBDS);
 if (req->port->inline_data_size)
  id->sgls |= cpu_to_le32(NVME_CTRL_SGLS_SAOS);

 strscpy(id->subnqn, ctrl->subsys->subsysnqn, sizeof(id->subnqn));

 /*
 * Max command capsule size is sqe + in-capsule data size.
 * Disable in-capsule data for Metadata capable controllers.
 */
 cmd_capsule_size = sizeof(struct nvme_command);
 if (!ctrl->pi_support)
  cmd_capsule_size += req->port->inline_data_size;
 id->ioccsz = cpu_to_le32(cmd_capsule_size / 16);

 /* Max response capsule size is cqe */
 id->iorcsz = cpu_to_le32(sizeof(struct nvme_completion) / 16);

 id->msdbd = ctrl->ops->msdbd;

 /*
 * Endurance group identifier is 16 bits, so we can't let namespaces
 * overflow that since we reuse the nsid
 */
 BUILD_BUG_ON(NVMET_MAX_NAMESPACES > USHRT_MAX);
 id->endgidmax = cpu_to_le16(NVMET_MAX_NAMESPACES);

 id->anacap = (1 << 0) | (1 << 1) | (1 << 2) | (1 << 3) | (1 << 4);
 id->anatt = 10; /* random value */
 id->anagrpmax = cpu_to_le32(NVMET_MAX_ANAGRPS);
 id->nanagrpid = cpu_to_le32(NVMET_MAX_ANAGRPS);

 /*
 * Meh, we don't really support any power state.  Fake up the same
 * values that qemu does.
 */
 id->psd[0].max_power = cpu_to_le16(0x9c4);
 id->psd[0].entry_lat = cpu_to_le32(0x10);
 id->psd[0].exit_lat = cpu_to_le32(0x4);

 id->nwpc = 1 << 0; /* write protect and no write protect */

 status = nvmet_copy_to_sgl(req, 0, id, sizeof(*id));

 kfree(id);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_identify_ns(struct nvmet_req *req)
{
 struct nvme_id_ns *id;
 u16 status;

 if (le32_to_cpu(req->cmd->identify.nsid) == NVME_NSID_ALL) {
  req->error_loc = offsetof(struct nvme_identify, nsid);
  status = NVME_SC_INVALID_NS | NVME_STATUS_DNR;
  goto out;
 }

 id = kzalloc(sizeof(*id), GFP_KERNEL);
 if (!id) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 /* return an all zeroed buffer if we can't find an active namespace */
 status = nvmet_req_find_ns(req);
 if (status) {
  status = 0;
  goto done;
 }

 if (nvmet_ns_revalidate(req->ns)) {
  mutex_lock(&req->ns->subsys->lock);
  nvmet_ns_changed(req->ns->subsys, req->ns->nsid);
  mutex_unlock(&req->ns->subsys->lock);
 }

 /*
 * nuse = ncap = nsze isn't always true, but we have no way to find
 * that out from the underlying device.
 */
 id->ncap = id->nsze =
  cpu_to_le64(req->ns->size >> req->ns->blksize_shift);
 switch (req->port->ana_state[req->ns->anagrpid]) {
 case NVME_ANA_INACCESSIBLE:
 case NVME_ANA_PERSISTENT_LOSS:
  break;
 default:
  id->nuse = id->nsze;
  break;
 }

 if (req->ns->bdev)
  nvmet_bdev_set_limits(req->ns->bdev, id);

 /*
 * We just provide a single LBA format that matches what the
 * underlying device reports.
 */
 id->nlbaf = 0;
 id->flbas = 0;

 /*
 * Our namespace might always be shared.  Not just with other
 * controllers, but also with any other user of the block device.
 */
 id->nmic = NVME_NS_NMIC_SHARED;
 id->anagrpid = cpu_to_le32(req->ns->anagrpid);

 if (req->ns->pr.enable)
  id->rescap = NVME_PR_SUPPORT_WRITE_EXCLUSIVE |
   NVME_PR_SUPPORT_EXCLUSIVE_ACCESS |
   NVME_PR_SUPPORT_WRITE_EXCLUSIVE_REG_ONLY |
   NVME_PR_SUPPORT_EXCLUSIVE_ACCESS_REG_ONLY |
   NVME_PR_SUPPORT_WRITE_EXCLUSIVE_ALL_REGS |
   NVME_PR_SUPPORT_EXCLUSIVE_ACCESS_ALL_REGS |
   NVME_PR_SUPPORT_IEKEY_VER_1_3_DEF;

 /*
 * Since we don't know any better, every namespace is its own endurance
 * group.
 */
 id->endgid = cpu_to_le16(req->ns->nsid);

 memcpy(&id->nguid, &req->ns->nguid, sizeof(id->nguid));

 id->lbaf[0].ds = req->ns->blksize_shift;

 if (req->sq->ctrl->pi_support && nvmet_ns_has_pi(req->ns)) {
  id->dpc = NVME_NS_DPC_PI_FIRST | NVME_NS_DPC_PI_LAST |
     NVME_NS_DPC_PI_TYPE1 | NVME_NS_DPC_PI_TYPE2 |
     NVME_NS_DPC_PI_TYPE3;
  id->mc = NVME_MC_EXTENDED_LBA;
  id->dps = req->ns->pi_type;
  id->flbas = NVME_NS_FLBAS_META_EXT;
  id->lbaf[0].ms = cpu_to_le16(req->ns->metadata_size);
 }

 if (req->ns->readonly)
  id->nsattr |= NVME_NS_ATTR_RO;
done:
 if (!status)
  status = nvmet_copy_to_sgl(req, 0, id, sizeof(*id));

 kfree(id);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_identify_endgrp_list(struct nvmet_req *req)
{
 u16 min_endgid = le16_to_cpu(req->cmd->identify.cnssid);
 static const int buf_size = NVME_IDENTIFY_DATA_SIZE;
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvmet_ns *ns;
 unsigned long idx;
 __le16 *list;
 u16 status;
 int i = 1;

 list = kzalloc(buf_size, GFP_KERNEL);
 if (!list) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 nvmet_for_each_enabled_ns(&ctrl->subsys->namespaces, idx, ns) {
  if (ns->nsid <= min_endgid)
   continue;

  list[i++] = cpu_to_le16(ns->nsid);
  if (i == buf_size / sizeof(__le16))
   break;
 }

 list[0] = cpu_to_le16(i - 1);
 status = nvmet_copy_to_sgl(req, 0, list, buf_size);
 kfree(list);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_identify_nslist(struct nvmet_req *req, bool match_css)
{
 static const int buf_size = NVME_IDENTIFY_DATA_SIZE;
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvmet_ns *ns;
 unsigned long idx;
 u32 min_nsid = le32_to_cpu(req->cmd->identify.nsid);
 __le32 *list;
 u16 status = 0;
 int i = 0;

 /*
 * NSID values 0xFFFFFFFE and NVME_NSID_ALL are invalid
 * See NVMe Base Specification, Active Namespace ID list (CNS 02h).
 */
 if (min_nsid == 0xFFFFFFFE || min_nsid == NVME_NSID_ALL) {
  req->error_loc = offsetof(struct nvme_identify, nsid);
  status = NVME_SC_INVALID_NS | NVME_STATUS_DNR;
  goto out;
 }

 list = kzalloc(buf_size, GFP_KERNEL);
 if (!list) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 nvmet_for_each_enabled_ns(&ctrl->subsys->namespaces, idx, ns) {
  if (ns->nsid <= min_nsid)
   continue;
  if (match_css && req->ns->csi != req->cmd->identify.csi)
   continue;
  list[i++] = cpu_to_le32(ns->nsid);
  if (i == buf_size / sizeof(__le32))
   break;
 }

 status = nvmet_copy_to_sgl(req, 0, list, buf_size);

 kfree(list);
out:
 nvmet_req_complete(req, status);
}

static u16 nvmet_copy_ns_identifier(struct nvmet_req *req, u8 type, u8 len,
        void *id, off_t *off)
{
 struct nvme_ns_id_desc desc = {
  .nidt = type,
  .nidl = len,
 };
 u16 status;

 status = nvmet_copy_to_sgl(req, *off, &desc, sizeof(desc));
 if (status)
  return status;
 *off += sizeof(desc);

 status = nvmet_copy_to_sgl(req, *off, id, len);
 if (status)
  return status;
 *off += len;

 return 0;
}

static void nvmet_execute_identify_desclist(struct nvmet_req *req)
{
 off_t off = 0;
 u16 status;

 status = nvmet_req_find_ns(req);
 if (status)
  goto out;

 if (memchr_inv(&req->ns->uuid, 0, sizeof(req->ns->uuid))) {
  status = nvmet_copy_ns_identifier(req, NVME_NIDT_UUID,
        NVME_NIDT_UUID_LEN,
        &req->ns->uuid, &off);
  if (status)
   goto out;
 }
 if (memchr_inv(req->ns->nguid, 0, sizeof(req->ns->nguid))) {
  status = nvmet_copy_ns_identifier(req, NVME_NIDT_NGUID,
        NVME_NIDT_NGUID_LEN,
        &req->ns->nguid, &off);
  if (status)
   goto out;
 }

 status = nvmet_copy_ns_identifier(req, NVME_NIDT_CSI,
       NVME_NIDT_CSI_LEN,
       &req->ns->csi, &off);
 if (status)
  goto out;

 if (sg_zero_buffer(req->sg, req->sg_cnt, NVME_IDENTIFY_DATA_SIZE - off,
   off) != NVME_IDENTIFY_DATA_SIZE - off)
  status = NVME_SC_INTERNAL | NVME_STATUS_DNR;

out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_identify_ctrl_nvm(struct nvmet_req *req)
{
 /* Not supported: return zeroes */
 nvmet_req_complete(req,
     nvmet_zero_sgl(req, 0, sizeof(struct nvme_id_ctrl_nvm)));
}

static void nvme_execute_identify_ns_nvm(struct nvmet_req *req)
{
 u16 status;
 struct nvme_id_ns_nvm *id;

 status = nvmet_req_find_ns(req);
 if (status)
  goto out;

 id = kzalloc(sizeof(*id), GFP_KERNEL);
 if (!id) {
  status = NVME_SC_INTERNAL;
  goto out;
 }
 status = nvmet_copy_to_sgl(req, 0, id, sizeof(*id));
 kfree(id);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_id_cs_indep(struct nvmet_req *req)
{
 struct nvme_id_ns_cs_indep *id;
 u16 status;

 status = nvmet_req_find_ns(req);
 if (status)
  goto out;

 id = kzalloc(sizeof(*id), GFP_KERNEL);
 if (!id) {
  status = NVME_SC_INTERNAL;
  goto out;
 }

 id->nstat = NVME_NSTAT_NRDY;
 id->anagrpid = cpu_to_le32(req->ns->anagrpid);
 id->nmic = NVME_NS_NMIC_SHARED;
 if (req->ns->readonly)
  id->nsattr |= NVME_NS_ATTR_RO;
 if (req->ns->bdev && !bdev_nonrot(req->ns->bdev))
  id->nsfeat |= NVME_NS_ROTATIONAL;
 /*
 * We need flush command to flush the file's metadata,
 * so report supporting vwc if backend is file, even
 * though buffered_io is disable.
 */
 if (req->ns->bdev && !bdev_write_cache(req->ns->bdev))
  id->nsfeat |= NVME_NS_VWC_NOT_PRESENT;

 status = nvmet_copy_to_sgl(req, 0, id, sizeof(*id));
 kfree(id);
out:
 nvmet_req_complete(req, status);
}

static void nvmet_execute_identify(struct nvmet_req *req)
{
 if (!nvmet_check_transfer_len(req, NVME_IDENTIFY_DATA_SIZE))
  return;

 switch (req->cmd->identify.cns) {
 case NVME_ID_CNS_NS:
  nvmet_execute_identify_ns(req);
  return;
 case NVME_ID_CNS_CTRL:
  nvmet_execute_identify_ctrl(req);
  return;
 case NVME_ID_CNS_NS_ACTIVE_LIST:
  nvmet_execute_identify_nslist(req, false);
  return;
 case NVME_ID_CNS_NS_DESC_LIST:
  nvmet_execute_identify_desclist(req);
  return;
 case NVME_ID_CNS_CS_NS:
  switch (req->cmd->identify.csi) {
  case NVME_CSI_NVM:
   nvme_execute_identify_ns_nvm(req);
   return;
  case NVME_CSI_ZNS:
   if (IS_ENABLED(CONFIG_BLK_DEV_ZONED)) {
    nvmet_execute_identify_ns_zns(req);
    return;
   }
   break;
  }
  break;
 case NVME_ID_CNS_CS_CTRL:
  switch (req->cmd->identify.csi) {
  case NVME_CSI_NVM:
   nvmet_execute_identify_ctrl_nvm(req);
   return;
  case NVME_CSI_ZNS:
   if (IS_ENABLED(CONFIG_BLK_DEV_ZONED)) {
    nvmet_execute_identify_ctrl_zns(req);
    return;
   }
   break;
  }
  break;
 case NVME_ID_CNS_NS_ACTIVE_LIST_CS:
  nvmet_execute_identify_nslist(req, true);
  return;
 case NVME_ID_CNS_NS_CS_INDEP:
  nvmet_execute_id_cs_indep(req);
  return;
 case NVME_ID_CNS_ENDGRP_LIST:
  nvmet_execute_identify_endgrp_list(req);
  return;
 }

 pr_debug("unhandled identify cns %d on qid %d\n",
        req->cmd->identify.cns, req->sq->qid);
 req->error_loc = offsetof(struct nvme_identify, cns);
 nvmet_req_complete(req, NVME_SC_INVALID_FIELD | NVME_STATUS_DNR);
}

/*
 * A "minimum viable" abort implementation: the command is mandatory in the
 * spec, but we are not required to do any useful work.  We couldn't really
 * do a useful abort, so don't bother even with waiting for the command
 * to be executed and return immediately telling the command to abort
 * wasn't found.
 */
static void nvmet_execute_abort(struct nvmet_req *req)
{
 if (!nvmet_check_transfer_len(req, 0))
  return;
 nvmet_set_result(req, 1);
 nvmet_req_complete(req, 0);
}

static u16 nvmet_write_protect_flush_sync(struct nvmet_req *req)
{
 u16 status;

 if (req->ns->file)
  status = nvmet_file_flush(req);
 else
  status = nvmet_bdev_flush(req);

 if (status)
  pr_err("write protect flush failed nsid: %u\n", req->ns->nsid);
 return status;
}

static u16 nvmet_set_feat_write_protect(struct nvmet_req *req)
{
 u32 write_protect = le32_to_cpu(req->cmd->common.cdw11);
 struct nvmet_subsys *subsys = nvmet_req_subsys(req);
 u16 status;

 status = nvmet_req_find_ns(req);
 if (status)
  return status;

 mutex_lock(&subsys->lock);
 switch (write_protect) {
 case NVME_NS_WRITE_PROTECT:
  req->ns->readonly = true;
  status = nvmet_write_protect_flush_sync(req);
  if (status)
   req->ns->readonly = false;
  break;
 case NVME_NS_NO_WRITE_PROTECT:
  req->ns->readonly = false;
  status = 0;
  break;
 default:
  break;
 }

 if (!status)
  nvmet_ns_changed(subsys, req->ns->nsid);
 mutex_unlock(&subsys->lock);
 return status;
}

u16 nvmet_set_feat_kato(struct nvmet_req *req)
{
 u32 val32 = le32_to_cpu(req->cmd->common.cdw11);

 nvmet_stop_keep_alive_timer(req->sq->ctrl);
 req->sq->ctrl->kato = DIV_ROUND_UP(val32, 1000);
 nvmet_start_keep_alive_timer(req->sq->ctrl);

 nvmet_set_result(req, req->sq->ctrl->kato);

 return 0;
}

u16 nvmet_set_feat_async_event(struct nvmet_req *req, u32 mask)
{
 u32 val32 = le32_to_cpu(req->cmd->common.cdw11);

 if (val32 & ~mask) {
  req->error_loc = offsetof(struct nvme_common_command, cdw11);
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
 }

 WRITE_ONCE(req->sq->ctrl->aen_enabled, val32);
 nvmet_set_result(req, val32);

 return 0;
}

static u16 nvmet_set_feat_host_id(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;

 if (!nvmet_is_pci_ctrl(ctrl))
  return NVME_SC_CMD_SEQ_ERROR | NVME_STATUS_DNR;

 /*
 * The NVMe base specifications v2.1 recommends supporting 128-bits host
 * IDs (section 5.1.25.1.28.1). However, that same section also says
 * that "The controller may support a 64-bit Host Identifier and/or an
 * extended 128-bit Host Identifier". So simplify this support and do
 * not support 64-bits host IDs to avoid needing to check that all
 * controllers associated with the same subsystem all use the same host
 * ID size.
 */
 if (!(req->cmd->common.cdw11 & cpu_to_le32(1 << 0))) {
  req->error_loc = offsetof(struct nvme_common_command, cdw11);
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
 }

 return nvmet_copy_from_sgl(req, 0, &req->sq->ctrl->hostid,
       sizeof(req->sq->ctrl->hostid));
}

static u16 nvmet_set_feat_irq_coalesce(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 u32 cdw11 = le32_to_cpu(req->cmd->common.cdw11);
 struct nvmet_feat_irq_coalesce irqc = {
  .time = (cdw11 >> 8) & 0xff,
  .thr = cdw11 & 0xff,
 };

 /*
 * This feature is not supported for fabrics controllers and mandatory
 * for PCI controllers.
 */
 if (!nvmet_is_pci_ctrl(ctrl)) {
  req->error_loc = offsetof(struct nvme_common_command, cdw10);
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
 }

 return ctrl->ops->set_feature(ctrl, NVME_FEAT_IRQ_COALESCE, &irqc);
}

static u16 nvmet_set_feat_irq_config(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 u32 cdw11 = le32_to_cpu(req->cmd->common.cdw11);
 struct nvmet_feat_irq_config irqcfg = {
  .iv = cdw11 & 0xffff,
  .cd = (cdw11 >> 16) & 0x1,
 };

 /*
 * This feature is not supported for fabrics controllers and mandatory
 * for PCI controllers.
 */
 if (!nvmet_is_pci_ctrl(ctrl)) {
  req->error_loc = offsetof(struct nvme_common_command, cdw10);
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
 }

 return ctrl->ops->set_feature(ctrl, NVME_FEAT_IRQ_CONFIG, &irqcfg);
}

static u16 nvmet_set_feat_arbitration(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 u32 cdw11 = le32_to_cpu(req->cmd->common.cdw11);
 struct nvmet_feat_arbitration arb = {
  .hpw = (cdw11 >> 24) & 0xff,
  .mpw = (cdw11 >> 16) & 0xff,
  .lpw = (cdw11 >> 8) & 0xff,
  .ab = cdw11 & 0x3,
 };

 if (!ctrl->ops->set_feature) {
  req->error_loc = offsetof(struct nvme_common_command, cdw10);
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
 }

 return ctrl->ops->set_feature(ctrl, NVME_FEAT_ARBITRATION, &arb);
}

void nvmet_execute_set_features(struct nvmet_req *req)
{
 struct nvmet_subsys *subsys = nvmet_req_subsys(req);
 u32 cdw10 = le32_to_cpu(req->cmd->common.cdw10);
 u32 cdw11 = le32_to_cpu(req->cmd->common.cdw11);
 u16 status = 0;
 u16 nsqr;
 u16 ncqr;

 if (!nvmet_check_data_len_lte(req, 0))
  return;

 switch (cdw10 & 0xff) {
 case NVME_FEAT_ARBITRATION:
  status = nvmet_set_feat_arbitration(req);
  break;
 case NVME_FEAT_NUM_QUEUES:
  ncqr = (cdw11 >> 16) & 0xffff;
  nsqr = cdw11 & 0xffff;
  if (ncqr == 0xffff || nsqr == 0xffff) {
   status = NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
   break;
  }
  nvmet_set_result(req,
   (subsys->max_qid - 1) | ((subsys->max_qid - 1) << 16));
  break;
 case NVME_FEAT_IRQ_COALESCE:
  status = nvmet_set_feat_irq_coalesce(req);
  break;
 case NVME_FEAT_IRQ_CONFIG:
  status = nvmet_set_feat_irq_config(req);
  break;
 case NVME_FEAT_KATO:
  status = nvmet_set_feat_kato(req);
  break;
 case NVME_FEAT_ASYNC_EVENT:
  status = nvmet_set_feat_async_event(req, NVMET_AEN_CFG_ALL);
  break;
 case NVME_FEAT_HOST_ID:
  status = nvmet_set_feat_host_id(req);
  break;
 case NVME_FEAT_WRITE_PROTECT:
  status = nvmet_set_feat_write_protect(req);
  break;
 case NVME_FEAT_RESV_MASK:
  status = nvmet_set_feat_resv_notif_mask(req, cdw11);
  break;
 default:
  req->error_loc = offsetof(struct nvme_common_command, cdw10);
  status = NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
  break;
 }

 nvmet_req_complete(req, status);
}

static u16 nvmet_get_feat_write_protect(struct nvmet_req *req)
{
 struct nvmet_subsys *subsys = nvmet_req_subsys(req);
 u32 result;

 result = nvmet_req_find_ns(req);
 if (result)
  return result;

 mutex_lock(&subsys->lock);
 if (req->ns->readonly == true)
  result = NVME_NS_WRITE_PROTECT;
 else
  result = NVME_NS_NO_WRITE_PROTECT;
 nvmet_set_result(req, result);
 mutex_unlock(&subsys->lock);

 return 0;
}

static u16 nvmet_get_feat_irq_coalesce(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvmet_feat_irq_coalesce irqc = { };
 u16 status;

 /*
 * This feature is not supported for fabrics controllers and mandatory
 * for PCI controllers.
 */
 if (!nvmet_is_pci_ctrl(ctrl)) {
  req->error_loc = offsetof(struct nvme_common_command, cdw10);
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
 }

 status = ctrl->ops->get_feature(ctrl, NVME_FEAT_IRQ_COALESCE, &irqc);
 if (status != NVME_SC_SUCCESS)
  return status;

 nvmet_set_result(req, ((u32)irqc.time << 8) | (u32)irqc.thr);

 return NVME_SC_SUCCESS;
}

static u16 nvmet_get_feat_irq_config(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 u32 iv = le32_to_cpu(req->cmd->common.cdw11) & 0xffff;
 struct nvmet_feat_irq_config irqcfg = { .iv = iv };
 u16 status;

 /*
 * This feature is not supported for fabrics controllers and mandatory
 * for PCI controllers.
 */
 if (!nvmet_is_pci_ctrl(ctrl)) {
  req->error_loc = offsetof(struct nvme_common_command, cdw10);
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
 }

 status = ctrl->ops->get_feature(ctrl, NVME_FEAT_IRQ_CONFIG, &irqcfg);
 if (status != NVME_SC_SUCCESS)
  return status;

 nvmet_set_result(req, ((u32)irqcfg.cd << 16) | iv);

 return NVME_SC_SUCCESS;
}

static u16 nvmet_get_feat_arbitration(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 struct nvmet_feat_arbitration arb = { };
 u16 status;

 if (!ctrl->ops->get_feature) {
  req->error_loc = offsetof(struct nvme_common_command, cdw10);
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
 }

 status = ctrl->ops->get_feature(ctrl, NVME_FEAT_ARBITRATION, &arb);
 if (status != NVME_SC_SUCCESS)
  return status;

 nvmet_set_result(req,
    ((u32)arb.hpw << 24) |
    ((u32)arb.mpw << 16) |
    ((u32)arb.lpw << 8) |
    (arb.ab & 0x3));

 return NVME_SC_SUCCESS;
}

void nvmet_get_feat_kato(struct nvmet_req *req)
{
 nvmet_set_result(req, req->sq->ctrl->kato * 1000);
}

void nvmet_get_feat_async_event(struct nvmet_req *req)
{
 nvmet_set_result(req, READ_ONCE(req->sq->ctrl->aen_enabled));
}

void nvmet_execute_get_features(struct nvmet_req *req)
{
 struct nvmet_subsys *subsys = nvmet_req_subsys(req);
 u32 cdw10 = le32_to_cpu(req->cmd->common.cdw10);
 u16 status = 0;

 if (!nvmet_check_transfer_len(req, nvmet_feat_data_len(req, cdw10)))
  return;

 switch (cdw10 & 0xff) {
 /*
 * These features are mandatory in the spec, but we don't
 * have a useful way to implement them.  We'll eventually
 * need to come up with some fake values for these.
 */
#if 0
 case NVME_FEAT_POWER_MGMT:
  break;
 case NVME_FEAT_TEMP_THRESH:
  break;
 case NVME_FEAT_ERR_RECOVERY:
  break;
 case NVME_FEAT_WRITE_ATOMIC:
  break;
#endif
 case NVME_FEAT_ARBITRATION:
  status = nvmet_get_feat_arbitration(req);
  break;
 case NVME_FEAT_IRQ_COALESCE:
  status = nvmet_get_feat_irq_coalesce(req);
  break;
 case NVME_FEAT_IRQ_CONFIG:
  status = nvmet_get_feat_irq_config(req);
  break;
 case NVME_FEAT_ASYNC_EVENT:
  nvmet_get_feat_async_event(req);
  break;
 case NVME_FEAT_VOLATILE_WC:
  nvmet_set_result(req, 1);
  break;
 case NVME_FEAT_NUM_QUEUES:
  nvmet_set_result(req,
   (subsys->max_qid-1) | ((subsys->max_qid-1) << 16));
  break;
 case NVME_FEAT_KATO:
  nvmet_get_feat_kato(req);
  break;
 case NVME_FEAT_HOST_ID:
  /* need 128-bit host identifier flag */
  if (!(req->cmd->common.cdw11 & cpu_to_le32(1 << 0))) {
   req->error_loc =
    offsetof(struct nvme_common_command, cdw11);
   status = NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
   break;
  }

  status = nvmet_copy_to_sgl(req, 0, &req->sq->ctrl->hostid,
    sizeof(req->sq->ctrl->hostid));
  break;
 case NVME_FEAT_WRITE_PROTECT:
  status = nvmet_get_feat_write_protect(req);
  break;
 case NVME_FEAT_RESV_MASK:
  status = nvmet_get_feat_resv_notif_mask(req);
  break;
 default:
  req->error_loc =
   offsetof(struct nvme_common_command, cdw10);
  status = NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;
  break;
 }

 nvmet_req_complete(req, status);
}

void nvmet_execute_async_event(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;

 if (!nvmet_check_transfer_len(req, 0))
  return;

 mutex_lock(&ctrl->lock);
 if (ctrl->nr_async_event_cmds >= NVMET_ASYNC_EVENTS) {
  mutex_unlock(&ctrl->lock);
  nvmet_req_complete(req, NVME_SC_ASYNC_LIMIT | NVME_STATUS_DNR);
  return;
 }
 ctrl->async_event_cmds[ctrl->nr_async_event_cmds++] = req;
 mutex_unlock(&ctrl->lock);

 queue_work(nvmet_wq, &ctrl->async_event_work);
}

void nvmet_execute_keep_alive(struct nvmet_req *req)
{
 struct nvmet_ctrl *ctrl = req->sq->ctrl;
 u16 status = 0;

 if (!nvmet_check_transfer_len(req, 0))
  return;

 if (!ctrl->kato) {
  status = NVME_SC_KA_TIMEOUT_INVALID;
  goto out;
 }

 pr_debug("ctrl %d update keep-alive timer for %d secs\n",
  ctrl->cntlid, ctrl->kato);
 mod_delayed_work(system_wq, &ctrl->ka_work, ctrl->kato * HZ);
out:
 nvmet_req_complete(req, status);
}

u32 nvmet_admin_cmd_data_len(struct nvmet_req *req)
{
 struct nvme_command *cmd = req->cmd;

 if (nvme_is_fabrics(cmd))
  return nvmet_fabrics_admin_cmd_data_len(req);
 if (nvmet_is_disc_subsys(nvmet_req_subsys(req)))
  return nvmet_discovery_cmd_data_len(req);

 switch (cmd->common.opcode) {
 case nvme_admin_get_log_page:
  return nvmet_get_log_page_len(cmd);
 case nvme_admin_identify:
  return NVME_IDENTIFY_DATA_SIZE;
 case nvme_admin_get_features:
  return nvmet_feat_data_len(req, le32_to_cpu(cmd->common.cdw10));
 default:
  return 0;
 }
}

u16 nvmet_parse_admin_cmd(struct nvmet_req *req)
{
 struct nvme_command *cmd = req->cmd;
 u16 ret;

 if (nvme_is_fabrics(cmd))
  return nvmet_parse_fabrics_admin_cmd(req);
 if (nvmet_is_disc_subsys(nvmet_req_subsys(req)))
  return nvmet_parse_discovery_cmd(req);

 ret = nvmet_check_ctrl_status(req);
 if (unlikely(ret))
  return ret;

 /* For PCI controllers, admin commands shall not use SGL. */
 if (nvmet_is_pci_ctrl(req->sq->ctrl) && !req->sq->qid &&
     cmd->common.flags & NVME_CMD_SGL_ALL)
  return NVME_SC_INVALID_FIELD | NVME_STATUS_DNR;

 if (nvmet_is_passthru_req(req))
  return nvmet_parse_passthru_admin_cmd(req);

 switch (cmd->common.opcode) {
 case nvme_admin_delete_sq:
  req->execute = nvmet_execute_delete_sq;
  return 0;
 case nvme_admin_create_sq:
  req->execute = nvmet_execute_create_sq;
  return 0;
 case nvme_admin_get_log_page:
  req->execute = nvmet_execute_get_log_page;
  return 0;
 case nvme_admin_delete_cq:
  req->execute = nvmet_execute_delete_cq;
  return 0;
 case nvme_admin_create_cq:
  req->execute = nvmet_execute_create_cq;
  return 0;
 case nvme_admin_identify:
  req->execute = nvmet_execute_identify;
  return 0;
 case nvme_admin_abort_cmd:
  req->execute = nvmet_execute_abort;
  return 0;
 case nvme_admin_set_features:
  req->execute = nvmet_execute_set_features;
  return 0;
 case nvme_admin_get_features:
  req->execute = nvmet_execute_get_features;
  return 0;
 case nvme_admin_async_event:
  req->execute = nvmet_execute_async_event;
  return 0;
 case nvme_admin_keep_alive:
  req->execute = nvmet_execute_keep_alive;
  return 0;
 default:
  return nvmet_report_invalid_opcode(req);
 }
}