[mirror_ubuntu-zesty-kernel.git] / drivers / nvdimm / blk.c

/*
 * NVDIMM Block Window Driver
 * Copyright (c) 2014, Intel Corporation.
 *
 * This program is free software; you can redistribute it and/or modify it
 * under the terms and conditions of the GNU General Public License,
 * version 2, as published by the Free Software Foundation.
 *
 * This program is distributed in the hope 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.
 */

#include <linux/blkdev.h>
#include <linux/fs.h>
#include <linux/genhd.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/nd.h>
#include <linux/sizes.h>
#include "nd.h"

static u32 nsblk_meta_size(struct nd_namespace_blk *nsblk)
{
	return nsblk->lbasize - ((nsblk->lbasize >= 4096) ? 4096 : 512);
}

static u32 nsblk_internal_lbasize(struct nd_namespace_blk *nsblk)
{
	return roundup(nsblk->lbasize, INT_LBASIZE_ALIGNMENT);
}

static u32 nsblk_sector_size(struct nd_namespace_blk *nsblk)
{
	return nsblk->lbasize - nsblk_meta_size(nsblk);
}

static resource_size_t to_dev_offset(struct nd_namespace_blk *nsblk,
				resource_size_t ns_offset, unsigned int len)
{
	int i;

	for (i = 0; i < nsblk->num_resources; i++) {
		if (ns_offset < resource_size(nsblk->res[i])) {
			if (ns_offset + len > resource_size(nsblk->res[i])) {
				dev_WARN_ONCE(&nsblk->common.dev, 1,
					"illegal request\n");
				return SIZE_MAX;
			}
			return nsblk->res[i]->start + ns_offset;
		}
		ns_offset -= resource_size(nsblk->res[i]);
	}

	dev_WARN_ONCE(&nsblk->common.dev, 1, "request out of range\n");
	return SIZE_MAX;
}

static struct nd_blk_region *to_ndbr(struct nd_namespace_blk *nsblk)
{
	struct nd_region *nd_region;
	struct device *parent;

	parent = nsblk->common.dev.parent;
	nd_region = container_of(parent, struct nd_region, dev);
	return container_of(nd_region, struct nd_blk_region, nd_region);
}

#ifdef CONFIG_BLK_DEV_INTEGRITY
static int nd_blk_rw_integrity(struct nd_namespace_blk *nsblk,
		struct bio_integrity_payload *bip, u64 lba, int rw)
{
	struct nd_blk_region *ndbr = to_ndbr(nsblk);
	unsigned int len = nsblk_meta_size(nsblk);
	resource_size_t	dev_offset, ns_offset;
	u32 internal_lbasize, sector_size;
	int err = 0;

	internal_lbasize = nsblk_internal_lbasize(nsblk);
	sector_size = nsblk_sector_size(nsblk);
	ns_offset = lba * internal_lbasize + sector_size;
	dev_offset = to_dev_offset(nsblk, ns_offset, len);
	if (dev_offset == SIZE_MAX)
		return -EIO;

	while (len) {
		unsigned int cur_len;
		struct bio_vec bv;
		void *iobuf;

		bv = bvec_iter_bvec(bip->bip_vec, bip->bip_iter);
		/*
		 * The 'bv' obtained from bvec_iter_bvec has its .bv_len and
		 * .bv_offset already adjusted for iter->bi_bvec_done, and we
		 * can use those directly
		 */

		cur_len = min(len, bv.bv_len);
		iobuf = kmap_atomic(bv.bv_page);
		err = ndbr->do_io(ndbr, dev_offset, iobuf + bv.bv_offset,
				cur_len, rw);
		kunmap_atomic(iobuf);
		if (err)
			return err;

		len -= cur_len;
		dev_offset += cur_len;
		bvec_iter_advance(bip->bip_vec, &bip->bip_iter, cur_len);
	}

	return err;
}

#else /* CONFIG_BLK_DEV_INTEGRITY */
static int nd_blk_rw_integrity(struct nd_namespace_blk *nsblk,
		struct bio_integrity_payload *bip, u64 lba, int rw)
{
	return 0;
}
#endif

