xref: /linux/drivers/iommu/iommu.c (revision 4413e16d9d21673bb5048a2e542f1aaa00015c2e)
1 /*
2  * Copyright (C) 2007-2008 Advanced Micro Devices, Inc.
3  * Author: Joerg Roedel <joerg.roedel@amd.com>
4  *
5  * This program is free software; you can redistribute it and/or modify it
6  * under the terms of the GNU General Public License version 2 as published
7  * by the Free Software Foundation.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software
16  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
17  */
18 
19 #define pr_fmt(fmt)    "%s: " fmt, __func__
20 
21 #include <linux/device.h>
22 #include <linux/kernel.h>
23 #include <linux/bug.h>
24 #include <linux/types.h>
25 #include <linux/module.h>
26 #include <linux/slab.h>
27 #include <linux/errno.h>
28 #include <linux/iommu.h>
29 #include <linux/idr.h>
30 #include <linux/notifier.h>
31 #include <linux/err.h>
32 
33 static struct kset *iommu_group_kset;
34 static struct ida iommu_group_ida;
35 static struct mutex iommu_group_mutex;
36 
37 struct iommu_group {
38 	struct kobject kobj;
39 	struct kobject *devices_kobj;
40 	struct list_head devices;
41 	struct mutex mutex;
42 	struct blocking_notifier_head notifier;
43 	void *iommu_data;
44 	void (*iommu_data_release)(void *iommu_data);
45 	char *name;
46 	int id;
47 };
48 
49 struct iommu_device {
50 	struct list_head list;
51 	struct device *dev;
52 	char *name;
53 };
54 
55 struct iommu_group_attribute {
56 	struct attribute attr;
57 	ssize_t (*show)(struct iommu_group *group, char *buf);
58 	ssize_t (*store)(struct iommu_group *group,
59 			 const char *buf, size_t count);
60 };
61 
62 #define IOMMU_GROUP_ATTR(_name, _mode, _show, _store)		\
63 struct iommu_group_attribute iommu_group_attr_##_name =		\
64 	__ATTR(_name, _mode, _show, _store)
65 
66 #define to_iommu_group_attr(_attr)	\
67 	container_of(_attr, struct iommu_group_attribute, attr)
68 #define to_iommu_group(_kobj)		\
69 	container_of(_kobj, struct iommu_group, kobj)
70 
71 static ssize_t iommu_group_attr_show(struct kobject *kobj,
72 				     struct attribute *__attr, char *buf)
73 {
74 	struct iommu_group_attribute *attr = to_iommu_group_attr(__attr);
75 	struct iommu_group *group = to_iommu_group(kobj);
76 	ssize_t ret = -EIO;
77 
78 	if (attr->show)
79 		ret = attr->show(group, buf);
80 	return ret;
81 }
82 
83 static ssize_t iommu_group_attr_store(struct kobject *kobj,
84 				      struct attribute *__attr,
85 				      const char *buf, size_t count)
86 {
87 	struct iommu_group_attribute *attr = to_iommu_group_attr(__attr);
88 	struct iommu_group *group = to_iommu_group(kobj);
89 	ssize_t ret = -EIO;
90 
91 	if (attr->store)
92 		ret = attr->store(group, buf, count);
93 	return ret;
94 }
95 
96 static const struct sysfs_ops iommu_group_sysfs_ops = {
97 	.show = iommu_group_attr_show,
98 	.store = iommu_group_attr_store,
99 };
100 
101 static int iommu_group_create_file(struct iommu_group *group,
102 				   struct iommu_group_attribute *attr)
103 {
104 	return sysfs_create_file(&group->kobj, &attr->attr);
105 }
106 
107 static void iommu_group_remove_file(struct iommu_group *group,
108 				    struct iommu_group_attribute *attr)
109 {
110 	sysfs_remove_file(&group->kobj, &attr->attr);
111 }
112 
113 static ssize_t iommu_group_show_name(struct iommu_group *group, char *buf)
114 {
115 	return sprintf(buf, "%s\n", group->name);
116 }
117 
118 static IOMMU_GROUP_ATTR(name, S_IRUGO, iommu_group_show_name, NULL);
119 
120 static void iommu_group_release(struct kobject *kobj)
121 {
122 	struct iommu_group *group = to_iommu_group(kobj);
123 
124 	if (group->iommu_data_release)
125 		group->iommu_data_release(group->iommu_data);
126 
127 	mutex_lock(&iommu_group_mutex);
128 	ida_remove(&iommu_group_ida, group->id);
129 	mutex_unlock(&iommu_group_mutex);
130 
131 	kfree(group->name);
132 	kfree(group);
133 }
134 
135 static struct kobj_type iommu_group_ktype = {
136 	.sysfs_ops = &iommu_group_sysfs_ops,
137 	.release = iommu_group_release,
138 };
139 
140 /**
141  * iommu_group_alloc - Allocate a new group
142  * @name: Optional name to associate with group, visible in sysfs
143  *
144  * This function is called by an iommu driver to allocate a new iommu
145  * group.  The iommu group represents the minimum granularity of the iommu.
146  * Upon successful return, the caller holds a reference to the supplied
147  * group in order to hold the group until devices are added.  Use
148  * iommu_group_put() to release this extra reference count, allowing the
149  * group to be automatically reclaimed once it has no devices or external
150  * references.
151  */
152 struct iommu_group *iommu_group_alloc(void)
153 {
154 	struct iommu_group *group;
155 	int ret;
156 
157 	group = kzalloc(sizeof(*group), GFP_KERNEL);
158 	if (!group)
159 		return ERR_PTR(-ENOMEM);
160 
161 	group->kobj.kset = iommu_group_kset;
162 	mutex_init(&group->mutex);
163 	INIT_LIST_HEAD(&group->devices);
164 	BLOCKING_INIT_NOTIFIER_HEAD(&group->notifier);
165 
166 	mutex_lock(&iommu_group_mutex);
167 
168 again:
169 	if (unlikely(0 == ida_pre_get(&iommu_group_ida, GFP_KERNEL))) {
170 		kfree(group);
171 		mutex_unlock(&iommu_group_mutex);
172 		return ERR_PTR(-ENOMEM);
173 	}
174 
175 	if (-EAGAIN == ida_get_new(&iommu_group_ida, &group->id))
176 		goto again;
177 
178 	mutex_unlock(&iommu_group_mutex);
179 
180 	ret = kobject_init_and_add(&group->kobj, &iommu_group_ktype,
181 				   NULL, "%d", group->id);
182 	if (ret) {
183 		mutex_lock(&iommu_group_mutex);
184 		ida_remove(&iommu_group_ida, group->id);
185 		mutex_unlock(&iommu_group_mutex);
186 		kfree(group);
187 		return ERR_PTR(ret);
188 	}
189 
190 	group->devices_kobj = kobject_create_and_add("devices", &group->kobj);
191 	if (!group->devices_kobj) {
192 		kobject_put(&group->kobj); /* triggers .release & free */
193 		return ERR_PTR(-ENOMEM);
194 	}
195 
196 	/*
197 	 * The devices_kobj holds a reference on the group kobject, so
198 	 * as long as that exists so will the group.  We can therefore
199 	 * use the devices_kobj for reference counting.
200 	 */
201 	kobject_put(&group->kobj);
202 
203 	return group;
204 }
205 EXPORT_SYMBOL_GPL(iommu_group_alloc);
206 
207 /**
208  * iommu_group_get_iommudata - retrieve iommu_data registered for a group
209  * @group: the group
210  *
211  * iommu drivers can store data in the group for use when doing iommu
212  * operations.  This function provides a way to retrieve it.  Caller
213  * should hold a group reference.
214  */
215 void *iommu_group_get_iommudata(struct iommu_group *group)
216 {
217 	return group->iommu_data;
218 }
219 EXPORT_SYMBOL_GPL(iommu_group_get_iommudata);
220 
221 /**
222  * iommu_group_set_iommudata - set iommu_data for a group
223  * @group: the group
224  * @iommu_data: new data
225  * @release: release function for iommu_data
226  *
227  * iommu drivers can store data in the group for use when doing iommu
228  * operations.  This function provides a way to set the data after
229  * the group has been allocated.  Caller should hold a group reference.
230  */
231 void iommu_group_set_iommudata(struct iommu_group *group, void *iommu_data,
232 			       void (*release)(void *iommu_data))
233 {
234 	group->iommu_data = iommu_data;
235 	group->iommu_data_release = release;
236 }
237 EXPORT_SYMBOL_GPL(iommu_group_set_iommudata);
238 
239 /**
240  * iommu_group_set_name - set name for a group
241  * @group: the group
242  * @name: name
243  *
244  * Allow iommu driver to set a name for a group.  When set it will
245  * appear in a name attribute file under the group in sysfs.
246  */
247 int iommu_group_set_name(struct iommu_group *group, const char *name)
248 {
249 	int ret;
250 
251 	if (group->name) {
252 		iommu_group_remove_file(group, &iommu_group_attr_name);
253 		kfree(group->name);
254 		group->name = NULL;
255 		if (!name)
256 			return 0;
257 	}
258 
259 	group->name = kstrdup(name, GFP_KERNEL);
260 	if (!group->name)
261 		return -ENOMEM;
262 
263 	ret = iommu_group_create_file(group, &iommu_group_attr_name);
264 	if (ret) {
265 		kfree(group->name);
266 		group->name = NULL;
267 		return ret;
268 	}
269 
270 	return 0;
271 }
272 EXPORT_SYMBOL_GPL(iommu_group_set_name);
273 
274 /**
275  * iommu_group_add_device - add a device to an iommu group
276  * @group: the group into which to add the device (reference should be held)
277  * @dev: the device
278  *
279  * This function is called by an iommu driver to add a device into a
280  * group.  Adding a device increments the group reference count.
281  */
282 int iommu_group_add_device(struct iommu_group *group, struct device *dev)
283 {
284 	int ret, i = 0;
285 	struct iommu_device *device;
286 
287 	device = kzalloc(sizeof(*device), GFP_KERNEL);
288 	if (!device)
289 		return -ENOMEM;
290 
291 	device->dev = dev;
292 
293 	ret = sysfs_create_link(&dev->kobj, &group->kobj, "iommu_group");
294 	if (ret) {
295 		kfree(device);
296 		return ret;
297 	}
298 
299 	device->name = kasprintf(GFP_KERNEL, "%s", kobject_name(&dev->kobj));
300 rename:
301 	if (!device->name) {
302 		sysfs_remove_link(&dev->kobj, "iommu_group");
303 		kfree(device);
304 		return -ENOMEM;
305 	}
306 
307 	ret = sysfs_create_link_nowarn(group->devices_kobj,
308 				       &dev->kobj, device->name);
309 	if (ret) {
310 		kfree(device->name);
311 		if (ret == -EEXIST && i >= 0) {
312 			/*
313 			 * Account for the slim chance of collision
314 			 * and append an instance to the name.
315 			 */
316 			device->name = kasprintf(GFP_KERNEL, "%s.%d",
317 						 kobject_name(&dev->kobj), i++);
318 			goto rename;
319 		}
320 
321 		sysfs_remove_link(&dev->kobj, "iommu_group");
322 		kfree(device);
323 		return ret;
324 	}
325 
326 	kobject_get(group->devices_kobj);
327 
328 	dev->iommu_group = group;
329 
330 	mutex_lock(&group->mutex);
331 	list_add_tail(&device->list, &group->devices);
332 	mutex_unlock(&group->mutex);
333 
334 	/* Notify any listeners about change to group. */
335 	blocking_notifier_call_chain(&group->notifier,
336 				     IOMMU_GROUP_NOTIFY_ADD_DEVICE, dev);
337 	return 0;
338 }
339 EXPORT_SYMBOL_GPL(iommu_group_add_device);
340 
341 /**
342  * iommu_group_remove_device - remove a device from it's current group
343  * @dev: device to be removed
344  *
345  * This function is called by an iommu driver to remove the device from
346  * it's current group.  This decrements the iommu group reference count.
347  */
348 void iommu_group_remove_device(struct device *dev)
349 {
350 	struct iommu_group *group = dev->iommu_group;
351 	struct iommu_device *tmp_device, *device = NULL;
352 
353 	/* Pre-notify listeners that a device is being removed. */
354 	blocking_notifier_call_chain(&group->notifier,
355 				     IOMMU_GROUP_NOTIFY_DEL_DEVICE, dev);
356 
357 	mutex_lock(&group->mutex);
358 	list_for_each_entry(tmp_device, &group->devices, list) {
359 		if (tmp_device->dev == dev) {
360 			device = tmp_device;
361 			list_del(&device->list);
362 			break;
363 		}
364 	}
365 	mutex_unlock(&group->mutex);
366 
367 	if (!device)
368 		return;
369 
370 	sysfs_remove_link(group->devices_kobj, device->name);
371 	sysfs_remove_link(&dev->kobj, "iommu_group");
372 
373 	kfree(device->name);
374 	kfree(device);
375 	dev->iommu_group = NULL;
376 	kobject_put(group->devices_kobj);
377 }
378 EXPORT_SYMBOL_GPL(iommu_group_remove_device);
379 
380 /**
381  * iommu_group_for_each_dev - iterate over each device in the group
382  * @group: the group
383  * @data: caller opaque data to be passed to callback function
384  * @fn: caller supplied callback function
385  *
386  * This function is called by group users to iterate over group devices.
387  * Callers should hold a reference count to the group during callback.
388  * The group->mutex is held across callbacks, which will block calls to
389  * iommu_group_add/remove_device.
390  */
391 int iommu_group_for_each_dev(struct iommu_group *group, void *data,
392 			     int (*fn)(struct device *, void *))
393 {
394 	struct iommu_device *device;
395 	int ret = 0;
396 
397 	mutex_lock(&group->mutex);
398 	list_for_each_entry(device, &group->devices, list) {
399 		ret = fn(device->dev, data);
400 		if (ret)
401 			break;
402 	}
403 	mutex_unlock(&group->mutex);
404 	return ret;
405 }
406 EXPORT_SYMBOL_GPL(iommu_group_for_each_dev);
407 
408 /**
409  * iommu_group_get - Return the group for a device and increment reference
410  * @dev: get the group that this device belongs to
411  *
412  * This function is called by iommu drivers and users to get the group
413  * for the specified device.  If found, the group is returned and the group
414  * reference in incremented, else NULL.
415  */
416 struct iommu_group *iommu_group_get(struct device *dev)
417 {
418 	struct iommu_group *group = dev->iommu_group;
419 
420 	if (group)
421 		kobject_get(group->devices_kobj);
422 
423 	return group;
424 }
425 EXPORT_SYMBOL_GPL(iommu_group_get);
426 
427 /**
428  * iommu_group_put - Decrement group reference
429  * @group: the group to use
430  *
431  * This function is called by iommu drivers and users to release the
432  * iommu group.  Once the reference count is zero, the group is released.
433  */
434 void iommu_group_put(struct iommu_group *group)
435 {
436 	if (group)
437 		kobject_put(group->devices_kobj);
438 }
439 EXPORT_SYMBOL_GPL(iommu_group_put);
440 
441 /**
442  * iommu_group_register_notifier - Register a notifier for group changes
443  * @group: the group to watch
444  * @nb: notifier block to signal
445  *
446  * This function allows iommu group users to track changes in a group.
447  * See include/linux/iommu.h for actions sent via this notifier.  Caller
448  * should hold a reference to the group throughout notifier registration.
449  */
450 int iommu_group_register_notifier(struct iommu_group *group,
451 				  struct notifier_block *nb)
452 {
453 	return blocking_notifier_chain_register(&group->notifier, nb);
454 }
455 EXPORT_SYMBOL_GPL(iommu_group_register_notifier);
456 
457 /**
458  * iommu_group_unregister_notifier - Unregister a notifier
459  * @group: the group to watch
460  * @nb: notifier block to signal
461  *
462  * Unregister a previously registered group notifier block.
463  */
464 int iommu_group_unregister_notifier(struct iommu_group *group,
465 				    struct notifier_block *nb)
466 {
467 	return blocking_notifier_chain_unregister(&group->notifier, nb);
468 }
469 EXPORT_SYMBOL_GPL(iommu_group_unregister_notifier);
470 
471 /**
472  * iommu_group_id - Return ID for a group
473  * @group: the group to ID
474  *
475  * Return the unique ID for the group matching the sysfs group number.
476  */
477 int iommu_group_id(struct iommu_group *group)
478 {
479 	return group->id;
480 }
481 EXPORT_SYMBOL_GPL(iommu_group_id);
482 
483 static int add_iommu_group(struct device *dev, void *data)
484 {
485 	struct iommu_ops *ops = data;
486 
487 	if (!ops->add_device)
488 		return -ENODEV;
489 
490 	WARN_ON(dev->iommu_group);
491 
492 	ops->add_device(dev);
493 
494 	return 0;
495 }
496 
497 static int iommu_bus_notifier(struct notifier_block *nb,
498 			      unsigned long action, void *data)
499 {
500 	struct device *dev = data;
501 	struct iommu_ops *ops = dev->bus->iommu_ops;
502 	struct iommu_group *group;
503 	unsigned long group_action = 0;
504 
505 	/*
506 	 * ADD/DEL call into iommu driver ops if provided, which may
507 	 * result in ADD/DEL notifiers to group->notifier
508 	 */
509 	if (action == BUS_NOTIFY_ADD_DEVICE) {
510 		if (ops->add_device)
511 			return ops->add_device(dev);
512 	} else if (action == BUS_NOTIFY_DEL_DEVICE) {
513 		if (ops->remove_device && dev->iommu_group) {
514 			ops->remove_device(dev);
515 			return 0;
516 		}
517 	}
518 
519 	/*
520 	 * Remaining BUS_NOTIFYs get filtered and republished to the
521 	 * group, if anyone is listening
522 	 */
523 	group = iommu_group_get(dev);
524 	if (!group)
525 		return 0;
526 
527 	switch (action) {
528 	case BUS_NOTIFY_BIND_DRIVER:
529 		group_action = IOMMU_GROUP_NOTIFY_BIND_DRIVER;
530 		break;
531 	case BUS_NOTIFY_BOUND_DRIVER:
532 		group_action = IOMMU_GROUP_NOTIFY_BOUND_DRIVER;
533 		break;
534 	case BUS_NOTIFY_UNBIND_DRIVER:
535 		group_action = IOMMU_GROUP_NOTIFY_UNBIND_DRIVER;
536 		break;
537 	case BUS_NOTIFY_UNBOUND_DRIVER:
538 		group_action = IOMMU_GROUP_NOTIFY_UNBOUND_DRIVER;
539 		break;
540 	}
541 
542 	if (group_action)
543 		blocking_notifier_call_chain(&group->notifier,
544 					     group_action, dev);
545 
546 	iommu_group_put(group);
547 	return 0;
548 }
549 
550 static struct notifier_block iommu_bus_nb = {
551 	.notifier_call = iommu_bus_notifier,
552 };
553 
554 static void iommu_bus_init(struct bus_type *bus, struct iommu_ops *ops)
555 {
556 	bus_register_notifier(bus, &iommu_bus_nb);
557 	bus_for_each_dev(bus, NULL, ops, add_iommu_group);
558 }
559 
560 /**
561  * bus_set_iommu - set iommu-callbacks for the bus
562  * @bus: bus.
563  * @ops: the callbacks provided by the iommu-driver
564  *
565  * This function is called by an iommu driver to set the iommu methods
566  * used for a particular bus. Drivers for devices on that bus can use
567  * the iommu-api after these ops are registered.
568  * This special function is needed because IOMMUs are usually devices on
569  * the bus itself, so the iommu drivers are not initialized when the bus
570  * is set up. With this function the iommu-driver can set the iommu-ops
571  * afterwards.
572  */
573 int bus_set_iommu(struct bus_type *bus, struct iommu_ops *ops)
574 {
575 	if (bus->iommu_ops != NULL)
576 		return -EBUSY;
577 
578 	bus->iommu_ops = ops;
579 
580 	/* Do IOMMU specific setup for this bus-type */
581 	iommu_bus_init(bus, ops);
582 
583 	return 0;
584 }
585 EXPORT_SYMBOL_GPL(bus_set_iommu);
586 
587 bool iommu_present(struct bus_type *bus)
588 {
589 	return bus->iommu_ops != NULL;
590 }
591 EXPORT_SYMBOL_GPL(iommu_present);
592 
593 /**
594  * iommu_set_fault_handler() - set a fault handler for an iommu domain
595  * @domain: iommu domain
596  * @handler: fault handler
597  * @token: user data, will be passed back to the fault handler
598  *
599  * This function should be used by IOMMU users which want to be notified
600  * whenever an IOMMU fault happens.
601  *
602  * The fault handler itself should return 0 on success, and an appropriate
603  * error code otherwise.
604  */
605 void iommu_set_fault_handler(struct iommu_domain *domain,
606 					iommu_fault_handler_t handler,
607 					void *token)
608 {
609 	BUG_ON(!domain);
610 
611 	domain->handler = handler;
612 	domain->handler_token = token;
613 }
614 EXPORT_SYMBOL_GPL(iommu_set_fault_handler);
615 
616 struct iommu_domain *iommu_domain_alloc(struct bus_type *bus)
617 {
618 	struct iommu_domain *domain;
619 	int ret;
620 
621 	if (bus == NULL || bus->iommu_ops == NULL)
622 		return NULL;
623 
624 	domain = kzalloc(sizeof(*domain), GFP_KERNEL);
625 	if (!domain)
626 		return NULL;
627 
628 	domain->ops = bus->iommu_ops;
629 
630 	ret = domain->ops->domain_init(domain);
631 	if (ret)
632 		goto out_free;
633 
634 	return domain;
635 
636 out_free:
637 	kfree(domain);
638 
639 	return NULL;
640 }
641 EXPORT_SYMBOL_GPL(iommu_domain_alloc);
642 
643 void iommu_domain_free(struct iommu_domain *domain)
644 {
645 	if (likely(domain->ops->domain_destroy != NULL))
646 		domain->ops->domain_destroy(domain);
647 
648 	kfree(domain);
649 }
650 EXPORT_SYMBOL_GPL(iommu_domain_free);
651 
652 int iommu_attach_device(struct iommu_domain *domain, struct device *dev)
653 {
654 	if (unlikely(domain->ops->attach_dev == NULL))
655 		return -ENODEV;
656 
657 	return domain->ops->attach_dev(domain, dev);
658 }
659 EXPORT_SYMBOL_GPL(iommu_attach_device);
660 
661 void iommu_detach_device(struct iommu_domain *domain, struct device *dev)
662 {
663 	if (unlikely(domain->ops->detach_dev == NULL))
664 		return;
665 
666 	domain->ops->detach_dev(domain, dev);
667 }
668 EXPORT_SYMBOL_GPL(iommu_detach_device);
669 
670 /*
671  * IOMMU groups are really the natrual working unit of the IOMMU, but
672  * the IOMMU API works on domains and devices.  Bridge that gap by
673  * iterating over the devices in a group.  Ideally we'd have a single
674  * device which represents the requestor ID of the group, but we also
675  * allow IOMMU drivers to create policy defined minimum sets, where
676  * the physical hardware may be able to distiguish members, but we
677  * wish to group them at a higher level (ex. untrusted multi-function
678  * PCI devices).  Thus we attach each device.
679  */
680 static int iommu_group_do_attach_device(struct device *dev, void *data)
681 {
682 	struct iommu_domain *domain = data;
683 
684 	return iommu_attach_device(domain, dev);
685 }
686 
687 int iommu_attach_group(struct iommu_domain *domain, struct iommu_group *group)
688 {
689 	return iommu_group_for_each_dev(group, domain,
690 					iommu_group_do_attach_device);
691 }
692 EXPORT_SYMBOL_GPL(iommu_attach_group);
693 
694 static int iommu_group_do_detach_device(struct device *dev, void *data)
695 {
696 	struct iommu_domain *domain = data;
697 
698 	iommu_detach_device(domain, dev);
699 
700 	return 0;
701 }
702 
703 void iommu_detach_group(struct iommu_domain *domain, struct iommu_group *group)
704 {
705 	iommu_group_for_each_dev(group, domain, iommu_group_do_detach_device);
706 }
707 EXPORT_SYMBOL_GPL(iommu_detach_group);
708 
709 phys_addr_t iommu_iova_to_phys(struct iommu_domain *domain,
710 			       unsigned long iova)
711 {
712 	if (unlikely(domain->ops->iova_to_phys == NULL))
713 		return 0;
714 
715 	return domain->ops->iova_to_phys(domain, iova);
716 }
717 EXPORT_SYMBOL_GPL(iommu_iova_to_phys);
718 
719 int iommu_domain_has_cap(struct iommu_domain *domain,
720 			 unsigned long cap)
721 {
722 	if (unlikely(domain->ops->domain_has_cap == NULL))
723 		return 0;
724 
725 	return domain->ops->domain_has_cap(domain, cap);
726 }
727 EXPORT_SYMBOL_GPL(iommu_domain_has_cap);
728 
729 int iommu_map(struct iommu_domain *domain, unsigned long iova,
730 	      phys_addr_t paddr, size_t size, int prot)
731 {
732 	unsigned long orig_iova = iova;
733 	unsigned int min_pagesz;
734 	size_t orig_size = size;
735 	int ret = 0;
736 
737 	if (unlikely(domain->ops->map == NULL))
738 		return -ENODEV;
739 
740 	/* find out the minimum page size supported */
741 	min_pagesz = 1 << __ffs(domain->ops->pgsize_bitmap);
742 
743 	/*
744 	 * both the virtual address and the physical one, as well as
745 	 * the size of the mapping, must be aligned (at least) to the
746 	 * size of the smallest page supported by the hardware
747 	 */
748 	if (!IS_ALIGNED(iova | paddr | size, min_pagesz)) {
749 		pr_err("unaligned: iova 0x%lx pa 0x%lx size 0x%lx min_pagesz "
750 			"0x%x\n", iova, (unsigned long)paddr,
751 			(unsigned long)size, min_pagesz);
752 		return -EINVAL;
753 	}
754 
755 	pr_debug("map: iova 0x%lx pa 0x%lx size 0x%lx\n", iova,
756 				(unsigned long)paddr, (unsigned long)size);
757 
758 	while (size) {
759 		unsigned long pgsize, addr_merge = iova | paddr;
760 		unsigned int pgsize_idx;
761 
762 		/* Max page size that still fits into 'size' */
763 		pgsize_idx = __fls(size);
764 
765 		/* need to consider alignment requirements ? */
766 		if (likely(addr_merge)) {
767 			/* Max page size allowed by both iova and paddr */
768 			unsigned int align_pgsize_idx = __ffs(addr_merge);
769 
770 			pgsize_idx = min(pgsize_idx, align_pgsize_idx);
771 		}
772 
773 		/* build a mask of acceptable page sizes */
774 		pgsize = (1UL << (pgsize_idx + 1)) - 1;
775 
776 		/* throw away page sizes not supported by the hardware */
777 		pgsize &= domain->ops->pgsize_bitmap;
778 
779 		/* make sure we're still sane */
780 		BUG_ON(!pgsize);
781 
782 		/* pick the biggest page */
783 		pgsize_idx = __fls(pgsize);
784 		pgsize = 1UL << pgsize_idx;
785 
786 		pr_debug("mapping: iova 0x%lx pa 0x%lx pgsize %lu\n", iova,
787 					(unsigned long)paddr, pgsize);
788 
789 		ret = domain->ops->map(domain, iova, paddr, pgsize, prot);
790 		if (ret)
791 			break;
792 
793 		iova += pgsize;
794 		paddr += pgsize;
795 		size -= pgsize;
796 	}
797 
798 	/* unroll mapping in case something went wrong */
799 	if (ret)
800 		iommu_unmap(domain, orig_iova, orig_size - size);
801 
802 	return ret;
803 }
804 EXPORT_SYMBOL_GPL(iommu_map);
805 
806 size_t iommu_unmap(struct iommu_domain *domain, unsigned long iova, size_t size)
807 {
808 	size_t unmapped_page, unmapped = 0;
809 	unsigned int min_pagesz;
810 
811 	if (unlikely(domain->ops->unmap == NULL))
812 		return -ENODEV;
813 
814 	/* find out the minimum page size supported */
815 	min_pagesz = 1 << __ffs(domain->ops->pgsize_bitmap);
816 
817 	/*
818 	 * The virtual address, as well as the size of the mapping, must be
819 	 * aligned (at least) to the size of the smallest page supported
820 	 * by the hardware
821 	 */
822 	if (!IS_ALIGNED(iova | size, min_pagesz)) {
823 		pr_err("unaligned: iova 0x%lx size 0x%lx min_pagesz 0x%x\n",
824 					iova, (unsigned long)size, min_pagesz);
825 		return -EINVAL;
826 	}
827 
828 	pr_debug("unmap this: iova 0x%lx size 0x%lx\n", iova,
829 							(unsigned long)size);
830 
831 	/*
832 	 * Keep iterating until we either unmap 'size' bytes (or more)
833 	 * or we hit an area that isn't mapped.
834 	 */
835 	while (unmapped < size) {
836 		size_t left = size - unmapped;
837 
838 		unmapped_page = domain->ops->unmap(domain, iova, left);
839 		if (!unmapped_page)
840 			break;
841 
842 		pr_debug("unmapped: iova 0x%lx size %lx\n", iova,
843 					(unsigned long)unmapped_page);
844 
845 		iova += unmapped_page;
846 		unmapped += unmapped_page;
847 	}
848 
849 	return unmapped;
850 }
851 EXPORT_SYMBOL_GPL(iommu_unmap);
852 
853 static int __init iommu_init(void)
854 {
855 	iommu_group_kset = kset_create_and_add("iommu_groups",
856 					       NULL, kernel_kobj);
857 	ida_init(&iommu_group_ida);
858 	mutex_init(&iommu_group_mutex);
859 
860 	BUG_ON(!iommu_group_kset);
861 
862 	return 0;
863 }
864 subsys_initcall(iommu_init);
865 
866 int iommu_domain_get_attr(struct iommu_domain *domain,
867 			  enum iommu_attr attr, void *data)
868 {
869 	struct iommu_domain_geometry *geometry;
870 	int ret = 0;
871 
872 	switch (attr) {
873 	case DOMAIN_ATTR_GEOMETRY:
874 		geometry  = data;
875 		*geometry = domain->geometry;
876 
877 		break;
878 	default:
879 		if (!domain->ops->domain_get_attr)
880 			return -EINVAL;
881 
882 		ret = domain->ops->domain_get_attr(domain, attr, data);
883 	}
884 
885 	return ret;
886 }
887 EXPORT_SYMBOL_GPL(iommu_domain_get_attr);
888 
889 int iommu_domain_set_attr(struct iommu_domain *domain,
890 			  enum iommu_attr attr, void *data)
891 {
892 	if (!domain->ops->domain_set_attr)
893 		return -EINVAL;
894 
895 	return domain->ops->domain_set_attr(domain, attr, data);
896 }
897 EXPORT_SYMBOL_GPL(iommu_domain_set_attr);
898