[linux-2.6-block.git] / drivers / nvme / host / lightnvm.c

/*
 * nvme-lightnvm.c - LightNVM NVMe device
 *
 * Copyright (C) 2014-2015 IT University of Copenhagen
 * Initial release: Matias Bjorling <mb@lightnvm.io>
 *
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License version
 * 2 as published by the Free Software Foundation.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; see the file COPYING.  If not, write to
 * the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139,
 * USA.
 *
 */

#include "nvme.h"

#include <linux/nvme.h>
#include <linux/bitops.h>
#include <linux/lightnvm.h>
#include <linux/vmalloc.h>

enum nvme_nvm_admin_opcode {
	nvme_nvm_admin_identity		= 0xe2,
	nvme_nvm_admin_get_l2p_tbl	= 0xea,
	nvme_nvm_admin_get_bb_tbl	= 0xf2,
	nvme_nvm_admin_set_bb_tbl	= 0xf1,
};

struct nvme_nvm_hb_rw {
	__u8			opcode;
	__u8			flags;
	__u16			command_id;
	__le32			nsid;
	__u64			rsvd2;
	__le64			metadata;
	__le64			prp1;
	__le64			prp2;
	__le64			spba;
	__le16			length;
	__le16			control;
	__le32			dsmgmt;
	__le64			slba;
};

struct nvme_nvm_ph_rw {
	__u8			opcode;
	__u8			flags;
	__u16			command_id;
	__le32			nsid;
	__u64			rsvd2;
	__le64			metadata;
	__le64			prp1;
	__le64			prp2;
	__le64			spba;
	__le16			length;
	__le16			control;
	__le32			dsmgmt;
	__le64			resv;
};

struct nvme_nvm_identity {
	__u8			opcode;
	__u8			flags;
	__u16			command_id;
	__le32			nsid;
	__u64			rsvd[2];
	__le64			prp1;
	__le64			prp2;
	__le32			chnl_off;
	__u32			rsvd11[5];
};

struct nvme_nvm_l2ptbl {
	__u8			opcode;
	__u8			flags;
	__u16			command_id;
	__le32			nsid;
	__le32			cdw2[4];
	__le64			prp1;
	__le64			prp2;
	__le64			slba;
	__le32			nlb;
	__le16			cdw14[6];
};

struct nvme_nvm_getbbtbl {
	__u8			opcode;
	__u8			flags;
	__u16			command_id;
	__le32			nsid;
	__u64			rsvd[2];
	__le64			prp1;
	__le64			prp2;
	__le64			spba;
	__u32			rsvd4[4];
};

struct nvme_nvm_setbbtbl {
	__u8			opcode;
	__u8			flags;
	__u16			command_id;
	__le32			nsid;
	__le64			rsvd[2];
	__le64			prp1;
	__le64			prp2;
	__le64			spba;
	__le16			nlb;
	__u8			value;
	__u8			rsvd3;
	__u32			rsvd4[3];
};

struct nvme_nvm_erase_blk {
	__u8			opcode;
	__u8			flags;
	__u16			command_id;
	__le32			nsid;
	__u64			rsvd[2];
	__le64			prp1;
	__le64			prp2;
	__le64			spba;
	__le16			length;
	__le16			control;
	__le32			dsmgmt;
	__le64			resv;
};

struct nvme_nvm_command {
	union {
		struct nvme_common_command common;
		struct nvme_nvm_identity identity;
		struct nvme_nvm_hb_rw hb_rw;
		struct nvme_nvm_ph_rw ph_rw;
		struct nvme_nvm_l2ptbl l2p;
		struct nvme_nvm_getbbtbl get_bb;
		struct nvme_nvm_setbbtbl set_bb;
		struct nvme_nvm_erase_blk erase;
	};
};

struct nvme_nvm_completion {
	__le64	result;		/* Used by LightNVM to return ppa completions */
	__le16	sq_head;	/* how much of this queue may be reclaimed */
	__le16	sq_id;		/* submission queue that generated this entry */
	__u16	command_id;	/* of the command which completed */
	__le16	status;		/* did the command fail, and if so, why? */
};

#define NVME_NVM_LP_MLC_PAIRS 886
struct nvme_nvm_lp_mlc {
	__le16			num_pairs;
	__u8			pairs[NVME_NVM_LP_MLC_PAIRS];
};

struct nvme_nvm_lp_tbl {
	__u8			id[8];
	struct nvme_nvm_lp_mlc	mlc;
};

struct nvme_nvm_id_group {
	__u8			mtype;
	__u8			fmtype;
	__le16			res16;
	__u8			num_ch;
	__u8			num_lun;
	__u8			num_pln;
	__u8			rsvd1;
	__le16			num_blk;
	__le16			num_pg;
	__le16			fpg_sz;
	__le16			csecs;
	__le16			sos;
	__le16			rsvd2;
	__le32			trdt;
	__le32			trdm;
	__le32			tprt;
	__le32			tprm;
	__le32			tbet;
	__le32			tbem;
	__le32			mpos;
	__le32			mccap;
	__le16			cpar;
	__u8			reserved[10];
	struct nvme_nvm_lp_tbl lptbl;
} __packed;

struct nvme_nvm_addr_format {
	__u8			ch_offset;
	__u8			ch_len;
	__u8			lun_offset;
	__u8			lun_len;
	__u8			pln_offset;
	__u8			pln_len;
	__u8			blk_offset;
	__u8			blk_len;
	__u8			pg_offset;
	__u8			pg_len;
	__u8			sect_offset;
	__u8			sect_len;
	__u8			res[4];
} __packed;

struct nvme_nvm_id {
	__u8			ver_id;
	__u8			vmnt;
	__u8			cgrps;
	__u8			res;
	__le32			cap;
	__le32			dom;
	struct nvme_nvm_addr_format ppaf;
	__u8			resv[228];
	struct nvme_nvm_id_group groups[4];
} __packed;

struct nvme_nvm_bb_tbl {
	__u8	tblid[4];
	__le16	verid;
	__le16	revid;
	__le32	rvsd1;
	__le32	tblks;
	__le32	tfact;
	__le32	tgrown;
	__le32	tdresv;
	__le32	thresv;
	__le32	rsvd2[8];
	__u8	blk[0];
};

/*
 * Check we didn't inadvertently grow the command struct
 */
static inline void _nvme_nvm_check_size(void)
{
	BUILD_BUG_ON(sizeof(struct nvme_nvm_identity) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_hb_rw) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_ph_rw) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_getbbtbl) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_setbbtbl) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_l2ptbl) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_erase_blk) != 64);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_id_group) != 960);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_addr_format) != 128);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_id) != 4096);
	BUILD_BUG_ON(sizeof(struct nvme_nvm_bb_tbl) != 512);
}

