[linux-2.6-block.git] / drivers / nvdimm / dimm_devs.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.
 */
#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
#include <linux/vmalloc.h>
#include <linux/device.h>
#include <linux/ndctl.h>
#include <linux/slab.h>
#include <linux/io.h>
#include <linux/fs.h>
#include <linux/mm.h>
#include "nd-core.h"
#include "label.h"
#include "pmem.h"
#include "nd.h"

static DEFINE_IDA(dimm_ida);

/*
 * Retrieve bus and dimm handle and return if this bus supports
 * get_config_data commands
 */
int nvdimm_check_config_data(struct device *dev)
{
	struct nvdimm *nvdimm = to_nvdimm(dev);

	if (!nvdimm->cmd_mask ||
	    !test_bit(ND_CMD_GET_CONFIG_DATA, &nvdimm->cmd_mask)) {
		if (test_bit(NDD_ALIASING, &nvdimm->flags))
			return -ENXIO;
		else
			return -ENOTTY;
	}

	return 0;
}

static int validate_dimm(struct nvdimm_drvdata *ndd)
{
	int rc;

	if (!ndd)
		return -EINVAL;

	rc = nvdimm_check_config_data(ndd->dev);
	if (rc)
		dev_dbg(ndd->dev, "%pf: %s error: %d\n",
				__builtin_return_address(0), __func__, rc);
	return rc;
}

/**
 * nvdimm_init_nsarea - determine the geometry of a dimm's namespace area
 * @nvdimm: dimm to initialize
 */
int nvdimm_init_nsarea(struct nvdimm_drvdata *ndd)
{
	struct nd_cmd_get_config_size *cmd = &ndd->nsarea;
	struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
	struct nvdimm_bus_descriptor *nd_desc;
	int rc = validate_dimm(ndd);
	int cmd_rc = 0;

	if (rc)
		return rc;

	if (cmd->config_size)
		return 0; /* already valid */

	memset(cmd, 0, sizeof(*cmd));
	nd_desc = nvdimm_bus->nd_desc;
	rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
			ND_CMD_GET_CONFIG_SIZE, cmd, sizeof(*cmd), &cmd_rc);
	if (rc < 0)
		return rc;
	return cmd_rc;
}

int nvdimm_init_config_data(struct nvdimm_drvdata *ndd)
{
	struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
	int rc = validate_dimm(ndd), cmd_rc = 0;
	struct nd_cmd_get_config_data_hdr *cmd;
	struct nvdimm_bus_descriptor *nd_desc;
	u32 max_cmd_size, config_size;
	size_t offset;

	if (rc)
		return rc;

	if (ndd->data)
		return 0;

	if (ndd->nsarea.status || ndd->nsarea.max_xfer == 0
			|| ndd->nsarea.config_size < ND_LABEL_MIN_SIZE) {
		dev_dbg(ndd->dev, "failed to init config data area: (%d:%d)\n",
				ndd->nsarea.max_xfer, ndd->nsarea.config_size);
		return -ENXIO;
	}

	ndd->data = kvmalloc(ndd->nsarea.config_size, GFP_KERNEL);
	if (!ndd->data)
		return -ENOMEM;

	max_cmd_size = min_t(u32, PAGE_SIZE, ndd->nsarea.max_xfer);
	cmd = kzalloc(max_cmd_size + sizeof(*cmd), GFP_KERNEL);
	if (!cmd)
		return -ENOMEM;

	nd_desc = nvdimm_bus->nd_desc;
	for (config_size = ndd->nsarea.config_size, offset = 0;
			config_size; config_size -= cmd->in_length,
			offset += cmd->in_length) {
		cmd->in_length = min(config_size, max_cmd_size);
		cmd->in_offset = offset;
		rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
				ND_CMD_GET_CONFIG_DATA, cmd,
				cmd->in_length + sizeof(*cmd), &cmd_rc);
		if (rc < 0)
			break;
		if (cmd_rc < 0) {
			rc = cmd_rc;
			break;
		}
		memcpy(ndd->data + offset, cmd->out_buf, cmd->in_length);
	}
	dev_dbg(ndd->dev, "len: %zu rc: %d\n", offset, rc);
	kfree(cmd);

	return rc;
}

int nvdimm_set_config_data(struct nvdimm_drvdata *ndd, size_t offset,
		void *buf, size_t len)
{
	size_t max_cmd_size, buf_offset;
	struct nd_cmd_set_config_hdr *cmd;
	int rc = validate_dimm(ndd), cmd_rc = 0;
	struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
	struct nvdimm_bus_descriptor *nd_desc = nvdimm_bus->nd_desc;

	if (rc)
		return rc;

	if (!ndd->data)
		return -ENXIO;

	if (offset + len > ndd->nsarea.config_size)
		return -ENXIO;

	max_cmd_size = min_t(u32, PAGE_SIZE, len);
	max_cmd_size = min_t(u32, max_cmd_size, ndd->nsarea.max_xfer);
	cmd = kzalloc(max_cmd_size + sizeof(*cmd) + sizeof(u32), GFP_KERNEL);
	if (!cmd)
		return -ENOMEM;

	for (buf_offset = 0; len; len -= cmd->in_length,
			buf_offset += cmd->in_length) {
		size_t cmd_size;

		cmd->in_offset = offset + buf_offset;
		cmd->in_length = min(max_cmd_size, len);
		memcpy(cmd->in_buf, buf + buf_offset, cmd->in_length);

		/* status is output in the last 4-bytes of the command buffer */
		cmd_size = sizeof(*cmd) + cmd->in_length + sizeof(u32);

		rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
				ND_CMD_SET_CONFIG_DATA, cmd, cmd_size, &cmd_rc);
		if (rc < 0)
			break;
		if (cmd_rc < 0) {
			rc = cmd_rc;
			break;
		}
	}
	kfree(cmd);

	return rc;
}

void nvdimm_set_aliasing(struct device *dev)
{
	struct nvdimm *nvdimm = to_nvdimm(dev);

	set_bit(NDD_ALIASING, &nvdimm->flags);
}

void nvdimm_set_locked(struct device *dev)
{
	struct nvdimm *nvdimm = to_nvdimm(dev);

	set_bit(NDD_LOCKED, &nvdimm->flags);
}