static int nsblk_do_bvec(struct nd_namespace_blk *nsblk,
		struct bio_integrity_payload *bip, struct page *page,
		unsigned int len, unsigned int off, int rw, sector_t sector)
{
	struct nd_blk_region *ndbr = to_ndbr(nsblk);
	resource_size_t	dev_offset, ns_offset;
	u32 internal_lbasize, sector_size;
	int err = 0;
	void *iobuf;
	u64 lba;

	internal_lbasize = nsblk_internal_lbasize(nsblk);
	sector_size = nsblk_sector_size(nsblk);
	while (len) {
		unsigned int cur_len;

		/*
		 * If we don't have an integrity payload, we don't have to
		 * split the bvec into sectors, as this would cause unnecessary
		 * Block Window setup/move steps. the do_io routine is capable
		 * of handling len <= PAGE_SIZE.
		 */
		cur_len = bip ? min(len, sector_size) : len;

		lba = div_u64(sector << SECTOR_SHIFT, sector_size);
		ns_offset = lba * internal_lbasize;
		dev_offset = to_dev_offset(nsblk, ns_offset, cur_len);
		if (dev_offset == SIZE_MAX)
			return -EIO;

		iobuf = kmap_atomic(page);
		err = ndbr->do_io(ndbr, dev_offset, iobuf + off, cur_len, rw);
		kunmap_atomic(iobuf);
		if (err)
			return err;

		if (bip) {
			err = nd_blk_rw_integrity(nsblk, bip, lba, rw);
			if (err)
				return err;
		}
		len -= cur_len;
		off += cur_len;
		sector += sector_size >> SECTOR_SHIFT;
	}

	return err;
}

static blk_qc_t nd_blk_make_request(struct request_queue *q, struct bio *bio)
{
	struct bio_integrity_payload *bip;
	struct nd_namespace_blk *nsblk;
	struct bvec_iter iter;
	unsigned long start;
	struct bio_vec bvec;
	int err = 0, rw;
	bool do_acct;

	/*
	 * bio_integrity_enabled also checks if the bio already has an
	 * integrity payload attached. If it does, we *don't* do a
	 * bio_integrity_prep here - the payload has been generated by
	 * another kernel subsystem, and we just pass it through.
	 */
	if (bio_integrity_enabled(bio) && bio_integrity_prep(bio)) {
		bio->bi_error = -EIO;
		goto out;
	}

	bip = bio_integrity(bio);
	nsblk = q->queuedata;
	rw = bio_data_dir(bio);
	do_acct = nd_iostat_start(bio, &start);
	bio_for_each_segment(bvec, bio, iter) {
		unsigned int len = bvec.bv_len;

		BUG_ON(len > PAGE_SIZE);
		err = nsblk_do_bvec(nsblk, bip, bvec.bv_page, len,
				bvec.bv_offset, rw, iter.bi_sector);
		if (err) {
			dev_dbg(&nsblk->common.dev,
					"io error in %s sector %lld, len %d,\n",
					(rw == READ) ? "READ" : "WRITE",
					(unsigned long long) iter.bi_sector, len);
			bio->bi_error = err;
			break;
		}
	}
	if (do_acct)
		nd_iostat_end(bio, start);

 out:
	bio_endio(bio);
	return BLK_QC_T_NONE;
}

static int nsblk_rw_bytes(struct nd_namespace_common *ndns,
		resource_size_t offset, void *iobuf, size_t n, int rw)
{
	struct nd_namespace_blk *nsblk = to_nd_namespace_blk(&ndns->dev);
	struct nd_blk_region *ndbr = to_ndbr(nsblk);
	resource_size_t	dev_offset;

	dev_offset = to_dev_offset(nsblk, offset, n);

	if (unlikely(offset + n > nsblk->size)) {
		dev_WARN_ONCE(&ndns->dev, 1, "request out of range\n");
		return -EFAULT;
	}

	if (dev_offset == SIZE_MAX)
		return -EIO;

	return ndbr->do_io(ndbr, dev_offset, iobuf, n, rw);
}

static const struct block_device_operations nd_blk_fops = {
	.owner = THIS_MODULE,
	.revalidate_disk = nvdimm_revalidate_disk,
};