static int init_grps(struct nvm_id *nvm_id, struct nvme_nvm_id *nvme_nvm_id)
{
	struct nvme_nvm_id_group *src;
	struct nvm_id_group *dst;
	int i, end;

	end = min_t(u32, 4, nvm_id->cgrps);

	for (i = 0; i < end; i++) {
		src = &nvme_nvm_id->groups[i];
		dst = &nvm_id->groups[i];

		dst->mtype = src->mtype;
		dst->fmtype = src->fmtype;
		dst->num_ch = src->num_ch;
		dst->num_lun = src->num_lun;
		dst->num_pln = src->num_pln;

		dst->num_pg = le16_to_cpu(src->num_pg);
		dst->num_blk = le16_to_cpu(src->num_blk);
		dst->fpg_sz = le16_to_cpu(src->fpg_sz);
		dst->csecs = le16_to_cpu(src->csecs);
		dst->sos = le16_to_cpu(src->sos);

		dst->trdt = le32_to_cpu(src->trdt);
		dst->trdm = le32_to_cpu(src->trdm);
		dst->tprt = le32_to_cpu(src->tprt);
		dst->tprm = le32_to_cpu(src->tprm);
		dst->tbet = le32_to_cpu(src->tbet);
		dst->tbem = le32_to_cpu(src->tbem);
		dst->mpos = le32_to_cpu(src->mpos);
		dst->mccap = le32_to_cpu(src->mccap);

		dst->cpar = le16_to_cpu(src->cpar);

		if (dst->fmtype == NVM_ID_FMTYPE_MLC) {
			memcpy(dst->lptbl.id, src->lptbl.id, 8);
			dst->lptbl.mlc.num_pairs =
					le16_to_cpu(src->lptbl.mlc.num_pairs);

			if (dst->lptbl.mlc.num_pairs > NVME_NVM_LP_MLC_PAIRS) {
				pr_err("nvm: number of MLC pairs not supported\n");
				return -EINVAL;
			}

			memcpy(dst->lptbl.mlc.pairs, src->lptbl.mlc.pairs,
						dst->lptbl.mlc.num_pairs);
		}
	}

	return 0;
}

static int nvme_nvm_identity(struct nvm_dev *nvmdev, struct nvm_id *nvm_id)
{
	struct nvme_ns *ns = nvmdev->q->queuedata;
	struct nvme_nvm_id *nvme_nvm_id;
	struct nvme_nvm_command c = {};
	int ret;

	c.identity.opcode = nvme_nvm_admin_identity;
	c.identity.nsid = cpu_to_le32(ns->ns_id);
	c.identity.chnl_off = 0;

	nvme_nvm_id = kmalloc(sizeof(struct nvme_nvm_id), GFP_KERNEL);
	if (!nvme_nvm_id)
		return -ENOMEM;

	ret = nvme_submit_sync_cmd(ns->ctrl->admin_q, (struct nvme_command *)&c,
				nvme_nvm_id, sizeof(struct nvme_nvm_id));
	if (ret) {
		ret = -EIO;
		goto out;
	}

	nvm_id->ver_id = nvme_nvm_id->ver_id;
	nvm_id->vmnt = nvme_nvm_id->vmnt;
	nvm_id->cgrps = nvme_nvm_id->cgrps;
	nvm_id->cap = le32_to_cpu(nvme_nvm_id->cap);
	nvm_id->dom = le32_to_cpu(nvme_nvm_id->dom);
	memcpy(&nvm_id->ppaf, &nvme_nvm_id->ppaf,
					sizeof(struct nvme_nvm_addr_format));

	ret = init_grps(nvm_id, nvme_nvm_id);
out:
	kfree(nvme_nvm_id);
	return ret;
}

static int nvme_nvm_get_l2p_tbl(struct nvm_dev *nvmdev, u64 slba, u32 nlb,
				nvm_l2p_update_fn *update_l2p, void *priv)
{
	struct nvme_ns *ns = nvmdev->q->queuedata;
	struct nvme_nvm_command c = {};
	u32 len = queue_max_hw_sectors(ns->ctrl->admin_q) << 9;
	u32 nlb_pr_rq = len / sizeof(u64);
	u64 cmd_slba = slba;
	void *entries;
	int ret = 0;

	c.l2p.opcode = nvme_nvm_admin_get_l2p_tbl;
	c.l2p.nsid = cpu_to_le32(ns->ns_id);
	entries = kmalloc(len, GFP_KERNEL);
	if (!entries)
		return -ENOMEM;

	while (nlb) {
		u32 cmd_nlb = min(nlb_pr_rq, nlb);

		c.l2p.slba = cpu_to_le64(cmd_slba);
		c.l2p.nlb = cpu_to_le32(cmd_nlb);

		ret = nvme_submit_sync_cmd(ns->ctrl->admin_q,
				(struct nvme_command *)&c, entries, len);
		if (ret) {
			dev_err(ns->ctrl->device,
				"L2P table transfer failed (%d)\n", ret);
			ret = -EIO;
			goto out;
		}

		if (update_l2p(cmd_slba, cmd_nlb, entries, priv)) {
			ret = -EINTR;
			goto out;
		}

		cmd_slba += cmd_nlb;
		nlb -= cmd_nlb;
	}

out:
	kfree(entries);
	return ret;
}

