|  | /* | 
|  | * iommufd container backend | 
|  | * | 
|  | * Copyright (C) 2023 Intel Corporation. | 
|  | * Copyright Red Hat, Inc. 2023 | 
|  | * | 
|  | * Authors: Yi Liu <yi.l.liu@intel.com> | 
|  | *          Eric Auger <eric.auger@redhat.com> | 
|  | * | 
|  | * SPDX-License-Identifier: GPL-2.0-or-later | 
|  | */ | 
|  |  | 
|  | #include "qemu/osdep.h" | 
|  | #include <sys/ioctl.h> | 
|  | #include <linux/vfio.h> | 
|  | #include <linux/iommufd.h> | 
|  |  | 
|  | #include "hw/vfio/vfio-common.h" | 
|  | #include "qemu/error-report.h" | 
|  | #include "trace.h" | 
|  | #include "qapi/error.h" | 
|  | #include "system/iommufd.h" | 
|  | #include "hw/qdev-core.h" | 
|  | #include "system/reset.h" | 
|  | #include "qemu/cutils.h" | 
|  | #include "qemu/chardev_open.h" | 
|  | #include "pci.h" | 
|  | #include "exec/ram_addr.h" | 
|  |  | 
|  | static int iommufd_cdev_map(const VFIOContainerBase *bcontainer, hwaddr iova, | 
|  | ram_addr_t size, void *vaddr, bool readonly) | 
|  | { | 
|  | const VFIOIOMMUFDContainer *container = | 
|  | container_of(bcontainer, VFIOIOMMUFDContainer, bcontainer); | 
|  |  | 
|  | return iommufd_backend_map_dma(container->be, | 
|  | container->ioas_id, | 
|  | iova, size, vaddr, readonly); | 
|  | } | 
|  |  | 
|  | static int iommufd_cdev_unmap(const VFIOContainerBase *bcontainer, | 
|  | hwaddr iova, ram_addr_t size, | 
|  | IOMMUTLBEntry *iotlb) | 
|  | { | 
|  | const VFIOIOMMUFDContainer *container = | 
|  | container_of(bcontainer, VFIOIOMMUFDContainer, bcontainer); | 
|  |  | 
|  | /* TODO: Handle dma_unmap_bitmap with iotlb args (migration) */ | 
|  | return iommufd_backend_unmap_dma(container->be, | 
|  | container->ioas_id, iova, size); | 
|  | } | 
|  |  | 
|  | static bool iommufd_cdev_kvm_device_add(VFIODevice *vbasedev, Error **errp) | 
|  | { | 
|  | return !vfio_kvm_device_add_fd(vbasedev->fd, errp); | 
|  | } | 
|  |  | 
|  | static void iommufd_cdev_kvm_device_del(VFIODevice *vbasedev) | 
|  | { | 
|  | Error *err = NULL; | 
|  |  | 
|  | if (vfio_kvm_device_del_fd(vbasedev->fd, &err)) { | 
|  | error_report_err(err); | 
|  | } | 
|  | } | 
|  |  | 
|  | static bool iommufd_cdev_connect_and_bind(VFIODevice *vbasedev, Error **errp) | 
|  | { | 
|  | IOMMUFDBackend *iommufd = vbasedev->iommufd; | 
|  | struct vfio_device_bind_iommufd bind = { | 
|  | .argsz = sizeof(bind), | 
|  | .flags = 0, | 
|  | }; | 
|  |  | 
|  | if (!iommufd_backend_connect(iommufd, errp)) { | 
|  | return false; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * Add device to kvm-vfio to be prepared for the tracking | 
|  | * in KVM. Especially for some emulated devices, it requires | 
|  | * to have kvm information in the device open. | 
|  | */ | 
|  | if (!iommufd_cdev_kvm_device_add(vbasedev, errp)) { | 
|  | goto err_kvm_device_add; | 
|  | } | 
|  |  | 
|  | /* Bind device to iommufd */ | 
|  | bind.iommufd = iommufd->fd; | 
|  | if (ioctl(vbasedev->fd, VFIO_DEVICE_BIND_IOMMUFD, &bind)) { | 
|  | error_setg_errno(errp, errno, "error bind device fd=%d to iommufd=%d", | 
|  | vbasedev->fd, bind.iommufd); | 
|  | goto err_bind; | 
|  | } | 
|  |  | 
|  | vbasedev->devid = bind.out_devid; | 
|  | trace_iommufd_cdev_connect_and_bind(bind.iommufd, vbasedev->name, | 
|  | vbasedev->fd, vbasedev->devid); | 
|  | return true; | 
|  | err_bind: | 
|  | iommufd_cdev_kvm_device_del(vbasedev); | 
|  | err_kvm_device_add: | 
|  | iommufd_backend_disconnect(iommufd); | 
|  | return false; | 
|  | } | 
|  |  | 
|  | static void iommufd_cdev_unbind_and_disconnect(VFIODevice *vbasedev) | 
|  | { | 
|  | /* Unbind is automatically conducted when device fd is closed */ | 
|  | iommufd_cdev_kvm_device_del(vbasedev); | 
|  | iommufd_backend_disconnect(vbasedev->iommufd); | 
|  | } | 
|  |  | 
|  | static bool iommufd_hwpt_dirty_tracking(VFIOIOASHwpt *hwpt) | 
|  | { | 
|  | return hwpt && hwpt->hwpt_flags & IOMMU_HWPT_ALLOC_DIRTY_TRACKING; | 
|  | } | 
|  |  | 
|  | static int iommufd_set_dirty_page_tracking(const VFIOContainerBase *bcontainer, | 
|  | bool start, Error **errp) | 
|  | { | 
|  | const VFIOIOMMUFDContainer *container = | 
|  | container_of(bcontainer, VFIOIOMMUFDContainer, bcontainer); | 
|  | VFIOIOASHwpt *hwpt; | 
|  |  | 
|  | QLIST_FOREACH(hwpt, &container->hwpt_list, next) { | 
|  | if (!iommufd_hwpt_dirty_tracking(hwpt)) { | 
|  | continue; | 
|  | } | 
|  |  | 
|  | if (!iommufd_backend_set_dirty_tracking(container->be, | 
|  | hwpt->hwpt_id, start, errp)) { | 
|  | goto err; | 
|  | } | 
|  | } | 
|  |  | 
|  | return 0; | 
|  |  | 
|  | err: | 
|  | QLIST_FOREACH(hwpt, &container->hwpt_list, next) { | 
|  | if (!iommufd_hwpt_dirty_tracking(hwpt)) { | 
|  | continue; | 
|  | } | 
|  | iommufd_backend_set_dirty_tracking(container->be, | 
|  | hwpt->hwpt_id, !start, NULL); | 
|  | } | 
|  | return -EINVAL; | 
|  | } | 
|  |  | 
|  | static int iommufd_query_dirty_bitmap(const VFIOContainerBase *bcontainer, | 
|  | VFIOBitmap *vbmap, hwaddr iova, | 
|  | hwaddr size, Error **errp) | 
|  | { | 
|  | VFIOIOMMUFDContainer *container = container_of(bcontainer, | 
|  | VFIOIOMMUFDContainer, | 
|  | bcontainer); | 
|  | unsigned long page_size = qemu_real_host_page_size(); | 
|  | VFIOIOASHwpt *hwpt; | 
|  |  | 
|  | QLIST_FOREACH(hwpt, &container->hwpt_list, next) { | 
|  | if (!iommufd_hwpt_dirty_tracking(hwpt)) { | 
|  | continue; | 
|  | } | 
|  |  | 
|  | if (!iommufd_backend_get_dirty_bitmap(container->be, hwpt->hwpt_id, | 
|  | iova, size, page_size, | 
|  | (uint64_t *)vbmap->bitmap, | 
|  | errp)) { | 
|  | return -EINVAL; | 
|  | } | 
|  | } | 
|  |  | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | static int iommufd_cdev_getfd(const char *sysfs_path, Error **errp) | 
|  | { | 
|  | ERRP_GUARD(); | 
|  | long int ret = -ENOTTY; | 
|  | g_autofree char *path = NULL; | 
|  | g_autofree char *vfio_dev_path = NULL; | 
|  | g_autofree char *vfio_path = NULL; | 
|  | DIR *dir = NULL; | 
|  | struct dirent *dent; | 
|  | g_autofree gchar *contents = NULL; | 
|  | gsize length; | 
|  | int major, minor; | 
|  | dev_t vfio_devt; | 
|  |  | 
|  | path = g_strdup_printf("%s/vfio-dev", sysfs_path); | 
|  | dir = opendir(path); | 
|  | if (!dir) { | 
|  | error_setg_errno(errp, errno, "couldn't open directory %s", path); | 
|  | goto out; | 
|  | } | 
|  |  | 
|  | while ((dent = readdir(dir))) { | 
|  | if (!strncmp(dent->d_name, "vfio", 4)) { | 
|  | vfio_dev_path = g_strdup_printf("%s/%s/dev", path, dent->d_name); | 
|  | break; | 
|  | } | 
|  | } | 
|  |  | 
|  | if (!vfio_dev_path) { | 
|  | error_setg(errp, "failed to find vfio-dev/vfioX/dev"); | 
|  | goto out_close_dir; | 
|  | } | 
|  |  | 
|  | if (!g_file_get_contents(vfio_dev_path, &contents, &length, NULL)) { | 
|  | error_setg(errp, "failed to load \"%s\"", vfio_dev_path); | 
|  | goto out_close_dir; | 
|  | } | 
|  |  | 
|  | if (sscanf(contents, "%d:%d", &major, &minor) != 2) { | 
|  | error_setg(errp, "failed to get major:minor for \"%s\"", vfio_dev_path); | 
|  | goto out_close_dir; | 
|  | } | 
|  | vfio_devt = makedev(major, minor); | 
|  |  | 
|  | vfio_path = g_strdup_printf("/dev/vfio/devices/%s", dent->d_name); | 
|  | ret = open_cdev(vfio_path, vfio_devt); | 
|  | if (ret < 0) { | 
|  | error_setg(errp, "Failed to open %s", vfio_path); | 
|  | } | 
|  |  | 
|  | trace_iommufd_cdev_getfd(vfio_path, ret); | 
|  |  | 
|  | out_close_dir: | 
|  | closedir(dir); | 
|  | out: | 
|  | if (*errp) { | 
|  | error_prepend(errp, VFIO_MSG_PREFIX, path); | 
|  | } | 
|  |  | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | static int iommufd_cdev_attach_ioas_hwpt(VFIODevice *vbasedev, uint32_t id, | 
|  | Error **errp) | 
|  | { | 
|  | int iommufd = vbasedev->iommufd->fd; | 
|  | struct vfio_device_attach_iommufd_pt attach_data = { | 
|  | .argsz = sizeof(attach_data), | 
|  | .flags = 0, | 
|  | .pt_id = id, | 
|  | }; | 
|  |  | 
|  | /* Attach device to an IOAS or hwpt within iommufd */ | 
|  | if (ioctl(vbasedev->fd, VFIO_DEVICE_ATTACH_IOMMUFD_PT, &attach_data)) { | 
|  | error_setg_errno(errp, errno, | 
|  | "[iommufd=%d] error attach %s (%d) to id=%d", | 
|  | iommufd, vbasedev->name, vbasedev->fd, id); | 
|  | return -errno; | 
|  | } | 
|  |  | 
|  | trace_iommufd_cdev_attach_ioas_hwpt(iommufd, vbasedev->name, | 
|  | vbasedev->fd, id); | 
|  | return 0; | 
|  | } | 
|  |  | 
|  | static bool iommufd_cdev_detach_ioas_hwpt(VFIODevice *vbasedev, Error **errp) | 
|  | { | 
|  | int iommufd = vbasedev->iommufd->fd; | 
|  | struct vfio_device_detach_iommufd_pt detach_data = { | 
|  | .argsz = sizeof(detach_data), | 
|  | .flags = 0, | 
|  | }; | 
|  |  | 
|  | if (ioctl(vbasedev->fd, VFIO_DEVICE_DETACH_IOMMUFD_PT, &detach_data)) { | 
|  | error_setg_errno(errp, errno, "detach %s failed", vbasedev->name); | 
|  | return false; | 
|  | } | 
|  |  | 
|  | trace_iommufd_cdev_detach_ioas_hwpt(iommufd, vbasedev->name); | 
|  | return true; | 
|  | } | 
|  |  | 
|  | static bool iommufd_cdev_autodomains_get(VFIODevice *vbasedev, | 
|  | VFIOIOMMUFDContainer *container, | 
|  | Error **errp) | 
|  | { | 
|  | ERRP_GUARD(); | 
|  | IOMMUFDBackend *iommufd = vbasedev->iommufd; | 
|  | uint32_t flags = 0; | 
|  | VFIOIOASHwpt *hwpt; | 
|  | uint32_t hwpt_id; | 
|  | int ret; | 
|  |  | 
|  | /* Try to find a domain */ | 
|  | QLIST_FOREACH(hwpt, &container->hwpt_list, next) { | 
|  | ret = iommufd_cdev_attach_ioas_hwpt(vbasedev, hwpt->hwpt_id, errp); | 
|  | if (ret) { | 
|  | /* -EINVAL means the domain is incompatible with the device. */ | 
|  | if (ret == -EINVAL) { | 
|  | /* | 
|  | * It is an expected failure and it just means we will try | 
|  | * another domain, or create one if no existing compatible | 
|  | * domain is found. Hence why the error is discarded below. | 
|  | */ | 
|  | error_free(*errp); | 
|  | *errp = NULL; | 
|  | continue; | 
|  | } | 
|  |  | 
|  | return false; | 
|  | } else { | 
|  | vbasedev->hwpt = hwpt; | 
|  | QLIST_INSERT_HEAD(&hwpt->device_list, vbasedev, hwpt_next); | 
|  | vbasedev->iommu_dirty_tracking = iommufd_hwpt_dirty_tracking(hwpt); | 
|  | return true; | 
|  | } | 
|  | } | 
|  |  | 
|  | /* | 
|  | * This is quite early and VFIO Migration state isn't yet fully | 
|  | * initialized, thus rely only on IOMMU hardware capabilities as to | 
|  | * whether IOMMU dirty tracking is going to be requested. Later | 
|  | * vfio_migration_realize() may decide to use VF dirty tracking | 
|  | * instead. | 
|  | */ | 
|  | if (vbasedev->hiod->caps.hw_caps & IOMMU_HW_CAP_DIRTY_TRACKING) { | 
|  | flags = IOMMU_HWPT_ALLOC_DIRTY_TRACKING; | 
|  | } | 
|  |  | 
|  | if (!iommufd_backend_alloc_hwpt(iommufd, vbasedev->devid, | 
|  | container->ioas_id, flags, | 
|  | IOMMU_HWPT_DATA_NONE, 0, NULL, | 
|  | &hwpt_id, errp)) { | 
|  | return false; | 
|  | } | 
|  |  | 
|  | hwpt = g_malloc0(sizeof(*hwpt)); | 
|  | hwpt->hwpt_id = hwpt_id; | 
|  | hwpt->hwpt_flags = flags; | 
|  | QLIST_INIT(&hwpt->device_list); | 
|  |  | 
|  | ret = iommufd_cdev_attach_ioas_hwpt(vbasedev, hwpt->hwpt_id, errp); | 
|  | if (ret) { | 
|  | iommufd_backend_free_id(container->be, hwpt->hwpt_id); | 
|  | g_free(hwpt); | 
|  | return false; | 
|  | } | 
|  |  | 
|  | vbasedev->hwpt = hwpt; | 
|  | vbasedev->iommu_dirty_tracking = iommufd_hwpt_dirty_tracking(hwpt); | 
|  | QLIST_INSERT_HEAD(&hwpt->device_list, vbasedev, hwpt_next); | 
|  | QLIST_INSERT_HEAD(&container->hwpt_list, hwpt, next); | 
|  | container->bcontainer.dirty_pages_supported |= | 
|  | vbasedev->iommu_dirty_tracking; | 
|  | if (container->bcontainer.dirty_pages_supported && | 
|  | !vbasedev->iommu_dirty_tracking) { | 
|  | warn_report("IOMMU instance for device %s doesn't support dirty tracking", | 
|  | vbasedev->name); | 
|  | } | 
|  | return true; | 
|  | } | 
|  |  | 
|  | static void iommufd_cdev_autodomains_put(VFIODevice *vbasedev, | 
|  | VFIOIOMMUFDContainer *container) | 
|  | { | 
|  | VFIOIOASHwpt *hwpt = vbasedev->hwpt; | 
|  |  | 
|  | QLIST_REMOVE(vbasedev, hwpt_next); | 
|  | vbasedev->hwpt = NULL; | 
|  |  | 
|  | if (QLIST_EMPTY(&hwpt->device_list)) { | 
|  | QLIST_REMOVE(hwpt, next); | 
|  | iommufd_backend_free_id(container->be, hwpt->hwpt_id); | 
|  | g_free(hwpt); | 
|  | } | 
|  | } | 
|  |  | 
|  | static bool iommufd_cdev_attach_container(VFIODevice *vbasedev, | 
|  | VFIOIOMMUFDContainer *container, | 
|  | Error **errp) | 
|  | { | 
|  | /* mdevs aren't physical devices and will fail with auto domains */ | 
|  | if (!vbasedev->mdev) { | 
|  | return iommufd_cdev_autodomains_get(vbasedev, container, errp); | 
|  | } | 
|  |  | 
|  | return !iommufd_cdev_attach_ioas_hwpt(vbasedev, container->ioas_id, errp); | 
|  | } | 
|  |  | 
|  | static void iommufd_cdev_detach_container(VFIODevice *vbasedev, | 
|  | VFIOIOMMUFDContainer *container) | 
|  | { | 
|  | Error *err = NULL; | 
|  |  | 
|  | if (!iommufd_cdev_detach_ioas_hwpt(vbasedev, &err)) { | 
|  | error_report_err(err); | 
|  | } | 
|  |  | 
|  | if (vbasedev->hwpt) { | 
|  | iommufd_cdev_autodomains_put(vbasedev, container); | 
|  | } | 
|  |  | 
|  | } | 
|  |  | 
|  | static void iommufd_cdev_container_destroy(VFIOIOMMUFDContainer *container) | 
|  | { | 
|  | VFIOContainerBase *bcontainer = &container->bcontainer; | 
|  |  | 
|  | if (!QLIST_EMPTY(&bcontainer->device_list)) { | 
|  | return; | 
|  | } | 
|  | memory_listener_unregister(&bcontainer->listener); | 
|  | iommufd_backend_free_id(container->be, container->ioas_id); | 
|  | object_unref(container); | 
|  | } | 
|  |  | 
|  | static int iommufd_cdev_ram_block_discard_disable(bool state) | 
|  | { | 
|  | /* | 
|  | * We support coordinated discarding of RAM via the RamDiscardManager. | 
|  | */ | 
|  | return ram_block_uncoordinated_discard_disable(state); | 
|  | } | 
|  |  | 
|  | static bool iommufd_cdev_get_info_iova_range(VFIOIOMMUFDContainer *container, | 
|  | uint32_t ioas_id, Error **errp) | 
|  | { | 
|  | VFIOContainerBase *bcontainer = &container->bcontainer; | 
|  | g_autofree struct iommu_ioas_iova_ranges *info = NULL; | 
|  | struct iommu_iova_range *iova_ranges; | 
|  | int sz, fd = container->be->fd; | 
|  |  | 
|  | info = g_malloc0(sizeof(*info)); | 
|  | info->size = sizeof(*info); | 
|  | info->ioas_id = ioas_id; | 
|  |  | 
|  | if (ioctl(fd, IOMMU_IOAS_IOVA_RANGES, info) && errno != EMSGSIZE) { | 
|  | goto error; | 
|  | } | 
|  |  | 
|  | sz = info->num_iovas * sizeof(struct iommu_iova_range); | 
|  | info = g_realloc(info, sizeof(*info) + sz); | 
|  | info->allowed_iovas = (uintptr_t)(info + 1); | 
|  |  | 
|  | if (ioctl(fd, IOMMU_IOAS_IOVA_RANGES, info)) { | 
|  | goto error; | 
|  | } | 
|  |  | 
|  | iova_ranges = (struct iommu_iova_range *)(uintptr_t)info->allowed_iovas; | 
|  |  | 
|  | for (int i = 0; i < info->num_iovas; i++) { | 
|  | Range *range = g_new(Range, 1); | 
|  |  | 
|  | range_set_bounds(range, iova_ranges[i].start, iova_ranges[i].last); | 
|  | bcontainer->iova_ranges = | 
|  | range_list_insert(bcontainer->iova_ranges, range); | 
|  | } | 
|  | bcontainer->pgsizes = info->out_iova_alignment; | 
|  |  | 
|  | return true; | 
|  |  | 
|  | error: | 
|  | error_setg_errno(errp, errno, "Cannot get IOVA ranges"); | 
|  | return false; | 
|  | } | 
|  |  | 
|  | static bool iommufd_cdev_attach(const char *name, VFIODevice *vbasedev, | 
|  | AddressSpace *as, Error **errp) | 
|  | { | 
|  | VFIOContainerBase *bcontainer; | 
|  | VFIOIOMMUFDContainer *container; | 
|  | VFIOAddressSpace *space; | 
|  | struct vfio_device_info dev_info = { .argsz = sizeof(dev_info) }; | 
|  | int ret, devfd; | 
|  | uint32_t ioas_id; | 
|  | Error *err = NULL; | 
|  | const VFIOIOMMUClass *iommufd_vioc = | 
|  | VFIO_IOMMU_CLASS(object_class_by_name(TYPE_VFIO_IOMMU_IOMMUFD)); | 
|  |  | 
|  | if (vbasedev->fd < 0) { | 
|  | devfd = iommufd_cdev_getfd(vbasedev->sysfsdev, errp); | 
|  | if (devfd < 0) { | 
|  | return false; | 
|  | } | 
|  | vbasedev->fd = devfd; | 
|  | } else { | 
|  | devfd = vbasedev->fd; | 
|  | } | 
|  |  | 
|  | if (!iommufd_cdev_connect_and_bind(vbasedev, errp)) { | 
|  | goto err_connect_bind; | 
|  | } | 
|  |  | 
|  | space = vfio_get_address_space(as); | 
|  |  | 
|  | /* | 
|  | * The HostIOMMUDevice data from legacy backend is static and doesn't need | 
|  | * any information from the (type1-iommu) backend to be initialized. In | 
|  | * contrast however, the IOMMUFD HostIOMMUDevice data requires the iommufd | 
|  | * FD to be connected and having a devid to be able to successfully call | 
|  | * iommufd_backend_get_device_info(). | 
|  | */ | 
|  | if (!vfio_device_hiod_realize(vbasedev, errp)) { | 
|  | goto err_alloc_ioas; | 
|  | } | 
|  |  | 
|  | /* try to attach to an existing container in this space */ | 
|  | QLIST_FOREACH(bcontainer, &space->containers, next) { | 
|  | container = container_of(bcontainer, VFIOIOMMUFDContainer, bcontainer); | 
|  | if (VFIO_IOMMU_GET_CLASS(bcontainer) != iommufd_vioc || | 
|  | vbasedev->iommufd != container->be) { | 
|  | continue; | 
|  | } | 
|  | if (!iommufd_cdev_attach_container(vbasedev, container, &err)) { | 
|  | const char *msg = error_get_pretty(err); | 
|  |  | 
|  | trace_iommufd_cdev_fail_attach_existing_container(msg); | 
|  | error_free(err); | 
|  | err = NULL; | 
|  | } else { | 
|  | ret = iommufd_cdev_ram_block_discard_disable(true); | 
|  | if (ret) { | 
|  | error_setg_errno(errp, -ret, | 
|  | "Cannot set discarding of RAM broken"); | 
|  | goto err_discard_disable; | 
|  | } | 
|  | goto found_container; | 
|  | } | 
|  | } | 
|  |  | 
|  | /* Need to allocate a new dedicated container */ | 
|  | if (!iommufd_backend_alloc_ioas(vbasedev->iommufd, &ioas_id, errp)) { | 
|  | goto err_alloc_ioas; | 
|  | } | 
|  |  | 
|  | trace_iommufd_cdev_alloc_ioas(vbasedev->iommufd->fd, ioas_id); | 
|  |  | 
|  | container = VFIO_IOMMU_IOMMUFD(object_new(TYPE_VFIO_IOMMU_IOMMUFD)); | 
|  | container->be = vbasedev->iommufd; | 
|  | container->ioas_id = ioas_id; | 
|  | QLIST_INIT(&container->hwpt_list); | 
|  |  | 
|  | bcontainer = &container->bcontainer; | 
|  | vfio_address_space_insert(space, bcontainer); | 
|  |  | 
|  | if (!iommufd_cdev_attach_container(vbasedev, container, errp)) { | 
|  | goto err_attach_container; | 
|  | } | 
|  |  | 
|  | ret = iommufd_cdev_ram_block_discard_disable(true); | 
|  | if (ret) { | 
|  | error_setg_errno(errp, -ret, "Cannot set discarding of RAM broken"); | 
|  | goto err_discard_disable; | 
|  | } | 
|  |  | 
|  | if (!iommufd_cdev_get_info_iova_range(container, ioas_id, &err)) { | 
|  | error_append_hint(&err, | 
|  | "Fallback to default 64bit IOVA range and 4K page size\n"); | 
|  | warn_report_err(err); | 
|  | err = NULL; | 
|  | bcontainer->pgsizes = qemu_real_host_page_size(); | 
|  | } | 
|  |  | 
|  | bcontainer->listener = vfio_memory_listener; | 
|  | memory_listener_register(&bcontainer->listener, bcontainer->space->as); | 
|  |  | 
|  | if (bcontainer->error) { | 
|  | error_propagate_prepend(errp, bcontainer->error, | 
|  | "memory listener initialization failed: "); | 
|  | goto err_listener_register; | 
|  | } | 
|  |  | 
|  | bcontainer->initialized = true; | 
|  |  | 
|  | found_container: | 
|  | ret = ioctl(devfd, VFIO_DEVICE_GET_INFO, &dev_info); | 
|  | if (ret) { | 
|  | error_setg_errno(errp, errno, "error getting device info"); | 
|  | goto err_listener_register; | 
|  | } | 
|  |  | 
|  | if (!vfio_cpr_register_container(bcontainer, errp)) { | 
|  | goto err_listener_register; | 
|  | } | 
|  |  | 
|  | /* | 
|  | * TODO: examine RAM_BLOCK_DISCARD stuff, should we do group level | 
|  | * for discarding incompatibility check as well? | 
|  | */ | 
|  | if (vbasedev->ram_block_discard_allowed) { | 
|  | iommufd_cdev_ram_block_discard_disable(false); | 
|  | } | 
|  |  | 
|  | vbasedev->group = 0; | 
|  | vbasedev->num_irqs = dev_info.num_irqs; | 
|  | vbasedev->num_regions = dev_info.num_regions; | 
|  | vbasedev->flags = dev_info.flags; | 
|  | vbasedev->reset_works = !!(dev_info.flags & VFIO_DEVICE_FLAGS_RESET); | 
|  | vbasedev->bcontainer = bcontainer; | 
|  | QLIST_INSERT_HEAD(&bcontainer->device_list, vbasedev, container_next); | 
|  | QLIST_INSERT_HEAD(&vfio_device_list, vbasedev, global_next); | 
|  |  | 
|  | trace_iommufd_cdev_device_info(vbasedev->name, devfd, vbasedev->num_irqs, | 
|  | vbasedev->num_regions, vbasedev->flags); | 
|  | return true; | 
|  |  | 
|  | err_listener_register: | 
|  | iommufd_cdev_ram_block_discard_disable(false); | 
|  | err_discard_disable: | 
|  | iommufd_cdev_detach_container(vbasedev, container); | 
|  | err_attach_container: | 
|  | iommufd_cdev_container_destroy(container); | 
|  | err_alloc_ioas: | 
|  | vfio_put_address_space(space); | 
|  | iommufd_cdev_unbind_and_disconnect(vbasedev); | 
|  | err_connect_bind: | 
|  | close(vbasedev->fd); | 
|  | return false; | 
|  | } | 
|  |  | 
|  | static void iommufd_cdev_detach(VFIODevice *vbasedev) | 
|  | { | 
|  | VFIOContainerBase *bcontainer = vbasedev->bcontainer; | 
|  | VFIOAddressSpace *space = bcontainer->space; | 
|  | VFIOIOMMUFDContainer *container = container_of(bcontainer, | 
|  | VFIOIOMMUFDContainer, | 
|  | bcontainer); | 
|  | QLIST_REMOVE(vbasedev, global_next); | 
|  | QLIST_REMOVE(vbasedev, container_next); | 
|  | vbasedev->bcontainer = NULL; | 
|  |  | 
|  | if (!vbasedev->ram_block_discard_allowed) { | 
|  | iommufd_cdev_ram_block_discard_disable(false); | 
|  | } | 
|  |  | 
|  | vfio_cpr_unregister_container(bcontainer); | 
|  | iommufd_cdev_detach_container(vbasedev, container); | 
|  | iommufd_cdev_container_destroy(container); | 
|  | vfio_put_address_space(space); | 
|  |  | 
|  | iommufd_cdev_unbind_and_disconnect(vbasedev); | 
|  | close(vbasedev->fd); | 
|  | } | 
|  |  | 
|  | static VFIODevice *iommufd_cdev_pci_find_by_devid(__u32 devid) | 
|  | { | 
|  | VFIODevice *vbasedev_iter; | 
|  | const VFIOIOMMUClass *iommufd_vioc = | 
|  | VFIO_IOMMU_CLASS(object_class_by_name(TYPE_VFIO_IOMMU_IOMMUFD)); | 
|  |  | 
|  | QLIST_FOREACH(vbasedev_iter, &vfio_device_list, global_next) { | 
|  | if (VFIO_IOMMU_GET_CLASS(vbasedev_iter->bcontainer) != iommufd_vioc) { | 
|  | continue; | 
|  | } | 
|  | if (devid == vbasedev_iter->devid) { | 
|  | return vbasedev_iter; | 
|  | } | 
|  | } | 
|  | return NULL; | 
|  | } | 
|  |  | 
|  | static VFIOPCIDevice * | 
|  | iommufd_cdev_dep_get_realized_vpdev(struct vfio_pci_dependent_device *dep_dev, | 
|  | VFIODevice *reset_dev) | 
|  | { | 
|  | VFIODevice *vbasedev_tmp; | 
|  |  | 
|  | if (dep_dev->devid == reset_dev->devid || | 
|  | dep_dev->devid == VFIO_PCI_DEVID_OWNED) { | 
|  | return NULL; | 
|  | } | 
|  |  | 
|  | vbasedev_tmp = iommufd_cdev_pci_find_by_devid(dep_dev->devid); | 
|  | if (!vbasedev_tmp || !vbasedev_tmp->dev->realized || | 
|  | vbasedev_tmp->type != VFIO_DEVICE_TYPE_PCI) { | 
|  | return NULL; | 
|  | } | 
|  |  | 
|  | return container_of(vbasedev_tmp, VFIOPCIDevice, vbasedev); | 
|  | } | 
|  |  | 
|  | static int iommufd_cdev_pci_hot_reset(VFIODevice *vbasedev, bool single) | 
|  | { | 
|  | VFIOPCIDevice *vdev = container_of(vbasedev, VFIOPCIDevice, vbasedev); | 
|  | struct vfio_pci_hot_reset_info *info = NULL; | 
|  | struct vfio_pci_dependent_device *devices; | 
|  | struct vfio_pci_hot_reset *reset; | 
|  | int ret, i; | 
|  | bool multi = false; | 
|  |  | 
|  | trace_vfio_pci_hot_reset(vdev->vbasedev.name, single ? "one" : "multi"); | 
|  |  | 
|  | if (!single) { | 
|  | vfio_pci_pre_reset(vdev); | 
|  | } | 
|  | vdev->vbasedev.needs_reset = false; | 
|  |  | 
|  | ret = vfio_pci_get_pci_hot_reset_info(vdev, &info); | 
|  |  | 
|  | if (ret) { | 
|  | goto out_single; | 
|  | } | 
|  |  | 
|  | assert(info->flags & VFIO_PCI_HOT_RESET_FLAG_DEV_ID); | 
|  |  | 
|  | devices = &info->devices[0]; | 
|  |  | 
|  | if (!(info->flags & VFIO_PCI_HOT_RESET_FLAG_DEV_ID_OWNED)) { | 
|  | if (!vdev->has_pm_reset) { | 
|  | for (i = 0; i < info->count; i++) { | 
|  | if (devices[i].devid == VFIO_PCI_DEVID_NOT_OWNED) { | 
|  | error_report("vfio: Cannot reset device %s, " | 
|  | "depends on device %04x:%02x:%02x.%x " | 
|  | "which is not owned.", | 
|  | vdev->vbasedev.name, devices[i].segment, | 
|  | devices[i].bus, PCI_SLOT(devices[i].devfn), | 
|  | PCI_FUNC(devices[i].devfn)); | 
|  | } | 
|  | } | 
|  | } | 
|  | ret = -EPERM; | 
|  | goto out_single; | 
|  | } | 
|  |  | 
|  | trace_vfio_pci_hot_reset_has_dep_devices(vdev->vbasedev.name); | 
|  |  | 
|  | for (i = 0; i < info->count; i++) { | 
|  | VFIOPCIDevice *tmp; | 
|  |  | 
|  | trace_iommufd_cdev_pci_hot_reset_dep_devices(devices[i].segment, | 
|  | devices[i].bus, | 
|  | PCI_SLOT(devices[i].devfn), | 
|  | PCI_FUNC(devices[i].devfn), | 
|  | devices[i].devid); | 
|  |  | 
|  | /* | 
|  | * If a VFIO cdev device is resettable, all the dependent devices | 
|  | * are either bound to same iommufd or within same iommu_groups as | 
|  | * one of the iommufd bound devices. | 
|  | */ | 
|  | assert(devices[i].devid != VFIO_PCI_DEVID_NOT_OWNED); | 
|  |  | 
|  | tmp = iommufd_cdev_dep_get_realized_vpdev(&devices[i], &vdev->vbasedev); | 
|  | if (!tmp) { | 
|  | continue; | 
|  | } | 
|  |  | 
|  | if (single) { | 
|  | ret = -EINVAL; | 
|  | goto out_single; | 
|  | } | 
|  | vfio_pci_pre_reset(tmp); | 
|  | tmp->vbasedev.needs_reset = false; | 
|  | multi = true; | 
|  | } | 
|  |  | 
|  | if (!single && !multi) { | 
|  | ret = -EINVAL; | 
|  | goto out_single; | 
|  | } | 
|  |  | 
|  | /* Use zero length array for hot reset with iommufd backend */ | 
|  | reset = g_malloc0(sizeof(*reset)); | 
|  | reset->argsz = sizeof(*reset); | 
|  |  | 
|  | /* Bus reset! */ | 
|  | ret = ioctl(vdev->vbasedev.fd, VFIO_DEVICE_PCI_HOT_RESET, reset); | 
|  | g_free(reset); | 
|  | if (ret) { | 
|  | ret = -errno; | 
|  | } | 
|  |  | 
|  | trace_vfio_pci_hot_reset_result(vdev->vbasedev.name, | 
|  | ret ? strerror(errno) : "Success"); | 
|  |  | 
|  | /* Re-enable INTx on affected devices */ | 
|  | for (i = 0; i < info->count; i++) { | 
|  | VFIOPCIDevice *tmp; | 
|  |  | 
|  | tmp = iommufd_cdev_dep_get_realized_vpdev(&devices[i], &vdev->vbasedev); | 
|  | if (!tmp) { | 
|  | continue; | 
|  | } | 
|  | vfio_pci_post_reset(tmp); | 
|  | } | 
|  | out_single: | 
|  | if (!single) { | 
|  | vfio_pci_post_reset(vdev); | 
|  | } | 
|  | g_free(info); | 
|  |  | 
|  | return ret; | 
|  | } | 
|  |  | 
|  | static void vfio_iommu_iommufd_class_init(ObjectClass *klass, void *data) | 
|  | { | 
|  | VFIOIOMMUClass *vioc = VFIO_IOMMU_CLASS(klass); | 
|  |  | 
|  | vioc->hiod_typename = TYPE_HOST_IOMMU_DEVICE_IOMMUFD_VFIO; | 
|  |  | 
|  | vioc->dma_map = iommufd_cdev_map; | 
|  | vioc->dma_unmap = iommufd_cdev_unmap; | 
|  | vioc->attach_device = iommufd_cdev_attach; | 
|  | vioc->detach_device = iommufd_cdev_detach; | 
|  | vioc->pci_hot_reset = iommufd_cdev_pci_hot_reset; | 
|  | vioc->set_dirty_page_tracking = iommufd_set_dirty_page_tracking; | 
|  | vioc->query_dirty_bitmap = iommufd_query_dirty_bitmap; | 
|  | }; | 
|  |  | 
|  | static bool hiod_iommufd_vfio_realize(HostIOMMUDevice *hiod, void *opaque, | 
|  | Error **errp) | 
|  | { | 
|  | VFIODevice *vdev = opaque; | 
|  | HostIOMMUDeviceCaps *caps = &hiod->caps; | 
|  | enum iommu_hw_info_type type; | 
|  | union { | 
|  | struct iommu_hw_info_vtd vtd; | 
|  | } data; | 
|  | uint64_t hw_caps; | 
|  |  | 
|  | hiod->agent = opaque; | 
|  |  | 
|  | if (!iommufd_backend_get_device_info(vdev->iommufd, vdev->devid, | 
|  | &type, &data, sizeof(data), | 
|  | &hw_caps, errp)) { | 
|  | return false; | 
|  | } | 
|  |  | 
|  | hiod->name = g_strdup(vdev->name); | 
|  | caps->type = type; | 
|  | caps->hw_caps = hw_caps; | 
|  |  | 
|  | return true; | 
|  | } | 
|  |  | 
|  | static GList * | 
|  | hiod_iommufd_vfio_get_iova_ranges(HostIOMMUDevice *hiod) | 
|  | { | 
|  | VFIODevice *vdev = hiod->agent; | 
|  |  | 
|  | g_assert(vdev); | 
|  | return vfio_container_get_iova_ranges(vdev->bcontainer); | 
|  | } | 
|  |  | 
|  | static uint64_t | 
|  | hiod_iommufd_vfio_get_page_size_mask(HostIOMMUDevice *hiod) | 
|  | { | 
|  | VFIODevice *vdev = hiod->agent; | 
|  |  | 
|  | g_assert(vdev); | 
|  | return vfio_container_get_page_size_mask(vdev->bcontainer); | 
|  | } | 
|  |  | 
|  |  | 
|  | static void hiod_iommufd_vfio_class_init(ObjectClass *oc, void *data) | 
|  | { | 
|  | HostIOMMUDeviceClass *hiodc = HOST_IOMMU_DEVICE_CLASS(oc); | 
|  |  | 
|  | hiodc->realize = hiod_iommufd_vfio_realize; | 
|  | hiodc->get_iova_ranges = hiod_iommufd_vfio_get_iova_ranges; | 
|  | hiodc->get_page_size_mask = hiod_iommufd_vfio_get_page_size_mask; | 
|  | }; | 
|  |  | 
|  | static const TypeInfo types[] = { | 
|  | { | 
|  | .name = TYPE_VFIO_IOMMU_IOMMUFD, | 
|  | .parent = TYPE_VFIO_IOMMU, | 
|  | .instance_size = sizeof(VFIOIOMMUFDContainer), | 
|  | .class_init = vfio_iommu_iommufd_class_init, | 
|  | }, { | 
|  | .name = TYPE_HOST_IOMMU_DEVICE_IOMMUFD_VFIO, | 
|  | .parent = TYPE_HOST_IOMMU_DEVICE_IOMMUFD, | 
|  | .class_init = hiod_iommufd_vfio_class_init, | 
|  | } | 
|  | }; | 
|  |  | 
|  | DEFINE_TYPES(types) |