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