static int nvme_nvm_get_bb_tbl(struct nvm_dev *nvmdev, struct ppa_addr ppa,
								u8 *blks)
{
	struct request_queue *q = nvmdev->q;
	struct nvme_ns *ns = q->queuedata;
	struct nvme_ctrl *ctrl = ns->ctrl;
	struct nvme_nvm_command c = {};
	struct nvme_nvm_bb_tbl *bb_tbl;
	int nr_blks = nvmdev->blks_per_lun * nvmdev->plane_mode;
	int tblsz = sizeof(struct nvme_nvm_bb_tbl) + nr_blks;
	int ret = 0;

	c.get_bb.opcode = nvme_nvm_admin_get_bb_tbl;
	c.get_bb.nsid = cpu_to_le32(ns->ns_id);
	c.get_bb.spba = cpu_to_le64(ppa.ppa);

	bb_tbl = kzalloc(tblsz, GFP_KERNEL);
	if (!bb_tbl)
		return -ENOMEM;

	ret = nvme_submit_sync_cmd(ctrl->admin_q, (struct nvme_command *)&c,
								bb_tbl, tblsz);
	if (ret) {
		dev_err(ctrl->device, "get bad block table failed (%d)\n", ret);
		ret = -EIO;
		goto out;
	}

	if (bb_tbl->tblid[0] != 'B' || bb_tbl->tblid[1] != 'B' ||
		bb_tbl->tblid[2] != 'L' || bb_tbl->tblid[3] != 'T') {
		dev_err(ctrl->device, "bbt format mismatch\n");
		ret = -EINVAL;
		goto out;
	}

	if (le16_to_cpu(bb_tbl->verid) != 1) {
		ret = -EINVAL;
		dev_err(ctrl->device, "bbt version not supported\n");
		goto out;
	}

	if (le32_to_cpu(bb_tbl->tblks) != nr_blks) {
		ret = -EINVAL;
		dev_err(ctrl->device,
				"bbt unsuspected blocks returned (%u!=%u)",
				le32_to_cpu(bb_tbl->tblks), nr_blks);
		goto out;
	}

	memcpy(blks, bb_tbl->blk, nvmdev->blks_per_lun * nvmdev->plane_mode);
out:
	kfree(bb_tbl);
	return ret;
}

static int nvme_nvm_set_bb_tbl(struct nvm_dev *nvmdev, struct ppa_addr *ppas,
							int nr_ppas, int type)
{
	struct nvme_ns *ns = nvmdev->q->queuedata;
	struct nvme_nvm_command c = {};
	int ret = 0;

	c.set_bb.opcode = nvme_nvm_admin_set_bb_tbl;
	c.set_bb.nsid = cpu_to_le32(ns->ns_id);
	c.set_bb.spba = cpu_to_le64(ppas->ppa);
	c.set_bb.nlb = cpu_to_le16(nr_ppas - 1);
	c.set_bb.value = type;

	ret = nvme_submit_sync_cmd(ns->ctrl->admin_q, (struct nvme_command *)&c,
								NULL, 0);
	if (ret)
		dev_err(ns->ctrl->device, "set bad block table failed (%d)\n",
									ret);
	return ret;
}

static inline void nvme_nvm_rqtocmd(struct request *rq, struct nvm_rq *rqd,
				struct nvme_ns *ns, struct nvme_nvm_command *c)
{
	c->ph_rw.opcode = rqd->opcode;
	c->ph_rw.nsid = cpu_to_le32(ns->ns_id);
	c->ph_rw.spba = cpu_to_le64(rqd->ppa_addr.ppa);
	c->ph_rw.metadata = cpu_to_le64(rqd->dma_meta_list);
	c->ph_rw.control = cpu_to_le16(rqd->flags);
	c->ph_rw.length = cpu_to_le16(rqd->nr_ppas - 1);

	if (rqd->opcode == NVM_OP_HBWRITE || rqd->opcode == NVM_OP_HBREAD)
		c->hb_rw.slba = cpu_to_le64(nvme_block_nr(ns,
					rqd->bio->bi_iter.bi_sector));
}

static void nvme_nvm_end_io(struct request *rq, int error)
{
	struct nvm_rq *rqd = rq->end_io_data;
	struct nvme_nvm_completion *cqe = rq->special;

	if (cqe)
		rqd->ppa_status = le64_to_cpu(cqe->result);

	nvm_end_io(rqd, error);

	kfree(rq->cmd);
	blk_mq_free_request(rq);
}

static int nvme_nvm_submit_io(struct nvm_dev *dev, struct nvm_rq *rqd)
{
	struct request_queue *q = dev->q;
	struct nvme_ns *ns = q->queuedata;
	struct request *rq;
	struct bio *bio = rqd->bio;
	struct nvme_nvm_command *cmd;

	rq = blk_mq_alloc_request(q, bio_data_dir(bio), 0);
	if (IS_ERR(rq))
		return -ENOMEM;

	cmd = kzalloc(sizeof(struct nvme_nvm_command) +
				sizeof(struct nvme_nvm_completion), GFP_KERNEL);
	if (!cmd) {
		blk_mq_free_request(rq);
		return -ENOMEM;
	}

	rq->cmd_type = REQ_TYPE_DRV_PRIV;
	rq->ioprio = bio_prio(bio);

	if (bio_has_data(bio))
		rq->nr_phys_segments = bio_phys_segments(q, bio);

	rq->__data_len = bio->bi_iter.bi_size;
	rq->bio = rq->biotail = bio;

	nvme_nvm_rqtocmd(rq, rqd, ns, cmd);

	rq->cmd = (unsigned char *)cmd;
	rq->cmd_len = sizeof(struct nvme_nvm_command);
	rq->special = cmd + 1;

	rq->end_io_data = rqd;

	blk_execute_rq_nowait(q, NULL, rq, 0, nvme_nvm_end_io);

	return 0;
}

static int nvme_nvm_erase_block(struct nvm_dev *dev, struct nvm_rq *rqd)
{
	struct request_queue *q = dev->q;
	struct nvme_ns *ns = q->queuedata;
	struct nvme_nvm_command c = {};

	c.erase.opcode = NVM_OP_ERASE;
	c.erase.nsid = cpu_to_le32(ns->ns_id);
	c.erase.spba = cpu_to_le64(rqd->ppa_addr.ppa);
	c.erase.length = cpu_to_le16(rqd->nr_ppas - 1);

	return nvme_submit_sync_cmd(q, (struct nvme_command *)&c, NULL, 0);
}

static void *nvme_nvm_create_dma_pool(struct nvm_dev *nvmdev, char *name)
{
	struct nvme_ns *ns = nvmdev->q->queuedata;

	return dma_pool_create(name, ns->ctrl->dev, PAGE_SIZE, PAGE_SIZE, 0);
}

static void nvme_nvm_destroy_dma_pool(void *pool)
{
	struct dma_pool *dma_pool = pool;

	dma_pool_destroy(dma_pool);
}

static void *nvme_nvm_dev_dma_alloc(struct nvm_dev *dev, void *pool,
				    gfp_t mem_flags, dma_addr_t *dma_handler)
{
	return dma_pool_alloc(pool, mem_flags, dma_handler);
}

static void nvme_nvm_dev_dma_free(void *pool, void *addr,
							dma_addr_t dma_handler)
{
	dma_pool_free(pool, addr, dma_handler);
}

static struct nvm_dev_ops nvme_nvm_dev_ops = {
	.identity		= nvme_nvm_identity,

	.get_l2p_tbl		= nvme_nvm_get_l2p_tbl,