static void nd_blk_release_queue(void *q)
{
	blk_cleanup_queue(q);
}

static void nd_blk_release_disk(void *disk)
{
	del_gendisk(disk);
	put_disk(disk);
}

static int nsblk_attach_disk(struct nd_namespace_blk *nsblk)
{
	struct device *dev = &nsblk->common.dev;
	resource_size_t available_disk_size;
	struct request_queue *q;
	struct gendisk *disk;
	u64 internal_nlba;

	internal_nlba = div_u64(nsblk->size, nsblk_internal_lbasize(nsblk));
	available_disk_size = internal_nlba * nsblk_sector_size(nsblk);

	q = blk_alloc_queue(GFP_KERNEL);
	if (!q)
		return -ENOMEM;
	if (devm_add_action_or_reset(dev, nd_blk_release_queue, q))
		return -ENOMEM;

	blk_queue_make_request(q, nd_blk_make_request);
	blk_queue_max_hw_sectors(q, UINT_MAX);
	blk_queue_bounce_limit(q, BLK_BOUNCE_ANY);
	blk_queue_logical_block_size(q, nsblk_sector_size(nsblk));
	queue_flag_set_unlocked(QUEUE_FLAG_NONROT, q);
	q->queuedata = nsblk;

	disk = alloc_disk(0);
	if (!disk)
		return -ENOMEM;

	disk->first_minor	= 0;
	disk->fops		= &nd_blk_fops;
	disk->queue		= q;
	disk->flags		= GENHD_FL_EXT_DEVT;
	nvdimm_namespace_disk_name(&nsblk->common, disk->disk_name);
	set_capacity(disk, 0);
	device_add_disk(dev, disk);

	if (devm_add_action_or_reset(dev, nd_blk_release_disk, disk))
		return -ENOMEM;

	if (nsblk_meta_size(nsblk)) {
		int rc = nd_integrity_init(disk, nsblk_meta_size(nsblk));

		if (rc)
			return rc;
	}

	set_capacity(disk, available_disk_size >> SECTOR_SHIFT);
	revalidate_disk(disk);
	return 0;
}

static int nd_blk_probe(struct device *dev)
{
	struct nd_namespace_common *ndns;
	struct nd_namespace_blk *nsblk;

	ndns = nvdimm_namespace_common_probe(dev);
	if (IS_ERR(ndns))
		return PTR_ERR(ndns);

	nsblk = to_nd_namespace_blk(&ndns->dev);
	nsblk->size = nvdimm_namespace_capacity(ndns);
	dev_set_drvdata(dev, nsblk);

	ndns->rw_bytes = nsblk_rw_bytes;
	if (is_nd_btt(dev))
		return nvdimm_namespace_attach_btt(ndns);
	else if (nd_btt_probe(dev, ndns) == 0) {
		/* we'll come back as btt-blk */
		return -ENXIO;
	} else
		return nsblk_attach_disk(nsblk);
}

static int nd_blk_remove(struct device *dev)
{
	if (is_nd_btt(dev))
		nvdimm_namespace_detach_btt(to_nd_btt(dev));
	return 0;
}

static struct nd_device_driver nd_blk_driver = {
	.probe = nd_blk_probe,
	.remove = nd_blk_remove,
	.drv = {
		.name = "nd_blk",
	},
	.type = ND_DRIVER_NAMESPACE_BLK,
};

static int __init nd_blk_init(void)
{
	return nd_driver_register(&nd_blk_driver);
}

static void __exit nd_blk_exit(void)
{
	driver_unregister(&nd_blk_driver.drv);
}

MODULE_AUTHOR("Ross Zwisler <ross.zwisler@linux.intel.com>");
MODULE_LICENSE("GPL v2");
MODULE_ALIAS_ND_DEVICE(ND_DEVICE_NAMESPACE_BLK);
module_init(nd_blk_init);
module_exit(nd_blk_exit);
Commit	Line	Data
047fc8a1 RZ	1	/*
	2	* NVDIMM Block Window Driver
	3	* Copyright (c) 2014, Intel Corporation.
	4	*
	5	* This program is free software; you can redistribute it and/or modify it
	6	* under the terms and conditions of the GNU General Public License,
	7	* version 2, as published by the Free Software Foundation.
	8	*
	9	* This program is distributed in the hope it will be useful, but WITHOUT
	10	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	11	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for
	12	* more details.
	13	*/
	14
	15	#include <linux/blkdev.h>
	16	#include <linux/fs.h>
	17	#include <linux/genhd.h>
	18	#include <linux/module.h>
	19	#include <linux/moduleparam.h>
	20	#include <linux/nd.h>
	21	#include <linux/sizes.h>
	22	#include "nd.h"
	23