void nvdimm_clear_locked(struct device *dev)
{
	struct nvdimm *nvdimm = to_nvdimm(dev);

	clear_bit(NDD_LOCKED, &nvdimm->flags);
}

static void nvdimm_release(struct device *dev)
{
	struct nvdimm *nvdimm = to_nvdimm(dev);

	ida_simple_remove(&dimm_ida, nvdimm->id);
	kfree(nvdimm);
}

static struct device_type nvdimm_device_type = {
	.name = "nvdimm",
	.release = nvdimm_release,
};

bool is_nvdimm(struct device *dev)
{
	return dev->type == &nvdimm_device_type;
}

struct nvdimm *to_nvdimm(struct device *dev)
{
	struct nvdimm *nvdimm = container_of(dev, struct nvdimm, dev);

	WARN_ON(!is_nvdimm(dev));
	return nvdimm;
}
EXPORT_SYMBOL_GPL(to_nvdimm);

struct nvdimm *nd_blk_region_to_dimm(struct nd_blk_region *ndbr)
{
	struct nd_region *nd_region = &ndbr->nd_region;
	struct nd_mapping *nd_mapping = &nd_region->mapping[0];

	return nd_mapping->nvdimm;
}
EXPORT_SYMBOL_GPL(nd_blk_region_to_dimm);

unsigned long nd_blk_memremap_flags(struct nd_blk_region *ndbr)
{
	/* pmem mapping properties are private to libnvdimm */
	return ARCH_MEMREMAP_PMEM;
}
EXPORT_SYMBOL_GPL(nd_blk_memremap_flags);

struct nvdimm_drvdata *to_ndd(struct nd_mapping *nd_mapping)
{
	struct nvdimm *nvdimm = nd_mapping->nvdimm;

	WARN_ON_ONCE(!is_nvdimm_bus_locked(&nvdimm->dev));

	return dev_get_drvdata(&nvdimm->dev);
}
EXPORT_SYMBOL(to_ndd);

void nvdimm_drvdata_release(struct kref *kref)
{
	struct nvdimm_drvdata *ndd = container_of(kref, typeof(*ndd), kref);
	struct device *dev = ndd->dev;
	struct resource *res, *_r;

	dev_dbg(dev, "trace\n");
	nvdimm_bus_lock(dev);
	for_each_dpa_resource_safe(ndd, res, _r)
		nvdimm_free_dpa(ndd, res);
	nvdimm_bus_unlock(dev);

	kvfree(ndd->data);
	kfree(ndd);
	put_device(dev);
}

void get_ndd(struct nvdimm_drvdata *ndd)
{
	kref_get(&ndd->kref);
}

void put_ndd(struct nvdimm_drvdata *ndd)
{
	if (ndd)
		kref_put(&ndd->kref, nvdimm_drvdata_release);
}

const char *nvdimm_name(struct nvdimm *nvdimm)
{
	return dev_name(&nvdimm->dev);
}
EXPORT_SYMBOL_GPL(nvdimm_name);

struct kobject *nvdimm_kobj(struct nvdimm *nvdimm)
{
	return &nvdimm->dev.kobj;
}
EXPORT_SYMBOL_GPL(nvdimm_kobj);

unsigned long nvdimm_cmd_mask(struct nvdimm *nvdimm)
{
	return nvdimm->cmd_mask;
}
EXPORT_SYMBOL_GPL(nvdimm_cmd_mask);

void *nvdimm_provider_data(struct nvdimm *nvdimm)
{
	if (nvdimm)
		return nvdimm->provider_data;
	return NULL;
}
EXPORT_SYMBOL_GPL(nvdimm_provider_data);

static ssize_t commands_show(struct device *dev,
		struct device_attribute *attr, char *buf)
{
	struct nvdimm *nvdimm = to_nvdimm(dev);
	int cmd, len = 0;

	if (!nvdimm->cmd_mask)
		return sprintf(buf, "\n");

	for_each_set_bit(cmd, &nvdimm->cmd_mask, BITS_PER_LONG)
		len += sprintf(buf + len, "%s ", nvdimm_cmd_name(cmd));
	len += sprintf(buf + len, "\n");
	return len;
}
static DEVICE_ATTR_RO(commands);

static ssize_t flags_show(struct device *dev,
		struct device_attribute *attr, char *buf)
{
	struct nvdimm *nvdimm = to_nvdimm(dev);

	return sprintf(buf, "%s%s\n",
			test_bit(NDD_ALIASING, &nvdimm->flags) ? "alias " : "",
			test_bit(NDD_LOCKED, &nvdimm->flags) ? "lock " : "");
}
static DEVICE_ATTR_RO(flags);

static ssize_t state_show(struct device *dev, struct device_attribute *attr,
		char *buf)
{
	struct nvdimm *nvdimm = to_nvdimm(dev);

	/*
	 * The state may be in the process of changing, userspace should
	 * quiesce probing if it wants a static answer
	 */
	nvdimm_bus_lock(dev);
	nvdimm_bus_unlock(dev);
	return sprintf(buf, "%s\n", atomic_read(&nvdimm->busy)
			? "active" : "idle");
}
static DEVICE_ATTR_RO(state);

static ssize_t available_slots_show(struct device *dev,
		struct device_attribute *attr, char *buf)
{
	struct nvdimm_drvdata *ndd = dev_get_drvdata(dev);
	ssize_t rc;
	u32 nfree;

	if (!ndd)
		return -ENXIO;

	nvdimm_bus_lock(dev);
	nfree = nd_label_nfree(ndd);
	if (nfree - 1 > nfree) {
		dev_WARN_ONCE(dev, 1, "we ate our last label?\n");
		nfree = 0;
	} else
		nfree--;
	rc = sprintf(buf, "%d\n", nfree);
	nvdimm_bus_unlock(dev);
	return rc;
}
static DEVICE_ATTR_RO(available_slots);

static struct attribute *nvdimm_attributes[] = {
	&dev_attr_state.attr,
	&dev_attr_flags.attr,
	&dev_attr_commands.attr,
	&dev_attr_available_slots.attr,
	NULL,
};

struct attribute_group nvdimm_attribute_group = {
	.attrs = nvdimm_attributes,
};
EXPORT_SYMBOL_GPL(nvdimm_attribute_group);