	.get_bb_tbl		= nvme_nvm_get_bb_tbl,
	.set_bb_tbl		= nvme_nvm_set_bb_tbl,

	.submit_io		= nvme_nvm_submit_io,
	.erase_block		= nvme_nvm_erase_block,

	.create_dma_pool	= nvme_nvm_create_dma_pool,
	.destroy_dma_pool	= nvme_nvm_destroy_dma_pool,
	.dev_dma_alloc		= nvme_nvm_dev_dma_alloc,
	.dev_dma_free		= nvme_nvm_dev_dma_free,

	.max_phys_sect		= 64,
};

int nvme_nvm_register(struct nvme_ns *ns, char *disk_name, int node,
		      const struct attribute_group *attrs)
{
	struct request_queue *q = ns->queue;
	struct nvm_dev *dev;
	int ret;

	dev = nvm_alloc_dev(node);
	if (!dev)
		return -ENOMEM;

	dev->q = q;
	memcpy(dev->name, disk_name, DISK_NAME_LEN);
	dev->ops = &nvme_nvm_dev_ops;
	dev->parent_dev = ns->ctrl->device;
	dev->private_data = ns;
	ns->ndev = dev;

	ret = nvm_register(dev);

	ns->lba_shift = ilog2(dev->sec_size) - 9;

	if (sysfs_create_group(&dev->dev.kobj, attrs))
		pr_warn("%s: failed to create sysfs group for identification\n",
			disk_name);
	return ret;
}

void nvme_nvm_unregister(struct nvme_ns *ns)
{
	nvm_unregister(ns->ndev);
}

/* move to shared place when used in multiple places. */
#define PCI_VENDOR_ID_CNEX 0x1d1d
#define PCI_DEVICE_ID_CNEX_WL 0x2807
#define PCI_DEVICE_ID_CNEX_QEMU 0x1f1f

int nvme_nvm_ns_supported(struct nvme_ns *ns, struct nvme_id_ns *id)
{
	struct nvme_ctrl *ctrl = ns->ctrl;
	/* XXX: this is poking into PCI structures from generic code! */
	struct pci_dev *pdev = to_pci_dev(ctrl->dev);

	/* QEMU NVMe simulator - PCI ID + Vendor specific bit */
	if (pdev->vendor == PCI_VENDOR_ID_CNEX &&
				pdev->device == PCI_DEVICE_ID_CNEX_QEMU &&
							id->vs[0] == 0x1)
		return 1;

	/* CNEX Labs - PCI ID + Vendor specific bit */
	if (pdev->vendor == PCI_VENDOR_ID_CNEX &&
				pdev->device == PCI_DEVICE_ID_CNEX_WL &&
							id->vs[0] == 0x1)
		return 1;

	return 0;
}
Commit	Line	Data
ca064085 MB	1	/*
	2	* nvme-lightnvm.c - LightNVM NVMe device
	3	*
	4	* Copyright (C) 2014-2015 IT University of Copenhagen
	5	* Initial release: Matias Bjorling <mb@lightnvm.io>
	6	*
	7	* This program is free software; you can redistribute it and/or
	8	* modify it under the terms of the GNU General Public License version
	9	* 2 as published by the Free Software Foundation.
	10	*
	11	* This program is distributed in the hope that it will be useful, but
	12	* WITHOUT ANY WARRANTY; without even the implied warranty of
	13	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	14	* General Public License for more details.
	15	*
	16	* You should have received a copy of the GNU General Public License
	17	* along with this program; see the file COPYING. If not, write to
	18	* the Free Software Foundation, 675 Mass Ave, Cambridge, MA 02139,
	19	* USA.
	20	*
	21	*/
	22
	23	#include "nvme.h"
	24
ca064085 MB	25	#include <linux/nvme.h>
	26	#include <linux/bitops.h>
	27	#include <linux/lightnvm.h>
	28	#include <linux/vmalloc.h>
	29
	30	enum nvme_nvm_admin_opcode {
	31	nvme_nvm_admin_identity = 0xe2,
	32	nvme_nvm_admin_get_l2p_tbl = 0xea,
	33	nvme_nvm_admin_get_bb_tbl = 0xf2,
	34	nvme_nvm_admin_set_bb_tbl = 0xf1,
	35	};
	36
	37	struct nvme_nvm_hb_rw {
	38	__u8 opcode;
	39	__u8 flags;
	40	__u16 command_id;
	41	__le32 nsid;
	42	__u64 rsvd2;
	43	__le64 metadata;
	44	__le64 prp1;
	45	__le64 prp2;
	46	__le64 spba;
	47	__le16 length;
	48	__le16 control;
	49	__le32 dsmgmt;
	50	__le64 slba;
	51	};
	52
	53	struct nvme_nvm_ph_rw {
	54	__u8 opcode;
	55	__u8 flags;
	56	__u16 command_id;
	57	__le32 nsid;
	58	__u64 rsvd2;
	59	__le64 metadata;
	60	__le64 prp1;
	61	__le64 prp2;
	62	__le64 spba;
	63	__le16 length;
	64	__le16 control;
	65	__le32 dsmgmt;
	66	__le64 resv;
	67	};
	68
	69	struct nvme_nvm_identity {
	70	__u8 opcode;
	71	__u8 flags;
	72	__u16 command_id;
	73	__le32 nsid;
	74	__u64 rsvd[2];
	75	__le64 prp1;
	76	__le64 prp2;
	77	__le32 chnl_off;
	78	__u32 rsvd11[5];
	79	};
	80
	81	struct nvme_nvm_l2ptbl {
	82	__u8 opcode;
	83	__u8 flags;
	84	__u16 command_id;
	85	__le32 nsid;
	86	__le32 cdw2[4];
	87	__le64 prp1;
	88	__le64 prp2;
89	__le64 slba;
90	__le32 nlb;
91	__le16 cdw14[6];
92	};
93
11450469	94	struct nvme_nvm_getbbtbl {
ca064085 MB	95	__u8 opcode;
	96	__u8 flags;
	97	__u16 command_id;
	98	__le32 nsid;
	99	__u64 rsvd[2];
	100	__le64 prp1;
	101	__le64 prp2;
11450469 MB	102	__le64 spba;
	103	__u32 rsvd4[4];
	104	};
	105
	106	struct nvme_nvm_setbbtbl {
	107	__u8 opcode;
	108	__u8 flags;
	109	__u16 command_id;
	110	__le32 nsid;
	111	__le64 rsvd[2];
	112	__le64 prp1;
	113	__le64 prp2;
	114	__le64 spba;
	115	__le16 nlb;
	116	__u8 value;
	117	__u8 rsvd3;
	118	__u32 rsvd4[3];
ca064085 MB	119	};
	120
	121	struct nvme_nvm_erase_blk {
	122	__u8 opcode;
	123	__u8 flags;
	124	__u16 command_id;
	125	__le32 nsid;
	126	__u64 rsvd[2];
	127	__le64 prp1;
	128	__le64 prp2;
	129	__le64 spba;
	130	__le16 length;
	131	__le16 control;
	132	__le32 dsmgmt;
	133	__le64 resv;
	134	};
	135
	136	struct nvme_nvm_command {
	137	union {
	138	struct nvme_common_command common;
	139	struct nvme_nvm_identity identity;
	140	struct nvme_nvm_hb_rw hb_rw;
	141	struct nvme_nvm_ph_rw ph_rw;
	142	struct nvme_nvm_l2ptbl l2p;
11450469 MB	143	struct nvme_nvm_getbbtbl get_bb;
11450469 MB	144	struct nvme_nvm_setbbtbl set_bb;
ca064085 MB	145	struct nvme_nvm_erase_blk erase;
	146	};
	147	};
	148