9d90725d DW	24	static u32 nsblk_meta_size(struct nd_namespace_blk *nsblk)
	25	{
	26	return nsblk->lbasize - ((nsblk->lbasize >= 4096) ? 4096 : 512);
	27	}
047fc8a1	28
9d90725d	29	static u32 nsblk_internal_lbasize(struct nd_namespace_blk *nsblk)
fcae6957	30	{
9d90725d DW	31	return roundup(nsblk->lbasize, INT_LBASIZE_ALIGNMENT);
	32	}
	33
	34	static u32 nsblk_sector_size(struct nd_namespace_blk *nsblk)
	35	{
	36	return nsblk->lbasize - nsblk_meta_size(nsblk);
fcae6957 VV	37	}
fcae6957 VV	38
047fc8a1 RZ	39	static resource_size_t to_dev_offset(struct nd_namespace_blk *nsblk,
	40	resource_size_t ns_offset, unsigned int len)
	41	{
	42	int i;
	43
	44	for (i = 0; i < nsblk->num_resources; i++) {
	45	if (ns_offset < resource_size(nsblk->res[i])) {
	46	if (ns_offset + len > resource_size(nsblk->res[i])) {
	47	dev_WARN_ONCE(&nsblk->common.dev, 1,
	48	"illegal request\n");
	49	return SIZE_MAX;
	50	}
	51	return nsblk->res[i]->start + ns_offset;
	52	}
	53	ns_offset -= resource_size(nsblk->res[i]);
	54	}
	55
	56	dev_WARN_ONCE(&nsblk->common.dev, 1, "request out of range\n");
	57	return SIZE_MAX;
	58	}
	59
9d90725d DW	60	static struct nd_blk_region to_ndbr(struct nd_namespace_blk nsblk)
	61	{
	62	struct nd_region *nd_region;
	63	struct device *parent;
	64
	65	parent = nsblk->common.dev.parent;
	66	nd_region = container_of(parent, struct nd_region, dev);
	67	return container_of(nd_region, struct nd_blk_region, nd_region);
	68	}
	69
fcae6957	70	#ifdef CONFIG_BLK_DEV_INTEGRITY
9d90725d DW	71	static int nd_blk_rw_integrity(struct nd_namespace_blk *nsblk,
9d90725d DW	72	struct bio_integrity_payload *bip, u64 lba, int rw)
fcae6957	73	{
9d90725d DW	74	struct nd_blk_region *ndbr = to_ndbr(nsblk);
9d90725d DW	75	unsigned int len = nsblk_meta_size(nsblk);
fcae6957	76	resource_size_t dev_offset, ns_offset;
9d90725d	77	u32 internal_lbasize, sector_size;
fcae6957 VV	78	int err = 0;
fcae6957 VV	79
9d90725d DW	80	internal_lbasize = nsblk_internal_lbasize(nsblk);
	81	sector_size = nsblk_sector_size(nsblk);
	82	ns_offset = lba * internal_lbasize + sector_size;
fcae6957 VV	83	dev_offset = to_dev_offset(nsblk, ns_offset, len);
	84	if (dev_offset == SIZE_MAX)
	85	return -EIO;
	86
	87	while (len) {
	88	unsigned int cur_len;
	89	struct bio_vec bv;
	90	void *iobuf;
	91
	92	bv = bvec_iter_bvec(bip->bip_vec, bip->bip_iter);
	93	/*
	94	* The 'bv' obtained from bvec_iter_bvec has its .bv_len and
	95	* .bv_offset already adjusted for iter->bi_bvec_done, and we
	96	* can use those directly
	97	*/
	98
	99	cur_len = min(len, bv.bv_len);
	100	iobuf = kmap_atomic(bv.bv_page);
	101	err = ndbr->do_io(ndbr, dev_offset, iobuf + bv.bv_offset,
	102	cur_len, rw);
	103	kunmap_atomic(iobuf);
	104	if (err)
	105	return err;
	106
	107	len -= cur_len;
	108	dev_offset += cur_len;
	109	bvec_iter_advance(bip->bip_vec, &bip->bip_iter, cur_len);
	110	}
	111
	112	return err;
	113	}
	114
	115	#else /* CONFIG_BLK_DEV_INTEGRITY */