struct nvdimm *nvdimm_create(struct nvdimm_bus *nvdimm_bus, void *provider_data,
		const struct attribute_group **groups, unsigned long flags,
		unsigned long cmd_mask, int num_flush,
		struct resource *flush_wpq)
{
	struct nvdimm *nvdimm = kzalloc(sizeof(*nvdimm), GFP_KERNEL);
	struct device *dev;

	if (!nvdimm)
		return NULL;

	nvdimm->id = ida_simple_get(&dimm_ida, 0, 0, GFP_KERNEL);
	if (nvdimm->id < 0) {
		kfree(nvdimm);
		return NULL;
	}
	nvdimm->provider_data = provider_data;
	nvdimm->flags = flags;
	nvdimm->cmd_mask = cmd_mask;
	nvdimm->num_flush = num_flush;
	nvdimm->flush_wpq = flush_wpq;
	atomic_set(&nvdimm->busy, 0);
	dev = &nvdimm->dev;
	dev_set_name(dev, "nmem%d", nvdimm->id);
	dev->parent = &nvdimm_bus->dev;
	dev->type = &nvdimm_device_type;
	dev->devt = MKDEV(nvdimm_major, nvdimm->id);
	dev->groups = groups;
	nd_device_register(dev);

	return nvdimm;
}
EXPORT_SYMBOL_GPL(nvdimm_create);

int alias_dpa_busy(struct device *dev, void *data)
{
	resource_size_t map_end, blk_start, new;
	struct blk_alloc_info *info = data;
	struct nd_mapping *nd_mapping;
	struct nd_region *nd_region;
	struct nvdimm_drvdata *ndd;
	struct resource *res;
	int i;

	if (!is_memory(dev))
		return 0;

	nd_region = to_nd_region(dev);
	for (i = 0; i < nd_region->ndr_mappings; i++) {
		nd_mapping  = &nd_region->mapping[i];
		if (nd_mapping->nvdimm == info->nd_mapping->nvdimm)
			break;
	}

	if (i >= nd_region->ndr_mappings)
		return 0;

	ndd = to_ndd(nd_mapping);
	map_end = nd_mapping->start + nd_mapping->size - 1;
	blk_start = nd_mapping->start;

	/*
	 * In the allocation case ->res is set to free space that we are
	 * looking to validate against PMEM aliasing collision rules
	 * (i.e. BLK is allocated after all aliased PMEM).
	 */
	if (info->res) {
		if (info->res->start >= nd_mapping->start
				&& info->res->start < map_end)
			/* pass */;
		else
			return 0;
	}

 retry:
	/*
	 * Find the free dpa from the end of the last pmem allocation to
	 * the end of the interleave-set mapping.
	 */
	for_each_dpa_resource(ndd, res) {
		if (strncmp(res->name, "pmem", 4) != 0)
			continue;
		if ((res->start >= blk_start && res->start < map_end)
				|| (res->end >= blk_start
					&& res->end <= map_end)) {
			new = max(blk_start, min(map_end + 1, res->end + 1));
			if (new != blk_start) {
				blk_start = new;
				goto retry;
			}
		}
	}

	/* update the free space range with the probed blk_start */
	if (info->res && blk_start > info->res->start) {
		info->res->start = max(info->res->start, blk_start);
		if (info->res->start > info->res->end)
			info->res->end = info->res->start - 1;
		return 1;
	}

	info->available -= blk_start - nd_mapping->start;

	return 0;
}

/**
 * nd_blk_available_dpa - account the unused dpa of BLK region
 * @nd_mapping: container of dpa-resource-root + labels
 *
 * Unlike PMEM, BLK namespaces can occupy discontiguous DPA ranges, but
 * we arrange for them to never start at an lower dpa than the last
 * PMEM allocation in an aliased region.
 */
resource_size_t nd_blk_available_dpa(struct nd_region *nd_region)
{
	struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(&nd_region->dev);
	struct nd_mapping *nd_mapping = &nd_region->mapping[0];
	struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
	struct blk_alloc_info info = {
		.nd_mapping = nd_mapping,
		.available = nd_mapping->size,
		.res = NULL,
	};
	struct resource *res;

	if (!ndd)
		return 0;

	device_for_each_child(&nvdimm_bus->dev, &info, alias_dpa_busy);

	/* now account for busy blk allocations in unaliased dpa */
	for_each_dpa_resource(ndd, res) {
		if (strncmp(res->name, "blk", 3) != 0)
			continue;
		info.available -= resource_size(res);
	}

	return info.available;
}

/**
 * nd_pmem_available_dpa - for the given dimm+region account unallocated dpa
 * @nd_mapping: container of dpa-resource-root + labels
 * @nd_region: constrain available space check to this reference region
 * @overlap: calculate available space assuming this level of overlap
 *
 * Validate that a PMEM label, if present, aligns with the start of an
 * interleave set and truncate the available size at the lowest BLK
 * overlap point.
 *
 * The expectation is that this routine is called multiple times as it
 * probes for the largest BLK encroachment for any single member DIMM of
 * the interleave set.  Once that value is determined the PMEM-limit for
 * the set can be established.
 */
resource_size_t nd_pmem_available_dpa(struct nd_region *nd_region,
		struct nd_mapping *nd_mapping, resource_size_t *overlap)
{
	resource_size_t map_start, map_end, busy = 0, available, blk_start;
	struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
	struct resource *res;
	const char *reason;

	if (!ndd)
		return 0;

	map_start = nd_mapping->start;
	map_end = map_start + nd_mapping->size - 1;
	blk_start = max(map_start, map_end + 1 - *overlap);
	for_each_dpa_resource(ndd, res) {
		if (res->start >= map_start && res->start < map_end) {
			if (strncmp(res->name, "blk", 3) == 0)
				blk_start = min(blk_start,
						max(map_start, res->start));
			else if (res->end > map_end) {
				reason = "misaligned to iset";
				goto err;
			} else
				busy += resource_size(res);
		} else if (res->end >= map_start && res->end <= map_end) {
			if (strncmp(res->name, "blk", 3) == 0) {
				/*
				 * If a BLK allocation overlaps the start of
				 * PMEM the entire interleave set may now only
				 * be used for BLK.
				 */
				blk_start = map_start;
			} else
				busy += resource_size(res);
		} else if (map_start > res->start && map_start < res->end) {
			/* total eclipse of the mapping */
			busy += nd_mapping->size;
			blk_start = map_start;
		}
	}

