[mirror_ubuntu-artful-kernel.git] / drivers / nvdimm / claim.c

/*
 * Copyright(c) 2013-2015 Intel Corporation. All rights reserved.
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of version 2 of the GNU General Public License 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.
 */
#include <linux/device.h>
#include <linux/sizes.h>
#include <linux/pmem.h>
#include "nd-core.h"
#include "pfn.h"
#include "btt.h"
#include "nd.h"

void __nd_detach_ndns(struct device *dev, struct nd_namespace_common **_ndns)
{
	struct nd_namespace_common *ndns = *_ndns;

	dev_WARN_ONCE(dev, !mutex_is_locked(&ndns->dev.mutex)
			|| ndns->claim != dev,
			"%s: invalid claim\n", __func__);
	ndns->claim = NULL;
	*_ndns = NULL;
	put_device(&ndns->dev);
}

void nd_detach_ndns(struct device *dev,
		struct nd_namespace_common **_ndns)
{
	struct nd_namespace_common *ndns = *_ndns;

	if (!ndns)
		return;
	get_device(&ndns->dev);
	device_lock(&ndns->dev);
	__nd_detach_ndns(dev, _ndns);
	device_unlock(&ndns->dev);
	put_device(&ndns->dev);
}

bool __nd_attach_ndns(struct device *dev, struct nd_namespace_common *attach,
		struct nd_namespace_common **_ndns)
{
	if (attach->claim)
		return false;
	dev_WARN_ONCE(dev, !mutex_is_locked(&attach->dev.mutex)
			|| *_ndns,
			"%s: invalid claim\n", __func__);
	attach->claim = dev;
	*_ndns = attach;
	get_device(&attach->dev);
	return true;
}

bool nd_attach_ndns(struct device *dev, struct nd_namespace_common *attach,
		struct nd_namespace_common **_ndns)
{
	bool claimed;

	device_lock(&attach->dev);
	claimed = __nd_attach_ndns(dev, attach, _ndns);
	device_unlock(&attach->dev);
	return claimed;
}

static int namespace_match(struct device *dev, void *data)
{
	char *name = data;

	return strcmp(name, dev_name(dev)) == 0;
}

static bool is_idle(struct device *dev, struct nd_namespace_common *ndns)
{
	struct nd_region *nd_region = to_nd_region(dev->parent);
	struct device *seed = NULL;

	if (is_nd_btt(dev))
		seed = nd_region->btt_seed;
	else if (is_nd_pfn(dev))
		seed = nd_region->pfn_seed;
	else if (is_nd_dax(dev))
		seed = nd_region->dax_seed;

	if (seed == dev || ndns || dev->driver)
		return false;
	return true;
}

static void nd_detach_and_reset(struct device *dev,
		struct nd_namespace_common **_ndns)
{
	/* detach the namespace and destroy / reset the device */
	nd_detach_ndns(dev, _ndns);
	if (is_idle(dev, *_ndns)) {
		nd_device_unregister(dev, ND_ASYNC);
	} else if (is_nd_btt(dev)) {
		struct nd_btt *nd_btt = to_nd_btt(dev);

		nd_btt->lbasize = 0;
		kfree(nd_btt->uuid);
		nd_btt->uuid = NULL;
	} else if (is_nd_pfn(dev)) {
		struct nd_pfn *nd_pfn = to_nd_pfn(dev);

		kfree(nd_pfn->uuid);
		nd_pfn->uuid = NULL;
		nd_pfn->mode = PFN_MODE_NONE;
	}
}

ssize_t nd_namespace_store(struct device *dev,
		struct nd_namespace_common **_ndns, const char *buf,
		size_t len)
{
	struct nd_namespace_common *ndns;
	struct device *found;
	char *name;

	if (dev->driver) {
		dev_dbg(dev, "%s: -EBUSY\n", __func__);
		return -EBUSY;
	}

	name = kstrndup(buf, len, GFP_KERNEL);
	if (!name)
		return -ENOMEM;
	strim(name);

	if (strncmp(name, "namespace", 9) == 0 || strcmp(name, "") == 0)
		/* pass */;
	else {
		len = -EINVAL;
		goto out;
	}

	ndns = *_ndns;
	if (strcmp(name, "") == 0) {
		nd_detach_and_reset(dev, _ndns);
		goto out;
	} else if (ndns) {
		dev_dbg(dev, "namespace already set to: %s\n",
				dev_name(&ndns->dev));
		len = -EBUSY;
		goto out;
	}

	found = device_find_child(dev->parent, name, namespace_match);
	if (!found) {
		dev_dbg(dev, "'%s' not found under %s\n", name,
				dev_name(dev->parent));
		len = -ENODEV;
		goto out;
	}

	ndns = to_ndns(found);
	if (__nvdimm_namespace_capacity(ndns) < SZ_16M) {
		dev_dbg(dev, "%s too small to host\n", name);
		len = -ENXIO;
		goto out_attach;
	}

	WARN_ON_ONCE(!is_nvdimm_bus_locked(dev));
	if (!nd_attach_ndns(dev, ndns, _ndns)) {
		dev_dbg(dev, "%s already claimed\n",
				dev_name(&ndns->dev));
		len = -EBUSY;
	}

 out_attach:
	put_device(&ndns->dev); /* from device_find_child */
 out:
	kfree(name);
	return len;
}