9d90725d DW	116	static int nd_blk_rw_integrity(struct nd_namespace_blk *nsblk,
9d90725d DW	117	struct bio_integrity_payload *bip, u64 lba, int rw)
fcae6957 VV	118	{
	119	return 0;
	120	}
	121	#endif
	122
9d90725d DW	123	static int nsblk_do_bvec(struct nd_namespace_blk *nsblk,
	124	struct bio_integrity_payload bip, struct page page,
	125	unsigned int len, unsigned int off, int rw, sector_t sector)
fcae6957	126	{
9d90725d	127	struct nd_blk_region *ndbr = to_ndbr(nsblk);
fcae6957	128	resource_size_t dev_offset, ns_offset;
9d90725d	129	u32 internal_lbasize, sector_size;
fcae6957 VV	130	int err = 0;
	131	void *iobuf;
	132	u64 lba;
	133
9d90725d DW	134	internal_lbasize = nsblk_internal_lbasize(nsblk);
9d90725d DW	135	sector_size = nsblk_sector_size(nsblk);
fcae6957 VV	136	while (len) {
	137	unsigned int cur_len;
	138
	139	/*
	140	* If we don't have an integrity payload, we don't have to
	141	* split the bvec into sectors, as this would cause unnecessary
	142	* Block Window setup/move steps. the do_io routine is capable
	143	* of handling len <= PAGE_SIZE.
	144	*/
9d90725d	145	cur_len = bip ? min(len, sector_size) : len;
fcae6957	146
9d90725d DW	147	lba = div_u64(sector << SECTOR_SHIFT, sector_size);
	148	ns_offset = lba * internal_lbasize;
	149	dev_offset = to_dev_offset(nsblk, ns_offset, cur_len);
fcae6957 VV	150	if (dev_offset == SIZE_MAX)
	151	return -EIO;
	152
	153	iobuf = kmap_atomic(page);
	154	err = ndbr->do_io(ndbr, dev_offset, iobuf + off, cur_len, rw);
	155	kunmap_atomic(iobuf);
	156	if (err)
	157	return err;
	158
	159	if (bip) {
9d90725d	160	err = nd_blk_rw_integrity(nsblk, bip, lba, rw);
fcae6957 VV	161	if (err)
	162	return err;
	163	}
	164	len -= cur_len;
	165	off += cur_len;
9d90725d	166	sector += sector_size >> SECTOR_SHIFT;
fcae6957 VV	167	}
	168
	169	return err;
	170	}
	171