	*overlap = map_end + 1 - blk_start;
	available = blk_start - map_start;
	if (busy < available)
		return available - busy;
	return 0;

 err:
	nd_dbg_dpa(nd_region, ndd, res, "%s\n", reason);
	return 0;
}

void nvdimm_free_dpa(struct nvdimm_drvdata *ndd, struct resource *res)
{
	WARN_ON_ONCE(!is_nvdimm_bus_locked(ndd->dev));
	kfree(res->name);
	__release_region(&ndd->dpa, res->start, resource_size(res));
}

struct resource *nvdimm_allocate_dpa(struct nvdimm_drvdata *ndd,
		struct nd_label_id *label_id, resource_size_t start,
		resource_size_t n)
{
	char *name = kmemdup(label_id, sizeof(*label_id), GFP_KERNEL);
	struct resource *res;

	if (!name)
		return NULL;

	WARN_ON_ONCE(!is_nvdimm_bus_locked(ndd->dev));
	res = __request_region(&ndd->dpa, start, n, name, 0);
	if (!res)
		kfree(name);
	return res;
}

/**
 * nvdimm_allocated_dpa - sum up the dpa currently allocated to this label_id
 * @nvdimm: container of dpa-resource-root + labels
 * @label_id: dpa resource name of the form {pmem|blk}-<human readable uuid>
 */
resource_size_t nvdimm_allocated_dpa(struct nvdimm_drvdata *ndd,
		struct nd_label_id *label_id)
{
	resource_size_t allocated = 0;
	struct resource *res;

	for_each_dpa_resource(ndd, res)
		if (strcmp(res->name, label_id->id) == 0)
			allocated += resource_size(res);

	return allocated;
}

static int count_dimms(struct device *dev, void *c)
{
	int *count = c;

	if (is_nvdimm(dev))
		(*count)++;
	return 0;
}

int nvdimm_bus_check_dimm_count(struct nvdimm_bus *nvdimm_bus, int dimm_count)
{
	int count = 0;
	/* Flush any possible dimm registration failures */
	nd_synchronize();

	device_for_each_child(&nvdimm_bus->dev, &count, count_dimms);
	dev_dbg(&nvdimm_bus->dev, "count: %d\n", count);
	if (count != dimm_count)
		return -ENXIO;
	return 0;
}
EXPORT_SYMBOL_GPL(nvdimm_bus_check_dimm_count);

void __exit nvdimm_devs_exit(void)
{
	ida_destroy(&dimm_ida);
}
Commit	Line	Data
e6dfb2de 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	#define pr_fmt(fmt) KBUILD_MODNAME ": " fmt
4d88a97a	14	#include <linux/vmalloc.h>
e6dfb2de	15	#include <linux/device.h>
62232e45	16	#include <linux/ndctl.h>
e6dfb2de DW	17	#include <linux/slab.h>
	18	#include <linux/io.h>
	19	#include <linux/fs.h>
	20	#include <linux/mm.h>
	21	#include "nd-core.h"
0ba1c634	22	#include "label.h"
ca6a4657	23	#include "pmem.h"
4d88a97a	24	#include "nd.h"
e6dfb2de DW	25
	26	static DEFINE_IDA(dimm_ida);
	27
4d88a97a DW	28	/*
	29	* Retrieve bus and dimm handle and return if this bus supports
	30	* get_config_data commands
	31	*/
aee65987	32	int nvdimm_check_config_data(struct device *dev)
4d88a97a	33	{
aee65987	34	struct nvdimm *nvdimm = to_nvdimm(dev);
4d88a97a	35
aee65987 TK	36	if (!nvdimm->cmd_mask \|\|
aee65987 TK	37	!test_bit(ND_CMD_GET_CONFIG_DATA, &nvdimm->cmd_mask)) {
8f078b38	38	if (test_bit(NDD_ALIASING, &nvdimm->flags))
aee65987 TK	39	return -ENXIO;
	40	else
	41	return -ENOTTY;
	42	}
4d88a97a DW	43
	44	return 0;
	45	}
	46
	47	static int validate_dimm(struct nvdimm_drvdata *ndd)
	48	{
aee65987	49	int rc;
4d88a97a	50
aee65987 TK	51	if (!ndd)
	52	return -EINVAL;
	53
	54	rc = nvdimm_check_config_data(ndd->dev);
	55	if (rc)
4d88a97a DW	56	dev_dbg(ndd->dev, "%pf: %s error: %d\n",
	57	__builtin_return_address(0), __func__, rc);
	58	return rc;
	59	}
	60
	61	/**
	62	* nvdimm_init_nsarea - determine the geometry of a dimm's namespace area
	63	* @nvdimm: dimm to initialize
	64	*/
	65	int nvdimm_init_nsarea(struct nvdimm_drvdata *ndd)
	66	{
	67	struct nd_cmd_get_config_size *cmd = &ndd->nsarea;
	68	struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
	69	struct nvdimm_bus_descriptor *nd_desc;
	70	int rc = validate_dimm(ndd);
9d62ed96	71	int cmd_rc = 0;
4d88a97a DW	72
	73	if (rc)
	74	return rc;
	75
	76	if (cmd->config_size)
	77	return 0; /* already valid */
	78
	79	memset(cmd, 0, sizeof(*cmd));
	80	nd_desc = nvdimm_bus->nd_desc;
9d62ed96 DW	81	rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
	82	ND_CMD_GET_CONFIG_SIZE, cmd, sizeof(*cmd), &cmd_rc);
	83	if (rc < 0)
	84	return rc;
	85	return cmd_rc;
