Linux Audio

Check our new training course

Loading...
v6.8
   1// SPDX-License-Identifier: GPL-2.0
   2/* Copyright(c) 2017-2018 Intel Corporation. All rights reserved. */
   3#include <linux/memremap.h>
   4#include <linux/device.h>
   5#include <linux/mutex.h>
   6#include <linux/list.h>
   7#include <linux/slab.h>
   8#include <linux/dax.h>
   9#include <linux/io.h>
  10#include "dax-private.h"
  11#include "bus.h"
  12
  13static DEFINE_MUTEX(dax_bus_lock);
  14
 
 
 
 
 
 
 
 
 
 
 
 
  15#define DAX_NAME_LEN 30
  16struct dax_id {
  17	struct list_head list;
  18	char dev_name[DAX_NAME_LEN];
  19};
  20
  21static int dax_bus_uevent(const struct device *dev, struct kobj_uevent_env *env)
  22{
  23	/*
  24	 * We only ever expect to handle device-dax instances, i.e. the
  25	 * @type argument to MODULE_ALIAS_DAX_DEVICE() is always zero
  26	 */
  27	return add_uevent_var(env, "MODALIAS=" DAX_DEVICE_MODALIAS_FMT, 0);
  28}
  29
  30static struct dax_device_driver *to_dax_drv(struct device_driver *drv)
  31{
  32	return container_of(drv, struct dax_device_driver, drv);
  33}
  34
  35static struct dax_id *__dax_match_id(struct dax_device_driver *dax_drv,
  36		const char *dev_name)
  37{
  38	struct dax_id *dax_id;
  39
  40	lockdep_assert_held(&dax_bus_lock);
  41
  42	list_for_each_entry(dax_id, &dax_drv->ids, list)
  43		if (sysfs_streq(dax_id->dev_name, dev_name))
  44			return dax_id;
  45	return NULL;
  46}
  47
  48static int dax_match_id(struct dax_device_driver *dax_drv, struct device *dev)
  49{
  50	int match;
  51
  52	mutex_lock(&dax_bus_lock);
  53	match = !!__dax_match_id(dax_drv, dev_name(dev));
  54	mutex_unlock(&dax_bus_lock);
  55
  56	return match;
  57}
  58
  59static int dax_match_type(struct dax_device_driver *dax_drv, struct device *dev)
  60{
  61	enum dax_driver_type type = DAXDRV_DEVICE_TYPE;
  62	struct dev_dax *dev_dax = to_dev_dax(dev);
  63
  64	if (dev_dax->region->res.flags & IORESOURCE_DAX_KMEM)
  65		type = DAXDRV_KMEM_TYPE;
  66
  67	if (dax_drv->type == type)
  68		return 1;
  69
  70	/* default to device mode if dax_kmem is disabled */
  71	if (dax_drv->type == DAXDRV_DEVICE_TYPE &&
  72	    !IS_ENABLED(CONFIG_DEV_DAX_KMEM))
  73		return 1;
  74
  75	return 0;
  76}
  77
  78enum id_action {
  79	ID_REMOVE,
  80	ID_ADD,
  81};
  82
  83static ssize_t do_id_store(struct device_driver *drv, const char *buf,
  84		size_t count, enum id_action action)
  85{
  86	struct dax_device_driver *dax_drv = to_dax_drv(drv);
  87	unsigned int region_id, id;
  88	char devname[DAX_NAME_LEN];
  89	struct dax_id *dax_id;
  90	ssize_t rc = count;
  91	int fields;
  92
  93	fields = sscanf(buf, "dax%d.%d", &region_id, &id);
  94	if (fields != 2)
  95		return -EINVAL;
  96	sprintf(devname, "dax%d.%d", region_id, id);
  97	if (!sysfs_streq(buf, devname))
  98		return -EINVAL;
  99
 100	mutex_lock(&dax_bus_lock);
 101	dax_id = __dax_match_id(dax_drv, buf);
 102	if (!dax_id) {
 103		if (action == ID_ADD) {
 104			dax_id = kzalloc(sizeof(*dax_id), GFP_KERNEL);
 105			if (dax_id) {
 106				strscpy(dax_id->dev_name, buf, DAX_NAME_LEN);
 107				list_add(&dax_id->list, &dax_drv->ids);
 108			} else
 109				rc = -ENOMEM;
 110		}
 111	} else if (action == ID_REMOVE) {
 112		list_del(&dax_id->list);
 113		kfree(dax_id);
 114	}
 115	mutex_unlock(&dax_bus_lock);
 116
 117	if (rc < 0)
 118		return rc;
 119	if (action == ID_ADD)
 120		rc = driver_attach(drv);
 121	if (rc)
 122		return rc;
 123	return count;
 124}
 125
 126static ssize_t new_id_store(struct device_driver *drv, const char *buf,
 127		size_t count)
 128{
 129	return do_id_store(drv, buf, count, ID_ADD);
 130}
 131static DRIVER_ATTR_WO(new_id);
 132
 133static ssize_t remove_id_store(struct device_driver *drv, const char *buf,
 134		size_t count)
 135{
 136	return do_id_store(drv, buf, count, ID_REMOVE);
 137}
 138static DRIVER_ATTR_WO(remove_id);
 139
 140static struct attribute *dax_drv_attrs[] = {
 141	&driver_attr_new_id.attr,
 142	&driver_attr_remove_id.attr,
 143	NULL,
 144};
 145ATTRIBUTE_GROUPS(dax_drv);
 146
 147static int dax_bus_match(struct device *dev, struct device_driver *drv);
 148
 149/*
 150 * Static dax regions are regions created by an external subsystem
 151 * nvdimm where a single range is assigned. Its boundaries are by the external
 152 * subsystem and are usually limited to one physical memory range. For example,
 153 * for PMEM it is usually defined by NVDIMM Namespace boundaries (i.e. a
 154 * single contiguous range)
 155 *
 156 * On dynamic dax regions, the assigned region can be partitioned by dax core
 157 * into multiple subdivisions. A subdivision is represented into one
 158 * /dev/daxN.M device composed by one or more potentially discontiguous ranges.
 159 *
 160 * When allocating a dax region, drivers must set whether it's static
 161 * (IORESOURCE_DAX_STATIC).  On static dax devices, the @pgmap is pre-assigned
 162 * to dax core when calling devm_create_dev_dax(), whereas in dynamic dax
 163 * devices it is NULL but afterwards allocated by dax core on device ->probe().
 164 * Care is needed to make sure that dynamic dax devices are torn down with a
 165 * cleared @pgmap field (see kill_dev_dax()).
 166 */
 167static bool is_static(struct dax_region *dax_region)
 168{
 169	return (dax_region->res.flags & IORESOURCE_DAX_STATIC) != 0;
 170}
 171
 172bool static_dev_dax(struct dev_dax *dev_dax)
 173{
 174	return is_static(dev_dax->region);
 175}
 176EXPORT_SYMBOL_GPL(static_dev_dax);
 177
 178static u64 dev_dax_size(struct dev_dax *dev_dax)
 179{
 180	u64 size = 0;
 181	int i;
 182
 183	device_lock_assert(&dev_dax->dev);
 184
 185	for (i = 0; i < dev_dax->nr_range; i++)
 186		size += range_len(&dev_dax->ranges[i].range);
 187
 188	return size;
 189}
 190
 191static int dax_bus_probe(struct device *dev)
 192{
 193	struct dax_device_driver *dax_drv = to_dax_drv(dev->driver);
 194	struct dev_dax *dev_dax = to_dev_dax(dev);
 195	struct dax_region *dax_region = dev_dax->region;
 196	int rc;
 
 197
 198	if (dev_dax_size(dev_dax) == 0 || dev_dax->id < 0)
 
 
 
 
 
 
 199		return -ENXIO;
 200
 201	rc = dax_drv->probe(dev_dax);
 202
 203	if (rc || is_static(dax_region))
 204		return rc;
 205
 206	/*
 207	 * Track new seed creation only after successful probe of the
 208	 * previous seed.
 209	 */
 210	if (dax_region->seed == dev)
 211		dax_region->seed = NULL;
 212
 213	return 0;
 214}
 215
 216static void dax_bus_remove(struct device *dev)
 217{
 218	struct dax_device_driver *dax_drv = to_dax_drv(dev->driver);
 219	struct dev_dax *dev_dax = to_dev_dax(dev);
 220
 221	if (dax_drv->remove)
 222		dax_drv->remove(dev_dax);
 223}
 224
 225static struct bus_type dax_bus_type = {
 226	.name = "dax",
 227	.uevent = dax_bus_uevent,
 228	.match = dax_bus_match,
 229	.probe = dax_bus_probe,
 230	.remove = dax_bus_remove,
 231	.drv_groups = dax_drv_groups,
 232};
 233
 234static int dax_bus_match(struct device *dev, struct device_driver *drv)
 235{
 236	struct dax_device_driver *dax_drv = to_dax_drv(drv);
 237
 238	if (dax_match_id(dax_drv, dev))
 239		return 1;
 240	return dax_match_type(dax_drv, dev);
 241}
 242
 243/*
 244 * Rely on the fact that drvdata is set before the attributes are
 245 * registered, and that the attributes are unregistered before drvdata
 246 * is cleared to assume that drvdata is always valid.
 247 */
 248static ssize_t id_show(struct device *dev,
 249		struct device_attribute *attr, char *buf)
 250{
 251	struct dax_region *dax_region = dev_get_drvdata(dev);
 252
 253	return sprintf(buf, "%d\n", dax_region->id);
 254}
 255static DEVICE_ATTR_RO(id);
 256
 257static ssize_t region_size_show(struct device *dev,
 258		struct device_attribute *attr, char *buf)
 259{
 260	struct dax_region *dax_region = dev_get_drvdata(dev);
 261
 262	return sprintf(buf, "%llu\n", (unsigned long long)
 263			resource_size(&dax_region->res));
 264}
 265static struct device_attribute dev_attr_region_size = __ATTR(size, 0444,
 266		region_size_show, NULL);
 267
 268static ssize_t region_align_show(struct device *dev,
 269		struct device_attribute *attr, char *buf)
 270{
 271	struct dax_region *dax_region = dev_get_drvdata(dev);
 272
 273	return sprintf(buf, "%u\n", dax_region->align);
 274}
 275static struct device_attribute dev_attr_region_align =
 276		__ATTR(align, 0400, region_align_show, NULL);
 277
 278#define for_each_dax_region_resource(dax_region, res) \
 279	for (res = (dax_region)->res.child; res; res = res->sibling)
 280
 281static unsigned long long dax_region_avail_size(struct dax_region *dax_region)
 282{
 283	resource_size_t size = resource_size(&dax_region->res);
 284	struct resource *res;
 285
 286	device_lock_assert(dax_region->dev);
 287
 288	for_each_dax_region_resource(dax_region, res)
 289		size -= resource_size(res);
 290	return size;
 291}
 292
 293static ssize_t available_size_show(struct device *dev,
 294		struct device_attribute *attr, char *buf)
 295{
 296	struct dax_region *dax_region = dev_get_drvdata(dev);
 297	unsigned long long size;
 
 298
 299	device_lock(dev);
 
 
 300	size = dax_region_avail_size(dax_region);
 301	device_unlock(dev);
 302
 303	return sprintf(buf, "%llu\n", size);
 304}
 305static DEVICE_ATTR_RO(available_size);
 306
 307static ssize_t seed_show(struct device *dev,
 308		struct device_attribute *attr, char *buf)
 309{
 310	struct dax_region *dax_region = dev_get_drvdata(dev);
 311	struct device *seed;
 312	ssize_t rc;
 313
 314	if (is_static(dax_region))
 315		return -EINVAL;
 316
 317	device_lock(dev);
 
 
 318	seed = dax_region->seed;
 319	rc = sprintf(buf, "%s\n", seed ? dev_name(seed) : "");
 320	device_unlock(dev);
 321
 322	return rc;
 323}
 324static DEVICE_ATTR_RO(seed);
 325
 326static ssize_t create_show(struct device *dev,
 327		struct device_attribute *attr, char *buf)
 328{
 329	struct dax_region *dax_region = dev_get_drvdata(dev);
 330	struct device *youngest;
 331	ssize_t rc;
 332
 333	if (is_static(dax_region))
 334		return -EINVAL;
 335
 336	device_lock(dev);
 
 
 337	youngest = dax_region->youngest;
 338	rc = sprintf(buf, "%s\n", youngest ? dev_name(youngest) : "");
 339	device_unlock(dev);
 340
 341	return rc;
 342}
 343
 
 
 344static ssize_t create_store(struct device *dev, struct device_attribute *attr,
 345		const char *buf, size_t len)
 346{
 347	struct dax_region *dax_region = dev_get_drvdata(dev);
 348	unsigned long long avail;
 349	ssize_t rc;
 350	int val;
 351
 352	if (is_static(dax_region))
 353		return -EINVAL;
 354
 355	rc = kstrtoint(buf, 0, &val);
 356	if (rc)
 357		return rc;
 358	if (val != 1)
 359		return -EINVAL;
 360
 361	device_lock(dev);
 
 
 362	avail = dax_region_avail_size(dax_region);
 363	if (avail == 0)
 364		rc = -ENOSPC;
 365	else {
 366		struct dev_dax_data data = {
 367			.dax_region = dax_region,
 368			.size = 0,
 369			.id = -1,
 370			.memmap_on_memory = false,
 371		};
 372		struct dev_dax *dev_dax = devm_create_dev_dax(&data);
 373
 374		if (IS_ERR(dev_dax))
 375			rc = PTR_ERR(dev_dax);
 376		else {
 377			/*
 378			 * In support of crafting multiple new devices
 379			 * simultaneously multiple seeds can be created,
 380			 * but only the first one that has not been
 381			 * successfully bound is tracked as the region
 382			 * seed.
 383			 */
 384			if (!dax_region->seed)
 385				dax_region->seed = &dev_dax->dev;
 386			dax_region->youngest = &dev_dax->dev;
 387			rc = len;
 388		}
 389	}
 390	device_unlock(dev);
 391
 392	return rc;
 393}
 394static DEVICE_ATTR_RW(create);
 395
 396void kill_dev_dax(struct dev_dax *dev_dax)
 397{
 398	struct dax_device *dax_dev = dev_dax->dax_dev;
 399	struct inode *inode = dax_inode(dax_dev);
 400
 401	kill_dax(dax_dev);
 402	unmap_mapping_range(inode->i_mapping, 0, 0, 1);
 403
 404	/*
 405	 * Dynamic dax region have the pgmap allocated via dev_kzalloc()
 406	 * and thus freed by devm. Clear the pgmap to not have stale pgmap
 407	 * ranges on probe() from previous reconfigurations of region devices.
 408	 */
 409	if (!static_dev_dax(dev_dax))
 410		dev_dax->pgmap = NULL;
 411}
 412EXPORT_SYMBOL_GPL(kill_dev_dax);
 413
 414static void trim_dev_dax_range(struct dev_dax *dev_dax)
 415{
 416	int i = dev_dax->nr_range - 1;
 417	struct range *range = &dev_dax->ranges[i].range;
 418	struct dax_region *dax_region = dev_dax->region;
 419
 420	device_lock_assert(dax_region->dev);
 421	dev_dbg(&dev_dax->dev, "delete range[%d]: %#llx:%#llx\n", i,
 422		(unsigned long long)range->start,
 423		(unsigned long long)range->end);
 424
 425	__release_region(&dax_region->res, range->start, range_len(range));
 426	if (--dev_dax->nr_range == 0) {
 427		kfree(dev_dax->ranges);
 428		dev_dax->ranges = NULL;
 429	}
 430}
 431
 432static void free_dev_dax_ranges(struct dev_dax *dev_dax)
 433{
 434	while (dev_dax->nr_range)
 435		trim_dev_dax_range(dev_dax);
 436}
 437
 438static void unregister_dev_dax(void *dev)
 439{
 440	struct dev_dax *dev_dax = to_dev_dax(dev);
 441
 442	dev_dbg(dev, "%s\n", __func__);
 443
 
 444	kill_dev_dax(dev_dax);
 445	device_del(dev);
 446	free_dev_dax_ranges(dev_dax);
 447	put_device(dev);
 
 448}
 449
 450static void dax_region_free(struct kref *kref)
 451{
 452	struct dax_region *dax_region;
 453
 454	dax_region = container_of(kref, struct dax_region, kref);
 455	kfree(dax_region);
 456}
 457
 458static void dax_region_put(struct dax_region *dax_region)
 459{
 460	kref_put(&dax_region->kref, dax_region_free);
 461}
 462
 463/* a return value >= 0 indicates this invocation invalidated the id */
 464static int __free_dev_dax_id(struct dev_dax *dev_dax)
 465{
 466	struct device *dev = &dev_dax->dev;
 467	struct dax_region *dax_region;
 468	int rc = dev_dax->id;
 469
 470	device_lock_assert(dev);
 471
 472	if (!dev_dax->dyn_id || dev_dax->id < 0)
 473		return -1;
 474	dax_region = dev_dax->region;
 475	ida_free(&dax_region->ida, dev_dax->id);
 476	dax_region_put(dax_region);
 477	dev_dax->id = -1;
 478	return rc;
 479}
 480
 481static int free_dev_dax_id(struct dev_dax *dev_dax)
 482{
 483	struct device *dev = &dev_dax->dev;
 484	int rc;
 485
 486	device_lock(dev);
 
 
 487	rc = __free_dev_dax_id(dev_dax);
 488	device_unlock(dev);
 489	return rc;
 490}
 491
 492static int alloc_dev_dax_id(struct dev_dax *dev_dax)
 493{
 494	struct dax_region *dax_region = dev_dax->region;
 495	int id;
 496
 497	id = ida_alloc(&dax_region->ida, GFP_KERNEL);
 498	if (id < 0)
 499		return id;
 500	kref_get(&dax_region->kref);
 501	dev_dax->dyn_id = true;
 502	dev_dax->id = id;
 503	return id;
 504}
 505
 506static ssize_t delete_store(struct device *dev, struct device_attribute *attr,
 507		const char *buf, size_t len)
 508{
 509	struct dax_region *dax_region = dev_get_drvdata(dev);
 510	struct dev_dax *dev_dax;
 511	struct device *victim;
 512	bool do_del = false;
 513	int rc;
 514
 515	if (is_static(dax_region))
 516		return -EINVAL;
 517
 518	victim = device_find_child_by_name(dax_region->dev, buf);
 519	if (!victim)
 520		return -ENXIO;
 521
 522	device_lock(dev);
 523	device_lock(victim);
 524	dev_dax = to_dev_dax(victim);
 
 525	if (victim->driver || dev_dax_size(dev_dax))
 526		rc = -EBUSY;
 527	else {
 528		/*
 529		 * Invalidate the device so it does not become active
 530		 * again, but always preserve device-id-0 so that
 531		 * /sys/bus/dax/ is guaranteed to be populated while any
 532		 * dax_region is registered.
 533		 */
 534		if (dev_dax->id > 0) {
 535			do_del = __free_dev_dax_id(dev_dax) >= 0;
 536			rc = len;
 537			if (dax_region->seed == victim)
 538				dax_region->seed = NULL;
 539			if (dax_region->youngest == victim)
 540				dax_region->youngest = NULL;
 541		} else
 542			rc = -EBUSY;
 543	}
 
 544	device_unlock(victim);
 545
 546	/* won the race to invalidate the device, clean it up */
 547	if (do_del)
 548		devm_release_action(dev, unregister_dev_dax, victim);
 549	device_unlock(dev);
 550	put_device(victim);
 551
 552	return rc;
 553}
 554static DEVICE_ATTR_WO(delete);
 555
 556static umode_t dax_region_visible(struct kobject *kobj, struct attribute *a,
 557		int n)
 558{
 559	struct device *dev = container_of(kobj, struct device, kobj);
 560	struct dax_region *dax_region = dev_get_drvdata(dev);
 561
 562	if (is_static(dax_region))
 563		if (a == &dev_attr_available_size.attr
 564				|| a == &dev_attr_create.attr
 565				|| a == &dev_attr_seed.attr
 566				|| a == &dev_attr_delete.attr)
 567			return 0;
 568	return a->mode;
 569}
 570
 571static struct attribute *dax_region_attributes[] = {
 572	&dev_attr_available_size.attr,
 573	&dev_attr_region_size.attr,
 574	&dev_attr_region_align.attr,
 575	&dev_attr_create.attr,
 576	&dev_attr_seed.attr,
 577	&dev_attr_delete.attr,
 578	&dev_attr_id.attr,
 579	NULL,
 580};
 581
 582static const struct attribute_group dax_region_attribute_group = {
 583	.name = "dax_region",
 584	.attrs = dax_region_attributes,
 585	.is_visible = dax_region_visible,
 586};
 587
 588static const struct attribute_group *dax_region_attribute_groups[] = {
 589	&dax_region_attribute_group,
 590	NULL,
 591};
 592
 593static void dax_region_unregister(void *region)
 594{
 595	struct dax_region *dax_region = region;
 596
 597	sysfs_remove_groups(&dax_region->dev->kobj,
 598			dax_region_attribute_groups);
 599	dax_region_put(dax_region);
 600}
 601
 602struct dax_region *alloc_dax_region(struct device *parent, int region_id,
 603		struct range *range, int target_node, unsigned int align,
 604		unsigned long flags)
 605{
 606	struct dax_region *dax_region;
 607
 608	/*
 609	 * The DAX core assumes that it can store its private data in
 610	 * parent->driver_data. This WARN is a reminder / safeguard for
 611	 * developers of device-dax drivers.
 612	 */
 613	if (dev_get_drvdata(parent)) {
 614		dev_WARN(parent, "dax core failed to setup private data\n");
 615		return NULL;
 616	}
 617
 618	if (!IS_ALIGNED(range->start, align)
 619			|| !IS_ALIGNED(range_len(range), align))
 620		return NULL;
 621
 622	dax_region = kzalloc(sizeof(*dax_region), GFP_KERNEL);
 623	if (!dax_region)
 624		return NULL;
 625
 626	dev_set_drvdata(parent, dax_region);
 627	kref_init(&dax_region->kref);
 628	dax_region->id = region_id;
 629	dax_region->align = align;
 630	dax_region->dev = parent;
 631	dax_region->target_node = target_node;
 632	ida_init(&dax_region->ida);
 633	dax_region->res = (struct resource) {
 634		.start = range->start,
 635		.end = range->end,
 636		.flags = IORESOURCE_MEM | flags,
 637	};
 638
 639	if (sysfs_create_groups(&parent->kobj, dax_region_attribute_groups)) {
 640		kfree(dax_region);
 641		return NULL;
 642	}
 643
 644	if (devm_add_action_or_reset(parent, dax_region_unregister, dax_region))
 645		return NULL;
 646	return dax_region;
 647}
 648EXPORT_SYMBOL_GPL(alloc_dax_region);
 649
 650static void dax_mapping_release(struct device *dev)
 651{
 652	struct dax_mapping *mapping = to_dax_mapping(dev);
 653	struct device *parent = dev->parent;
 654	struct dev_dax *dev_dax = to_dev_dax(parent);
 655
 656	ida_free(&dev_dax->ida, mapping->id);
 657	kfree(mapping);
 658	put_device(parent);
 659}
 660
 661static void unregister_dax_mapping(void *data)
 662{
 663	struct device *dev = data;
 664	struct dax_mapping *mapping = to_dax_mapping(dev);
 665	struct dev_dax *dev_dax = to_dev_dax(dev->parent);
 666	struct dax_region *dax_region = dev_dax->region;
 667
 668	dev_dbg(dev, "%s\n", __func__);
 669
 670	device_lock_assert(dax_region->dev);
 671
 672	dev_dax->ranges[mapping->range_id].mapping = NULL;
 673	mapping->range_id = -1;
 674
 675	device_unregister(dev);
 676}
 677
 678static struct dev_dax_range *get_dax_range(struct device *dev)
 679{
 680	struct dax_mapping *mapping = to_dax_mapping(dev);
 681	struct dev_dax *dev_dax = to_dev_dax(dev->parent);
 682	struct dax_region *dax_region = dev_dax->region;
 683
 684	device_lock(dax_region->dev);
 
 
 685	if (mapping->range_id < 0) {
 686		device_unlock(dax_region->dev);
 687		return NULL;
 688	}
 689
 690	return &dev_dax->ranges[mapping->range_id];
 691}
 692
 693static void put_dax_range(struct dev_dax_range *dax_range)
 694{
 695	struct dax_mapping *mapping = dax_range->mapping;
 696	struct dev_dax *dev_dax = to_dev_dax(mapping->dev.parent);
 697	struct dax_region *dax_region = dev_dax->region;
 698
 699	device_unlock(dax_region->dev);
 700}
 701
 702static ssize_t start_show(struct device *dev,
 703		struct device_attribute *attr, char *buf)
 704{
 705	struct dev_dax_range *dax_range;
 706	ssize_t rc;
 707
 708	dax_range = get_dax_range(dev);
 709	if (!dax_range)
 710		return -ENXIO;
 711	rc = sprintf(buf, "%#llx\n", dax_range->range.start);
 712	put_dax_range(dax_range);
 713
 714	return rc;
 715}
 716static DEVICE_ATTR(start, 0400, start_show, NULL);
 717
 718static ssize_t end_show(struct device *dev,
 719		struct device_attribute *attr, char *buf)
 720{
 721	struct dev_dax_range *dax_range;
 722	ssize_t rc;
 723
 724	dax_range = get_dax_range(dev);
 725	if (!dax_range)
 726		return -ENXIO;
 727	rc = sprintf(buf, "%#llx\n", dax_range->range.end);
 728	put_dax_range(dax_range);
 729
 730	return rc;
 731}
 732static DEVICE_ATTR(end, 0400, end_show, NULL);
 733
 734static ssize_t pgoff_show(struct device *dev,
 735		struct device_attribute *attr, char *buf)
 736{
 737	struct dev_dax_range *dax_range;
 738	ssize_t rc;
 739
 740	dax_range = get_dax_range(dev);
 741	if (!dax_range)
 742		return -ENXIO;
 743	rc = sprintf(buf, "%#lx\n", dax_range->pgoff);
 744	put_dax_range(dax_range);
 745
 746	return rc;
 747}
 748static DEVICE_ATTR(page_offset, 0400, pgoff_show, NULL);
 749
 750static struct attribute *dax_mapping_attributes[] = {
 751	&dev_attr_start.attr,
 752	&dev_attr_end.attr,
 753	&dev_attr_page_offset.attr,
 754	NULL,
 755};
 756
 757static const struct attribute_group dax_mapping_attribute_group = {
 758	.attrs = dax_mapping_attributes,
 759};
 760
 761static const struct attribute_group *dax_mapping_attribute_groups[] = {
 762	&dax_mapping_attribute_group,
 763	NULL,
 764};
 765
 766static struct device_type dax_mapping_type = {
 767	.release = dax_mapping_release,
 768	.groups = dax_mapping_attribute_groups,
 769};
 770
 771static int devm_register_dax_mapping(struct dev_dax *dev_dax, int range_id)
 772{
 773	struct dax_region *dax_region = dev_dax->region;
 774	struct dax_mapping *mapping;
 775	struct device *dev;
 776	int rc;
 777
 778	device_lock_assert(dax_region->dev);
 779
 780	if (dev_WARN_ONCE(&dev_dax->dev, !dax_region->dev->driver,
 781				"region disabled\n"))
 782		return -ENXIO;
 783
 784	mapping = kzalloc(sizeof(*mapping), GFP_KERNEL);
 785	if (!mapping)
 786		return -ENOMEM;
 787	mapping->range_id = range_id;
 788	mapping->id = ida_alloc(&dev_dax->ida, GFP_KERNEL);
 789	if (mapping->id < 0) {
 790		kfree(mapping);
 791		return -ENOMEM;
 792	}
 793	dev_dax->ranges[range_id].mapping = mapping;
 794	dev = &mapping->dev;
 795	device_initialize(dev);
 796	dev->parent = &dev_dax->dev;
 797	get_device(dev->parent);
 798	dev->type = &dax_mapping_type;
 799	dev_set_name(dev, "mapping%d", mapping->id);
 800	rc = device_add(dev);
 801	if (rc) {
 802		put_device(dev);
 803		return rc;
 804	}
 805
 806	rc = devm_add_action_or_reset(dax_region->dev, unregister_dax_mapping,
 807			dev);
 808	if (rc)
 809		return rc;
 810	return 0;
 811}
 812
 813static int alloc_dev_dax_range(struct dev_dax *dev_dax, u64 start,
 814		resource_size_t size)
 815{
 816	struct dax_region *dax_region = dev_dax->region;
 817	struct resource *res = &dax_region->res;
 818	struct device *dev = &dev_dax->dev;
 819	struct dev_dax_range *ranges;
 820	unsigned long pgoff = 0;
 821	struct resource *alloc;
 822	int i, rc;
 823
 824	device_lock_assert(dax_region->dev);
 825
 826	/* handle the seed alloc special case */
 827	if (!size) {
 828		if (dev_WARN_ONCE(dev, dev_dax->nr_range,
 829					"0-size allocation must be first\n"))
 830			return -EBUSY;
 831		/* nr_range == 0 is elsewhere special cased as 0-size device */
 832		return 0;
 833	}
 834
 835	alloc = __request_region(res, start, size, dev_name(dev), 0);
 836	if (!alloc)
 837		return -ENOMEM;
 838
 839	ranges = krealloc(dev_dax->ranges, sizeof(*ranges)
 840			* (dev_dax->nr_range + 1), GFP_KERNEL);
 841	if (!ranges) {
 842		__release_region(res, alloc->start, resource_size(alloc));
 843		return -ENOMEM;
 844	}
 845
 846	for (i = 0; i < dev_dax->nr_range; i++)
 847		pgoff += PHYS_PFN(range_len(&ranges[i].range));
 848	dev_dax->ranges = ranges;
 849	ranges[dev_dax->nr_range++] = (struct dev_dax_range) {
 850		.pgoff = pgoff,
 851		.range = {
 852			.start = alloc->start,
 853			.end = alloc->end,
 854		},
 855	};
 856
 857	dev_dbg(dev, "alloc range[%d]: %pa:%pa\n", dev_dax->nr_range - 1,
 858			&alloc->start, &alloc->end);
 859	/*
 860	 * A dev_dax instance must be registered before mapping device
 861	 * children can be added. Defer to devm_create_dev_dax() to add
 862	 * the initial mapping device.
 863	 */
 864	if (!device_is_registered(&dev_dax->dev))
 865		return 0;
 866
 867	rc = devm_register_dax_mapping(dev_dax, dev_dax->nr_range - 1);
 868	if (rc)
 869		trim_dev_dax_range(dev_dax);
 870
 871	return rc;
 872}
 873
 874static int adjust_dev_dax_range(struct dev_dax *dev_dax, struct resource *res, resource_size_t size)
 875{
 876	int last_range = dev_dax->nr_range - 1;
 877	struct dev_dax_range *dax_range = &dev_dax->ranges[last_range];
 878	struct dax_region *dax_region = dev_dax->region;
 879	bool is_shrink = resource_size(res) > size;
 880	struct range *range = &dax_range->range;
 881	struct device *dev = &dev_dax->dev;
 882	int rc;
 883
 884	device_lock_assert(dax_region->dev);
 885
 886	if (dev_WARN_ONCE(dev, !size, "deletion is handled by dev_dax_shrink\n"))
 887		return -EINVAL;
 888
 889	rc = adjust_resource(res, range->start, size);
 890	if (rc)
 891		return rc;
 892
 893	*range = (struct range) {
 894		.start = range->start,
 895		.end = range->start + size - 1,
 896	};
 897
 898	dev_dbg(dev, "%s range[%d]: %#llx:%#llx\n", is_shrink ? "shrink" : "extend",
 899			last_range, (unsigned long long) range->start,
 900			(unsigned long long) range->end);
 901
 902	return 0;
 903}
 904
 905static ssize_t size_show(struct device *dev,
 906		struct device_attribute *attr, char *buf)
 907{
 908	struct dev_dax *dev_dax = to_dev_dax(dev);
 909	unsigned long long size;
 
 910
 911	device_lock(dev);
 
 
 912	size = dev_dax_size(dev_dax);
 913	device_unlock(dev);
 914
 915	return sprintf(buf, "%llu\n", size);
 916}
 917
 918static bool alloc_is_aligned(struct dev_dax *dev_dax, resource_size_t size)
 919{
 920	/*
 921	 * The minimum mapping granularity for a device instance is a
 922	 * single subsection, unless the arch says otherwise.
 923	 */
 924	return IS_ALIGNED(size, max_t(unsigned long, dev_dax->align, memremap_compat_align()));
 925}
 926
 927static int dev_dax_shrink(struct dev_dax *dev_dax, resource_size_t size)
 928{
 929	resource_size_t to_shrink = dev_dax_size(dev_dax) - size;
 930	struct dax_region *dax_region = dev_dax->region;
 931	struct device *dev = &dev_dax->dev;
 932	int i;
 933
 934	for (i = dev_dax->nr_range - 1; i >= 0; i--) {
 935		struct range *range = &dev_dax->ranges[i].range;
 936		struct dax_mapping *mapping = dev_dax->ranges[i].mapping;
 937		struct resource *adjust = NULL, *res;
 938		resource_size_t shrink;
 939
 940		shrink = min_t(u64, to_shrink, range_len(range));
 941		if (shrink >= range_len(range)) {
 942			devm_release_action(dax_region->dev,
 943					unregister_dax_mapping, &mapping->dev);
 944			trim_dev_dax_range(dev_dax);
 945			to_shrink -= shrink;
 946			if (!to_shrink)
 947				break;
 948			continue;
 949		}
 950
 951		for_each_dax_region_resource(dax_region, res)
 952			if (strcmp(res->name, dev_name(dev)) == 0
 953					&& res->start == range->start) {
 954				adjust = res;
 955				break;
 956			}
 957
 958		if (dev_WARN_ONCE(dev, !adjust || i != dev_dax->nr_range - 1,
 959					"failed to find matching resource\n"))
 960			return -ENXIO;
 961		return adjust_dev_dax_range(dev_dax, adjust, range_len(range)
 962				- shrink);
 963	}
 964	return 0;
 965}
 966
 967/*
 968 * Only allow adjustments that preserve the relative pgoff of existing
 969 * allocations. I.e. the dev_dax->ranges array is ordered by increasing pgoff.
 970 */
 971static bool adjust_ok(struct dev_dax *dev_dax, struct resource *res)
 972{
 973	struct dev_dax_range *last;
 974	int i;
 975
 976	if (dev_dax->nr_range == 0)
 977		return false;
 978	if (strcmp(res->name, dev_name(&dev_dax->dev)) != 0)
 979		return false;
 980	last = &dev_dax->ranges[dev_dax->nr_range - 1];
 981	if (last->range.start != res->start || last->range.end != res->end)
 982		return false;
 983	for (i = 0; i < dev_dax->nr_range - 1; i++) {
 984		struct dev_dax_range *dax_range = &dev_dax->ranges[i];
 985
 986		if (dax_range->pgoff > last->pgoff)
 987			return false;
 988	}
 989
 990	return true;
 991}
 992
 993static ssize_t dev_dax_resize(struct dax_region *dax_region,
 994		struct dev_dax *dev_dax, resource_size_t size)
 995{
 996	resource_size_t avail = dax_region_avail_size(dax_region), to_alloc;
 997	resource_size_t dev_size = dev_dax_size(dev_dax);
 998	struct resource *region_res = &dax_region->res;
 999	struct device *dev = &dev_dax->dev;
1000	struct resource *res, *first;
1001	resource_size_t alloc = 0;
1002	int rc;
1003
1004	if (dev->driver)
1005		return -EBUSY;
1006	if (size == dev_size)
1007		return 0;
1008	if (size > dev_size && size - dev_size > avail)
1009		return -ENOSPC;
1010	if (size < dev_size)
1011		return dev_dax_shrink(dev_dax, size);
1012
1013	to_alloc = size - dev_size;
1014	if (dev_WARN_ONCE(dev, !alloc_is_aligned(dev_dax, to_alloc),
1015			"resize of %pa misaligned\n", &to_alloc))
1016		return -ENXIO;
1017
1018	/*
1019	 * Expand the device into the unused portion of the region. This
1020	 * may involve adjusting the end of an existing resource, or
1021	 * allocating a new resource.
1022	 */
1023retry:
1024	first = region_res->child;
1025	if (!first)
1026		return alloc_dev_dax_range(dev_dax, dax_region->res.start, to_alloc);
1027
1028	rc = -ENOSPC;
1029	for (res = first; res; res = res->sibling) {
1030		struct resource *next = res->sibling;
1031
1032		/* space at the beginning of the region */
1033		if (res == first && res->start > dax_region->res.start) {
1034			alloc = min(res->start - dax_region->res.start, to_alloc);
1035			rc = alloc_dev_dax_range(dev_dax, dax_region->res.start, alloc);
1036			break;
1037		}
1038
1039		alloc = 0;
1040		/* space between allocations */
1041		if (next && next->start > res->end + 1)
1042			alloc = min(next->start - (res->end + 1), to_alloc);
1043
1044		/* space at the end of the region */
1045		if (!alloc && !next && res->end < region_res->end)
1046			alloc = min(region_res->end - res->end, to_alloc);
1047
1048		if (!alloc)
1049			continue;
1050
1051		if (adjust_ok(dev_dax, res)) {
1052			rc = adjust_dev_dax_range(dev_dax, res, resource_size(res) + alloc);
1053			break;
1054		}
1055		rc = alloc_dev_dax_range(dev_dax, res->end + 1, alloc);
1056		break;
1057	}
1058	if (rc)
1059		return rc;
1060	to_alloc -= alloc;
1061	if (to_alloc)
1062		goto retry;
1063	return 0;
1064}
1065
1066static ssize_t size_store(struct device *dev, struct device_attribute *attr,
1067		const char *buf, size_t len)
1068{
1069	ssize_t rc;
1070	unsigned long long val;
1071	struct dev_dax *dev_dax = to_dev_dax(dev);
1072	struct dax_region *dax_region = dev_dax->region;
1073
1074	rc = kstrtoull(buf, 0, &val);
1075	if (rc)
1076		return rc;
1077
1078	if (!alloc_is_aligned(dev_dax, val)) {
1079		dev_dbg(dev, "%s: size: %lld misaligned\n", __func__, val);
1080		return -EINVAL;
1081	}
1082
1083	device_lock(dax_region->dev);
 
 
1084	if (!dax_region->dev->driver) {
1085		device_unlock(dax_region->dev);
1086		return -ENXIO;
1087	}
1088	device_lock(dev);
 
 
 
1089	rc = dev_dax_resize(dax_region, dev_dax, val);
1090	device_unlock(dev);
1091	device_unlock(dax_region->dev);
1092
1093	return rc == 0 ? len : rc;
 
 
 
 
 
 
 
1094}
1095static DEVICE_ATTR_RW(size);
1096
1097static ssize_t range_parse(const char *opt, size_t len, struct range *range)
1098{
1099	unsigned long long addr = 0;
1100	char *start, *end, *str;
1101	ssize_t rc = -EINVAL;
1102
1103	str = kstrdup(opt, GFP_KERNEL);
1104	if (!str)
1105		return rc;
1106
1107	end = str;
1108	start = strsep(&end, "-");
1109	if (!start || !end)
1110		goto err;
1111
1112	rc = kstrtoull(start, 16, &addr);
1113	if (rc)
1114		goto err;
1115	range->start = addr;
1116
1117	rc = kstrtoull(end, 16, &addr);
1118	if (rc)
1119		goto err;
1120	range->end = addr;
1121
1122err:
1123	kfree(str);
1124	return rc;
1125}
1126
1127static ssize_t mapping_store(struct device *dev, struct device_attribute *attr,
1128		const char *buf, size_t len)
1129{
1130	struct dev_dax *dev_dax = to_dev_dax(dev);
1131	struct dax_region *dax_region = dev_dax->region;
1132	size_t to_alloc;
1133	struct range r;
1134	ssize_t rc;
1135
1136	rc = range_parse(buf, len, &r);
1137	if (rc)
1138		return rc;
1139
1140	rc = -ENXIO;
1141	device_lock(dax_region->dev);
 
1142	if (!dax_region->dev->driver) {
1143		device_unlock(dax_region->dev);
 
 
 
 
 
1144		return rc;
1145	}
1146	device_lock(dev);
1147
1148	to_alloc = range_len(&r);
1149	if (alloc_is_aligned(dev_dax, to_alloc))
1150		rc = alloc_dev_dax_range(dev_dax, r.start, to_alloc);
1151	device_unlock(dev);
1152	device_unlock(dax_region->dev);
1153
1154	return rc == 0 ? len : rc;
1155}
1156static DEVICE_ATTR_WO(mapping);
1157
1158static ssize_t align_show(struct device *dev,
1159		struct device_attribute *attr, char *buf)
1160{
1161	struct dev_dax *dev_dax = to_dev_dax(dev);
1162
1163	return sprintf(buf, "%d\n", dev_dax->align);
1164}
1165
1166static ssize_t dev_dax_validate_align(struct dev_dax *dev_dax)
1167{
1168	struct device *dev = &dev_dax->dev;
1169	int i;
1170
1171	for (i = 0; i < dev_dax->nr_range; i++) {
1172		size_t len = range_len(&dev_dax->ranges[i].range);
1173
1174		if (!alloc_is_aligned(dev_dax, len)) {
1175			dev_dbg(dev, "%s: align %u invalid for range %d\n",
1176				__func__, dev_dax->align, i);
1177			return -EINVAL;
1178		}
1179	}
1180
1181	return 0;
1182}
1183
1184static ssize_t align_store(struct device *dev, struct device_attribute *attr,
1185		const char *buf, size_t len)
1186{
1187	struct dev_dax *dev_dax = to_dev_dax(dev);
1188	struct dax_region *dax_region = dev_dax->region;
1189	unsigned long val, align_save;
1190	ssize_t rc;
1191
1192	rc = kstrtoul(buf, 0, &val);
1193	if (rc)
1194		return -ENXIO;
1195
1196	if (!dax_align_valid(val))
1197		return -EINVAL;
1198
1199	device_lock(dax_region->dev);
 
 
1200	if (!dax_region->dev->driver) {
1201		device_unlock(dax_region->dev);
1202		return -ENXIO;
1203	}
1204
1205	device_lock(dev);
 
 
 
 
1206	if (dev->driver) {
1207		rc = -EBUSY;
1208		goto out_unlock;
1209	}
1210
1211	align_save = dev_dax->align;
1212	dev_dax->align = val;
1213	rc = dev_dax_validate_align(dev_dax);
1214	if (rc)
1215		dev_dax->align = align_save;
1216out_unlock:
1217	device_unlock(dev);
1218	device_unlock(dax_region->dev);
1219	return rc == 0 ? len : rc;
1220}
1221static DEVICE_ATTR_RW(align);
1222
1223static int dev_dax_target_node(struct dev_dax *dev_dax)
1224{
1225	struct dax_region *dax_region = dev_dax->region;
1226
1227	return dax_region->target_node;
1228}
1229
1230static ssize_t target_node_show(struct device *dev,
1231		struct device_attribute *attr, char *buf)
1232{
1233	struct dev_dax *dev_dax = to_dev_dax(dev);
1234
1235	return sprintf(buf, "%d\n", dev_dax_target_node(dev_dax));
1236}
1237static DEVICE_ATTR_RO(target_node);
1238
1239static ssize_t resource_show(struct device *dev,
1240		struct device_attribute *attr, char *buf)
1241{
1242	struct dev_dax *dev_dax = to_dev_dax(dev);
1243	struct dax_region *dax_region = dev_dax->region;
1244	unsigned long long start;
1245
1246	if (dev_dax->nr_range < 1)
1247		start = dax_region->res.start;
1248	else
1249		start = dev_dax->ranges[0].range.start;
1250
1251	return sprintf(buf, "%#llx\n", start);
1252}
1253static DEVICE_ATTR(resource, 0400, resource_show, NULL);
1254
1255static ssize_t modalias_show(struct device *dev, struct device_attribute *attr,
1256		char *buf)
1257{
1258	/*
1259	 * We only ever expect to handle device-dax instances, i.e. the
1260	 * @type argument to MODULE_ALIAS_DAX_DEVICE() is always zero
1261	 */
1262	return sprintf(buf, DAX_DEVICE_MODALIAS_FMT "\n", 0);
1263}
1264static DEVICE_ATTR_RO(modalias);
1265
1266static ssize_t numa_node_show(struct device *dev,
1267		struct device_attribute *attr, char *buf)
1268{
1269	return sprintf(buf, "%d\n", dev_to_node(dev));
1270}
1271static DEVICE_ATTR_RO(numa_node);
1272
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1273static umode_t dev_dax_visible(struct kobject *kobj, struct attribute *a, int n)
1274{
1275	struct device *dev = container_of(kobj, struct device, kobj);
1276	struct dev_dax *dev_dax = to_dev_dax(dev);
1277	struct dax_region *dax_region = dev_dax->region;
1278
1279	if (a == &dev_attr_target_node.attr && dev_dax_target_node(dev_dax) < 0)
1280		return 0;
1281	if (a == &dev_attr_numa_node.attr && !IS_ENABLED(CONFIG_NUMA))
1282		return 0;
1283	if (a == &dev_attr_mapping.attr && is_static(dax_region))
1284		return 0;
1285	if ((a == &dev_attr_align.attr ||
1286	     a == &dev_attr_size.attr) && is_static(dax_region))
1287		return 0444;
1288	return a->mode;
1289}
1290
1291static struct attribute *dev_dax_attributes[] = {
1292	&dev_attr_modalias.attr,
1293	&dev_attr_size.attr,
1294	&dev_attr_mapping.attr,
1295	&dev_attr_target_node.attr,
1296	&dev_attr_align.attr,
1297	&dev_attr_resource.attr,
1298	&dev_attr_numa_node.attr,
 
1299	NULL,
1300};
1301
1302static const struct attribute_group dev_dax_attribute_group = {
1303	.attrs = dev_dax_attributes,
1304	.is_visible = dev_dax_visible,
1305};
1306
1307static const struct attribute_group *dax_attribute_groups[] = {
1308	&dev_dax_attribute_group,
1309	NULL,
1310};
1311
1312static void dev_dax_release(struct device *dev)
1313{
1314	struct dev_dax *dev_dax = to_dev_dax(dev);
1315	struct dax_device *dax_dev = dev_dax->dax_dev;
1316
1317	put_dax(dax_dev);
1318	free_dev_dax_id(dev_dax);
1319	kfree(dev_dax->pgmap);
1320	kfree(dev_dax);
1321}
1322
1323static const struct device_type dev_dax_type = {
1324	.release = dev_dax_release,
1325	.groups = dax_attribute_groups,
1326};
1327
1328struct dev_dax *devm_create_dev_dax(struct dev_dax_data *data)
1329{
1330	struct dax_region *dax_region = data->dax_region;
1331	struct device *parent = dax_region->dev;
1332	struct dax_device *dax_dev;
1333	struct dev_dax *dev_dax;
1334	struct inode *inode;
1335	struct device *dev;
1336	int rc;
1337
1338	dev_dax = kzalloc(sizeof(*dev_dax), GFP_KERNEL);
1339	if (!dev_dax)
1340		return ERR_PTR(-ENOMEM);
1341
1342	dev_dax->region = dax_region;
1343	if (is_static(dax_region)) {
1344		if (dev_WARN_ONCE(parent, data->id < 0,
1345				"dynamic id specified to static region\n")) {
1346			rc = -EINVAL;
1347			goto err_id;
1348		}
1349
1350		dev_dax->id = data->id;
1351	} else {
1352		if (dev_WARN_ONCE(parent, data->id >= 0,
1353				"static id specified to dynamic region\n")) {
1354			rc = -EINVAL;
1355			goto err_id;
1356		}
1357
1358		rc = alloc_dev_dax_id(dev_dax);
1359		if (rc < 0)
1360			goto err_id;
1361	}
1362
1363	dev = &dev_dax->dev;
1364	device_initialize(dev);
1365	dev_set_name(dev, "dax%d.%d", dax_region->id, dev_dax->id);
1366
1367	rc = alloc_dev_dax_range(dev_dax, dax_region->res.start, data->size);
1368	if (rc)
1369		goto err_range;
1370
1371	if (data->pgmap) {
1372		dev_WARN_ONCE(parent, !is_static(dax_region),
1373			"custom dev_pagemap requires a static dax_region\n");
1374
1375		dev_dax->pgmap = kmemdup(data->pgmap,
1376				sizeof(struct dev_pagemap), GFP_KERNEL);
1377		if (!dev_dax->pgmap) {
1378			rc = -ENOMEM;
1379			goto err_pgmap;
1380		}
1381	}
1382
1383	/*
1384	 * No dax_operations since there is no access to this device outside of
1385	 * mmap of the resulting character device.
1386	 */
1387	dax_dev = alloc_dax(dev_dax, NULL);
1388	if (IS_ERR(dax_dev)) {
1389		rc = PTR_ERR(dax_dev);
1390		goto err_alloc_dax;
1391	}
1392	set_dax_synchronous(dax_dev);
1393	set_dax_nocache(dax_dev);
1394	set_dax_nomc(dax_dev);
1395
1396	/* a device_dax instance is dead while the driver is not attached */
1397	kill_dax(dax_dev);
1398
1399	dev_dax->dax_dev = dax_dev;
1400	dev_dax->target_node = dax_region->target_node;
1401	dev_dax->align = dax_region->align;
1402	ida_init(&dev_dax->ida);
1403
1404	dev_dax->memmap_on_memory = data->memmap_on_memory;
1405
1406	inode = dax_inode(dax_dev);
1407	dev->devt = inode->i_rdev;
1408	dev->bus = &dax_bus_type;
1409	dev->parent = parent;
1410	dev->type = &dev_dax_type;
1411
1412	rc = device_add(dev);
1413	if (rc) {
1414		kill_dev_dax(dev_dax);
1415		put_device(dev);
1416		return ERR_PTR(rc);
1417	}
1418
1419	rc = devm_add_action_or_reset(dax_region->dev, unregister_dev_dax, dev);
1420	if (rc)
1421		return ERR_PTR(rc);
1422
1423	/* register mapping device for the initial allocation range */
1424	if (dev_dax->nr_range && range_len(&dev_dax->ranges[0].range)) {
1425		rc = devm_register_dax_mapping(dev_dax, 0);
1426		if (rc)
1427			return ERR_PTR(rc);
1428	}
1429
1430	return dev_dax;
1431
1432err_alloc_dax:
1433	kfree(dev_dax->pgmap);
1434err_pgmap:
1435	free_dev_dax_ranges(dev_dax);
1436err_range:
1437	free_dev_dax_id(dev_dax);
1438err_id:
1439	kfree(dev_dax);
1440
1441	return ERR_PTR(rc);
 
 
 
 
 
 
 
 
 
 
 
1442}
1443EXPORT_SYMBOL_GPL(devm_create_dev_dax);
1444
1445int __dax_driver_register(struct dax_device_driver *dax_drv,
1446		struct module *module, const char *mod_name)
1447{
1448	struct device_driver *drv = &dax_drv->drv;
1449
1450	/*
1451	 * dax_bus_probe() calls dax_drv->probe() unconditionally.
1452	 * So better be safe than sorry and ensure it is provided.
1453	 */
1454	if (!dax_drv->probe)
1455		return -EINVAL;
1456
1457	INIT_LIST_HEAD(&dax_drv->ids);
1458	drv->owner = module;
1459	drv->name = mod_name;
1460	drv->mod_name = mod_name;
1461	drv->bus = &dax_bus_type;
1462
1463	return driver_register(drv);
1464}
1465EXPORT_SYMBOL_GPL(__dax_driver_register);
1466
1467void dax_driver_unregister(struct dax_device_driver *dax_drv)
1468{
1469	struct device_driver *drv = &dax_drv->drv;
1470	struct dax_id *dax_id, *_id;
1471
1472	mutex_lock(&dax_bus_lock);
1473	list_for_each_entry_safe(dax_id, _id, &dax_drv->ids, list) {
1474		list_del(&dax_id->list);
1475		kfree(dax_id);
1476	}
1477	mutex_unlock(&dax_bus_lock);
1478	driver_unregister(drv);
1479}
1480EXPORT_SYMBOL_GPL(dax_driver_unregister);
1481
1482int __init dax_bus_init(void)
1483{
1484	return bus_register(&dax_bus_type);
1485}
1486
1487void __exit dax_bus_exit(void)
1488{
1489	bus_unregister(&dax_bus_type);
1490}
v6.13.7
   1// SPDX-License-Identifier: GPL-2.0
   2/* Copyright(c) 2017-2018 Intel Corporation. All rights reserved. */
   3#include <linux/memremap.h>
   4#include <linux/device.h>
   5#include <linux/mutex.h>
   6#include <linux/list.h>
   7#include <linux/slab.h>
   8#include <linux/dax.h>
   9#include <linux/io.h>
  10#include "dax-private.h"
  11#include "bus.h"
  12
  13static DEFINE_MUTEX(dax_bus_lock);
  14
  15/*
  16 * All changes to the dax region configuration occur with this lock held
  17 * for write.
  18 */
  19DECLARE_RWSEM(dax_region_rwsem);
  20
  21/*
  22 * All changes to the dax device configuration occur with this lock held
  23 * for write.
  24 */
  25DECLARE_RWSEM(dax_dev_rwsem);
  26
  27#define DAX_NAME_LEN 30
  28struct dax_id {
  29	struct list_head list;
  30	char dev_name[DAX_NAME_LEN];
  31};
  32
  33static int dax_bus_uevent(const struct device *dev, struct kobj_uevent_env *env)
  34{
  35	/*
  36	 * We only ever expect to handle device-dax instances, i.e. the
  37	 * @type argument to MODULE_ALIAS_DAX_DEVICE() is always zero
  38	 */
  39	return add_uevent_var(env, "MODALIAS=" DAX_DEVICE_MODALIAS_FMT, 0);
  40}
  41
  42#define to_dax_drv(__drv)	container_of_const(__drv, struct dax_device_driver, drv)
 
 
 
  43
  44static struct dax_id *__dax_match_id(const struct dax_device_driver *dax_drv,
  45		const char *dev_name)
  46{
  47	struct dax_id *dax_id;
  48
  49	lockdep_assert_held(&dax_bus_lock);
  50
  51	list_for_each_entry(dax_id, &dax_drv->ids, list)
  52		if (sysfs_streq(dax_id->dev_name, dev_name))
  53			return dax_id;
  54	return NULL;
  55}
  56
  57static int dax_match_id(const struct dax_device_driver *dax_drv, struct device *dev)
  58{
  59	int match;
  60
  61	mutex_lock(&dax_bus_lock);
  62	match = !!__dax_match_id(dax_drv, dev_name(dev));
  63	mutex_unlock(&dax_bus_lock);
  64
  65	return match;
  66}
  67
  68static int dax_match_type(const struct dax_device_driver *dax_drv, struct device *dev)
  69{
  70	enum dax_driver_type type = DAXDRV_DEVICE_TYPE;
  71	struct dev_dax *dev_dax = to_dev_dax(dev);
  72
  73	if (dev_dax->region->res.flags & IORESOURCE_DAX_KMEM)
  74		type = DAXDRV_KMEM_TYPE;
  75
  76	if (dax_drv->type == type)
  77		return 1;
  78
  79	/* default to device mode if dax_kmem is disabled */
  80	if (dax_drv->type == DAXDRV_DEVICE_TYPE &&
  81	    !IS_ENABLED(CONFIG_DEV_DAX_KMEM))
  82		return 1;
  83
  84	return 0;
  85}
  86
  87enum id_action {
  88	ID_REMOVE,
  89	ID_ADD,
  90};
  91
  92static ssize_t do_id_store(struct device_driver *drv, const char *buf,
  93		size_t count, enum id_action action)
  94{
  95	struct dax_device_driver *dax_drv = to_dax_drv(drv);
  96	unsigned int region_id, id;
  97	char devname[DAX_NAME_LEN];
  98	struct dax_id *dax_id;
  99	ssize_t rc = count;
 100	int fields;
 101
 102	fields = sscanf(buf, "dax%d.%d", &region_id, &id);
 103	if (fields != 2)
 104		return -EINVAL;
 105	sprintf(devname, "dax%d.%d", region_id, id);
 106	if (!sysfs_streq(buf, devname))
 107		return -EINVAL;
 108
 109	mutex_lock(&dax_bus_lock);
 110	dax_id = __dax_match_id(dax_drv, buf);
 111	if (!dax_id) {
 112		if (action == ID_ADD) {
 113			dax_id = kzalloc(sizeof(*dax_id), GFP_KERNEL);
 114			if (dax_id) {
 115				strscpy(dax_id->dev_name, buf, DAX_NAME_LEN);
 116				list_add(&dax_id->list, &dax_drv->ids);
 117			} else
 118				rc = -ENOMEM;
 119		}
 120	} else if (action == ID_REMOVE) {
 121		list_del(&dax_id->list);
 122		kfree(dax_id);
 123	}
 124	mutex_unlock(&dax_bus_lock);
 125
 126	if (rc < 0)
 127		return rc;
 128	if (action == ID_ADD)
 129		rc = driver_attach(drv);
 130	if (rc)
 131		return rc;
 132	return count;
 133}
 134
 135static ssize_t new_id_store(struct device_driver *drv, const char *buf,
 136		size_t count)
 137{
 138	return do_id_store(drv, buf, count, ID_ADD);
 139}
 140static DRIVER_ATTR_WO(new_id);
 141
 142static ssize_t remove_id_store(struct device_driver *drv, const char *buf,
 143		size_t count)
 144{
 145	return do_id_store(drv, buf, count, ID_REMOVE);
 146}
 147static DRIVER_ATTR_WO(remove_id);
 148
 149static struct attribute *dax_drv_attrs[] = {
 150	&driver_attr_new_id.attr,
 151	&driver_attr_remove_id.attr,
 152	NULL,
 153};
 154ATTRIBUTE_GROUPS(dax_drv);
 155
 156static int dax_bus_match(struct device *dev, const struct device_driver *drv);
 157
 158/*
 159 * Static dax regions are regions created by an external subsystem
 160 * nvdimm where a single range is assigned. Its boundaries are by the external
 161 * subsystem and are usually limited to one physical memory range. For example,
 162 * for PMEM it is usually defined by NVDIMM Namespace boundaries (i.e. a
 163 * single contiguous range)
 164 *
 165 * On dynamic dax regions, the assigned region can be partitioned by dax core
 166 * into multiple subdivisions. A subdivision is represented into one
 167 * /dev/daxN.M device composed by one or more potentially discontiguous ranges.
 168 *
 169 * When allocating a dax region, drivers must set whether it's static
 170 * (IORESOURCE_DAX_STATIC).  On static dax devices, the @pgmap is pre-assigned
 171 * to dax core when calling devm_create_dev_dax(), whereas in dynamic dax
 172 * devices it is NULL but afterwards allocated by dax core on device ->probe().
 173 * Care is needed to make sure that dynamic dax devices are torn down with a
 174 * cleared @pgmap field (see kill_dev_dax()).
 175 */
 176static bool is_static(struct dax_region *dax_region)
 177{
 178	return (dax_region->res.flags & IORESOURCE_DAX_STATIC) != 0;
 179}
 180
 181bool static_dev_dax(struct dev_dax *dev_dax)
 182{
 183	return is_static(dev_dax->region);
 184}
 185EXPORT_SYMBOL_GPL(static_dev_dax);
 186
 187static u64 dev_dax_size(struct dev_dax *dev_dax)
 188{
 189	u64 size = 0;
 190	int i;
 191
 192	lockdep_assert_held(&dax_dev_rwsem);
 193
 194	for (i = 0; i < dev_dax->nr_range; i++)
 195		size += range_len(&dev_dax->ranges[i].range);
 196
 197	return size;
 198}
 199
 200static int dax_bus_probe(struct device *dev)
 201{
 202	struct dax_device_driver *dax_drv = to_dax_drv(dev->driver);
 203	struct dev_dax *dev_dax = to_dev_dax(dev);
 204	struct dax_region *dax_region = dev_dax->region;
 205	int rc;
 206	u64 size;
 207
 208	rc = down_read_interruptible(&dax_dev_rwsem);
 209	if (rc)
 210		return rc;
 211	size = dev_dax_size(dev_dax);
 212	up_read(&dax_dev_rwsem);
 213
 214	if (size == 0 || dev_dax->id < 0)
 215		return -ENXIO;
 216
 217	rc = dax_drv->probe(dev_dax);
 218
 219	if (rc || is_static(dax_region))
 220		return rc;
 221
 222	/*
 223	 * Track new seed creation only after successful probe of the
 224	 * previous seed.
 225	 */
 226	if (dax_region->seed == dev)
 227		dax_region->seed = NULL;
 228
 229	return 0;
 230}
 231
 232static void dax_bus_remove(struct device *dev)
 233{
 234	struct dax_device_driver *dax_drv = to_dax_drv(dev->driver);
 235	struct dev_dax *dev_dax = to_dev_dax(dev);
 236
 237	if (dax_drv->remove)
 238		dax_drv->remove(dev_dax);
 239}
 240
 241static const struct bus_type dax_bus_type = {
 242	.name = "dax",
 243	.uevent = dax_bus_uevent,
 244	.match = dax_bus_match,
 245	.probe = dax_bus_probe,
 246	.remove = dax_bus_remove,
 247	.drv_groups = dax_drv_groups,
 248};
 249
 250static int dax_bus_match(struct device *dev, const struct device_driver *drv)
 251{
 252	const struct dax_device_driver *dax_drv = to_dax_drv(drv);
 253
 254	if (dax_match_id(dax_drv, dev))
 255		return 1;
 256	return dax_match_type(dax_drv, dev);
 257}
 258
 259/*
 260 * Rely on the fact that drvdata is set before the attributes are
 261 * registered, and that the attributes are unregistered before drvdata
 262 * is cleared to assume that drvdata is always valid.
 263 */
 264static ssize_t id_show(struct device *dev,
 265		struct device_attribute *attr, char *buf)
 266{
 267	struct dax_region *dax_region = dev_get_drvdata(dev);
 268
 269	return sysfs_emit(buf, "%d\n", dax_region->id);
 270}
 271static DEVICE_ATTR_RO(id);
 272
 273static ssize_t region_size_show(struct device *dev,
 274		struct device_attribute *attr, char *buf)
 275{
 276	struct dax_region *dax_region = dev_get_drvdata(dev);
 277
 278	return sysfs_emit(buf, "%llu\n",
 279			  (unsigned long long)resource_size(&dax_region->res));
 280}
 281static struct device_attribute dev_attr_region_size = __ATTR(size, 0444,
 282		region_size_show, NULL);
 283
 284static ssize_t region_align_show(struct device *dev,
 285		struct device_attribute *attr, char *buf)
 286{
 287	struct dax_region *dax_region = dev_get_drvdata(dev);
 288
 289	return sysfs_emit(buf, "%u\n", dax_region->align);
 290}
 291static struct device_attribute dev_attr_region_align =
 292		__ATTR(align, 0400, region_align_show, NULL);
 293
 294#define for_each_dax_region_resource(dax_region, res) \
 295	for (res = (dax_region)->res.child; res; res = res->sibling)
 296
 297static unsigned long long dax_region_avail_size(struct dax_region *dax_region)
 298{
 299	resource_size_t size = resource_size(&dax_region->res);
 300	struct resource *res;
 301
 302	lockdep_assert_held(&dax_region_rwsem);
 303
 304	for_each_dax_region_resource(dax_region, res)
 305		size -= resource_size(res);
 306	return size;
 307}
 308
 309static ssize_t available_size_show(struct device *dev,
 310		struct device_attribute *attr, char *buf)
 311{
 312	struct dax_region *dax_region = dev_get_drvdata(dev);
 313	unsigned long long size;
 314	int rc;
 315
 316	rc = down_read_interruptible(&dax_region_rwsem);
 317	if (rc)
 318		return rc;
 319	size = dax_region_avail_size(dax_region);
 320	up_read(&dax_region_rwsem);
 321
 322	return sysfs_emit(buf, "%llu\n", size);
 323}
 324static DEVICE_ATTR_RO(available_size);
 325
 326static ssize_t seed_show(struct device *dev,
 327		struct device_attribute *attr, char *buf)
 328{
 329	struct dax_region *dax_region = dev_get_drvdata(dev);
 330	struct device *seed;
 331	ssize_t rc;
 332
 333	if (is_static(dax_region))
 334		return -EINVAL;
 335
 336	rc = down_read_interruptible(&dax_region_rwsem);
 337	if (rc)
 338		return rc;
 339	seed = dax_region->seed;
 340	rc = sysfs_emit(buf, "%s\n", seed ? dev_name(seed) : "");
 341	up_read(&dax_region_rwsem);
 342
 343	return rc;
 344}
 345static DEVICE_ATTR_RO(seed);
 346
 347static ssize_t create_show(struct device *dev,
 348		struct device_attribute *attr, char *buf)
 349{
 350	struct dax_region *dax_region = dev_get_drvdata(dev);
 351	struct device *youngest;
 352	ssize_t rc;
 353
 354	if (is_static(dax_region))
 355		return -EINVAL;
 356
 357	rc = down_read_interruptible(&dax_region_rwsem);
 358	if (rc)
 359		return rc;
 360	youngest = dax_region->youngest;
 361	rc = sysfs_emit(buf, "%s\n", youngest ? dev_name(youngest) : "");
 362	up_read(&dax_region_rwsem);
 363
 364	return rc;
 365}
 366
 367static struct dev_dax *__devm_create_dev_dax(struct dev_dax_data *data);
 368
 369static ssize_t create_store(struct device *dev, struct device_attribute *attr,
 370		const char *buf, size_t len)
 371{
 372	struct dax_region *dax_region = dev_get_drvdata(dev);
 373	unsigned long long avail;
 374	ssize_t rc;
 375	int val;
 376
 377	if (is_static(dax_region))
 378		return -EINVAL;
 379
 380	rc = kstrtoint(buf, 0, &val);
 381	if (rc)
 382		return rc;
 383	if (val != 1)
 384		return -EINVAL;
 385
 386	rc = down_write_killable(&dax_region_rwsem);
 387	if (rc)
 388		return rc;
 389	avail = dax_region_avail_size(dax_region);
 390	if (avail == 0)
 391		rc = -ENOSPC;
 392	else {
 393		struct dev_dax_data data = {
 394			.dax_region = dax_region,
 395			.size = 0,
 396			.id = -1,
 397			.memmap_on_memory = false,
 398		};
 399		struct dev_dax *dev_dax = __devm_create_dev_dax(&data);
 400
 401		if (IS_ERR(dev_dax))
 402			rc = PTR_ERR(dev_dax);
 403		else {
 404			/*
 405			 * In support of crafting multiple new devices
 406			 * simultaneously multiple seeds can be created,
 407			 * but only the first one that has not been
 408			 * successfully bound is tracked as the region
 409			 * seed.
 410			 */
 411			if (!dax_region->seed)
 412				dax_region->seed = &dev_dax->dev;
 413			dax_region->youngest = &dev_dax->dev;
 414			rc = len;
 415		}
 416	}
 417	up_write(&dax_region_rwsem);
 418
 419	return rc;
 420}
 421static DEVICE_ATTR_RW(create);
 422
 423void kill_dev_dax(struct dev_dax *dev_dax)
 424{
 425	struct dax_device *dax_dev = dev_dax->dax_dev;
 426	struct inode *inode = dax_inode(dax_dev);
 427
 428	kill_dax(dax_dev);
 429	unmap_mapping_range(inode->i_mapping, 0, 0, 1);
 430
 431	/*
 432	 * Dynamic dax region have the pgmap allocated via dev_kzalloc()
 433	 * and thus freed by devm. Clear the pgmap to not have stale pgmap
 434	 * ranges on probe() from previous reconfigurations of region devices.
 435	 */
 436	if (!static_dev_dax(dev_dax))
 437		dev_dax->pgmap = NULL;
 438}
 439EXPORT_SYMBOL_GPL(kill_dev_dax);
 440
 441static void trim_dev_dax_range(struct dev_dax *dev_dax)
 442{
 443	int i = dev_dax->nr_range - 1;
 444	struct range *range = &dev_dax->ranges[i].range;
 445	struct dax_region *dax_region = dev_dax->region;
 446
 447	lockdep_assert_held_write(&dax_region_rwsem);
 448	dev_dbg(&dev_dax->dev, "delete range[%d]: %#llx:%#llx\n", i,
 449		(unsigned long long)range->start,
 450		(unsigned long long)range->end);
 451
 452	__release_region(&dax_region->res, range->start, range_len(range));
 453	if (--dev_dax->nr_range == 0) {
 454		kfree(dev_dax->ranges);
 455		dev_dax->ranges = NULL;
 456	}
 457}
 458
 459static void free_dev_dax_ranges(struct dev_dax *dev_dax)
 460{
 461	while (dev_dax->nr_range)
 462		trim_dev_dax_range(dev_dax);
 463}
 464
 465static void unregister_dev_dax(void *dev)
 466{
 467	struct dev_dax *dev_dax = to_dev_dax(dev);
 468
 469	dev_dbg(dev, "%s\n", __func__);
 470
 471	down_write(&dax_region_rwsem);
 472	kill_dev_dax(dev_dax);
 473	device_del(dev);
 474	free_dev_dax_ranges(dev_dax);
 475	put_device(dev);
 476	up_write(&dax_region_rwsem);
 477}
 478
 479static void dax_region_free(struct kref *kref)
 480{
 481	struct dax_region *dax_region;
 482
 483	dax_region = container_of(kref, struct dax_region, kref);
 484	kfree(dax_region);
 485}
 486
 487static void dax_region_put(struct dax_region *dax_region)
 488{
 489	kref_put(&dax_region->kref, dax_region_free);
 490}
 491
 492/* a return value >= 0 indicates this invocation invalidated the id */
 493static int __free_dev_dax_id(struct dev_dax *dev_dax)
 494{
 
 495	struct dax_region *dax_region;
 496	int rc = dev_dax->id;
 497
 498	lockdep_assert_held_write(&dax_dev_rwsem);
 499
 500	if (!dev_dax->dyn_id || dev_dax->id < 0)
 501		return -1;
 502	dax_region = dev_dax->region;
 503	ida_free(&dax_region->ida, dev_dax->id);
 504	dax_region_put(dax_region);
 505	dev_dax->id = -1;
 506	return rc;
 507}
 508
 509static int free_dev_dax_id(struct dev_dax *dev_dax)
 510{
 
 511	int rc;
 512
 513	rc = down_write_killable(&dax_dev_rwsem);
 514	if (rc)
 515		return rc;
 516	rc = __free_dev_dax_id(dev_dax);
 517	up_write(&dax_dev_rwsem);
 518	return rc;
 519}
 520
 521static int alloc_dev_dax_id(struct dev_dax *dev_dax)
 522{
 523	struct dax_region *dax_region = dev_dax->region;
 524	int id;
 525
 526	id = ida_alloc(&dax_region->ida, GFP_KERNEL);
 527	if (id < 0)
 528		return id;
 529	kref_get(&dax_region->kref);
 530	dev_dax->dyn_id = true;
 531	dev_dax->id = id;
 532	return id;
 533}
 534
 535static ssize_t delete_store(struct device *dev, struct device_attribute *attr,
 536		const char *buf, size_t len)
 537{
 538	struct dax_region *dax_region = dev_get_drvdata(dev);
 539	struct dev_dax *dev_dax;
 540	struct device *victim;
 541	bool do_del = false;
 542	int rc;
 543
 544	if (is_static(dax_region))
 545		return -EINVAL;
 546
 547	victim = device_find_child_by_name(dax_region->dev, buf);
 548	if (!victim)
 549		return -ENXIO;
 550
 551	device_lock(dev);
 552	device_lock(victim);
 553	dev_dax = to_dev_dax(victim);
 554	down_write(&dax_dev_rwsem);
 555	if (victim->driver || dev_dax_size(dev_dax))
 556		rc = -EBUSY;
 557	else {
 558		/*
 559		 * Invalidate the device so it does not become active
 560		 * again, but always preserve device-id-0 so that
 561		 * /sys/bus/dax/ is guaranteed to be populated while any
 562		 * dax_region is registered.
 563		 */
 564		if (dev_dax->id > 0) {
 565			do_del = __free_dev_dax_id(dev_dax) >= 0;
 566			rc = len;
 567			if (dax_region->seed == victim)
 568				dax_region->seed = NULL;
 569			if (dax_region->youngest == victim)
 570				dax_region->youngest = NULL;
 571		} else
 572			rc = -EBUSY;
 573	}
 574	up_write(&dax_dev_rwsem);
 575	device_unlock(victim);
 576
 577	/* won the race to invalidate the device, clean it up */
 578	if (do_del)
 579		devm_release_action(dev, unregister_dev_dax, victim);
 580	device_unlock(dev);
 581	put_device(victim);
 582
 583	return rc;
 584}
 585static DEVICE_ATTR_WO(delete);
 586
 587static umode_t dax_region_visible(struct kobject *kobj, struct attribute *a,
 588		int n)
 589{
 590	struct device *dev = container_of(kobj, struct device, kobj);
 591	struct dax_region *dax_region = dev_get_drvdata(dev);
 592
 593	if (is_static(dax_region))
 594		if (a == &dev_attr_available_size.attr
 595				|| a == &dev_attr_create.attr
 596				|| a == &dev_attr_seed.attr
 597				|| a == &dev_attr_delete.attr)
 598			return 0;
 599	return a->mode;
 600}
 601
 602static struct attribute *dax_region_attributes[] = {
 603	&dev_attr_available_size.attr,
 604	&dev_attr_region_size.attr,
 605	&dev_attr_region_align.attr,
 606	&dev_attr_create.attr,
 607	&dev_attr_seed.attr,
 608	&dev_attr_delete.attr,
 609	&dev_attr_id.attr,
 610	NULL,
 611};
 612
 613static const struct attribute_group dax_region_attribute_group = {
 614	.name = "dax_region",
 615	.attrs = dax_region_attributes,
 616	.is_visible = dax_region_visible,
 617};
 618
 619static const struct attribute_group *dax_region_attribute_groups[] = {
 620	&dax_region_attribute_group,
 621	NULL,
 622};
 623
 624static void dax_region_unregister(void *region)
 625{
 626	struct dax_region *dax_region = region;
 627
 628	sysfs_remove_groups(&dax_region->dev->kobj,
 629			dax_region_attribute_groups);
 630	dax_region_put(dax_region);
 631}
 632
 633struct dax_region *alloc_dax_region(struct device *parent, int region_id,
 634		struct range *range, int target_node, unsigned int align,
 635		unsigned long flags)
 636{
 637	struct dax_region *dax_region;
 638
 639	/*
 640	 * The DAX core assumes that it can store its private data in
 641	 * parent->driver_data. This WARN is a reminder / safeguard for
 642	 * developers of device-dax drivers.
 643	 */
 644	if (dev_get_drvdata(parent)) {
 645		dev_WARN(parent, "dax core failed to setup private data\n");
 646		return NULL;
 647	}
 648
 649	if (!IS_ALIGNED(range->start, align)
 650			|| !IS_ALIGNED(range_len(range), align))
 651		return NULL;
 652
 653	dax_region = kzalloc(sizeof(*dax_region), GFP_KERNEL);
 654	if (!dax_region)
 655		return NULL;
 656
 657	dev_set_drvdata(parent, dax_region);
 658	kref_init(&dax_region->kref);
 659	dax_region->id = region_id;
 660	dax_region->align = align;
 661	dax_region->dev = parent;
 662	dax_region->target_node = target_node;
 663	ida_init(&dax_region->ida);
 664	dax_region->res = (struct resource) {
 665		.start = range->start,
 666		.end = range->end,
 667		.flags = IORESOURCE_MEM | flags,
 668	};
 669
 670	if (sysfs_create_groups(&parent->kobj, dax_region_attribute_groups)) {
 671		kfree(dax_region);
 672		return NULL;
 673	}
 674
 675	if (devm_add_action_or_reset(parent, dax_region_unregister, dax_region))
 676		return NULL;
 677	return dax_region;
 678}
 679EXPORT_SYMBOL_GPL(alloc_dax_region);
 680
 681static void dax_mapping_release(struct device *dev)
 682{
 683	struct dax_mapping *mapping = to_dax_mapping(dev);
 684	struct device *parent = dev->parent;
 685	struct dev_dax *dev_dax = to_dev_dax(parent);
 686
 687	ida_free(&dev_dax->ida, mapping->id);
 688	kfree(mapping);
 689	put_device(parent);
 690}
 691
 692static void unregister_dax_mapping(void *data)
 693{
 694	struct device *dev = data;
 695	struct dax_mapping *mapping = to_dax_mapping(dev);
 696	struct dev_dax *dev_dax = to_dev_dax(dev->parent);
 
 697
 698	dev_dbg(dev, "%s\n", __func__);
 699
 
 
 700	dev_dax->ranges[mapping->range_id].mapping = NULL;
 701	mapping->range_id = -1;
 702
 703	device_unregister(dev);
 704}
 705
 706static struct dev_dax_range *get_dax_range(struct device *dev)
 707{
 708	struct dax_mapping *mapping = to_dax_mapping(dev);
 709	struct dev_dax *dev_dax = to_dev_dax(dev->parent);
 710	int rc;
 711
 712	rc = down_write_killable(&dax_region_rwsem);
 713	if (rc)
 714		return NULL;
 715	if (mapping->range_id < 0) {
 716		up_write(&dax_region_rwsem);
 717		return NULL;
 718	}
 719
 720	return &dev_dax->ranges[mapping->range_id];
 721}
 722
 723static void put_dax_range(void)
 724{
 725	up_write(&dax_region_rwsem);
 
 
 
 
 726}
 727
 728static ssize_t start_show(struct device *dev,
 729		struct device_attribute *attr, char *buf)
 730{
 731	struct dev_dax_range *dax_range;
 732	ssize_t rc;
 733
 734	dax_range = get_dax_range(dev);
 735	if (!dax_range)
 736		return -ENXIO;
 737	rc = sysfs_emit(buf, "%#llx\n", dax_range->range.start);
 738	put_dax_range();
 739
 740	return rc;
 741}
 742static DEVICE_ATTR(start, 0400, start_show, NULL);
 743
 744static ssize_t end_show(struct device *dev,
 745		struct device_attribute *attr, char *buf)
 746{
 747	struct dev_dax_range *dax_range;
 748	ssize_t rc;
 749
 750	dax_range = get_dax_range(dev);
 751	if (!dax_range)
 752		return -ENXIO;
 753	rc = sysfs_emit(buf, "%#llx\n", dax_range->range.end);
 754	put_dax_range();
 755
 756	return rc;
 757}
 758static DEVICE_ATTR(end, 0400, end_show, NULL);
 759
 760static ssize_t pgoff_show(struct device *dev,
 761		struct device_attribute *attr, char *buf)
 762{
 763	struct dev_dax_range *dax_range;
 764	ssize_t rc;
 765
 766	dax_range = get_dax_range(dev);
 767	if (!dax_range)
 768		return -ENXIO;
 769	rc = sysfs_emit(buf, "%#lx\n", dax_range->pgoff);
 770	put_dax_range();
 771
 772	return rc;
 773}
 774static DEVICE_ATTR(page_offset, 0400, pgoff_show, NULL);
 775
 776static struct attribute *dax_mapping_attributes[] = {
 777	&dev_attr_start.attr,
 778	&dev_attr_end.attr,
 779	&dev_attr_page_offset.attr,
 780	NULL,
 781};
 782
 783static const struct attribute_group dax_mapping_attribute_group = {
 784	.attrs = dax_mapping_attributes,
 785};
 786
 787static const struct attribute_group *dax_mapping_attribute_groups[] = {
 788	&dax_mapping_attribute_group,
 789	NULL,
 790};
 791
 792static const struct device_type dax_mapping_type = {
 793	.release = dax_mapping_release,
 794	.groups = dax_mapping_attribute_groups,
 795};
 796
 797static int devm_register_dax_mapping(struct dev_dax *dev_dax, int range_id)
 798{
 799	struct dax_region *dax_region = dev_dax->region;
 800	struct dax_mapping *mapping;
 801	struct device *dev;
 802	int rc;
 803
 804	lockdep_assert_held_write(&dax_region_rwsem);
 805
 806	if (dev_WARN_ONCE(&dev_dax->dev, !dax_region->dev->driver,
 807				"region disabled\n"))
 808		return -ENXIO;
 809
 810	mapping = kzalloc(sizeof(*mapping), GFP_KERNEL);
 811	if (!mapping)
 812		return -ENOMEM;
 813	mapping->range_id = range_id;
 814	mapping->id = ida_alloc(&dev_dax->ida, GFP_KERNEL);
 815	if (mapping->id < 0) {
 816		kfree(mapping);
 817		return -ENOMEM;
 818	}
 819	dev_dax->ranges[range_id].mapping = mapping;
 820	dev = &mapping->dev;
 821	device_initialize(dev);
 822	dev->parent = &dev_dax->dev;
 823	get_device(dev->parent);
 824	dev->type = &dax_mapping_type;
 825	dev_set_name(dev, "mapping%d", mapping->id);
 826	rc = device_add(dev);
 827	if (rc) {
 828		put_device(dev);
 829		return rc;
 830	}
 831
 832	rc = devm_add_action_or_reset(dax_region->dev, unregister_dax_mapping,
 833			dev);
 834	if (rc)
 835		return rc;
 836	return 0;
 837}
 838
 839static int alloc_dev_dax_range(struct dev_dax *dev_dax, u64 start,
 840		resource_size_t size)
 841{
 842	struct dax_region *dax_region = dev_dax->region;
 843	struct resource *res = &dax_region->res;
 844	struct device *dev = &dev_dax->dev;
 845	struct dev_dax_range *ranges;
 846	unsigned long pgoff = 0;
 847	struct resource *alloc;
 848	int i, rc;
 849
 850	lockdep_assert_held_write(&dax_region_rwsem);
 851
 852	/* handle the seed alloc special case */
 853	if (!size) {
 854		if (dev_WARN_ONCE(dev, dev_dax->nr_range,
 855					"0-size allocation must be first\n"))
 856			return -EBUSY;
 857		/* nr_range == 0 is elsewhere special cased as 0-size device */
 858		return 0;
 859	}
 860
 861	alloc = __request_region(res, start, size, dev_name(dev), 0);
 862	if (!alloc)
 863		return -ENOMEM;
 864
 865	ranges = krealloc(dev_dax->ranges, sizeof(*ranges)
 866			* (dev_dax->nr_range + 1), GFP_KERNEL);
 867	if (!ranges) {
 868		__release_region(res, alloc->start, resource_size(alloc));
 869		return -ENOMEM;
 870	}
 871
 872	for (i = 0; i < dev_dax->nr_range; i++)
 873		pgoff += PHYS_PFN(range_len(&ranges[i].range));
 874	dev_dax->ranges = ranges;
 875	ranges[dev_dax->nr_range++] = (struct dev_dax_range) {
 876		.pgoff = pgoff,
 877		.range = {
 878			.start = alloc->start,
 879			.end = alloc->end,
 880		},
 881	};
 882
 883	dev_dbg(dev, "alloc range[%d]: %pa:%pa\n", dev_dax->nr_range - 1,
 884			&alloc->start, &alloc->end);
 885	/*
 886	 * A dev_dax instance must be registered before mapping device
 887	 * children can be added. Defer to devm_create_dev_dax() to add
 888	 * the initial mapping device.
 889	 */
 890	if (!device_is_registered(&dev_dax->dev))
 891		return 0;
 892
 893	rc = devm_register_dax_mapping(dev_dax, dev_dax->nr_range - 1);
 894	if (rc)
 895		trim_dev_dax_range(dev_dax);
 896
 897	return rc;
 898}
 899
 900static int adjust_dev_dax_range(struct dev_dax *dev_dax, struct resource *res, resource_size_t size)
 901{
 902	int last_range = dev_dax->nr_range - 1;
 903	struct dev_dax_range *dax_range = &dev_dax->ranges[last_range];
 
 904	bool is_shrink = resource_size(res) > size;
 905	struct range *range = &dax_range->range;
 906	struct device *dev = &dev_dax->dev;
 907	int rc;
 908
 909	lockdep_assert_held_write(&dax_region_rwsem);
 910
 911	if (dev_WARN_ONCE(dev, !size, "deletion is handled by dev_dax_shrink\n"))
 912		return -EINVAL;
 913
 914	rc = adjust_resource(res, range->start, size);
 915	if (rc)
 916		return rc;
 917
 918	*range = (struct range) {
 919		.start = range->start,
 920		.end = range->start + size - 1,
 921	};
 922
 923	dev_dbg(dev, "%s range[%d]: %#llx:%#llx\n", is_shrink ? "shrink" : "extend",
 924			last_range, (unsigned long long) range->start,
 925			(unsigned long long) range->end);
 926
 927	return 0;
 928}
 929
 930static ssize_t size_show(struct device *dev,
 931		struct device_attribute *attr, char *buf)
 932{
 933	struct dev_dax *dev_dax = to_dev_dax(dev);
 934	unsigned long long size;
 935	int rc;
 936
 937	rc = down_read_interruptible(&dax_dev_rwsem);
 938	if (rc)
 939		return rc;
 940	size = dev_dax_size(dev_dax);
 941	up_read(&dax_dev_rwsem);
 942
 943	return sysfs_emit(buf, "%llu\n", size);
 944}
 945
 946static bool alloc_is_aligned(struct dev_dax *dev_dax, resource_size_t size)
 947{
 948	/*
 949	 * The minimum mapping granularity for a device instance is a
 950	 * single subsection, unless the arch says otherwise.
 951	 */
 952	return IS_ALIGNED(size, max_t(unsigned long, dev_dax->align, memremap_compat_align()));
 953}
 954
 955static int dev_dax_shrink(struct dev_dax *dev_dax, resource_size_t size)
 956{
 957	resource_size_t to_shrink = dev_dax_size(dev_dax) - size;
 958	struct dax_region *dax_region = dev_dax->region;
 959	struct device *dev = &dev_dax->dev;
 960	int i;
 961
 962	for (i = dev_dax->nr_range - 1; i >= 0; i--) {
 963		struct range *range = &dev_dax->ranges[i].range;
 964		struct dax_mapping *mapping = dev_dax->ranges[i].mapping;
 965		struct resource *adjust = NULL, *res;
 966		resource_size_t shrink;
 967
 968		shrink = min_t(u64, to_shrink, range_len(range));
 969		if (shrink >= range_len(range)) {
 970			devm_release_action(dax_region->dev,
 971					unregister_dax_mapping, &mapping->dev);
 972			trim_dev_dax_range(dev_dax);
 973			to_shrink -= shrink;
 974			if (!to_shrink)
 975				break;
 976			continue;
 977		}
 978
 979		for_each_dax_region_resource(dax_region, res)
 980			if (strcmp(res->name, dev_name(dev)) == 0
 981					&& res->start == range->start) {
 982				adjust = res;
 983				break;
 984			}
 985
 986		if (dev_WARN_ONCE(dev, !adjust || i != dev_dax->nr_range - 1,
 987					"failed to find matching resource\n"))
 988			return -ENXIO;
 989		return adjust_dev_dax_range(dev_dax, adjust, range_len(range)
 990				- shrink);
 991	}
 992	return 0;
 993}
 994
 995/*
 996 * Only allow adjustments that preserve the relative pgoff of existing
 997 * allocations. I.e. the dev_dax->ranges array is ordered by increasing pgoff.
 998 */
 999static bool adjust_ok(struct dev_dax *dev_dax, struct resource *res)