/*
 * nd_sb_checksum: compute checksum for a generic info block
 *
 * Returns a fletcher64 checksum of everything in the given info block
 * except the last field (since that's where the checksum lives).
 */
u64 nd_sb_checksum(struct nd_gen_sb *nd_gen_sb)
{
	u64 sum;
	__le64 sum_save;

	BUILD_BUG_ON(sizeof(struct btt_sb) != SZ_4K);
	BUILD_BUG_ON(sizeof(struct nd_pfn_sb) != SZ_4K);
	BUILD_BUG_ON(sizeof(struct nd_gen_sb) != SZ_4K);

	sum_save = nd_gen_sb->checksum;
	nd_gen_sb->checksum = 0;
	sum = nd_fletcher64(nd_gen_sb, sizeof(*nd_gen_sb), 1);
	nd_gen_sb->checksum = sum_save;
	return sum;
}
EXPORT_SYMBOL(nd_sb_checksum);

static int nsio_rw_bytes(struct nd_namespace_common *ndns,
		resource_size_t offset, void *buf, size_t size, int rw)
{
	struct nd_namespace_io *nsio = to_nd_namespace_io(&ndns->dev);

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

	if (rw == READ) {
		unsigned int sz_align = ALIGN(size + (offset & (512 - 1)), 512);

		if (unlikely(is_bad_pmem(&nsio->bb, offset / 512, sz_align)))
			return -EIO;
		return memcpy_from_pmem(buf, nsio->addr + offset, size);
	} else {
		memcpy_to_pmem(nsio->addr + offset, buf, size);
		wmb_pmem();
	}

	return 0;
}

int devm_nsio_enable(struct device *dev, struct nd_namespace_io *nsio)
{
	struct resource *res = &nsio->res;
	struct nd_namespace_common *ndns = &nsio->common;

	nsio->size = resource_size(res);
	if (!devm_request_mem_region(dev, res->start, resource_size(res),
				dev_name(dev))) {
		dev_warn(dev, "could not reserve region %pR\n", res);
		return -EBUSY;
	}

	ndns->rw_bytes = nsio_rw_bytes;
	if (devm_init_badblocks(dev, &nsio->bb))
		return -ENOMEM;
	nvdimm_badblocks_populate(to_nd_region(ndns->dev.parent), &nsio->bb,
			&nsio->res);

	nsio->addr = devm_memremap(dev, res->start, resource_size(res),
			ARCH_MEMREMAP_PMEM);
	if (IS_ERR(nsio->addr))
		return PTR_ERR(nsio->addr);
	return 0;
}
EXPORT_SYMBOL_GPL(devm_nsio_enable);

void devm_nsio_disable(struct device *dev, struct nd_namespace_io *nsio)
{
	struct resource *res = &nsio->res;

	devm_memunmap(dev, nsio->addr);
	devm_exit_badblocks(dev, &nsio->bb);
	devm_release_mem_region(dev, res->start, resource_size(res));
}
EXPORT_SYMBOL_GPL(devm_nsio_disable);
Commit	Line	Data
e1455744 DW	1	/*
	2	* Copyright(c) 2013-2015 Intel Corporation. All rights reserved.
	3	*
	4	* This program is free software; you can redistribute it and/or modify
	5	* it under the terms of version 2 of the GNU General Public License as
	6	* published by the Free Software Foundation.
	7	*
	8	* This program is distributed in the hope that it will be useful, but
	9	* WITHOUT ANY WARRANTY; without even the implied warranty of
	10	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
	11	* General Public License for more details.
	12	*/
	13	#include <linux/device.h>
	14	#include <linux/sizes.h>