4d88a97a DW	86	}
	87
	88	int nvdimm_init_config_data(struct nvdimm_drvdata *ndd)
	89	{
	90	struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
e7c5a571	91	int rc = validate_dimm(ndd), cmd_rc = 0;
4d88a97a DW	92	struct nd_cmd_get_config_data_hdr *cmd;
4d88a97a DW	93	struct nvdimm_bus_descriptor *nd_desc;
4d88a97a DW	94	u32 max_cmd_size, config_size;
	95	size_t offset;
	96
	97	if (rc)
	98	return rc;
	99
	100	if (ndd->data)
	101	return 0;
	102
4a826c83 DW	103	if (ndd->nsarea.status \|\| ndd->nsarea.max_xfer == 0
	104	\|\| ndd->nsarea.config_size < ND_LABEL_MIN_SIZE) {
	105	dev_dbg(ndd->dev, "failed to init config data area: (%d:%d)\n",
	106	ndd->nsarea.max_xfer, ndd->nsarea.config_size);
4d88a97a	107	return -ENXIO;
4a826c83	108	}
4d88a97a	109
752ade68	110	ndd->data = kvmalloc(ndd->nsarea.config_size, GFP_KERNEL);
4d88a97a DW	111	if (!ndd->data)
	112	return -ENOMEM;
	113
	114	max_cmd_size = min_t(u32, PAGE_SIZE, ndd->nsarea.max_xfer);
	115	cmd = kzalloc(max_cmd_size + sizeof(*cmd), GFP_KERNEL);
	116	if (!cmd)
	117	return -ENOMEM;
	118
	119	nd_desc = nvdimm_bus->nd_desc;
	120	for (config_size = ndd->nsarea.config_size, offset = 0;
	121	config_size; config_size -= cmd->in_length,
	122	offset += cmd->in_length) {
	123	cmd->in_length = min(config_size, max_cmd_size);
	124	cmd->in_offset = offset;
	125	rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
	126	ND_CMD_GET_CONFIG_DATA, cmd,
e7c5a571 DW	127	cmd->in_length + sizeof(*cmd), &cmd_rc);
	128	if (rc < 0)
	129	break;
	130	if (cmd_rc < 0) {
	131	rc = cmd_rc;
4d88a97a DW	132	break;
	133	}
	134	memcpy(ndd->data + offset, cmd->out_buf, cmd->in_length);
	135	}
426824d6	136	dev_dbg(ndd->dev, "len: %zu rc: %d\n", offset, rc);
4d88a97a DW	137	kfree(cmd);
	138
	139	return rc;
	140	}
	141
f524bf27 DW	142	int nvdimm_set_config_data(struct nvdimm_drvdata *ndd, size_t offset,
	143	void *buf, size_t len)
	144	{
f524bf27 DW	145	size_t max_cmd_size, buf_offset;
f524bf27 DW	146	struct nd_cmd_set_config_hdr *cmd;
e7c5a571	147	int rc = validate_dimm(ndd), cmd_rc = 0;
f524bf27 DW	148	struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(ndd->dev);
	149	struct nvdimm_bus_descriptor *nd_desc = nvdimm_bus->nd_desc;
	150
	151	if (rc)
	152	return rc;
	153
	154	if (!ndd->data)
	155	return -ENXIO;
	156
	157	if (offset + len > ndd->nsarea.config_size)
	158	return -ENXIO;
	159
	160	max_cmd_size = min_t(u32, PAGE_SIZE, len);
	161	max_cmd_size = min_t(u32, max_cmd_size, ndd->nsarea.max_xfer);
	162	cmd = kzalloc(max_cmd_size + sizeof(*cmd) + sizeof(u32), GFP_KERNEL);
	163	if (!cmd)
	164	return -ENOMEM;
	165
	166	for (buf_offset = 0; len; len -= cmd->in_length,
	167	buf_offset += cmd->in_length) {
	168	size_t cmd_size;
f524bf27 DW	169
	170	cmd->in_offset = offset + buf_offset;
	171	cmd->in_length = min(max_cmd_size, len);
	172	memcpy(cmd->in_buf, buf + buf_offset, cmd->in_length);
	173
	174	/* status is output in the last 4-bytes of the command buffer */
	175	cmd_size = sizeof(*cmd) + cmd->in_length + sizeof(u32);
f524bf27 DW	176
f524bf27 DW	177	rc = nd_desc->ndctl(nd_desc, to_nvdimm(ndd->dev),
e7c5a571 DW	178	ND_CMD_SET_CONFIG_DATA, cmd, cmd_size, &cmd_rc);
	179	if (rc < 0)
	180	break;
	181	if (cmd_rc < 0) {
	182	rc = cmd_rc;
f524bf27 DW	183	break;
	184	}
	185	}
	186	kfree(cmd);
	187
	188	return rc;
	189	}
	190
42237e39 DW	191	void nvdimm_set_aliasing(struct device *dev)
	192	{
	193	struct nvdimm *nvdimm = to_nvdimm(dev);
	194
8f078b38 DW	195	set_bit(NDD_ALIASING, &nvdimm->flags);
	196	}
	197
	198	void nvdimm_set_locked(struct device *dev)
	199	{
	200	struct nvdimm *nvdimm = to_nvdimm(dev);
	201
	202	set_bit(NDD_LOCKED, &nvdimm->flags);
42237e39 DW	203	}
42237e39 DW	204
d34cb808 DW	205	void nvdimm_clear_locked(struct device *dev)
	206	{
	207	struct nvdimm *nvdimm = to_nvdimm(dev);
	208
	209	clear_bit(NDD_LOCKED, &nvdimm->flags);
	210	}
	211
e6dfb2de DW	212	static void nvdimm_release(struct device *dev)
	213	{
	214	struct nvdimm *nvdimm = to_nvdimm(dev);
	215
	216	ida_simple_remove(&dimm_ida, nvdimm->id);
	217	kfree(nvdimm);
	218	}
	219
	220	static struct device_type nvdimm_device_type = {
	221	.name = "nvdimm",
	222	.release = nvdimm_release,
	223	};
	224
62232e45	225	bool is_nvdimm(struct device *dev)
e6dfb2de DW	226	{
	227	return dev->type == &nvdimm_device_type;
	228	}
	229
	230	struct nvdimm to_nvdimm(struct device dev)
	231	{
	232	struct nvdimm *nvdimm = container_of(dev, struct nvdimm, dev);
	233
	234	WARN_ON(!is_nvdimm(dev));
	235	return nvdimm;
	236	}
	237	EXPORT_SYMBOL_GPL(to_nvdimm);
	238
