static void domain_update_iommu_cap(struct dmar_domain *domain)
 {
        domain_update_iommu_coherency(domain);
-       domain->iommu_snooping = domain_update_iommu_snooping(NULL);
        domain->iommu_superpage = domain_update_iommu_superpage(domain, NULL);
 
        /*
        domain->agaw = width_to_agaw(adjust_width);
 
        domain->iommu_coherency = false;
-       domain->iommu_snooping = false;
        domain->iommu_superpage = 0;
        domain->max_addr = 0;
 
                prot |= DMA_PTE_READ;
        if (iommu_prot & IOMMU_WRITE)
                prot |= DMA_PTE_WRITE;
-       if (((iommu_prot & IOMMU_CACHE) && dmar_domain->iommu_snooping) ||
-           dmar_domain->force_snooping)
+       if (dmar_domain->force_snooping)
                prot |= DMA_PTE_SNP;
 
        max_addr = iova + size;
 {
        struct dmar_domain *dmar_domain = to_dmar_domain(domain);
 
-       if (!dmar_domain->iommu_snooping)
+       if (!domain_update_iommu_snooping(NULL))
                return false;
        dmar_domain->force_snooping = true;
        return true;
 
        struct iommu_domain     *domain;
        struct list_head        next;
        struct list_head        group_list;
-       int                     prot;           /* IOMMU_CACHE */
-       bool                    fgsp;           /* Fine-grained super pages */
+       bool                    fgsp : 1;       /* Fine-grained super pages */
+       bool                    enforce_cache_coherency : 1;
 };
 
 struct vfio_dma {
 
        list_for_each_entry(d, &iommu->domain_list, next) {
                ret = iommu_map(d->domain, iova, (phys_addr_t)pfn << PAGE_SHIFT,
-                               npage << PAGE_SHIFT, prot | d->prot);
+                               npage << PAGE_SHIFT, prot | IOMMU_CACHE);
                if (ret)
                        goto unwind;
 
                        }
 
                        ret = iommu_map(domain->domain, iova, phys,
-                                       size, dma->prot | domain->prot);
+                                       size, dma->prot | IOMMU_CACHE);
                        if (ret) {
                                if (!dma->iommu_mapped) {
                                        vfio_unpin_pages_remote(dma, iova,
                return;
 
        ret = iommu_map(domain->domain, 0, page_to_phys(pages), PAGE_SIZE * 2,
-                       IOMMU_READ | IOMMU_WRITE | domain->prot);
+                       IOMMU_READ | IOMMU_WRITE | IOMMU_CACHE);
        if (!ret) {
                size_t unmapped = iommu_unmap(domain->domain, 0, PAGE_SIZE);
 
                goto out_detach;
        }
 
-       if (iommu_capable(bus, IOMMU_CAP_CACHE_COHERENCY))
-               domain->prot |= IOMMU_CACHE;
+       /*
+        * If the IOMMU can block non-coherent operations (ie PCIe TLPs with
+        * no-snoop set) then VFIO always turns this feature on because on Intel
+        * platforms it optimizes KVM to disable wbinvd emulation.
+        */
+       if (domain->domain->ops->enforce_cache_coherency)
+               domain->enforce_cache_coherency =
+                       domain->domain->ops->enforce_cache_coherency(
+                               domain->domain);
 
        /*
         * Try to match an existing compatible domain.  We don't want to
         */
        list_for_each_entry(d, &iommu->domain_list, next) {
                if (d->domain->ops == domain->domain->ops &&
-                   d->prot == domain->prot) {
+                   d->enforce_cache_coherency ==
+                           domain->enforce_cache_coherency) {
                        iommu_detach_group(domain->domain, group->iommu_group);
                        if (!iommu_attach_group(d->domain,
                                                group->iommu_group)) {
        kfree(iommu);
 }
 
-static int vfio_domains_have_iommu_cache(struct vfio_iommu *iommu)
+static int vfio_domains_have_enforce_cache_coherency(struct vfio_iommu *iommu)
 {
        struct vfio_domain *domain;
        int ret = 1;
 
        mutex_lock(&iommu->lock);
        list_for_each_entry(domain, &iommu->domain_list, next) {
-               if (!(domain->prot & IOMMU_CACHE)) {
+               if (!(domain->enforce_cache_coherency)) {
                        ret = 0;
                        break;
                }
        case VFIO_DMA_CC_IOMMU:
                if (!iommu)
                        return 0;
-               return vfio_domains_have_iommu_cache(iommu);
+               return vfio_domains_have_enforce_cache_coherency(iommu);
        default:
                return 0;
        }