200c79da	15	#include <linux/pmem.h>
e1455744 DW	16	#include "nd-core.h"
	17	#include "pfn.h"
	18	#include "btt.h"
	19	#include "nd.h"
	20
	21	void __nd_detach_ndns(struct device dev, struct nd_namespace_common *_ndns)
	22	{
	23	struct nd_namespace_common ndns = _ndns;
	24
	25	dev_WARN_ONCE(dev, !mutex_is_locked(&ndns->dev.mutex)
	26	\|\| ndns->claim != dev,
	27	"%s: invalid claim\n", __func__);
	28	ndns->claim = NULL;
	29	*_ndns = NULL;
	30	put_device(&ndns->dev);
	31	}
	32
	33	void nd_detach_ndns(struct device *dev,
	34	struct nd_namespace_common **_ndns)
	35	{
	36	struct nd_namespace_common ndns = _ndns;
	37
	38	if (!ndns)
	39	return;
	40	get_device(&ndns->dev);
	41	device_lock(&ndns->dev);
	42	__nd_detach_ndns(dev, _ndns);
	43	device_unlock(&ndns->dev);
	44	put_device(&ndns->dev);
	45	}
	46
	47	bool __nd_attach_ndns(struct device dev, struct nd_namespace_common attach,
	48	struct nd_namespace_common **_ndns)
	49	{
	50	if (attach->claim)
	51	return false;
	52	dev_WARN_ONCE(dev, !mutex_is_locked(&attach->dev.mutex)
	53	\|\| *_ndns,
	54	"%s: invalid claim\n", __func__);
	55	attach->claim = dev;
	56	*_ndns = attach;
	57	get_device(&attach->dev);
	58	return true;
	59	}
	60
	61	bool nd_attach_ndns(struct device dev, struct nd_namespace_common attach,
	62	struct nd_namespace_common **_ndns)
	63	{
	64	bool claimed;
	65
	66	device_lock(&attach->dev);
	67	claimed = __nd_attach_ndns(dev, attach, _ndns);
	68	device_unlock(&attach->dev);
	69	return claimed;
	70	}
	71
	72	static int namespace_match(struct device dev, void data)
	73	{
	74	char *name = data;
	75
	76	return strcmp(name, dev_name(dev)) == 0;
	77	}
	78
	79	static bool is_idle(struct device dev, struct nd_namespace_common ndns)