047fc8a1 RZ	239	struct nvdimm nd_blk_region_to_dimm(struct nd_blk_region ndbr)
	240	{
	241	struct nd_region *nd_region = &ndbr->nd_region;
	242	struct nd_mapping *nd_mapping = &nd_region->mapping[0];
	243
	244	return nd_mapping->nvdimm;
	245	}
	246	EXPORT_SYMBOL_GPL(nd_blk_region_to_dimm);
	247
ca6a4657 DW	248	unsigned long nd_blk_memremap_flags(struct nd_blk_region *ndbr)
	249	{
	250	/* pmem mapping properties are private to libnvdimm */
	251	return ARCH_MEMREMAP_PMEM;
	252	}
	253	EXPORT_SYMBOL_GPL(nd_blk_memremap_flags);
	254
bf9bccc1 DW	255	struct nvdimm_drvdata to_ndd(struct nd_mapping nd_mapping)
	256	{
	257	struct nvdimm *nvdimm = nd_mapping->nvdimm;
	258
	259	WARN_ON_ONCE(!is_nvdimm_bus_locked(&nvdimm->dev));
	260
	261	return dev_get_drvdata(&nvdimm->dev);
	262	}
	263	EXPORT_SYMBOL(to_ndd);
	264
	265	void nvdimm_drvdata_release(struct kref *kref)
	266	{
	267	struct nvdimm_drvdata ndd = container_of(kref, typeof(ndd), kref);
	268	struct device *dev = ndd->dev;
	269	struct resource res, _r;
	270
426824d6	271	dev_dbg(dev, "trace\n");
bf9bccc1 DW	272	nvdimm_bus_lock(dev);
	273	for_each_dpa_resource_safe(ndd, res, _r)
	274	nvdimm_free_dpa(ndd, res);
	275	nvdimm_bus_unlock(dev);
	276
a06a7576	277	kvfree(ndd->data);
bf9bccc1 DW	278	kfree(ndd);
	279	put_device(dev);
	280	}
	281
	282	void get_ndd(struct nvdimm_drvdata *ndd)
	283	{
	284	kref_get(&ndd->kref);
	285	}
	286
	287	void put_ndd(struct nvdimm_drvdata *ndd)
	288	{
	289	if (ndd)
	290	kref_put(&ndd->kref, nvdimm_drvdata_release);
	291	}
	292
e6dfb2de DW	293	const char nvdimm_name(struct nvdimm nvdimm)
	294	{
	295	return dev_name(&nvdimm->dev);
	296	}
	297	EXPORT_SYMBOL_GPL(nvdimm_name);
	298
ba9c8dd3 DW	299	struct kobject nvdimm_kobj(struct nvdimm nvdimm)
	300	{
	301	return &nvdimm->dev.kobj;
	302	}
	303	EXPORT_SYMBOL_GPL(nvdimm_kobj);
	304
e3654eca DW	305	unsigned long nvdimm_cmd_mask(struct nvdimm *nvdimm)
	306	{
	307	return nvdimm->cmd_mask;
	308	}
	309	EXPORT_SYMBOL_GPL(nvdimm_cmd_mask);
	310
e6dfb2de DW	311	void nvdimm_provider_data(struct nvdimm nvdimm)
e6dfb2de DW	312	{
62232e45 DW	313	if (nvdimm)
	314	return nvdimm->provider_data;
	315	return NULL;
e6dfb2de DW	316	}
	317	EXPORT_SYMBOL_GPL(nvdimm_provider_data);
	318
62232e45 DW	319	static ssize_t commands_show(struct device *dev,
	320	struct device_attribute attr, char buf)
	321	{
	322	struct nvdimm *nvdimm = to_nvdimm(dev);
	323	int cmd, len = 0;
	324
e3654eca	325	if (!nvdimm->cmd_mask)
62232e45 DW	326	return sprintf(buf, "\n");
62232e45 DW	327
e3654eca	328	for_each_set_bit(cmd, &nvdimm->cmd_mask, BITS_PER_LONG)
62232e45 DW	329	len += sprintf(buf + len, "%s ", nvdimm_cmd_name(cmd));
	330	len += sprintf(buf + len, "\n");
	331	return len;
	332	}
	333	static DEVICE_ATTR_RO(commands);
	334
efbf6f50 DW	335	static ssize_t flags_show(struct device *dev,
	336	struct device_attribute attr, char buf)
	337	{
	338	struct nvdimm *nvdimm = to_nvdimm(dev);
	339
	340	return sprintf(buf, "%s%s\n",
	341	test_bit(NDD_ALIASING, &nvdimm->flags) ? "alias " : "",
	342	test_bit(NDD_LOCKED, &nvdimm->flags) ? "lock " : "");
	343	}
	344	static DEVICE_ATTR_RO(flags);
	345
eaf96153 DW	346	static ssize_t state_show(struct device dev, struct device_attribute attr,
	347	char *buf)
	348	{
	349	struct nvdimm *nvdimm = to_nvdimm(dev);
	350
	351	/*
	352	* The state may be in the process of changing, userspace should
	353	* quiesce probing if it wants a static answer
	354	*/
	355	nvdimm_bus_lock(dev);
	356	nvdimm_bus_unlock(dev);
	357	return sprintf(buf, "%s\n", atomic_read(&nvdimm->busy)
	358	? "active" : "idle");
	359	}
	360	static DEVICE_ATTR_RO(state);
	361
0ba1c634 DW	362	static ssize_t available_slots_show(struct device *dev,
	363	struct device_attribute attr, char buf)
	364	{
	365	struct nvdimm_drvdata *ndd = dev_get_drvdata(dev);
	366	ssize_t rc;
	367	u32 nfree;
	368
	369	if (!ndd)
	370	return -ENXIO;
	371
	372	nvdimm_bus_lock(dev);
	373	nfree = nd_label_nfree(ndd);
	374	if (nfree - 1 > nfree) {
	375	dev_WARN_ONCE(dev, 1, "we ate our last label?\n");
	376	nfree = 0;
	377	} else
	378	nfree--;
	379	rc = sprintf(buf, "%d\n", nfree);
	380	nvdimm_bus_unlock(dev);
	381	return rc;
	382	}
	383	static DEVICE_ATTR_RO(available_slots);
	384
