137 static u64 dev_dax_size(struct dev_dax *dev_dax)  in dev_dax_size()  argument
142 	device_lock_assert(&dev_dax->dev);  in dev_dax_size()
144 	for (i = 0; i < dev_dax->nr_range; i++)  in dev_dax_size()
145 		size += range_len(&dev_dax->ranges[i].range);  in dev_dax_size()
153 	struct dev_dax *dev_dax = to_dev_dax(dev);  in dax_bus_probe()  local
154 	struct dax_region *dax_region = dev_dax->region;  in dax_bus_probe()
157 	if (dev_dax_size(dev_dax) == 0 || dev_dax->id < 0)  in dax_bus_probe()
160 	rc = dax_drv->probe(dev_dax);  in dax_bus_probe()
178 	struct dev_dax *dev_dax = to_dev_dax(dev);  in dax_bus_remove()  local
181 		dax_drv->remove(dev_dax);  in dax_bus_remove()
335 		struct dev_dax *dev_dax = devm_create_dev_dax(&data);  in create_store()  local
337 		if (IS_ERR(dev_dax))  in create_store()
338 			rc = PTR_ERR(dev_dax);  in create_store()
348 				dax_region->seed = &dev_dax->dev;  in create_store()
349 			dax_region->youngest = &dev_dax->dev;  in create_store()
359 void kill_dev_dax(struct dev_dax *dev_dax)  in kill_dev_dax()  argument
361 	struct dax_device *dax_dev = dev_dax->dax_dev;  in kill_dev_dax()
369 static void trim_dev_dax_range(struct dev_dax *dev_dax)  in trim_dev_dax_range()  argument
371 	int i = dev_dax->nr_range - 1;  in trim_dev_dax_range()
372 	struct range *range = &dev_dax->ranges[i].range;  in trim_dev_dax_range()
373 	struct dax_region *dax_region = dev_dax->region;  in trim_dev_dax_range()
376 	dev_dbg(&dev_dax->dev, "delete range[%d]: %#llx:%#llx\n", i,  in trim_dev_dax_range()
381 	if (--dev_dax->nr_range == 0) {  in trim_dev_dax_range()
382 		kfree(dev_dax->ranges);  in trim_dev_dax_range()
383 		dev_dax->ranges = NULL;  in trim_dev_dax_range()
387 static void free_dev_dax_ranges(struct dev_dax *dev_dax)  in free_dev_dax_ranges()  argument
389 	while (dev_dax->nr_range)  in free_dev_dax_ranges()
390 		trim_dev_dax_range(dev_dax);  in free_dev_dax_ranges()
395 	struct dev_dax *dev_dax = to_dev_dax(dev);  in unregister_dev_dax()  local
399 	kill_dev_dax(dev_dax);  in unregister_dev_dax()
400 	free_dev_dax_ranges(dev_dax);  in unregister_dev_dax()
406 static int __free_dev_dax_id(struct dev_dax *dev_dax)  in __free_dev_dax_id()  argument
408 	struct dax_region *dax_region = dev_dax->region;  in __free_dev_dax_id()
409 	struct device *dev = &dev_dax->dev;  in __free_dev_dax_id()
410 	int rc = dev_dax->id;  in __free_dev_dax_id()
414 	if (is_static(dax_region) || dev_dax->id < 0)  in __free_dev_dax_id()
416 	ida_free(&dax_region->ida, dev_dax->id);  in __free_dev_dax_id()
417 	dev_dax->id = -1;  in __free_dev_dax_id()
421 static int free_dev_dax_id(struct dev_dax *dev_dax)  in free_dev_dax_id()  argument
423 	struct device *dev = &dev_dax->dev;  in free_dev_dax_id()
427 	rc = __free_dev_dax_id(dev_dax);  in free_dev_dax_id()
436 	struct dev_dax *dev_dax;  in delete_store()  local
450 	dev_dax = to_dev_dax(victim);  in delete_store()
451 	if (victim->driver || dev_dax_size(dev_dax))  in delete_store()
460 		if (dev_dax->id > 0) {  in delete_store()
461 			do_del = __free_dev_dax_id(dev_dax) >= 0;  in delete_store()
594 	struct dev_dax *dev_dax = to_dev_dax(dev->parent);  in dax_mapping_release()  local
596 	ida_free(&dev_dax->ida, mapping->id);  in dax_mapping_release()
604 	struct dev_dax *dev_dax = to_dev_dax(dev->parent);  in unregister_dax_mapping()  local
605 	struct dax_region *dax_region = dev_dax->region;  in unregister_dax_mapping()
611 	dev_dax->ranges[mapping->range_id].mapping = NULL;  in unregister_dax_mapping()
621 	struct dev_dax *dev_dax = to_dev_dax(dev->parent);  in get_dax_range()  local
622 	struct dax_region *dax_region = dev_dax->region;  in get_dax_range()
630 	return &dev_dax->ranges[mapping->range_id];  in get_dax_range()
636 	struct dev_dax *dev_dax = to_dev_dax(mapping->dev.parent);  in put_dax_range()  local
637 	struct dax_region *dax_region = dev_dax->region;  in put_dax_range()
711 static int devm_register_dax_mapping(struct dev_dax *dev_dax, int range_id)  in devm_register_dax_mapping()  argument
713 	struct dax_region *dax_region = dev_dax->region;  in devm_register_dax_mapping()
720 	if (dev_WARN_ONCE(&dev_dax->dev, !dax_region->dev->driver,  in devm_register_dax_mapping()
728 	mapping->id = ida_alloc(&dev_dax->ida, GFP_KERNEL);  in devm_register_dax_mapping()
733 	dev_dax->ranges[range_id].mapping = mapping;  in devm_register_dax_mapping()
736 	dev->parent = &dev_dax->dev;  in devm_register_dax_mapping()
752 static int alloc_dev_dax_range(struct dev_dax *dev_dax, u64 start,  in alloc_dev_dax_range()  argument
755 	struct dax_region *dax_region = dev_dax->region;  in alloc_dev_dax_range()
757 	struct device *dev = &dev_dax->dev;  in alloc_dev_dax_range()
767 		if (dev_WARN_ONCE(dev, dev_dax->nr_range,  in alloc_dev_dax_range()
778 	ranges = krealloc(dev_dax->ranges, sizeof(*ranges)  in alloc_dev_dax_range()
779 			* (dev_dax->nr_range + 1), GFP_KERNEL);  in alloc_dev_dax_range()
785 	for (i = 0; i < dev_dax->nr_range; i++)  in alloc_dev_dax_range()
787 	dev_dax->ranges = ranges;  in alloc_dev_dax_range()
788 	ranges[dev_dax->nr_range++] = (struct dev_dax_range) {  in alloc_dev_dax_range()
796 	dev_dbg(dev, "alloc range[%d]: %pa:%pa\n", dev_dax->nr_range - 1,  in alloc_dev_dax_range()
803 	if (!device_is_registered(&dev_dax->dev))  in alloc_dev_dax_range()
806 	rc = devm_register_dax_mapping(dev_dax, dev_dax->nr_range - 1);  in alloc_dev_dax_range()
808 		trim_dev_dax_range(dev_dax);  in alloc_dev_dax_range()
813 static int adjust_dev_dax_range(struct dev_dax *dev_dax, struct resource *res, resource_size_t size)  in adjust_dev_dax_range()  argument
815 	int last_range = dev_dax->nr_range - 1;  in adjust_dev_dax_range()
816 	struct dev_dax_range *dax_range = &dev_dax->ranges[last_range];  in adjust_dev_dax_range()
817 	struct dax_region *dax_region = dev_dax->region;  in adjust_dev_dax_range()
820 	struct device *dev = &dev_dax->dev;  in adjust_dev_dax_range()
847 	struct dev_dax *dev_dax = to_dev_dax(dev);  in size_show()  local
851 	size = dev_dax_size(dev_dax);  in size_show()
857 static bool alloc_is_aligned(struct dev_dax *dev_dax, resource_size_t size)  in alloc_is_aligned()  argument
863 	return IS_ALIGNED(size, max_t(unsigned long, dev_dax->align, memremap_compat_align()));  in alloc_is_aligned()
866 static int dev_dax_shrink(struct dev_dax *dev_dax, resource_size_t size)  in dev_dax_shrink()  argument
868 	resource_size_t to_shrink = dev_dax_size(dev_dax) - size;  in dev_dax_shrink()
869 	struct dax_region *dax_region = dev_dax->region;  in dev_dax_shrink()
870 	struct device *dev = &dev_dax->dev;  in dev_dax_shrink()
873 	for (i = dev_dax->nr_range - 1; i >= 0; i--) {  in dev_dax_shrink()
874 		struct range *range = &dev_dax->ranges[i].range;  in dev_dax_shrink()
875 		struct dax_mapping *mapping = dev_dax->ranges[i].mapping;  in dev_dax_shrink()
883 			trim_dev_dax_range(dev_dax);  in dev_dax_shrink()
897 		if (dev_WARN_ONCE(dev, !adjust || i != dev_dax->nr_range - 1,  in dev_dax_shrink()
900 		return adjust_dev_dax_range(dev_dax, adjust, range_len(range)  in dev_dax_shrink()
910 static bool adjust_ok(struct dev_dax *dev_dax, struct resource *res)  in adjust_ok()  argument
915 	if (dev_dax->nr_range == 0)  in adjust_ok()
917 	if (strcmp(res->name, dev_name(&dev_dax->dev)) != 0)  in adjust_ok()
919 	last = &dev_dax->ranges[dev_dax->nr_range - 1];  in adjust_ok()
922 	for (i = 0; i < dev_dax->nr_range - 1; i++) {  in adjust_ok()
923 		struct dev_dax_range *dax_range = &dev_dax->ranges[i];  in adjust_ok()
933 		struct dev_dax *dev_dax, resource_size_t size)  in dev_dax_resize()  argument
936 	resource_size_t dev_size = dev_dax_size(dev_dax);  in dev_dax_resize()
938 	struct device *dev = &dev_dax->dev;  in dev_dax_resize()
950 		return dev_dax_shrink(dev_dax, size);  in dev_dax_resize()
953 	if (dev_WARN_ONCE(dev, !alloc_is_aligned(dev_dax, to_alloc),  in dev_dax_resize()
965 		return alloc_dev_dax_range(dev_dax, dax_region->res.start, to_alloc);  in dev_dax_resize()
974 			rc = alloc_dev_dax_range(dev_dax, dax_region->res.start, alloc);  in dev_dax_resize()
990 		if (adjust_ok(dev_dax, res)) {  in dev_dax_resize()
991 			rc = adjust_dev_dax_range(dev_dax, res, resource_size(res) + alloc);  in dev_dax_resize()
994 		rc = alloc_dev_dax_range(dev_dax, res->end + 1, alloc);  in dev_dax_resize()
1010 	struct dev_dax *dev_dax = to_dev_dax(dev);  in size_store()  local
1011 	struct dax_region *dax_region = dev_dax->region;  in size_store()
1017 	if (!alloc_is_aligned(dev_dax, val)) {  in size_store()
1028 	rc = dev_dax_resize(dax_region, dev_dax, val);  in size_store()
1069 	struct dev_dax *dev_dax = to_dev_dax(dev);  in mapping_store()  local
1070 	struct dax_region *dax_region = dev_dax->region;  in mapping_store()
1088 	if (alloc_is_aligned(dev_dax, to_alloc))  in mapping_store()
1089 		rc = alloc_dev_dax_range(dev_dax, r.start, to_alloc);  in mapping_store()
1100 	struct dev_dax *dev_dax = to_dev_dax(dev);  in align_show()  local
1102 	return sprintf(buf, "%d\n", dev_dax->align);  in align_show()
1105 static ssize_t dev_dax_validate_align(struct dev_dax *dev_dax)  in dev_dax_validate_align()  argument
1107 	struct device *dev = &dev_dax->dev;  in dev_dax_validate_align()
1110 	for (i = 0; i < dev_dax->nr_range; i++) {  in dev_dax_validate_align()
1111 		size_t len = range_len(&dev_dax->ranges[i].range);  in dev_dax_validate_align()
1113 		if (!alloc_is_aligned(dev_dax, len)) {  in dev_dax_validate_align()
1115 				__func__, dev_dax->align, i);  in dev_dax_validate_align()
1126 	struct dev_dax *dev_dax = to_dev_dax(dev);  in align_store()  local
1127 	struct dax_region *dax_region = dev_dax->region;  in align_store()
1150 	align_save = dev_dax->align;  in align_store()
1151 	dev_dax->align = val;  in align_store()
1152 	rc = dev_dax_validate_align(dev_dax);  in align_store()
1154 		dev_dax->align = align_save;  in align_store()
1162 static int dev_dax_target_node(struct dev_dax *dev_dax)  in dev_dax_target_node()  argument
1164 	struct dax_region *dax_region = dev_dax->region;  in dev_dax_target_node()
1172 	struct dev_dax *dev_dax = to_dev_dax(dev);  in target_node_show()  local
1174 	return sprintf(buf, "%d\n", dev_dax_target_node(dev_dax));  in target_node_show()
1181 	struct dev_dax *dev_dax = to_dev_dax(dev);  in resource_show()  local
1182 	struct dax_region *dax_region = dev_dax->region;  in resource_show()
1185 	if (dev_dax->nr_range < 1)  in resource_show()
1188 		start = dev_dax->ranges[0].range.start;  in resource_show()
1215 	struct dev_dax *dev_dax = to_dev_dax(dev);  in dev_dax_visible()  local
1216 	struct dax_region *dax_region = dev_dax->region;  in dev_dax_visible()
1218 	if (a == &dev_attr_target_node.attr && dev_dax_target_node(dev_dax) < 0)  in dev_dax_visible()
1253 	struct dev_dax *dev_dax = to_dev_dax(dev);  in dev_dax_release()  local
1254 	struct dax_region *dax_region = dev_dax->region;  in dev_dax_release()
1255 	struct dax_device *dax_dev = dev_dax->dax_dev;  in dev_dax_release()
1258 	free_dev_dax_id(dev_dax);  in dev_dax_release()
1260 	kfree(dev_dax->pgmap);  in dev_dax_release()
1261 	kfree(dev_dax);  in dev_dax_release()
1269 struct dev_dax *devm_create_dev_dax(struct dev_dax_data *data)  in devm_create_dev_dax()
1274 	struct dev_dax *dev_dax;  in devm_create_dev_dax()  local
1279 	dev_dax = kzalloc(sizeof(*dev_dax), GFP_KERNEL);  in devm_create_dev_dax()
1280 	if (!dev_dax)  in devm_create_dev_dax()
1290 		dev_dax->id = data->id;  in devm_create_dev_dax()
1301 		dev_dax->id = rc;  in devm_create_dev_dax()
1304 	dev_dax->region = dax_region;  in devm_create_dev_dax()
1305 	dev = &dev_dax->dev;  in devm_create_dev_dax()
1307 	dev_set_name(dev, "dax%d.%d", dax_region->id, dev_dax->id);  in devm_create_dev_dax()
1309 	rc = alloc_dev_dax_range(dev_dax, dax_region->res.start, data->size);  in devm_create_dev_dax()
1317 		dev_dax->pgmap = kmemdup(data->pgmap,  in devm_create_dev_dax()
1319 		if (!dev_dax->pgmap) {  in devm_create_dev_dax()
1329 	dax_dev = alloc_dax(dev_dax, NULL, NULL, DAXDEV_F_SYNC);  in devm_create_dev_dax()
1338 	dev_dax->dax_dev = dax_dev;  in devm_create_dev_dax()
1339 	dev_dax->target_node = dax_region->target_node;  in devm_create_dev_dax()
1340 	dev_dax->align = dax_region->align;  in devm_create_dev_dax()
1341 	ida_init(&dev_dax->ida);  in devm_create_dev_dax()
1355 		kill_dev_dax(dev_dax);  in devm_create_dev_dax()
1365 	if (dev_dax->nr_range && range_len(&dev_dax->ranges[0].range)) {  in devm_create_dev_dax()
1366 		rc = devm_register_dax_mapping(dev_dax, 0);  in devm_create_dev_dax()
1371 	return dev_dax;  in devm_create_dev_dax()
1374 	kfree(dev_dax->pgmap);  in devm_create_dev_dax()
1376 	free_dev_dax_ranges(dev_dax);  in devm_create_dev_dax()
1378 	free_dev_dax_id(dev_dax);  in devm_create_dev_dax()
1380 	kfree(dev_dax);  in devm_create_dev_dax()