xref: /openbmc/qemu/hw/vfio/iommufd.c (revision 7d87775f)
1 /*
2  * iommufd container backend
3  *
4  * Copyright (C) 2023 Intel Corporation.
5  * Copyright Red Hat, Inc. 2023
6  *
7  * Authors: Yi Liu <yi.l.liu@intel.com>
8  *          Eric Auger <eric.auger@redhat.com>
9  *
10  * SPDX-License-Identifier: GPL-2.0-or-later
11  */
12 
13 #include "qemu/osdep.h"
14 #include <sys/ioctl.h>
15 #include <linux/vfio.h>
16 #include <linux/iommufd.h>
17 
18 #include "hw/vfio/vfio-common.h"
19 #include "qemu/error-report.h"
20 #include "trace.h"
21 #include "qapi/error.h"
22 #include "sysemu/iommufd.h"
23 #include "hw/qdev-core.h"
24 #include "sysemu/reset.h"
25 #include "qemu/cutils.h"
26 #include "qemu/chardev_open.h"
27 #include "pci.h"
28 #include "exec/ram_addr.h"
29 
30 static int iommufd_cdev_map(const VFIOContainerBase *bcontainer, hwaddr iova,
31                             ram_addr_t size, void *vaddr, bool readonly)
32 {
33     const VFIOIOMMUFDContainer *container =
34         container_of(bcontainer, VFIOIOMMUFDContainer, bcontainer);
35 
36     return iommufd_backend_map_dma(container->be,
37                                    container->ioas_id,
38                                    iova, size, vaddr, readonly);
39 }
40 
41 static int iommufd_cdev_unmap(const VFIOContainerBase *bcontainer,
42                               hwaddr iova, ram_addr_t size,
43                               IOMMUTLBEntry *iotlb)
44 {
45     const VFIOIOMMUFDContainer *container =
46         container_of(bcontainer, VFIOIOMMUFDContainer, bcontainer);
47 
48     /* TODO: Handle dma_unmap_bitmap with iotlb args (migration) */
49     return iommufd_backend_unmap_dma(container->be,
50                                      container->ioas_id, iova, size);
51 }
52 
53 static bool iommufd_cdev_kvm_device_add(VFIODevice *vbasedev, Error **errp)
54 {
55     return !vfio_kvm_device_add_fd(vbasedev->fd, errp);
56 }
57 
58 static void iommufd_cdev_kvm_device_del(VFIODevice *vbasedev)
59 {
60     Error *err = NULL;
61 
62     if (vfio_kvm_device_del_fd(vbasedev->fd, &err)) {
63         error_report_err(err);
64     }
65 }
66 
67 static bool iommufd_cdev_connect_and_bind(VFIODevice *vbasedev, Error **errp)
68 {
69     IOMMUFDBackend *iommufd = vbasedev->iommufd;
70     struct vfio_device_bind_iommufd bind = {
71         .argsz = sizeof(bind),
72         .flags = 0,
73     };
74 
75     if (!iommufd_backend_connect(iommufd, errp)) {
76         return false;
77     }
78 
79     /*
80      * Add device to kvm-vfio to be prepared for the tracking
81      * in KVM. Especially for some emulated devices, it requires
82      * to have kvm information in the device open.
83      */
84     if (!iommufd_cdev_kvm_device_add(vbasedev, errp)) {
85         goto err_kvm_device_add;
86     }
87 
88     /* Bind device to iommufd */
89     bind.iommufd = iommufd->fd;
90     if (ioctl(vbasedev->fd, VFIO_DEVICE_BIND_IOMMUFD, &bind)) {
91         error_setg_errno(errp, errno, "error bind device fd=%d to iommufd=%d",
92                          vbasedev->fd, bind.iommufd);
93         goto err_bind;
94     }
95 
96     vbasedev->devid = bind.out_devid;
97     trace_iommufd_cdev_connect_and_bind(bind.iommufd, vbasedev->name,
98                                         vbasedev->fd, vbasedev->devid);
99     return true;
100 err_bind:
101     iommufd_cdev_kvm_device_del(vbasedev);
102 err_kvm_device_add:
103     iommufd_backend_disconnect(iommufd);
104     return false;
105 }
106 
107 static void iommufd_cdev_unbind_and_disconnect(VFIODevice *vbasedev)
108 {
109     /* Unbind is automatically conducted when device fd is closed */
110     iommufd_cdev_kvm_device_del(vbasedev);
111     iommufd_backend_disconnect(vbasedev->iommufd);
112 }
113 
114 static bool iommufd_hwpt_dirty_tracking(VFIOIOASHwpt *hwpt)
115 {
116     return hwpt && hwpt->hwpt_flags & IOMMU_HWPT_ALLOC_DIRTY_TRACKING;
117 }
118 
119 static int iommufd_set_dirty_page_tracking(const VFIOContainerBase *bcontainer,
120                                            bool start, Error **errp)
121 {
122     const VFIOIOMMUFDContainer *container =
123         container_of(bcontainer, VFIOIOMMUFDContainer, bcontainer);
124     VFIOIOASHwpt *hwpt;
125 
126     QLIST_FOREACH(hwpt, &container->hwpt_list, next) {
127         if (!iommufd_hwpt_dirty_tracking(hwpt)) {
128             continue;
129         }
130 
131         if (!iommufd_backend_set_dirty_tracking(container->be,
132                                                 hwpt->hwpt_id, start, errp)) {
133             goto err;
134         }
135     }
136 
137     return 0;
138 
139 err:
140     QLIST_FOREACH(hwpt, &container->hwpt_list, next) {
141         if (!iommufd_hwpt_dirty_tracking(hwpt)) {
142             continue;
143         }
144         iommufd_backend_set_dirty_tracking(container->be,
145                                            hwpt->hwpt_id, !start, NULL);
146     }
147     return -EINVAL;
148 }
149 
150 static int iommufd_query_dirty_bitmap(const VFIOContainerBase *bcontainer,
151                                       VFIOBitmap *vbmap, hwaddr iova,
152                                       hwaddr size, Error **errp)
153 {
154     VFIOIOMMUFDContainer *container = container_of(bcontainer,
155                                                    VFIOIOMMUFDContainer,
156                                                    bcontainer);
157     unsigned long page_size = qemu_real_host_page_size();
158     VFIOIOASHwpt *hwpt;
159 
160     QLIST_FOREACH(hwpt, &container->hwpt_list, next) {
161         if (!iommufd_hwpt_dirty_tracking(hwpt)) {
162             continue;
163         }
164 
165         if (!iommufd_backend_get_dirty_bitmap(container->be, hwpt->hwpt_id,
166                                               iova, size, page_size,
167                                               (uint64_t *)vbmap->bitmap,
168                                               errp)) {
169             return -EINVAL;
170         }
171     }
172 
173     return 0;
174 }
175 
176 static int iommufd_cdev_getfd(const char *sysfs_path, Error **errp)
177 {
178     ERRP_GUARD();
179     long int ret = -ENOTTY;
180     g_autofree char *path = NULL;
181     g_autofree char *vfio_dev_path = NULL;
182     g_autofree char *vfio_path = NULL;
183     DIR *dir = NULL;
184     struct dirent *dent;
185     g_autofree gchar *contents = NULL;
186     gsize length;
187     int major, minor;
188     dev_t vfio_devt;
189 
190     path = g_strdup_printf("%s/vfio-dev", sysfs_path);
191     dir = opendir(path);
192     if (!dir) {
193         error_setg_errno(errp, errno, "couldn't open directory %s", path);
194         goto out;
195     }
196 
197     while ((dent = readdir(dir))) {
198         if (!strncmp(dent->d_name, "vfio", 4)) {
199             vfio_dev_path = g_strdup_printf("%s/%s/dev", path, dent->d_name);
200             break;
201         }
202     }
203 
204     if (!vfio_dev_path) {
205         error_setg(errp, "failed to find vfio-dev/vfioX/dev");
206         goto out_close_dir;
207     }
208 
209     if (!g_file_get_contents(vfio_dev_path, &contents, &length, NULL)) {
210         error_setg(errp, "failed to load \"%s\"", vfio_dev_path);
211         goto out_close_dir;
212     }
213 
214     if (sscanf(contents, "%d:%d", &major, &minor) != 2) {
215         error_setg(errp, "failed to get major:minor for \"%s\"", vfio_dev_path);
216         goto out_close_dir;
217     }
218     vfio_devt = makedev(major, minor);
219 
220     vfio_path = g_strdup_printf("/dev/vfio/devices/%s", dent->d_name);
221     ret = open_cdev(vfio_path, vfio_devt);
222     if (ret < 0) {
223         error_setg(errp, "Failed to open %s", vfio_path);
224     }
225 
226     trace_iommufd_cdev_getfd(vfio_path, ret);
227 
228 out_close_dir:
229     closedir(dir);
230 out:
231     if (*errp) {
232         error_prepend(errp, VFIO_MSG_PREFIX, path);
233     }
234 
235     return ret;
236 }
237 
238 static int iommufd_cdev_attach_ioas_hwpt(VFIODevice *vbasedev, uint32_t id,
239                                          Error **errp)
240 {
241     int iommufd = vbasedev->iommufd->fd;
242     struct vfio_device_attach_iommufd_pt attach_data = {
243         .argsz = sizeof(attach_data),
244         .flags = 0,
245         .pt_id = id,
246     };
247 
248     /* Attach device to an IOAS or hwpt within iommufd */
249     if (ioctl(vbasedev->fd, VFIO_DEVICE_ATTACH_IOMMUFD_PT, &attach_data)) {
250         error_setg_errno(errp, errno,
251                          "[iommufd=%d] error attach %s (%d) to id=%d",
252                          iommufd, vbasedev->name, vbasedev->fd, id);
253         return -errno;
254     }
255 
256     trace_iommufd_cdev_attach_ioas_hwpt(iommufd, vbasedev->name,
257                                         vbasedev->fd, id);
258     return 0;
259 }
260 
261 static bool iommufd_cdev_detach_ioas_hwpt(VFIODevice *vbasedev, Error **errp)
262 {
263     int iommufd = vbasedev->iommufd->fd;
264     struct vfio_device_detach_iommufd_pt detach_data = {
265         .argsz = sizeof(detach_data),
266         .flags = 0,
267     };
268 
269     if (ioctl(vbasedev->fd, VFIO_DEVICE_DETACH_IOMMUFD_PT, &detach_data)) {
270         error_setg_errno(errp, errno, "detach %s failed", vbasedev->name);
271         return false;
272     }
273 
274     trace_iommufd_cdev_detach_ioas_hwpt(iommufd, vbasedev->name);
275     return true;
276 }
277 
278 static bool iommufd_cdev_autodomains_get(VFIODevice *vbasedev,
279                                          VFIOIOMMUFDContainer *container,
280                                          Error **errp)
281 {
282     ERRP_GUARD();
283     IOMMUFDBackend *iommufd = vbasedev->iommufd;
284     uint32_t flags = 0;
285     VFIOIOASHwpt *hwpt;
286     uint32_t hwpt_id;
287     int ret;
288 
289     /* Try to find a domain */
290     QLIST_FOREACH(hwpt, &container->hwpt_list, next) {
291         ret = iommufd_cdev_attach_ioas_hwpt(vbasedev, hwpt->hwpt_id, errp);
292         if (ret) {
293             /* -EINVAL means the domain is incompatible with the device. */
294             if (ret == -EINVAL) {
295                 /*
296                  * It is an expected failure and it just means we will try
297                  * another domain, or create one if no existing compatible
298                  * domain is found. Hence why the error is discarded below.
299                  */
300                 error_free(*errp);
301                 *errp = NULL;
302                 continue;
303             }
304 
305             return false;
306         } else {
307             vbasedev->hwpt = hwpt;
308             QLIST_INSERT_HEAD(&hwpt->device_list, vbasedev, hwpt_next);
309             vbasedev->iommu_dirty_tracking = iommufd_hwpt_dirty_tracking(hwpt);
310             return true;
311         }
312     }
313 
314     /*
315      * This is quite early and VFIO Migration state isn't yet fully
316      * initialized, thus rely only on IOMMU hardware capabilities as to
317      * whether IOMMU dirty tracking is going to be requested. Later
318      * vfio_migration_realize() may decide to use VF dirty tracking
319      * instead.
320      */
321     if (vbasedev->hiod->caps.hw_caps & IOMMU_HW_CAP_DIRTY_TRACKING) {
322         flags = IOMMU_HWPT_ALLOC_DIRTY_TRACKING;
323     }
324 
325     if (!iommufd_backend_alloc_hwpt(iommufd, vbasedev->devid,
326                                     container->ioas_id, flags,
327                                     IOMMU_HWPT_DATA_NONE, 0, NULL,
328                                     &hwpt_id, errp)) {
329         return false;
330     }
331 
332     hwpt = g_malloc0(sizeof(*hwpt));
333     hwpt->hwpt_id = hwpt_id;
334     hwpt->hwpt_flags = flags;
335     QLIST_INIT(&hwpt->device_list);
336 
337     ret = iommufd_cdev_attach_ioas_hwpt(vbasedev, hwpt->hwpt_id, errp);
338     if (ret) {
339         iommufd_backend_free_id(container->be, hwpt->hwpt_id);
340         g_free(hwpt);
341         return false;
342     }
343 
344     vbasedev->hwpt = hwpt;
345     vbasedev->iommu_dirty_tracking = iommufd_hwpt_dirty_tracking(hwpt);
346     QLIST_INSERT_HEAD(&hwpt->device_list, vbasedev, hwpt_next);
347     QLIST_INSERT_HEAD(&container->hwpt_list, hwpt, next);
348     container->bcontainer.dirty_pages_supported |=
349                                 vbasedev->iommu_dirty_tracking;
350     if (container->bcontainer.dirty_pages_supported &&
351         !vbasedev->iommu_dirty_tracking) {
352         warn_report("IOMMU instance for device %s doesn't support dirty tracking",
353                     vbasedev->name);
354     }
355     return true;
356 }
357 
358 static void iommufd_cdev_autodomains_put(VFIODevice *vbasedev,
359                                          VFIOIOMMUFDContainer *container)
360 {
361     VFIOIOASHwpt *hwpt = vbasedev->hwpt;
362 
363     QLIST_REMOVE(vbasedev, hwpt_next);
364     vbasedev->hwpt = NULL;
365 
366     if (QLIST_EMPTY(&hwpt->device_list)) {
367         QLIST_REMOVE(hwpt, next);
368         iommufd_backend_free_id(container->be, hwpt->hwpt_id);
369         g_free(hwpt);
370     }
371 }
372 
373 static bool iommufd_cdev_attach_container(VFIODevice *vbasedev,
374                                           VFIOIOMMUFDContainer *container,
375                                           Error **errp)
376 {
377     /* mdevs aren't physical devices and will fail with auto domains */
378     if (!vbasedev->mdev) {
379         return iommufd_cdev_autodomains_get(vbasedev, container, errp);
380     }
381 
382     return !iommufd_cdev_attach_ioas_hwpt(vbasedev, container->ioas_id, errp);
383 }
384 
385 static void iommufd_cdev_detach_container(VFIODevice *vbasedev,
386                                           VFIOIOMMUFDContainer *container)
387 {
388     Error *err = NULL;
389 
390     if (!iommufd_cdev_detach_ioas_hwpt(vbasedev, &err)) {
391         error_report_err(err);
392     }
393 
394     if (vbasedev->hwpt) {
395         iommufd_cdev_autodomains_put(vbasedev, container);
396     }
397 
398 }
399 
400 static void iommufd_cdev_container_destroy(VFIOIOMMUFDContainer *container)
401 {
402     VFIOContainerBase *bcontainer = &container->bcontainer;
403 
404     if (!QLIST_EMPTY(&bcontainer->device_list)) {
405         return;
406     }
407     memory_listener_unregister(&bcontainer->listener);
408     iommufd_backend_free_id(container->be, container->ioas_id);
409     object_unref(container);
410 }
411 
412 static int iommufd_cdev_ram_block_discard_disable(bool state)
413 {
414     /*
415      * We support coordinated discarding of RAM via the RamDiscardManager.
416      */
417     return ram_block_uncoordinated_discard_disable(state);
418 }
419 
420 static bool iommufd_cdev_get_info_iova_range(VFIOIOMMUFDContainer *container,
421                                              uint32_t ioas_id, Error **errp)
422 {
423     VFIOContainerBase *bcontainer = &container->bcontainer;
424     g_autofree struct iommu_ioas_iova_ranges *info = NULL;
425     struct iommu_iova_range *iova_ranges;
426     int sz, fd = container->be->fd;
427 
428     info = g_malloc0(sizeof(*info));
429     info->size = sizeof(*info);
430     info->ioas_id = ioas_id;
431 
432     if (ioctl(fd, IOMMU_IOAS_IOVA_RANGES, info) && errno != EMSGSIZE) {
433         goto error;
434     }
435 
436     sz = info->num_iovas * sizeof(struct iommu_iova_range);
437     info = g_realloc(info, sizeof(*info) + sz);
438     info->allowed_iovas = (uintptr_t)(info + 1);
439 
440     if (ioctl(fd, IOMMU_IOAS_IOVA_RANGES, info)) {
441         goto error;
442     }
443 
444     iova_ranges = (struct iommu_iova_range *)(uintptr_t)info->allowed_iovas;
445 
446     for (int i = 0; i < info->num_iovas; i++) {
447         Range *range = g_new(Range, 1);
448 
449         range_set_bounds(range, iova_ranges[i].start, iova_ranges[i].last);
450         bcontainer->iova_ranges =
451             range_list_insert(bcontainer->iova_ranges, range);
452     }
453     bcontainer->pgsizes = info->out_iova_alignment;
454 
455     return true;
456 
457 error:
458     error_setg_errno(errp, errno, "Cannot get IOVA ranges");
459     return false;
460 }
461 
462 static bool iommufd_cdev_attach(const char *name, VFIODevice *vbasedev,
463                                 AddressSpace *as, Error **errp)
464 {
465     VFIOContainerBase *bcontainer;
466     VFIOIOMMUFDContainer *container;
467     VFIOAddressSpace *space;
468     struct vfio_device_info dev_info = { .argsz = sizeof(dev_info) };
469     int ret, devfd;
470     uint32_t ioas_id;
471     Error *err = NULL;
472     const VFIOIOMMUClass *iommufd_vioc =
473         VFIO_IOMMU_CLASS(object_class_by_name(TYPE_VFIO_IOMMU_IOMMUFD));
474 
475     if (vbasedev->fd < 0) {
476         devfd = iommufd_cdev_getfd(vbasedev->sysfsdev, errp);
477         if (devfd < 0) {
478             return false;
479         }
480         vbasedev->fd = devfd;
481     } else {
482         devfd = vbasedev->fd;
483     }
484 
485     if (!iommufd_cdev_connect_and_bind(vbasedev, errp)) {
486         goto err_connect_bind;
487     }
488 
489     space = vfio_get_address_space(as);
490 
491     /*
492      * The HostIOMMUDevice data from legacy backend is static and doesn't need
493      * any information from the (type1-iommu) backend to be initialized. In
494      * contrast however, the IOMMUFD HostIOMMUDevice data requires the iommufd
495      * FD to be connected and having a devid to be able to successfully call
496      * iommufd_backend_get_device_info().
497      */
498     if (!vfio_device_hiod_realize(vbasedev, errp)) {
499         goto err_alloc_ioas;
500     }
501 
502     /* try to attach to an existing container in this space */
503     QLIST_FOREACH(bcontainer, &space->containers, next) {
504         container = container_of(bcontainer, VFIOIOMMUFDContainer, bcontainer);
505         if (VFIO_IOMMU_GET_CLASS(bcontainer) != iommufd_vioc ||
506             vbasedev->iommufd != container->be) {
507             continue;
508         }
509         if (!iommufd_cdev_attach_container(vbasedev, container, &err)) {
510             const char *msg = error_get_pretty(err);
511 
512             trace_iommufd_cdev_fail_attach_existing_container(msg);
513             error_free(err);
514             err = NULL;
515         } else {
516             ret = iommufd_cdev_ram_block_discard_disable(true);
517             if (ret) {
518                 error_setg(errp,
519                               "Cannot set discarding of RAM broken (%d)", ret);
520                 goto err_discard_disable;
521             }
522             goto found_container;
523         }
524     }
525 
526     /* Need to allocate a new dedicated container */
527     if (!iommufd_backend_alloc_ioas(vbasedev->iommufd, &ioas_id, errp)) {
528         goto err_alloc_ioas;
529     }
530 
531     trace_iommufd_cdev_alloc_ioas(vbasedev->iommufd->fd, ioas_id);
532 
533     container = VFIO_IOMMU_IOMMUFD(object_new(TYPE_VFIO_IOMMU_IOMMUFD));
534     container->be = vbasedev->iommufd;
535     container->ioas_id = ioas_id;
536     QLIST_INIT(&container->hwpt_list);
537 
538     bcontainer = &container->bcontainer;
539     vfio_address_space_insert(space, bcontainer);
540 
541     if (!iommufd_cdev_attach_container(vbasedev, container, errp)) {
542         goto err_attach_container;
543     }
544 
545     ret = iommufd_cdev_ram_block_discard_disable(true);
546     if (ret) {
547         goto err_discard_disable;
548     }
549 
550     if (!iommufd_cdev_get_info_iova_range(container, ioas_id, &err)) {
551         error_append_hint(&err,
552                    "Fallback to default 64bit IOVA range and 4K page size\n");
553         warn_report_err(err);
554         err = NULL;
555         bcontainer->pgsizes = qemu_real_host_page_size();
556     }
557 
558     bcontainer->listener = vfio_memory_listener;
559     memory_listener_register(&bcontainer->listener, bcontainer->space->as);
560 
561     if (bcontainer->error) {
562         error_propagate_prepend(errp, bcontainer->error,
563                                 "memory listener initialization failed: ");
564         goto err_listener_register;
565     }
566 
567     bcontainer->initialized = true;
568 
569 found_container:
570     ret = ioctl(devfd, VFIO_DEVICE_GET_INFO, &dev_info);
571     if (ret) {
572         error_setg_errno(errp, errno, "error getting device info");
573         goto err_listener_register;
574     }
575 
576     if (!vfio_cpr_register_container(bcontainer, errp)) {
577         goto err_listener_register;
578     }
579 
580     /*
581      * TODO: examine RAM_BLOCK_DISCARD stuff, should we do group level
582      * for discarding incompatibility check as well?
583      */
584     if (vbasedev->ram_block_discard_allowed) {
585         iommufd_cdev_ram_block_discard_disable(false);
586     }
587 
588     vbasedev->group = 0;
589     vbasedev->num_irqs = dev_info.num_irqs;
590     vbasedev->num_regions = dev_info.num_regions;
591     vbasedev->flags = dev_info.flags;
592     vbasedev->reset_works = !!(dev_info.flags & VFIO_DEVICE_FLAGS_RESET);
593     vbasedev->bcontainer = bcontainer;
594     QLIST_INSERT_HEAD(&bcontainer->device_list, vbasedev, container_next);
595     QLIST_INSERT_HEAD(&vfio_device_list, vbasedev, global_next);
596 
597     trace_iommufd_cdev_device_info(vbasedev->name, devfd, vbasedev->num_irqs,
598                                    vbasedev->num_regions, vbasedev->flags);
599     return true;
600 
601 err_listener_register:
602     iommufd_cdev_ram_block_discard_disable(false);
603 err_discard_disable:
604     iommufd_cdev_detach_container(vbasedev, container);
605 err_attach_container:
606     iommufd_cdev_container_destroy(container);
607 err_alloc_ioas:
608     vfio_put_address_space(space);
609     iommufd_cdev_unbind_and_disconnect(vbasedev);
610 err_connect_bind:
611     close(vbasedev->fd);
612     return false;
613 }
614 
615 static void iommufd_cdev_detach(VFIODevice *vbasedev)
616 {
617     VFIOContainerBase *bcontainer = vbasedev->bcontainer;
618     VFIOAddressSpace *space = bcontainer->space;
619     VFIOIOMMUFDContainer *container = container_of(bcontainer,
620                                                    VFIOIOMMUFDContainer,
621                                                    bcontainer);
622     QLIST_REMOVE(vbasedev, global_next);
623     QLIST_REMOVE(vbasedev, container_next);
624     vbasedev->bcontainer = NULL;
625 
626     if (!vbasedev->ram_block_discard_allowed) {
627         iommufd_cdev_ram_block_discard_disable(false);
628     }
629 
630     vfio_cpr_unregister_container(bcontainer);
631     iommufd_cdev_detach_container(vbasedev, container);
632     iommufd_cdev_container_destroy(container);
633     vfio_put_address_space(space);
634 
635     iommufd_cdev_unbind_and_disconnect(vbasedev);
636     close(vbasedev->fd);
637 }
638 
639 static VFIODevice *iommufd_cdev_pci_find_by_devid(__u32 devid)
640 {
641     VFIODevice *vbasedev_iter;
642     const VFIOIOMMUClass *iommufd_vioc =
643         VFIO_IOMMU_CLASS(object_class_by_name(TYPE_VFIO_IOMMU_IOMMUFD));
644 
645     QLIST_FOREACH(vbasedev_iter, &vfio_device_list, global_next) {
646         if (VFIO_IOMMU_GET_CLASS(vbasedev_iter->bcontainer) != iommufd_vioc) {
647             continue;
648         }
649         if (devid == vbasedev_iter->devid) {
650             return vbasedev_iter;
651         }
652     }
653     return NULL;
654 }
655 
656 static VFIOPCIDevice *
657 iommufd_cdev_dep_get_realized_vpdev(struct vfio_pci_dependent_device *dep_dev,
658                                     VFIODevice *reset_dev)
659 {
660     VFIODevice *vbasedev_tmp;
661 
662     if (dep_dev->devid == reset_dev->devid ||
663         dep_dev->devid == VFIO_PCI_DEVID_OWNED) {
664         return NULL;
665     }
666 
667     vbasedev_tmp = iommufd_cdev_pci_find_by_devid(dep_dev->devid);
668     if (!vbasedev_tmp || !vbasedev_tmp->dev->realized ||
669         vbasedev_tmp->type != VFIO_DEVICE_TYPE_PCI) {
670         return NULL;
671     }
672 
673     return container_of(vbasedev_tmp, VFIOPCIDevice, vbasedev);
674 }
675 
676 static int iommufd_cdev_pci_hot_reset(VFIODevice *vbasedev, bool single)
677 {
678     VFIOPCIDevice *vdev = container_of(vbasedev, VFIOPCIDevice, vbasedev);
679     struct vfio_pci_hot_reset_info *info = NULL;
680     struct vfio_pci_dependent_device *devices;
681     struct vfio_pci_hot_reset *reset;
682     int ret, i;
683     bool multi = false;
684 
685     trace_vfio_pci_hot_reset(vdev->vbasedev.name, single ? "one" : "multi");
686 
687     if (!single) {
688         vfio_pci_pre_reset(vdev);
689     }
690     vdev->vbasedev.needs_reset = false;
691 
692     ret = vfio_pci_get_pci_hot_reset_info(vdev, &info);
693 
694     if (ret) {
695         goto out_single;
696     }
697 
698     assert(info->flags & VFIO_PCI_HOT_RESET_FLAG_DEV_ID);
699 
700     devices = &info->devices[0];
701 
702     if (!(info->flags & VFIO_PCI_HOT_RESET_FLAG_DEV_ID_OWNED)) {
703         if (!vdev->has_pm_reset) {
704             for (i = 0; i < info->count; i++) {
705                 if (devices[i].devid == VFIO_PCI_DEVID_NOT_OWNED) {
706                     error_report("vfio: Cannot reset device %s, "
707                                  "depends on device %04x:%02x:%02x.%x "
708                                  "which is not owned.",
709                                  vdev->vbasedev.name, devices[i].segment,
710                                  devices[i].bus, PCI_SLOT(devices[i].devfn),
711                                  PCI_FUNC(devices[i].devfn));
712                 }
713             }
714         }
715         ret = -EPERM;
716         goto out_single;
717     }
718 
719     trace_vfio_pci_hot_reset_has_dep_devices(vdev->vbasedev.name);
720 
721     for (i = 0; i < info->count; i++) {
722         VFIOPCIDevice *tmp;
723 
724         trace_iommufd_cdev_pci_hot_reset_dep_devices(devices[i].segment,
725                                                      devices[i].bus,
726                                                      PCI_SLOT(devices[i].devfn),
727                                                      PCI_FUNC(devices[i].devfn),
728                                                      devices[i].devid);
729 
730         /*
731          * If a VFIO cdev device is resettable, all the dependent devices
732          * are either bound to same iommufd or within same iommu_groups as
733          * one of the iommufd bound devices.
734          */
735         assert(devices[i].devid != VFIO_PCI_DEVID_NOT_OWNED);
736 
737         tmp = iommufd_cdev_dep_get_realized_vpdev(&devices[i], &vdev->vbasedev);
738         if (!tmp) {
739             continue;
740         }
741 
742         if (single) {
743             ret = -EINVAL;
744             goto out_single;
745         }
746         vfio_pci_pre_reset(tmp);
747         tmp->vbasedev.needs_reset = false;
748         multi = true;
749     }
750 
751     if (!single && !multi) {
752         ret = -EINVAL;
753         goto out_single;
754     }
755 
756     /* Use zero length array for hot reset with iommufd backend */
757     reset = g_malloc0(sizeof(*reset));
758     reset->argsz = sizeof(*reset);
759 
760      /* Bus reset! */
761     ret = ioctl(vdev->vbasedev.fd, VFIO_DEVICE_PCI_HOT_RESET, reset);
762     g_free(reset);
763     if (ret) {
764         ret = -errno;
765     }
766 
767     trace_vfio_pci_hot_reset_result(vdev->vbasedev.name,
768                                     ret ? strerror(errno) : "Success");
769 
770     /* Re-enable INTx on affected devices */
771     for (i = 0; i < info->count; i++) {
772         VFIOPCIDevice *tmp;
773 
774         tmp = iommufd_cdev_dep_get_realized_vpdev(&devices[i], &vdev->vbasedev);
775         if (!tmp) {
776             continue;
777         }
778         vfio_pci_post_reset(tmp);
779     }
780 out_single:
781     if (!single) {
782         vfio_pci_post_reset(vdev);
783     }
784     g_free(info);
785 
786     return ret;
787 }
788 
789 static void vfio_iommu_iommufd_class_init(ObjectClass *klass, void *data)
790 {
791     VFIOIOMMUClass *vioc = VFIO_IOMMU_CLASS(klass);
792 
793     vioc->hiod_typename = TYPE_HOST_IOMMU_DEVICE_IOMMUFD_VFIO;
794 
795     vioc->dma_map = iommufd_cdev_map;
796     vioc->dma_unmap = iommufd_cdev_unmap;
797     vioc->attach_device = iommufd_cdev_attach;
798     vioc->detach_device = iommufd_cdev_detach;
799     vioc->pci_hot_reset = iommufd_cdev_pci_hot_reset;
800     vioc->set_dirty_page_tracking = iommufd_set_dirty_page_tracking;
801     vioc->query_dirty_bitmap = iommufd_query_dirty_bitmap;
802 };
803 
804 static bool hiod_iommufd_vfio_realize(HostIOMMUDevice *hiod, void *opaque,
805                                       Error **errp)
806 {
807     VFIODevice *vdev = opaque;
808     HostIOMMUDeviceCaps *caps = &hiod->caps;
809     enum iommu_hw_info_type type;
810     union {
811         struct iommu_hw_info_vtd vtd;
812     } data;
813     uint64_t hw_caps;
814 
815     hiod->agent = opaque;
816 
817     if (!iommufd_backend_get_device_info(vdev->iommufd, vdev->devid,
818                                          &type, &data, sizeof(data),
819                                          &hw_caps, errp)) {
820         return false;
821     }
822 
823     hiod->name = g_strdup(vdev->name);
824     caps->type = type;
825     caps->hw_caps = hw_caps;
826 
827     return true;
828 }
829 
830 static GList *
831 hiod_iommufd_vfio_get_iova_ranges(HostIOMMUDevice *hiod)
832 {
833     VFIODevice *vdev = hiod->agent;
834 
835     g_assert(vdev);
836     return vfio_container_get_iova_ranges(vdev->bcontainer);
837 }
838 
839 static uint64_t
840 hiod_iommufd_vfio_get_page_size_mask(HostIOMMUDevice *hiod)
841 {
842     VFIODevice *vdev = hiod->agent;
843 
844     g_assert(vdev);
845     return vfio_container_get_page_size_mask(vdev->bcontainer);
846 }
847 
848 
849 static void hiod_iommufd_vfio_class_init(ObjectClass *oc, void *data)
850 {
851     HostIOMMUDeviceClass *hiodc = HOST_IOMMU_DEVICE_CLASS(oc);
852 
853     hiodc->realize = hiod_iommufd_vfio_realize;
854     hiodc->get_iova_ranges = hiod_iommufd_vfio_get_iova_ranges;
855     hiodc->get_page_size_mask = hiod_iommufd_vfio_get_page_size_mask;
856 };
857 
858 static const TypeInfo types[] = {
859     {
860         .name = TYPE_VFIO_IOMMU_IOMMUFD,
861         .parent = TYPE_VFIO_IOMMU,
862         .instance_size = sizeof(VFIOIOMMUFDContainer),
863         .class_init = vfio_iommu_iommufd_class_init,
864     }, {
865         .name = TYPE_HOST_IOMMU_DEVICE_IOMMUFD_VFIO,
866         .parent = TYPE_HOST_IOMMU_DEVICE_IOMMUFD,
867         .class_init = hiod_iommufd_vfio_class_init,
868     }
869 };
870 
871 DEFINE_TYPES(types)
872