dece1635	172	static blk_qc_t nd_blk_make_request(struct request_queue q, struct bio bio)
047fc8a1	173	{
fcae6957	174	struct bio_integrity_payload *bip;
9d90725d	175	struct nd_namespace_blk *nsblk;
047fc8a1	176	struct bvec_iter iter;
f0dc089c	177	unsigned long start;
047fc8a1 RZ	178	struct bio_vec bvec;
047fc8a1 RZ	179	int err = 0, rw;
f0dc089c	180	bool do_acct;
047fc8a1	181
fcae6957 VV	182	/*
	183	* bio_integrity_enabled also checks if the bio already has an
	184	* integrity payload attached. If it does, we don't do a
	185	* bio_integrity_prep here - the payload has been generated by
	186	* another kernel subsystem, and we just pass it through.
	187	*/
	188	if (bio_integrity_enabled(bio) && bio_integrity_prep(bio)) {
4246a0b6	189	bio->bi_error = -EIO;
fcae6957 VV	190	goto out;
	191	}
	192
	193	bip = bio_integrity(bio);
9d90725d	194	nsblk = q->queuedata;
047fc8a1	195	rw = bio_data_dir(bio);
f0dc089c	196	do_acct = nd_iostat_start(bio, &start);
047fc8a1 RZ	197	bio_for_each_segment(bvec, bio, iter) {
047fc8a1 RZ	198	unsigned int len = bvec.bv_len;
047fc8a1 RZ	199
047fc8a1 RZ	200	BUG_ON(len > PAGE_SIZE);
9d90725d DW	201	err = nsblk_do_bvec(nsblk, bip, bvec.bv_page, len,
9d90725d DW	202	bvec.bv_offset, rw, iter.bi_sector);
fcae6957	203	if (err) {
9d90725d	204	dev_dbg(&nsblk->common.dev,
fcae6957 VV	205	"io error in %s sector %lld, len %d,\n",
	206	(rw == READ) ? "READ" : "WRITE",
	207	(unsigned long long) iter.bi_sector, len);
4246a0b6	208	bio->bi_error = err;
f0dc089c	209	break;
047fc8a1	210	}
047fc8a1	211	}
f0dc089c DW	212	if (do_acct)
f0dc089c DW	213	nd_iostat_end(bio, start);
047fc8a1 RZ	214
047fc8a1 RZ	215	out:
4246a0b6	216	bio_endio(bio);
dece1635	217	return BLK_QC_T_NONE;
047fc8a1 RZ	218	}
047fc8a1 RZ	219
9d90725d	220	static int nsblk_rw_bytes(struct nd_namespace_common *ndns,
047fc8a1 RZ	221	resource_size_t offset, void *iobuf, size_t n, int rw)
047fc8a1 RZ	222	{
9d90725d DW	223	struct nd_namespace_blk *nsblk = to_nd_namespace_blk(&ndns->dev);
9d90725d DW	224	struct nd_blk_region *ndbr = to_ndbr(nsblk);
047fc8a1 RZ	225	resource_size_t dev_offset;
	226
	227	dev_offset = to_dev_offset(nsblk, offset, n);
	228
9d90725d	229	if (unlikely(offset + n > nsblk->size)) {
047fc8a1 RZ	230	dev_WARN_ONCE(&ndns->dev, 1, "request out of range\n");
	231	return -EFAULT;
	232	}
	233
	234	if (dev_offset == SIZE_MAX)
	235	return -EIO;
	236
	237	return ndbr->do_io(ndbr, dev_offset, iobuf, n, rw);
	238	}
	239
	240	static const struct block_device_operations nd_blk_fops = {
	241	.owner = THIS_MODULE,
58138820	242	.revalidate_disk = nvdimm_revalidate_disk,
047fc8a1 RZ	243	};
047fc8a1 RZ	244
d29cee12 DW	245	static void nd_blk_release_queue(void *q)
	246	{
	247	blk_cleanup_queue(q);
	248	}
	249
	250	static void nd_blk_release_disk(void *disk)
	251	{
	252	del_gendisk(disk);
	253	put_disk(disk);
	254	}
	255