9f867268 MB	149	struct nvme_nvm_completion {
	150	__le64 result; /* Used by LightNVM to return ppa completions */
	151	__le16 sq_head; /* how much of this queue may be reclaimed */
	152	__le16 sq_id; /* submission queue that generated this entry */
	153	__u16 command_id; /* of the command which completed */
	154	__le16 status; /* did the command fail, and if so, why? */
	155	};
	156
6dde1d6c	157	#define NVME_NVM_LP_MLC_PAIRS 886
ca5927e7	158	struct nvme_nvm_lp_mlc {
6f929702	159	__le16 num_pairs;
6dde1d6c	160	__u8 pairs[NVME_NVM_LP_MLC_PAIRS];
ca5927e7 MB	161	};
	162
	163	struct nvme_nvm_lp_tbl {
	164	__u8 id[8];
	165	struct nvme_nvm_lp_mlc mlc;
	166	};
	167
ca064085 MB	168	struct nvme_nvm_id_group {
	169	__u8 mtype;
	170	__u8 fmtype;
	171	__le16 res16;
	172	__u8 num_ch;
	173	__u8 num_lun;
	174	__u8 num_pln;
36d5dbc6	175	__u8 rsvd1;
ca064085 MB	176	__le16 num_blk;
	177	__le16 num_pg;
	178	__le16 fpg_sz;
	179	__le16 csecs;
	180	__le16 sos;
36d5dbc6	181	__le16 rsvd2;
ca064085 MB	182	__le32 trdt;
	183	__le32 trdm;
	184	__le32 tprt;
	185	__le32 tprm;
	186	__le32 tbet;
	187	__le32 tbem;
	188	__le32 mpos;
12be5edf	189	__le32 mccap;
ca064085	190	__le16 cpar;
ca5927e7 MB	191	__u8 reserved[10];
ca5927e7 MB	192	struct nvme_nvm_lp_tbl lptbl;
ca064085 MB	193	} __packed;
	194
	195	struct nvme_nvm_addr_format {
	196	__u8 ch_offset;
	197	__u8 ch_len;
	198	__u8 lun_offset;
	199	__u8 lun_len;
	200	__u8 pln_offset;
	201	__u8 pln_len;
	202	__u8 blk_offset;
	203	__u8 blk_len;
	204	__u8 pg_offset;
	205	__u8 pg_len;
	206	__u8 sect_offset;
	207	__u8 sect_len;
	208	__u8 res[4];
	209	} __packed;
	210
	211	struct nvme_nvm_id {
	212	__u8 ver_id;
	213	__u8 vmnt;
	214	__u8 cgrps;
dad1b009	215	__u8 res;
ca064085 MB	216	__le32 cap;
	217	__le32 dom;
	218	struct nvme_nvm_addr_format ppaf;
dad1b009	219	__u8 resv[228];
ca064085 MB	220	struct nvme_nvm_id_group groups[4];
	221	} __packed;
	222
11450469 MB	223	struct nvme_nvm_bb_tbl {
	224	__u8 tblid[4];
	225	__le16 verid;
	226	__le16 revid;
	227	__le32 rvsd1;
	228	__le32 tblks;
	229	__le32 tfact;
	230	__le32 tgrown;
	231	__le32 tdresv;
	232	__le32 thresv;
	233	__le32 rsvd2[8];
	234	__u8 blk[0];
	235	};
	236
ca064085 MB	237	/*
	238	* Check we didn't inadvertently grow the command struct
	239	*/
	240	static inline void _nvme_nvm_check_size(void)
	241	{
	242	BUILD_BUG_ON(sizeof(struct nvme_nvm_identity) != 64);
	243	BUILD_BUG_ON(sizeof(struct nvme_nvm_hb_rw) != 64);
	244	BUILD_BUG_ON(sizeof(struct nvme_nvm_ph_rw) != 64);
11450469 MB	245	BUILD_BUG_ON(sizeof(struct nvme_nvm_getbbtbl) != 64);
11450469 MB	246	BUILD_BUG_ON(sizeof(struct nvme_nvm_setbbtbl) != 64);
ca064085 MB	247	BUILD_BUG_ON(sizeof(struct nvme_nvm_l2ptbl) != 64);
	248	BUILD_BUG_ON(sizeof(struct nvme_nvm_erase_blk) != 64);
	249	BUILD_BUG_ON(sizeof(struct nvme_nvm_id_group) != 960);
	250	BUILD_BUG_ON(sizeof(struct nvme_nvm_addr_format) != 128);
	251	BUILD_BUG_ON(sizeof(struct nvme_nvm_id) != 4096);
11450469	252	BUILD_BUG_ON(sizeof(struct nvme_nvm_bb_tbl) != 512);
ca064085 MB	253	}
	254
	255	static int init_grps(struct nvm_id nvm_id, struct nvme_nvm_id nvme_nvm_id)
	256	{
	257	struct nvme_nvm_id_group *src;
	258	struct nvm_id_group *dst;
	259	int i, end;
	260
	261	end = min_t(u32, 4, nvm_id->cgrps);
	262
	263	for (i = 0; i < end; i++) {
	264	src = &nvme_nvm_id->groups[i];
	265	dst = &nvm_id->groups[i];
	266
	267	dst->mtype = src->mtype;
	268	dst->fmtype = src->fmtype;
	269	dst->num_ch = src->num_ch;
	270	dst->num_lun = src->num_lun;
	271	dst->num_pln = src->num_pln;
	272
	273	dst->num_pg = le16_to_cpu(src->num_pg);
	274	dst->num_blk = le16_to_cpu(src->num_blk);
	275	dst->fpg_sz = le16_to_cpu(src->fpg_sz);
	276	dst->csecs = le16_to_cpu(src->csecs);
	277	dst->sos = le16_to_cpu(src->sos);
	278
	279	dst->trdt = le32_to_cpu(src->trdt);
	280	dst->trdm = le32_to_cpu(src->trdm);
	281	dst->tprt = le32_to_cpu(src->tprt);
	282	dst->tprm = le32_to_cpu(src->tprm);
	283	dst->tbet = le32_to_cpu(src->tbet);
	284	dst->tbem = le32_to_cpu(src->tbem);
	285	dst->mpos = le32_to_cpu(src->mpos);
12be5edf	286	dst->mccap = le32_to_cpu(src->mccap);
ca064085 MB	287
ca064085 MB	288	dst->cpar = le16_to_cpu(src->cpar);
ca5927e7 MB	289
	290	if (dst->fmtype == NVM_ID_FMTYPE_MLC) {
	291	memcpy(dst->lptbl.id, src->lptbl.id, 8);
	292	dst->lptbl.mlc.num_pairs =
	293	le16_to_cpu(src->lptbl.mlc.num_pairs);
6dde1d6c MB	294
	295	if (dst->lptbl.mlc.num_pairs > NVME_NVM_LP_MLC_PAIRS) {
	296	pr_err("nvm: number of MLC pairs not supported\n");
	297	return -EINVAL;
	298	}
	299
ca5927e7	300	memcpy(dst->lptbl.mlc.pairs, src->lptbl.mlc.pairs,
6dde1d6c	301	dst->lptbl.mlc.num_pairs);
ca5927e7	302	}
ca064085 MB	303	}
	304
	305	return 0;
	306	}
	307