62232e45	385	static struct attribute *nvdimm_attributes[] = {
eaf96153	386	&dev_attr_state.attr,
efbf6f50	387	&dev_attr_flags.attr,
62232e45	388	&dev_attr_commands.attr,
0ba1c634	389	&dev_attr_available_slots.attr,
62232e45 DW	390	NULL,
	391	};
	392
	393	struct attribute_group nvdimm_attribute_group = {
	394	.attrs = nvdimm_attributes,
	395	};
	396	EXPORT_SYMBOL_GPL(nvdimm_attribute_group);
	397
e6dfb2de	398	struct nvdimm nvdimm_create(struct nvdimm_bus nvdimm_bus, void *provider_data,
62232e45	399	const struct attribute_group **groups, unsigned long flags,
e5ae3b25 DW	400	unsigned long cmd_mask, int num_flush,
e5ae3b25 DW	401	struct resource *flush_wpq)
e6dfb2de DW	402	{
	403	struct nvdimm nvdimm = kzalloc(sizeof(nvdimm), GFP_KERNEL);
	404	struct device *dev;
	405
	406	if (!nvdimm)
	407	return NULL;
	408
	409	nvdimm->id = ida_simple_get(&dimm_ida, 0, 0, GFP_KERNEL);
	410	if (nvdimm->id < 0) {
	411	kfree(nvdimm);
	412	return NULL;
	413	}
	414	nvdimm->provider_data = provider_data;
	415	nvdimm->flags = flags;
e3654eca	416	nvdimm->cmd_mask = cmd_mask;
e5ae3b25 DW	417	nvdimm->num_flush = num_flush;
e5ae3b25 DW	418	nvdimm->flush_wpq = flush_wpq;
eaf96153	419	atomic_set(&nvdimm->busy, 0);
e6dfb2de DW	420	dev = &nvdimm->dev;
	421	dev_set_name(dev, "nmem%d", nvdimm->id);
	422	dev->parent = &nvdimm_bus->dev;
	423	dev->type = &nvdimm_device_type;
62232e45	424	dev->devt = MKDEV(nvdimm_major, nvdimm->id);
e6dfb2de	425	dev->groups = groups;
4d88a97a	426	nd_device_register(dev);
e6dfb2de DW	427
	428	return nvdimm;
	429	}
	430	EXPORT_SYMBOL_GPL(nvdimm_create);
4d88a97a	431
762d067d	432	int alias_dpa_busy(struct device dev, void data)
a1f3e4d6	433	{
fe514739	434	resource_size_t map_end, blk_start, new;
a1f3e4d6 DW	435	struct blk_alloc_info *info = data;
	436	struct nd_mapping *nd_mapping;
	437	struct nd_region *nd_region;
	438	struct nvdimm_drvdata *ndd;
	439	struct resource *res;
	440	int i;
	441
c9e582aa	442	if (!is_memory(dev))
a1f3e4d6 DW	443	return 0;
	444
	445	nd_region = to_nd_region(dev);
	446	for (i = 0; i < nd_region->ndr_mappings; i++) {
	447	nd_mapping = &nd_region->mapping[i];
	448	if (nd_mapping->nvdimm == info->nd_mapping->nvdimm)
	449	break;
	450	}
	451
	452	if (i >= nd_region->ndr_mappings)
	453	return 0;
	454
	455	ndd = to_ndd(nd_mapping);
	456	map_end = nd_mapping->start + nd_mapping->size - 1;
	457	blk_start = nd_mapping->start;
762d067d DW	458
	459	/*
	460	* In the allocation case ->res is set to free space that we are
	461	* looking to validate against PMEM aliasing collision rules
	462	* (i.e. BLK is allocated after all aliased PMEM).
	463	*/
	464	if (info->res) {
	465	if (info->res->start >= nd_mapping->start
	466	&& info->res->start < map_end)
	467	/* pass */;
	468	else
	469	return 0;
	470	}
	471
a1f3e4d6 DW	472	retry:
	473	/*
	474	* Find the free dpa from the end of the last pmem allocation to
fe514739	475	* the end of the interleave-set mapping.
a1f3e4d6	476	*/
a1f3e4d6	477	for_each_dpa_resource(ndd, res) {
fe514739 DW	478	if (strncmp(res->name, "pmem", 4) != 0)
fe514739 DW	479	continue;
a1f3e4d6 DW	480	if ((res->start >= blk_start && res->start < map_end)
	481	\|\| (res->end >= blk_start
	482	&& res->end <= map_end)) {
fe514739 DW	483	new = max(blk_start, min(map_end + 1, res->end + 1));
	484	if (new != blk_start) {
	485	blk_start = new;
	486	goto retry;
	487	}
a1f3e4d6 DW	488	}
	489	}
	490
762d067d DW	491	/* update the free space range with the probed blk_start */
	492	if (info->res && blk_start > info->res->start) {
	493	info->res->start = max(info->res->start, blk_start);
	494	if (info->res->start > info->res->end)
	495	info->res->end = info->res->start - 1;
	496	return 1;
	497	}
	498
fe514739	499	info->available -= blk_start - nd_mapping->start;
762d067d	500
a1f3e4d6 DW	501	return 0;
	502	}
	503