9d90725d	256	static int nsblk_attach_disk(struct nd_namespace_blk *nsblk)
047fc8a1	257	{
9d90725d	258	struct device *dev = &nsblk->common.dev;
fcae6957	259	resource_size_t available_disk_size;
d29cee12	260	struct request_queue *q;
047fc8a1	261	struct gendisk *disk;
fcae6957 VV	262	u64 internal_nlba;
fcae6957 VV	263
9d90725d DW	264	internal_nlba = div_u64(nsblk->size, nsblk_internal_lbasize(nsblk));
9d90725d DW	265	available_disk_size = internal_nlba * nsblk_sector_size(nsblk);
047fc8a1	266
d29cee12 DW	267	q = blk_alloc_queue(GFP_KERNEL);
	268	if (!q)
	269	return -ENOMEM;
f02716db	270	if (devm_add_action_or_reset(dev, nd_blk_release_queue, q))
047fc8a1 RZ	271	return -ENOMEM;
047fc8a1 RZ	272
d29cee12 DW	273	blk_queue_make_request(q, nd_blk_make_request);
	274	blk_queue_max_hw_sectors(q, UINT_MAX);
	275	blk_queue_bounce_limit(q, BLK_BOUNCE_ANY);
9d90725d	276	blk_queue_logical_block_size(q, nsblk_sector_size(nsblk));
d29cee12	277	queue_flag_set_unlocked(QUEUE_FLAG_NONROT, q);
9d90725d	278	q->queuedata = nsblk;
047fc8a1	279
d29cee12 DW	280	disk = alloc_disk(0);
	281	if (!disk)
	282	return -ENOMEM;
047fc8a1	283
047fc8a1 RZ	284	disk->first_minor = 0;
047fc8a1 RZ	285	disk->fops = &nd_blk_fops;
d29cee12	286	disk->queue = q;
047fc8a1	287	disk->flags = GENHD_FL_EXT_DEVT;
9d90725d	288	nvdimm_namespace_disk_name(&nsblk->common, disk->disk_name);
fcae6957	289	set_capacity(disk, 0);
0d52c756	290	device_add_disk(dev, disk);
047fc8a1	291
f02716db DW	292	if (devm_add_action_or_reset(dev, nd_blk_release_disk, disk))
	293	return -ENOMEM;
	294
9d90725d DW	295	if (nsblk_meta_size(nsblk)) {
9d90725d DW	296	int rc = nd_integrity_init(disk, nsblk_meta_size(nsblk));
fcae6957	297
d29cee12	298	if (rc)
fcae6957	299	return rc;
fcae6957 VV	300	}
	301
	302	set_capacity(disk, available_disk_size >> SECTOR_SHIFT);
58138820	303	revalidate_disk(disk);
047fc8a1 RZ	304	return 0;
	305	}
	306
	307	static int nd_blk_probe(struct device *dev)
	308	{
	309	struct nd_namespace_common *ndns;
fcae6957	310	struct nd_namespace_blk *nsblk;
047fc8a1 RZ	311
	312	ndns = nvdimm_namespace_common_probe(dev);
	313	if (IS_ERR(ndns))
	314	return PTR_ERR(ndns);
	315
fcae6957	316	nsblk = to_nd_namespace_blk(&ndns->dev);
9d90725d DW	317	nsblk->size = nvdimm_namespace_capacity(ndns);
	318	dev_set_drvdata(dev, nsblk);
	319
	320	ndns->rw_bytes = nsblk_rw_bytes;
047fc8a1	321	if (is_nd_btt(dev))
d29cee12	322	return nvdimm_namespace_attach_btt(ndns);
200c79da	323	else if (nd_btt_probe(dev, ndns) == 0) {
047fc8a1	324	/* we'll come back as btt-blk */
d29cee12	325	return -ENXIO;
047fc8a1	326	} else
9d90725d	327	return nsblk_attach_disk(nsblk);
047fc8a1 RZ	328	}
	329
	330	static int nd_blk_remove(struct device *dev)
	331	{
047fc8a1	332	if (is_nd_btt(dev))
298f2bc5	333	nvdimm_namespace_detach_btt(to_nd_btt(dev));
047fc8a1 RZ	334	return 0;
	335	}
	336
	337	static struct nd_device_driver nd_blk_driver = {
	338	.probe = nd_blk_probe,
	339	.remove = nd_blk_remove,
	340	.drv = {
	341	.name = "nd_blk",
	342	},
	343	.type = ND_DRIVER_NAMESPACE_BLK,
	344	};
	345
	346	static int __init nd_blk_init(void)
	347	{
ec56151d	348	return nd_driver_register(&nd_blk_driver);
047fc8a1 RZ	349	}
	350
	351	static void __exit nd_blk_exit(void)
	352	{
	353	driver_unregister(&nd_blk_driver.drv);
047fc8a1 RZ	354	}
	355
	356	MODULE_AUTHOR("Ross Zwisler <ross.zwisler@linux.intel.com>");
	357	MODULE_LICENSE("GPL v2");
	358	MODULE_ALIAS_ND_DEVICE(ND_DEVICE_NAMESPACE_BLK);
	359	module_init(nd_blk_init);
	360	module_exit(nd_blk_exit);