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