1b40e09a DW	504	/**
	505	* nd_blk_available_dpa - account the unused dpa of BLK region
	506	* @nd_mapping: container of dpa-resource-root + labels
	507	*
a1f3e4d6 DW	508	* Unlike PMEM, BLK namespaces can occupy discontiguous DPA ranges, but
	509	* we arrange for them to never start at an lower dpa than the last
	510	* PMEM allocation in an aliased region.
1b40e09a	511	*/
a1f3e4d6	512	resource_size_t nd_blk_available_dpa(struct nd_region *nd_region)
1b40e09a	513	{
a1f3e4d6 DW	514	struct nvdimm_bus *nvdimm_bus = walk_to_nvdimm_bus(&nd_region->dev);
a1f3e4d6 DW	515	struct nd_mapping *nd_mapping = &nd_region->mapping[0];
1b40e09a	516	struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
a1f3e4d6 DW	517	struct blk_alloc_info info = {
	518	.nd_mapping = nd_mapping,
	519	.available = nd_mapping->size,
762d067d	520	.res = NULL,
a1f3e4d6	521	};
1b40e09a DW	522	struct resource *res;
	523
	524	if (!ndd)
	525	return 0;
	526
a1f3e4d6	527	device_for_each_child(&nvdimm_bus->dev, &info, alias_dpa_busy);
1b40e09a	528
a1f3e4d6 DW	529	/* now account for busy blk allocations in unaliased dpa */
	530	for_each_dpa_resource(ndd, res) {
	531	if (strncmp(res->name, "blk", 3) != 0)
	532	continue;
fe514739	533	info.available -= resource_size(res);
a1f3e4d6 DW	534	}
	535
	536	return info.available;
1b40e09a DW	537	}
1b40e09a DW	538
bf9bccc1 DW	539	/**
	540	* nd_pmem_available_dpa - for the given dimm+region account unallocated dpa
	541	* @nd_mapping: container of dpa-resource-root + labels
	542	* @nd_region: constrain available space check to this reference region
	543	* @overlap: calculate available space assuming this level of overlap
	544	*
	545	* Validate that a PMEM label, if present, aligns with the start of an
	546	* interleave set and truncate the available size at the lowest BLK
	547	* overlap point.
	548	*
	549	* The expectation is that this routine is called multiple times as it
	550	* probes for the largest BLK encroachment for any single member DIMM of
	551	* the interleave set. Once that value is determined the PMEM-limit for
	552	* the set can be established.
	553	*/
	554	resource_size_t nd_pmem_available_dpa(struct nd_region *nd_region,
	555	struct nd_mapping nd_mapping, resource_size_t overlap)
	556	{
	557	resource_size_t map_start, map_end, busy = 0, available, blk_start;
	558	struct nvdimm_drvdata *ndd = to_ndd(nd_mapping);
	559	struct resource *res;
	560	const char *reason;
	561
	562	if (!ndd)
	563	return 0;
	564
	565	map_start = nd_mapping->start;
	566	map_end = map_start + nd_mapping->size - 1;
	567	blk_start = max(map_start, map_end + 1 - *overlap);
a1f3e4d6	568	for_each_dpa_resource(ndd, res) {
bf9bccc1 DW	569	if (res->start >= map_start && res->start < map_end) {
bf9bccc1 DW	570	if (strncmp(res->name, "blk", 3) == 0)
a1f3e4d6 DW	571	blk_start = min(blk_start,
	572	max(map_start, res->start));
	573	else if (res->end > map_end) {
bf9bccc1 DW	574	reason = "misaligned to iset";
bf9bccc1 DW	575	goto err;
a1f3e4d6	576	} else
bf9bccc1	577	busy += resource_size(res);
bf9bccc1 DW	578	} else if (res->end >= map_start && res->end <= map_end) {
	579	if (strncmp(res->name, "blk", 3) == 0) {
	580	/*
	581	* If a BLK allocation overlaps the start of
	582	* PMEM the entire interleave set may now only
	583	* be used for BLK.
	584	*/
	585	blk_start = map_start;
a1f3e4d6 DW	586	} else
a1f3e4d6 DW	587	busy += resource_size(res);
bf9bccc1 DW	588	} else if (map_start > res->start && map_start < res->end) {
	589	/* total eclipse of the mapping */
	590	busy += nd_mapping->size;
	591	blk_start = map_start;
	592	}
a1f3e4d6	593	}
bf9bccc1 DW	594
	595	*overlap = map_end + 1 - blk_start;
	596	available = blk_start - map_start;
	597	if (busy < available)
	598	return available - busy;
	599	return 0;
	600
	601	err:
bf9bccc1 DW	602	nd_dbg_dpa(nd_region, ndd, res, "%s\n", reason);
	603	return 0;
	604	}
	605
4a826c83 DW	606	void nvdimm_free_dpa(struct nvdimm_drvdata ndd, struct resource res)
	607	{
	608	WARN_ON_ONCE(!is_nvdimm_bus_locked(ndd->dev));
	609	kfree(res->name);
	610	__release_region(&ndd->dpa, res->start, resource_size(res));
	611	}
	612
	613	struct resource nvdimm_allocate_dpa(struct nvdimm_drvdata ndd,
	614	struct nd_label_id *label_id, resource_size_t start,
	615	resource_size_t n)
	616	{
	617	char name = kmemdup(label_id, sizeof(label_id), GFP_KERNEL);
	618	struct resource *res;
	619
	620	if (!name)
	621	return NULL;
	622
	623	WARN_ON_ONCE(!is_nvdimm_bus_locked(ndd->dev));
	624	res = __request_region(&ndd->dpa, start, n, name, 0);
	625	if (!res)
	626	kfree(name);
	627	return res;
	628	}
	629
bf9bccc1 DW	630	/**
	631	* nvdimm_allocated_dpa - sum up the dpa currently allocated to this label_id
	632	* @nvdimm: container of dpa-resource-root + labels
	633	* @label_id: dpa resource name of the form {pmem\|blk}-<human readable uuid>
	634	*/
	635	resource_size_t nvdimm_allocated_dpa(struct nvdimm_drvdata *ndd,
	636	struct nd_label_id *label_id)
	637	{
	638	resource_size_t allocated = 0;
	639	struct resource *res;
	640
	641	for_each_dpa_resource(ndd, res)
	642	if (strcmp(res->name, label_id->id) == 0)
	643	allocated += resource_size(res);
	644
	645	return allocated;
	646	}
	647
4d88a97a DW	648	static int count_dimms(struct device dev, void c)
	649	{
	650	int *count = c;
	651
	652	if (is_nvdimm(dev))
	653	(*count)++;
	654	return 0;
	655	}
	656
	657	int nvdimm_bus_check_dimm_count(struct nvdimm_bus *nvdimm_bus, int dimm_count)
	658	{
	659	int count = 0;
	660	/* Flush any possible dimm registration failures */
	661	nd_synchronize();
	662
	663	device_for_each_child(&nvdimm_bus->dev, &count, count_dimms);
426824d6	664	dev_dbg(&nvdimm_bus->dev, "count: %d\n", count);
4d88a97a DW	665	if (count != dimm_count)
	666	return -ENXIO;
	667	return 0;
	668	}
	669	EXPORT_SYMBOL_GPL(nvdimm_bus_check_dimm_count);
b354aba0 DW	670
	671	void __exit nvdimm_devs_exit(void)
	672	{
	673	ida_destroy(&dimm_ida);
	674	}