80	{
81	struct nd_region *nd_region = to_nd_region(dev->parent);
82	struct device *seed = NULL;
83
84	if (is_nd_btt(dev))
85	seed = nd_region->btt_seed;
86	else if (is_nd_pfn(dev))
87	seed = nd_region->pfn_seed;
cd03412a DW	88	else if (is_nd_dax(dev))
cd03412a DW	89	seed = nd_region->dax_seed;
e1455744 DW	90
	91	if (seed == dev \|\| ndns \|\| dev->driver)
	92	return false;
	93	return true;
	94	}
	95
	96	static void nd_detach_and_reset(struct device *dev,
	97	struct nd_namespace_common **_ndns)
	98	{
	99	/* detach the namespace and destroy / reset the device */
	100	nd_detach_ndns(dev, _ndns);
	101	if (is_idle(dev, *_ndns)) {
	102	nd_device_unregister(dev, ND_ASYNC);
	103	} else if (is_nd_btt(dev)) {
	104	struct nd_btt *nd_btt = to_nd_btt(dev);
	105
	106	nd_btt->lbasize = 0;
	107	kfree(nd_btt->uuid);
	108	nd_btt->uuid = NULL;
	109	} else if (is_nd_pfn(dev)) {
	110	struct nd_pfn *nd_pfn = to_nd_pfn(dev);
	111
	112	kfree(nd_pfn->uuid);
	113	nd_pfn->uuid = NULL;
	114	nd_pfn->mode = PFN_MODE_NONE;
	115	}
	116	}
	117
	118	ssize_t nd_namespace_store(struct device *dev,
	119	struct nd_namespace_common *_ndns, const char buf,
	120	size_t len)
	121	{
	122	struct nd_namespace_common *ndns;
	123	struct device *found;
	124	char *name;
	125
	126	if (dev->driver) {
	127	dev_dbg(dev, "%s: -EBUSY\n", __func__);
	128	return -EBUSY;
	129	}
	130
	131	name = kstrndup(buf, len, GFP_KERNEL);
	132	if (!name)
	133	return -ENOMEM;
	134	strim(name);
	135
	136	if (strncmp(name, "namespace", 9) == 0 \|\| strcmp(name, "") == 0)
	137	/* pass */;
	138	else {
	139	len = -EINVAL;
	140	goto out;
	141	}
	142
	143	ndns = *_ndns;
	144	if (strcmp(name, "") == 0) {
	145	nd_detach_and_reset(dev, _ndns);
	146	goto out;
	147	} else if (ndns) {
	148	dev_dbg(dev, "namespace already set to: %s\n",
	149	dev_name(&ndns->dev));
	150	len = -EBUSY;
	151	goto out;
	152	}
	153
154	found = device_find_child(dev->parent, name, namespace_match);
155	if (!found) {
156	dev_dbg(dev, "'%s' not found under %s\n", name,
157	dev_name(dev->parent));
158	len = -ENODEV;
159	goto out;
160	}
161
162	ndns = to_ndns(found);
163	if (__nvdimm_namespace_capacity(ndns) < SZ_16M) {
164	dev_dbg(dev, "%s too small to host\n", name);
165	len = -ENXIO;
166	goto out_attach;
167	}
168
169	WARN_ON_ONCE(!is_nvdimm_bus_locked(dev));
170	if (!nd_attach_ndns(dev, ndns, _ndns)) {
171	dev_dbg(dev, "%s already claimed\n",
172	dev_name(&ndns->dev));
173	len = -EBUSY;
174	}
175
176	out_attach:
177	put_device(&ndns->dev); /* from device_find_child */
178	out:
179	kfree(name);
180	return len;
181	}
182
183	/*
184	* nd_sb_checksum: compute checksum for a generic info block
185	*
186	* Returns a fletcher64 checksum of everything in the given info block
187	* except the last field (since that's where the checksum lives).
188	*/
189	u64 nd_sb_checksum(struct nd_gen_sb *nd_gen_sb)
190	{
191	u64 sum;
192	__le64 sum_save;
193
194	BUILD_BUG_ON(sizeof(struct btt_sb) != SZ_4K);
195	BUILD_BUG_ON(sizeof(struct nd_pfn_sb) != SZ_4K);
196	BUILD_BUG_ON(sizeof(struct nd_gen_sb) != SZ_4K);
197
198	sum_save = nd_gen_sb->checksum;
199	nd_gen_sb->checksum = 0;
200	sum = nd_fletcher64(nd_gen_sb, sizeof(*nd_gen_sb), 1);
201	nd_gen_sb->checksum = sum_save;
202	return sum;
203	}
204	EXPORT_SYMBOL(nd_sb_checksum);
200c79da DW	205
	206	static int nsio_rw_bytes(struct nd_namespace_common *ndns,
	207	resource_size_t offset, void *buf, size_t size, int rw)
	208	{
	209	struct nd_namespace_io *nsio = to_nd_namespace_io(&ndns->dev);
	210
	211	if (unlikely(offset + size > nsio->size)) {
	212	dev_WARN_ONCE(&ndns->dev, 1, "request out of range\n");
	213	return -EFAULT;
	214	}
	215
	216	if (rw == READ) {
	217	unsigned int sz_align = ALIGN(size + (offset & (512 - 1)), 512);
	218
	219	if (unlikely(is_bad_pmem(&nsio->bb, offset / 512, sz_align)))
	220	return -EIO;
	221	return memcpy_from_pmem(buf, nsio->addr + offset, size);
	222	} else {
	223	memcpy_to_pmem(nsio->addr + offset, buf, size);
	224	wmb_pmem();
	225	}
	226
	227	return 0;
	228	}
	229
	230	int devm_nsio_enable(struct device dev, struct nd_namespace_io nsio)
	231	{
	232	struct resource *res = &nsio->res;
	233	struct nd_namespace_common *ndns = &nsio->common;
	234
	235	nsio->size = resource_size(res);
	236	if (!devm_request_mem_region(dev, res->start, resource_size(res),
	237	dev_name(dev))) {
	238	dev_warn(dev, "could not reserve region %pR\n", res);
	239	return -EBUSY;
	240	}
	241
	242	ndns->rw_bytes = nsio_rw_bytes;
	243	if (devm_init_badblocks(dev, &nsio->bb))
	244	return -ENOMEM;
	245	nvdimm_badblocks_populate(to_nd_region(ndns->dev.parent), &nsio->bb,
	246	&nsio->res);
	247
	248	nsio->addr = devm_memremap(dev, res->start, resource_size(res),
	249	ARCH_MEMREMAP_PMEM);
	250	if (IS_ERR(nsio->addr))
	251	return PTR_ERR(nsio->addr);
	252	return 0;
	253	}
	254	EXPORT_SYMBOL_GPL(devm_nsio_enable);
	255
	256	void devm_nsio_disable(struct device dev, struct nd_namespace_io nsio)
	257	{
	258	struct resource *res = &nsio->res;
	259
	260	devm_memunmap(dev, nsio->addr);
	261	devm_exit_badblocks(dev, &nsio->bb);
	262	devm_release_mem_region(dev, res->start, resource_size(res));
	263	}
	264	EXPORT_SYMBOL_GPL(devm_nsio_disable);