16f26c3a	308	static int nvme_nvm_identity(struct nvm_dev nvmdev, struct nvm_id nvm_id)
ca064085	309	{
16f26c3a	310	struct nvme_ns *ns = nvmdev->q->queuedata;
ca064085 MB	311	struct nvme_nvm_id *nvme_nvm_id;
	312	struct nvme_nvm_command c = {};
	313	int ret;
	314
	315	c.identity.opcode = nvme_nvm_admin_identity;
	316	c.identity.nsid = cpu_to_le32(ns->ns_id);
	317	c.identity.chnl_off = 0;
	318
	319	nvme_nvm_id = kmalloc(sizeof(struct nvme_nvm_id), GFP_KERNEL);
	320	if (!nvme_nvm_id)
	321	return -ENOMEM;
	322
ac02ddde	323	ret = nvme_submit_sync_cmd(ns->ctrl->admin_q, (struct nvme_command *)&c,
47b3115a	324	nvme_nvm_id, sizeof(struct nvme_nvm_id));
ca064085 MB	325	if (ret) {
	326	ret = -EIO;
	327	goto out;
	328	}
	329
	330	nvm_id->ver_id = nvme_nvm_id->ver_id;
	331	nvm_id->vmnt = nvme_nvm_id->vmnt;
	332	nvm_id->cgrps = nvme_nvm_id->cgrps;
	333	nvm_id->cap = le32_to_cpu(nvme_nvm_id->cap);
	334	nvm_id->dom = le32_to_cpu(nvme_nvm_id->dom);
2393bd39 MB	335	memcpy(&nvm_id->ppaf, &nvme_nvm_id->ppaf,
2393bd39 MB	336	sizeof(struct nvme_nvm_addr_format));
ca064085 MB	337
	338	ret = init_grps(nvm_id, nvme_nvm_id);
	339	out:
	340	kfree(nvme_nvm_id);
	341	return ret;
	342	}
	343
16f26c3a	344	static int nvme_nvm_get_l2p_tbl(struct nvm_dev *nvmdev, u64 slba, u32 nlb,
ca064085 MB	345	nvm_l2p_update_fn update_l2p, void priv)
ca064085 MB	346	{
16f26c3a	347	struct nvme_ns *ns = nvmdev->q->queuedata;
ca064085	348	struct nvme_nvm_command c = {};
ac02ddde	349	u32 len = queue_max_hw_sectors(ns->ctrl->admin_q) << 9;
5f436e5e	350	u32 nlb_pr_rq = len / sizeof(u64);
ca064085 MB	351	u64 cmd_slba = slba;
	352	void *entries;
	353	int ret = 0;
	354
	355	c.l2p.opcode = nvme_nvm_admin_get_l2p_tbl;
	356	c.l2p.nsid = cpu_to_le32(ns->ns_id);
	357	entries = kmalloc(len, GFP_KERNEL);
	358	if (!entries)
	359	return -ENOMEM;
	360
	361	while (nlb) {
5f436e5e	362	u32 cmd_nlb = min(nlb_pr_rq, nlb);
ca064085 MB	363
	364	c.l2p.slba = cpu_to_le64(cmd_slba);
	365	c.l2p.nlb = cpu_to_le32(cmd_nlb);
	366
ac02ddde	367	ret = nvme_submit_sync_cmd(ns->ctrl->admin_q,
47b3115a	368	(struct nvme_command *)&c, entries, len);
ca064085	369	if (ret) {
b86d8d36 SG	370	dev_err(ns->ctrl->device,
b86d8d36 SG	371	"L2P table transfer failed (%d)\n", ret);
ca064085 MB	372	ret = -EIO;
	373	goto out;
	374	}
	375
	376	if (update_l2p(cmd_slba, cmd_nlb, entries, priv)) {
	377	ret = -EINTR;
	378	goto out;
	379	}
	380
	381	cmd_slba += cmd_nlb;
	382	nlb -= cmd_nlb;
	383	}
	384
	385	out:
	386	kfree(entries);
	387	return ret;
	388	}
	389