1000{
1001	struct dev_dax_range *last;
1002	int i;
1003
1004	if (dev_dax->nr_range == 0)
1005		return false;
1006	if (strcmp(res->name, dev_name(&dev_dax->dev)) != 0)
1007		return false;
1008	last = &dev_dax->ranges[dev_dax->nr_range - 1];
1009	if (last->range.start != res->start || last->range.end != res->end)
1010		return false;
1011	for (i = 0; i < dev_dax->nr_range - 1; i++) {
1012		struct dev_dax_range *dax_range = &dev_dax->ranges[i];
1013
1014		if (dax_range->pgoff > last->pgoff)
1015			return false;
1016	}
1017
1018	return true;
1019}
1020
1021static ssize_t dev_dax_resize(struct dax_region *dax_region,
1022		struct dev_dax *dev_dax, resource_size_t size)
1023{
1024	resource_size_t avail = dax_region_avail_size(dax_region), to_alloc;
1025	resource_size_t dev_size = dev_dax_size(dev_dax);
1026	struct resource *region_res = &dax_region->res;
1027	struct device *dev = &dev_dax->dev;
1028	struct resource *res, *first;
1029	resource_size_t alloc = 0;
1030	int rc;
1031
1032	if (dev->driver)
1033		return -EBUSY;
1034	if (size == dev_size)
1035		return 0;
1036	if (size > dev_size && size - dev_size > avail)
1037		return -ENOSPC;
1038	if (size < dev_size)
1039		return dev_dax_shrink(dev_dax, size);
1040
1041	to_alloc = size - dev_size;
1042	if (dev_WARN_ONCE(dev, !alloc_is_aligned(dev_dax, to_alloc),
1043			"resize of %pa misaligned\n", &to_alloc))
1044		return -ENXIO;
1045
1046	/*
1047	 * Expand the device into the unused portion of the region. This
1048	 * may involve adjusting the end of an existing resource, or
1049	 * allocating a new resource.
1050	 */
1051retry:
1052	first = region_res->child;
1053	if (!first)
1054		return alloc_dev_dax_range(dev_dax, dax_region->res.start, to_alloc);
1055
1056	rc = -ENOSPC;
1057	for (res = first; res; res = res->sibling) {
1058		struct resource *next = res->sibling;
1059
1060		/* space at the beginning of the region */
1061		if (res == first && res->start > dax_region->res.start) {
1062			alloc = min(res->start - dax_region->res.start, to_alloc);
1063			rc = alloc_dev_dax_range(dev_dax, dax_region->res.start, alloc);
1064			break;
1065		}
1066
1067		alloc = 0;
1068		/* space between allocations */
1069		if (next && next->start > res->end + 1)
1070			alloc = min(next->start - (res->end + 1), to_alloc);
1071
1072		/* space at the end of the region */
1073		if (!alloc && !next && res->end < region_res->end)
1074			alloc = min(region_res->end - res->end, to_alloc);
1075
1076		if (!alloc)
1077			continue;
1078
1079		if (adjust_ok(dev_dax, res)) {
1080			rc = adjust_dev_dax_range(dev_dax, res, resource_size(res) + alloc);
1081			break;
1082		}
1083		rc = alloc_dev_dax_range(dev_dax, res->end + 1, alloc);
1084		break;
1085	}
1086	if (rc)
1087		return rc;
1088	to_alloc -= alloc;
1089	if (to_alloc)
1090		goto retry;
1091	return 0;
1092}
1093
1094static ssize_t size_store(struct device *dev, struct device_attribute *attr,
1095		const char *buf, size_t len)
1096{
1097	ssize_t rc;
1098	unsigned long long val;
1099	struct dev_dax *dev_dax = to_dev_dax(dev);
1100	struct dax_region *dax_region = dev_dax->region;
1101
1102	rc = kstrtoull(buf, 0, &val);
1103	if (rc)
1104		return rc;
1105
1106	if (!alloc_is_aligned(dev_dax, val)) {
1107		dev_dbg(dev, "%s: size: %lld misaligned\n", __func__, val);
1108		return -EINVAL;
1109	}
1110
1111	rc = down_write_killable(&dax_region_rwsem);
1112	if (rc)
1113		return rc;
1114	if (!dax_region->dev->driver) {
1115		rc = -ENXIO;
1116		goto err_region;
1117	}
1118	rc = down_write_killable(&dax_dev_rwsem);
1119	if (rc)
1120		goto err_dev;
1121
1122	rc = dev_dax_resize(dax_region, dev_dax, val);
 
 
1123
1124err_dev:
1125	up_write(&dax_dev_rwsem);
1126err_region:
1127	up_write(&dax_region_rwsem);
1128
1129	if (rc == 0)
1130		return len;
1131	return rc;
1132}
1133static DEVICE_ATTR_RW(size);
1134
1135static ssize_t range_parse(const char *opt, size_t len, struct range *range)
1136{
1137	unsigned long long addr = 0;
1138	char *start, *end, *str;
1139	ssize_t rc = -EINVAL;
1140
1141	str = kstrdup(opt, GFP_KERNEL);
1142	if (!str)
1143		return rc;
1144
1145	end = str;
1146	start = strsep(&end, "-");
1147	if (!start || !end)
1148		goto err;
1149
1150	rc = kstrtoull(start, 16, &addr);
1151	if (rc)
1152		goto err;
1153	range->start = addr;
1154
1155	rc = kstrtoull(end, 16, &addr);
1156	if (rc)
1157		goto err;
1158	range->end = addr;
1159
1160err:
1161	kfree(str);
1162	return rc;
1163}
1164
1165static ssize_t mapping_store(struct device *dev, struct device_attribute *attr,
1166		const char *buf, size_t len)
1167{
1168	struct dev_dax *dev_dax = to_dev_dax(dev);
1169	struct dax_region *dax_region = dev_dax->region;
1170	size_t to_alloc;
1171	struct range r;
1172	ssize_t rc;
1173
1174	rc = range_parse(buf, len, &r);
1175	if (rc)
1176		return rc;
1177
1178	rc = down_write_killable(&dax_region_rwsem);
1179	if (rc)
1180		return rc;
1181	if (!dax_region->dev->driver) {
1182		up_write(&dax_region_rwsem);
1183		return rc;
1184	}
1185	rc = down_write_killable(&dax_dev_rwsem);
1186	if (rc) {
1187		up_write(&dax_region_rwsem);
1188		return rc;
1189	}
 
1190
1191	to_alloc = range_len(&r);
1192	if (alloc_is_aligned(dev_dax, to_alloc))
1193		rc = alloc_dev_dax_range(dev_dax, r.start, to_alloc);
1194	up_write(&dax_dev_rwsem);
1195	up_write(&dax_region_rwsem);
1196
1197	return rc == 0 ? len : rc;
1198}
1199static DEVICE_ATTR_WO(mapping);
1200
1201static ssize_t align_show(struct device *dev,
1202		struct device_attribute *attr, char *buf)
1203{
1204	struct dev_dax *dev_dax = to_dev_dax(dev);
1205
1206	return sysfs_emit(buf, "%d\n", dev_dax->align);
1207}
1208
1209static ssize_t dev_dax_validate_align(struct dev_dax *dev_dax)
1210{
1211	struct device *dev = &dev_dax->dev;
1212	int i;
1213
1214	for (i = 0; i < dev_dax->nr_range; i++) {
1215		size_t len = range_len(&dev_dax->ranges[i].range);
1216
1217		if (!alloc_is_aligned(dev_dax, len)) {
1218			dev_dbg(dev, "%s: align %u invalid for range %d\n",
1219				__func__, dev_dax->align, i);
1220			return -EINVAL;
1221		}
1222	}
1223
1224	return 0;
1225}
1226
1227static ssize_t align_store(struct device *dev, struct device_attribute *attr,
1228		const char *buf, size_t len)
1229{
1230	struct dev_dax *dev_dax = to_dev_dax(dev);
1231	struct dax_region *dax_region = dev_dax->region;
1232	unsigned long val, align_save;
1233	ssize_t rc;
1234
1235	rc = kstrtoul(buf, 0, &val);
1236	if (rc)
1237		return -ENXIO;
1238
1239	if (!dax_align_valid(val))
1240		return -EINVAL;
1241
1242	rc = down_write_killable(&dax_region_rwsem);
1243	if (rc)
1244		return rc;
1245	if (!dax_region->dev->driver) {
1246		up_write(&dax_region_rwsem);
1247		return -ENXIO;
1248	}
1249
1250	rc = down_write_killable(&dax_dev_rwsem);
1251	if (rc) {
1252		up_write(&dax_region_rwsem);
1253		return rc;
1254	}
1255	if (dev->driver) {
1256		rc = -EBUSY;
1257		goto out_unlock;
1258	}
1259
1260	align_save = dev_dax->align;
1261	dev_dax->align = val;
1262	rc = dev_dax_validate_align(dev_dax);
1263	if (rc)
1264		dev_dax->align = align_save;
1265out_unlock:
1266	up_write(&dax_dev_rwsem);
1267	up_write(&dax_region_rwsem);
1268	return rc == 0 ? len : rc;
1269}
1270static DEVICE_ATTR_RW(align);
1271
1272static int dev_dax_target_node(struct dev_dax *dev_dax)
1273{
1274	struct dax_region *dax_region = dev_dax->region;
1275
1276	return dax_region->target_node;
1277}
1278
1279static ssize_t target_node_show(struct device *dev,
1280		struct device_attribute *attr, char *buf)
1281{
1282	struct dev_dax *dev_dax = to_dev_dax(dev);
1283
1284	return sysfs_emit(buf, "%d\n", dev_dax_target_node(dev_dax));
1285}
1286static DEVICE_ATTR_RO(target_node);
1287
1288static ssize_t resource_show(struct device *dev,
1289		struct device_attribute *attr, char *buf)
1290{
1291	struct dev_dax *dev_dax = to_dev_dax(dev);
1292	struct dax_region *dax_region = dev_dax->region;
1293	unsigned long long start;
1294
1295	if (dev_dax->nr_range < 1)
1296		start = dax_region->res.start;
1297	else
1298		start = dev_dax->ranges[0].range.start;
1299
1300	return sysfs_emit(buf, "%#llx\n", start);
1301}
1302static DEVICE_ATTR(resource, 0400, resource_show, NULL);
1303
1304static ssize_t modalias_show(struct device *dev, struct device_attribute *attr,
1305		char *buf)
1306{
1307	/*
1308	 * We only ever expect to handle device-dax instances, i.e. the
1309	 * @type argument to MODULE_ALIAS_DAX_DEVICE() is always zero
1310	 */
1311	return sysfs_emit(buf, DAX_DEVICE_MODALIAS_FMT "\n", 0);
1312}
1313static DEVICE_ATTR_RO(modalias);
1314
1315static ssize_t numa_node_show(struct device *dev,
1316		struct device_attribute *attr, char *buf)
1317{
1318	return sysfs_emit(buf, "%d\n", dev_to_node(dev));
1319}
1320static DEVICE_ATTR_RO(numa_node);
1321
1322static ssize_t memmap_on_memory_show(struct device *dev,
1323				     struct device_attribute *attr, char *buf)
1324{
1325	struct dev_dax *dev_dax = to_dev_dax(dev);
1326
1327	return sysfs_emit(buf, "%d\n", dev_dax->memmap_on_memory);
1328}
1329
1330static ssize_t memmap_on_memory_store(struct device *dev,
1331				      struct device_attribute *attr,
1332				      const char *buf, size_t len)
1333{
1334	struct dev_dax *dev_dax = to_dev_dax(dev);
1335	bool val;
1336	int rc;
1337
1338	rc = kstrtobool(buf, &val);
1339	if (rc)
1340		return rc;
1341
1342	if (val == true && !mhp_supports_memmap_on_memory()) {
1343		dev_dbg(dev, "memmap_on_memory is not available\n");
1344		return -EOPNOTSUPP;
1345	}
1346
1347	rc = down_write_killable(&dax_dev_rwsem);
1348	if (rc)
1349		return rc;
1350
1351	if (dev_dax->memmap_on_memory != val && dev->driver &&
1352	    to_dax_drv(dev->driver)->type == DAXDRV_KMEM_TYPE) {
1353		up_write(&dax_dev_rwsem);
1354		return -EBUSY;
1355	}
1356
1357	dev_dax->memmap_on_memory = val;
1358	up_write(&dax_dev_rwsem);
1359
1360	return len;
1361}
1362static DEVICE_ATTR_RW(memmap_on_memory);
1363
1364static umode_t dev_dax_visible(struct kobject *kobj, struct attribute *a, int n)
1365{
1366	struct device *dev = container_of(kobj, struct device, kobj);
1367	struct dev_dax *dev_dax = to_dev_dax(dev);
1368	struct dax_region *dax_region = dev_dax->region;
1369
1370	if (a == &dev_attr_target_node.attr && dev_dax_target_node(dev_dax) < 0)
1371		return 0;
1372	if (a == &dev_attr_numa_node.attr && !IS_ENABLED(CONFIG_NUMA))
1373		return 0;
1374	if (a == &dev_attr_mapping.attr && is_static(dax_region))
1375		return 0;
1376	if ((a == &dev_attr_align.attr ||
1377	     a == &dev_attr_size.attr) && is_static(dax_region))
1378		return 0444;
1379	return a->mode;
1380}
1381
1382static struct attribute *dev_dax_attributes[] = {
1383	&dev_attr_modalias.attr,
1384	&dev_attr_size.attr,
1385	&dev_attr_mapping.attr,
1386	&dev_attr_target_node.attr,
1387	&dev_attr_align.attr,
1388	&dev_attr_resource.attr,
1389	&dev_attr_numa_node.attr,
1390	&dev_attr_memmap_on_memory.attr,
1391	NULL,
1392};
1393
1394static const struct attribute_group dev_dax_attribute_group = {
1395	.attrs = dev_dax_attributes,
1396	.is_visible = dev_dax_visible,
1397};
1398
1399static const struct attribute_group *dax_attribute_groups[] = {
1400	&dev_dax_attribute_group,
1401	NULL,
1402};
1403
1404static void dev_dax_release(struct device *dev)
1405{
1406	struct dev_dax *dev_dax = to_dev_dax(dev);
1407	struct dax_device *dax_dev = dev_dax->dax_dev;
1408
1409	put_dax(dax_dev);
1410	free_dev_dax_id(dev_dax);
1411	kfree(dev_dax->pgmap);
1412	kfree(dev_dax);
1413}
1414
1415static const struct device_type dev_dax_type = {
1416	.release = dev_dax_release,
1417	.groups = dax_attribute_groups,
1418};
1419
1420static struct dev_dax *__devm_create_dev_dax(struct dev_dax_data *data)
1421{
1422	struct dax_region *dax_region = data->dax_region;
1423	struct device *parent = dax_region->dev;
1424	struct dax_device *dax_dev;
1425	struct dev_dax *dev_dax;
1426	struct inode *inode;
1427	struct device *dev;
1428	int rc;
1429
1430	dev_dax = kzalloc(sizeof(*dev_dax), GFP_KERNEL);
1431	if (!dev_dax)
1432		return ERR_PTR(-ENOMEM);
1433
1434	dev_dax->region = dax_region;
1435	if (is_static(dax_region)) {
1436		if (dev_WARN_ONCE(parent, data->id < 0,
1437				"dynamic id specified to static region\n")) {
1438			rc = -EINVAL;
1439			goto err_id;
1440		}
1441
1442		dev_dax->id = data->id;
1443	} else {
1444		if (dev_WARN_ONCE(parent, data->id >= 0,
1445				"static id specified to dynamic region\n")) {
1446			rc = -EINVAL;
1447			goto err_id;
1448		}
1449
1450		rc = alloc_dev_dax_id(dev_dax);
1451		if (rc < 0)
1452			goto err_id;
1453	}
1454
1455	dev = &dev_dax->dev;
1456	device_initialize(dev);
1457	dev_set_name(dev, "dax%d.%d", dax_region->id, dev_dax->id);
1458
1459	rc = alloc_dev_dax_range(dev_dax, dax_region->res.start, data->size);
1460	if (rc)
1461		goto err_range;
1462
1463	if (data->pgmap) {
1464		dev_WARN_ONCE(parent, !is_static(dax_region),
1465			"custom dev_pagemap requires a static dax_region\n");
1466
1467		dev_dax->pgmap = kmemdup(data->pgmap,
1468				sizeof(struct dev_pagemap), GFP_KERNEL);
1469		if (!dev_dax->pgmap) {
1470			rc = -ENOMEM;
1471			goto err_pgmap;
1472		}
1473	}
1474
1475	/*
1476	 * No dax_operations since there is no access to this device outside of
1477	 * mmap of the resulting character device.
1478	 */
1479	dax_dev = alloc_dax(dev_dax, NULL);
1480	if (IS_ERR(dax_dev)) {
1481		rc = PTR_ERR(dax_dev);
1482		goto err_alloc_dax;
1483	}
1484	set_dax_synchronous(dax_dev);
1485	set_dax_nocache(dax_dev);
1486	set_dax_nomc(dax_dev);
1487
1488	/* a device_dax instance is dead while the driver is not attached */
1489	kill_dax(dax_dev);
1490
1491	dev_dax->dax_dev = dax_dev;
1492	dev_dax->target_node = dax_region->target_node;
1493	dev_dax->align = dax_region->align;
1494	ida_init(&dev_dax->ida);
1495
1496	dev_dax->memmap_on_memory = data->memmap_on_memory;
1497
1498	inode = dax_inode(dax_dev);
1499	dev->devt = inode->i_rdev;
1500	dev->bus = &dax_bus_type;
1501	dev->parent = parent;
1502	dev->type = &dev_dax_type;
1503
1504	rc = device_add(dev);
1505	if (rc) {
1506		kill_dev_dax(dev_dax);
1507		put_device(dev);
1508		return ERR_PTR(rc);
1509	}
1510
1511	rc = devm_add_action_or_reset(dax_region->dev, unregister_dev_dax, dev);
1512	if (rc)
1513		return ERR_PTR(rc);
1514
1515	/* register mapping device for the initial allocation range */
1516	if (dev_dax->nr_range && range_len(&dev_dax->ranges[0].range)) {
1517		rc = devm_register_dax_mapping(dev_dax, 0);
1518		if (rc)
1519			return ERR_PTR(rc);
1520	}
1521
1522	return dev_dax;
1523
1524err_alloc_dax:
1525	kfree(dev_dax->pgmap);
1526err_pgmap:
1527	free_dev_dax_ranges(dev_dax);
1528err_range:
1529	free_dev_dax_id(dev_dax);
1530err_id:
1531	kfree(dev_dax);
1532
1533	return ERR_PTR(rc);
1534}
1535
1536struct dev_dax *devm_create_dev_dax(struct dev_dax_data *data)
1537{
1538	struct dev_dax *dev_dax;
1539
1540	down_write(&dax_region_rwsem);
1541	dev_dax = __devm_create_dev_dax(data);
1542	up_write(&dax_region_rwsem);
1543
1544	return dev_dax;
1545}
1546EXPORT_SYMBOL_GPL(devm_create_dev_dax);
1547
1548int __dax_driver_register(struct dax_device_driver *dax_drv,
1549		struct module *module, const char *mod_name)
1550{
1551	struct device_driver *drv = &dax_drv->drv;
1552
1553	/*
1554	 * dax_bus_probe() calls dax_drv->probe() unconditionally.
1555	 * So better be safe than sorry and ensure it is provided.
1556	 */
1557	if (!dax_drv->probe)
1558		return -EINVAL;
1559
1560	INIT_LIST_HEAD(&dax_drv->ids);
1561	drv->owner = module;
1562	drv->name = mod_name;
1563	drv->mod_name = mod_name;
1564	drv->bus = &dax_bus_type;
1565
1566	return driver_register(drv);
1567}
1568EXPORT_SYMBOL_GPL(__dax_driver_register);
1569
1570void dax_driver_unregister(struct dax_device_driver *dax_drv)
1571{
1572	struct device_driver *drv = &dax_drv->drv;
1573	struct dax_id *dax_id, *_id;
1574
1575	mutex_lock(&dax_bus_lock);
1576	list_for_each_entry_safe(dax_id, _id, &dax_drv->ids, list) {
1577		list_del(&dax_id->list);
1578		kfree(dax_id);
1579	}
1580	mutex_unlock(&dax_bus_lock);
1581	driver_unregister(drv);
1582}
1583EXPORT_SYMBOL_GPL(dax_driver_unregister);
1584
1585int __init dax_bus_init(void)
1586{
1587	return bus_register(&dax_bus_type);
1588}
1589
1590void __exit dax_bus_exit(void)
1591{
1592	bus_unregister(&dax_bus_type);
1593}