1// SPDX-License-Identifier: GPL-2.0-only
  2/*
  3 * Copyright(c) 2013-2016 Intel Corporation. All rights reserved.
  4 */
  5#include <linux/device.h>
  6#include <linux/sizes.h>
  7#include <linux/slab.h>
  8#include <linux/mm.h>
  9#include "nd-core.h"
 10#include "pfn.h"
 11#include "nd.h"
 12
 13static void nd_dax_release(struct device *dev)
 14{
 15	struct nd_region *nd_region = to_nd_region(dev->parent);
 16	struct nd_dax *nd_dax = to_nd_dax(dev);
 17	struct nd_pfn *nd_pfn = &nd_dax->nd_pfn;
 18
 19	dev_dbg(dev, "trace\n");
 20	nd_detach_ndns(dev, &nd_pfn->ndns);
 21	ida_free(&nd_region->dax_ida, nd_pfn->id);
 22	kfree(nd_pfn->uuid);
 23	kfree(nd_dax);
 24}
 25
 26struct nd_dax *to_nd_dax(struct device *dev)
 27{
 28	struct nd_dax *nd_dax = container_of(dev, struct nd_dax, nd_pfn.dev);
 29
 30	WARN_ON(!is_nd_dax(dev));
 31	return nd_dax;
 32}
 33EXPORT_SYMBOL(to_nd_dax);
 34
 35static const struct device_type nd_dax_device_type = {
 36	.name = "nd_dax",
 37	.release = nd_dax_release,
 38	.groups = nd_pfn_attribute_groups,
 39};
 40
 41bool is_nd_dax(const struct device *dev)
 42{
 43	return dev ? dev->type == &nd_dax_device_type : false;
 44}
 45EXPORT_SYMBOL(is_nd_dax);
 46
 47static struct nd_dax *nd_dax_alloc(struct nd_region *nd_region)
 48{
 49	struct nd_pfn *nd_pfn;
 50	struct nd_dax *nd_dax;
 51	struct device *dev;
 52
 53	nd_dax = kzalloc(sizeof(*nd_dax), GFP_KERNEL);
 54	if (!nd_dax)
 55		return NULL;
 56
 57	nd_pfn = &nd_dax->nd_pfn;
 58	nd_pfn->id = ida_alloc(&nd_region->dax_ida, GFP_KERNEL);
 59	if (nd_pfn->id < 0) {
 60		kfree(nd_dax);
 61		return NULL;
 62	}
 63
 64	dev = &nd_pfn->dev;
 65	dev_set_name(dev, "dax%d.%d", nd_region->id, nd_pfn->id);
 66	dev->type = &nd_dax_device_type;
 67	dev->parent = &nd_region->dev;
 68
 69	return nd_dax;
 70}
 71
 72struct device *nd_dax_create(struct nd_region *nd_region)
 73{
 74	struct device *dev = NULL;
 75	struct nd_dax *nd_dax;
 76
 77	if (!is_memory(&nd_region->dev))
 78		return NULL;
 79
 80	nd_dax = nd_dax_alloc(nd_region);
 81	if (nd_dax)
 82		dev = nd_pfn_devinit(&nd_dax->nd_pfn, NULL);
 83	nd_device_register(dev);
 84	return dev;
 85}
 86
 87int nd_dax_probe(struct device *dev, struct nd_namespace_common *ndns)
 88{
 89	int rc;
 90	struct nd_dax *nd_dax;
 91	struct device *dax_dev;
 92	struct nd_pfn *nd_pfn;
 93	struct nd_pfn_sb *pfn_sb;
 94	struct nd_region *nd_region = to_nd_region(ndns->dev.parent);
 95
 96	if (ndns->force_raw)
 97		return -ENODEV;
 98
 99	switch (ndns->claim_class) {
100	case NVDIMM_CCLASS_NONE:
101	case NVDIMM_CCLASS_DAX:
102		break;
103	default:
104		return -ENODEV;
105	}
106
107	nvdimm_bus_lock(&ndns->dev);
108	nd_dax = nd_dax_alloc(nd_region);
109	nd_pfn = &nd_dax->nd_pfn;
110	dax_dev = nd_pfn_devinit(nd_pfn, ndns);
111	nvdimm_bus_unlock(&ndns->dev);
112	if (!dax_dev)
113		return -ENOMEM;
114	pfn_sb = devm_kmalloc(dev, sizeof(*pfn_sb), GFP_KERNEL);
115	nd_pfn->pfn_sb = pfn_sb;
116	rc = nd_pfn_validate(nd_pfn, DAX_SIG);
117	dev_dbg(dev, "dax: %s\n", rc == 0 ? dev_name(dax_dev) : "<none>");
118	if (rc < 0) {
119		nd_detach_ndns(dax_dev, &nd_pfn->ndns);
120		put_device(dax_dev);
121	} else
122		nd_device_register(dax_dev);
123
124	return rc;
125}
126EXPORT_SYMBOL(nd_dax_probe);