08236c6b	390	static int nvme_nvm_get_bb_tbl(struct nvm_dev *nvmdev, struct ppa_addr ppa,
e11903f5	391	u8 *blks)
ca064085	392	{
08236c6b	393	struct request_queue *q = nvmdev->q;
ca064085	394	struct nvme_ns *ns = q->queuedata;
ac02ddde	395	struct nvme_ctrl *ctrl = ns->ctrl;
ca064085	396	struct nvme_nvm_command c = {};
11450469	397	struct nvme_nvm_bb_tbl *bb_tbl;
22e8c976 MB	398	int nr_blks = nvmdev->blks_per_lun * nvmdev->plane_mode;
22e8c976 MB	399	int tblsz = sizeof(struct nvme_nvm_bb_tbl) + nr_blks;
ca064085 MB	400	int ret = 0;
	401
	402	c.get_bb.opcode = nvme_nvm_admin_get_bb_tbl;
	403	c.get_bb.nsid = cpu_to_le32(ns->ns_id);
11450469	404	c.get_bb.spba = cpu_to_le64(ppa.ppa);
ca064085	405
11450469 MB	406	bb_tbl = kzalloc(tblsz, GFP_KERNEL);
	407	if (!bb_tbl)
	408	return -ENOMEM;
ca064085	409
ac02ddde	410	ret = nvme_submit_sync_cmd(ctrl->admin_q, (struct nvme_command *)&c,
47b3115a	411	bb_tbl, tblsz);
ca064085	412	if (ret) {
b86d8d36	413	dev_err(ctrl->device, "get bad block table failed (%d)\n", ret);
ca064085 MB	414	ret = -EIO;
	415	goto out;
	416	}
	417
11450469 MB	418	if (bb_tbl->tblid[0] != 'B' \|\| bb_tbl->tblid[1] != 'B' \|\|
11450469 MB	419	bb_tbl->tblid[2] != 'L' \|\| bb_tbl->tblid[3] != 'T') {
b86d8d36	420	dev_err(ctrl->device, "bbt format mismatch\n");
11450469 MB	421	ret = -EINVAL;
	422	goto out;
	423	}
	424
	425	if (le16_to_cpu(bb_tbl->verid) != 1) {
	426	ret = -EINVAL;
b86d8d36	427	dev_err(ctrl->device, "bbt version not supported\n");
11450469 MB	428	goto out;
	429	}
	430
22e8c976	431	if (le32_to_cpu(bb_tbl->tblks) != nr_blks) {
11450469	432	ret = -EINVAL;
b86d8d36 SG	433	dev_err(ctrl->device,
b86d8d36 SG	434	"bbt unsuspected blocks returned (%u!=%u)",
22e8c976	435	le32_to_cpu(bb_tbl->tblks), nr_blks);
11450469 MB	436	goto out;
	437	}
	438
e11903f5	439	memcpy(blks, bb_tbl->blk, nvmdev->blks_per_lun * nvmdev->plane_mode);
ca064085	440	out:
11450469 MB	441	kfree(bb_tbl);
	442	return ret;
	443	}
	444
00ee6cc3 MB	445	static int nvme_nvm_set_bb_tbl(struct nvm_dev nvmdev, struct ppa_addr ppas,
00ee6cc3 MB	446	int nr_ppas, int type)
11450469	447	{
16f26c3a	448	struct nvme_ns *ns = nvmdev->q->queuedata;
11450469 MB	449	struct nvme_nvm_command c = {};
	450	int ret = 0;
	451
	452	c.set_bb.opcode = nvme_nvm_admin_set_bb_tbl;
	453	c.set_bb.nsid = cpu_to_le32(ns->ns_id);
00ee6cc3 MB	454	c.set_bb.spba = cpu_to_le64(ppas->ppa);
00ee6cc3 MB	455	c.set_bb.nlb = cpu_to_le16(nr_ppas - 1);
11450469 MB	456	c.set_bb.value = type;
11450469 MB	457
ac02ddde	458	ret = nvme_submit_sync_cmd(ns->ctrl->admin_q, (struct nvme_command *)&c,
47b3115a	459	NULL, 0);
11450469	460	if (ret)
b86d8d36 SG	461	dev_err(ns->ctrl->device, "set bad block table failed (%d)\n",
b86d8d36 SG	462	ret);
ca064085 MB	463	return ret;
	464	}
	465
	466	static inline void nvme_nvm_rqtocmd(struct request rq, struct nvm_rq rqd,
	467	struct nvme_ns ns, struct nvme_nvm_command c)
	468	{
	469	c->ph_rw.opcode = rqd->opcode;
	470	c->ph_rw.nsid = cpu_to_le32(ns->ns_id);
	471	c->ph_rw.spba = cpu_to_le64(rqd->ppa_addr.ppa);
45bbd052	472	c->ph_rw.metadata = cpu_to_le64(rqd->dma_meta_list);
ca064085	473	c->ph_rw.control = cpu_to_le16(rqd->flags);
6d5be959	474	c->ph_rw.length = cpu_to_le16(rqd->nr_ppas - 1);
ca064085 MB	475
ca064085 MB	476	if (rqd->opcode == NVM_OP_HBWRITE \|\| rqd->opcode == NVM_OP_HBREAD)
b0b4e09c MB	477	c->hb_rw.slba = cpu_to_le64(nvme_block_nr(ns,
b0b4e09c MB	478	rqd->bio->bi_iter.bi_sector));
ca064085 MB	479	}
	480
	481	static void nvme_nvm_end_io(struct request *rq, int error)
	482	{
	483	struct nvm_rq *rqd = rq->end_io_data;
9f867268 MB	484	struct nvme_nvm_completion *cqe = rq->special;
	485
	486	if (cqe)
	487	rqd->ppa_status = le64_to_cpu(cqe->result);
ca064085	488
91276162	489	nvm_end_io(rqd, error);
ca064085 MB	490
	491	kfree(rq->cmd);
	492	blk_mq_free_request(rq);
	493	}
	494
16f26c3a	495	static int nvme_nvm_submit_io(struct nvm_dev dev, struct nvm_rq rqd)
ca064085	496	{
16f26c3a	497	struct request_queue *q = dev->q;
ca064085 MB	498	struct nvme_ns *ns = q->queuedata;
	499	struct request *rq;
	500	struct bio *bio = rqd->bio;
	501	struct nvme_nvm_command *cmd;
	502
70246286	503	rq = blk_mq_alloc_request(q, bio_data_dir(bio), 0);
ca064085 MB	504	if (IS_ERR(rq))
	505	return -ENOMEM;
	506
9f867268 MB	507	cmd = kzalloc(sizeof(struct nvme_nvm_command) +
9f867268 MB	508	sizeof(struct nvme_nvm_completion), GFP_KERNEL);
ca064085 MB	509	if (!cmd) {
	510	blk_mq_free_request(rq);
	511	return -ENOMEM;
	512	}
	513
	514	rq->cmd_type = REQ_TYPE_DRV_PRIV;
	515	rq->ioprio = bio_prio(bio);
	516
	517	if (bio_has_data(bio))
	518	rq->nr_phys_segments = bio_phys_segments(q, bio);
	519
	520	rq->__data_len = bio->bi_iter.bi_size;
	521	rq->bio = rq->biotail = bio;
	522
	523	nvme_nvm_rqtocmd(rq, rqd, ns, cmd);
	524
	525	rq->cmd = (unsigned char *)cmd;
	526	rq->cmd_len = sizeof(struct nvme_nvm_command);
9f867268	527	rq->special = cmd + 1;
ca064085 MB	528
	529	rq->end_io_data = rqd;
	530
	531	blk_execute_rq_nowait(q, NULL, rq, 0, nvme_nvm_end_io);
	532
	533	return 0;
	534	}
	535
16f26c3a	536	static int nvme_nvm_erase_block(struct nvm_dev dev, struct nvm_rq rqd)
ca064085	537	{
16f26c3a	538	struct request_queue *q = dev->q;
ca064085 MB	539	struct nvme_ns *ns = q->queuedata;
	540	struct nvme_nvm_command c = {};
	541
	542	c.erase.opcode = NVM_OP_ERASE;
	543	c.erase.nsid = cpu_to_le32(ns->ns_id);
	544	c.erase.spba = cpu_to_le64(rqd->ppa_addr.ppa);
6d5be959	545	c.erase.length = cpu_to_le16(rqd->nr_ppas - 1);
ca064085 MB	546
	547	return nvme_submit_sync_cmd(q, (struct nvme_command *)&c, NULL, 0);
	548	}
	549
16f26c3a	550	static void nvme_nvm_create_dma_pool(struct nvm_dev nvmdev, char *name)
ca064085	551	{
16f26c3a	552	struct nvme_ns *ns = nvmdev->q->queuedata;
ca064085	553
ac02ddde	554	return dma_pool_create(name, ns->ctrl->dev, PAGE_SIZE, PAGE_SIZE, 0);
ca064085 MB	555	}
	556
	557	static void nvme_nvm_destroy_dma_pool(void *pool)
	558	{
	559	struct dma_pool *dma_pool = pool;
	560
	561	dma_pool_destroy(dma_pool);
	562	}
	563
16f26c3a	564	static void nvme_nvm_dev_dma_alloc(struct nvm_dev dev, void *pool,
ca064085 MB	565	gfp_t mem_flags, dma_addr_t *dma_handler)
	566	{
	567	return dma_pool_alloc(pool, mem_flags, dma_handler);
	568	}
	569
75b85649	570	static void nvme_nvm_dev_dma_free(void pool, void addr,
ca064085 MB	571	dma_addr_t dma_handler)
ca064085 MB	572	{
75b85649	573	dma_pool_free(pool, addr, dma_handler);
ca064085 MB	574	}
	575
	576	static struct nvm_dev_ops nvme_nvm_dev_ops = {
	577	.identity = nvme_nvm_identity,
	578
	579	.get_l2p_tbl = nvme_nvm_get_l2p_tbl,
	580
	581	.get_bb_tbl = nvme_nvm_get_bb_tbl,
11450469	582	.set_bb_tbl = nvme_nvm_set_bb_tbl,
ca064085 MB	583
	584	.submit_io = nvme_nvm_submit_io,
	585	.erase_block = nvme_nvm_erase_block,
	586
	587	.create_dma_pool = nvme_nvm_create_dma_pool,
	588	.destroy_dma_pool = nvme_nvm_destroy_dma_pool,
	589	.dev_dma_alloc = nvme_nvm_dev_dma_alloc,
	590	.dev_dma_free = nvme_nvm_dev_dma_free,
	591
	592	.max_phys_sect = 64,
	593	};
	594
40267efd SL	595	int nvme_nvm_register(struct nvme_ns ns, char disk_name, int node,
40267efd SL	596	const struct attribute_group *attrs)
ca064085	597	{
b0b4e09c MB	598	struct request_queue *q = ns->queue;
	599	struct nvm_dev *dev;
	600	int ret;
	601
	602	dev = nvm_alloc_dev(node);
	603	if (!dev)
	604	return -ENOMEM;
	605
	606	dev->q = q;
	607	memcpy(dev->name, disk_name, DISK_NAME_LEN);
	608	dev->ops = &nvme_nvm_dev_ops;
40267efd SL	609	dev->parent_dev = ns->ctrl->device;
40267efd SL	610	dev->private_data = ns;
b0b4e09c MB	611	ns->ndev = dev;
	612
	613	ret = nvm_register(dev);
	614
	615	ns->lba_shift = ilog2(dev->sec_size) - 9;
	616
40267efd SL	617	if (sysfs_create_group(&dev->dev.kobj, attrs))
	618	pr_warn("%s: failed to create sysfs group for identification\n",
	619	disk_name);
b0b4e09c	620	return ret;
ca064085 MB	621	}
ca064085 MB	622
b0b4e09c	623	void nvme_nvm_unregister(struct nvme_ns *ns)
ca064085	624	{
b0b4e09c	625	nvm_unregister(ns->ndev);
ca064085 MB	626	}
ca064085 MB	627
09f2e716 MB	628	/* move to shared place when used in multiple places. */
	629	#define PCI_VENDOR_ID_CNEX 0x1d1d
	630	#define PCI_DEVICE_ID_CNEX_WL 0x2807
	631	#define PCI_DEVICE_ID_CNEX_QEMU 0x1f1f
	632
ca064085 MB	633	int nvme_nvm_ns_supported(struct nvme_ns ns, struct nvme_id_ns id)
ca064085 MB	634	{
ac02ddde CH	635	struct nvme_ctrl *ctrl = ns->ctrl;
	636	/* XXX: this is poking into PCI structures from generic code! */
	637	struct pci_dev *pdev = to_pci_dev(ctrl->dev);
ca064085 MB	638
ca064085 MB	639	/* QEMU NVMe simulator - PCI ID + Vendor specific bit */
09f2e716 MB	640	if (pdev->vendor == PCI_VENDOR_ID_CNEX &&
09f2e716 MB	641	pdev->device == PCI_DEVICE_ID_CNEX_QEMU &&
ca064085 MB	642	id->vs[0] == 0x1)
	643	return 1;
	644
	645	/* CNEX Labs - PCI ID + Vendor specific bit */
09f2e716 MB	646	if (pdev->vendor == PCI_VENDOR_ID_CNEX &&
09f2e716 MB	647	pdev->device == PCI_DEVICE_ID_CNEX_WL &&
ca064085 MB	648	id->vs[0] == 0x1)
	649	return 1;
	650
	651	return 0;
	652	}