From: Manish Honap <[email protected]>

Some devices expose registers in a BAR that must be reached only through
a trap, not a direct guest mapping. A CXL Type-2 device's HDM decoder
block is one: mapping it would let userspace reprogram the physical
decoder that governs host memory decode. Give a provider a way to mark a
BAR sub-range off-limits to mmap; it is advertised as a sparse-mmap
region and refused in the mmap path, while the provider's own region
still serves it.

Signed-off-by: Manish Honap <[email protected]>
---
 drivers/vfio/pci/vfio_pci_core.c   | 72 ++++++++++++++++++++++++++++++
 drivers/vfio/pci/vfio_pci_dmabuf.c | 13 ++++++
 drivers/vfio/pci/vfio_pci_priv.h   | 13 ++++++
 include/linux/vfio_pci_core.h      |  6 +++
 4 files changed, 104 insertions(+)

diff --git a/drivers/vfio/pci/vfio_pci_core.c b/drivers/vfio/pci/vfio_pci_core.c
index 0f9b5dfeea66..49dfbdaf3f05 100644
--- a/drivers/vfio/pci/vfio_pci_core.c
+++ b/drivers/vfio/pci/vfio_pci_core.c
@@ -1009,6 +1009,67 @@ static int msix_mmappable_cap(struct 
vfio_pci_core_device *vdev,
        return vfio_info_add_capability(caps, &header, sizeof(header));
 }
 
+/*
+ * A provider can keep a BAR sub-range off mmap (for example a CXL device's
+ * trapped HDM decoder block). Callers hold the resource so /dev/mem is already
+ * blocked; this only governs the vfio mmap path.
+ */
+void vfio_pci_core_set_mmap_exclude(struct vfio_pci_core_device *vdev, int bar,
+                                   u64 start, u64 len)
+{
+       vdev->mmap_exclude_bar = bar;
+       vdev->mmap_exclude_start = start;
+       vdev->mmap_exclude_len = len;
+}
+EXPORT_SYMBOL_GPL(vfio_pci_core_set_mmap_exclude);
+
+/* Advertise the BAR as mmappable minus the excluded sub-range. */
+static int vfio_pci_mmap_exclude_cap(struct vfio_pci_core_device *vdev,
+                                    int index, struct vfio_info_cap *caps)
+{
+       u64 bar_len = pci_resource_len(vdev->pdev, index);
+       u64 excl_start = ALIGN_DOWN(vdev->mmap_exclude_start, PAGE_SIZE);
+       u64 excl_end = ALIGN(vdev->mmap_exclude_start + vdev->mmap_exclude_len,
+                            PAGE_SIZE);
+       struct vfio_region_info_cap_sparse_mmap *sparse;
+       int nr_areas = 0, i = 0, ret;
+       size_t size;
+
+       /*
+        * mmap is page granular, so the mmappable areas must stop at the page
+        * boundaries enclosing the excluded sub-range. The byte-granular
+        * exclusion still governs the fault and read/write paths; only the
+        * advertised mmap areas round out to whole pages.
+        */
+       if (excl_start > 0)
+               nr_areas++;
+       if (excl_end < bar_len)
+               nr_areas++;
+
+       size = struct_size(sparse, areas, nr_areas);
+       sparse = kzalloc(size, GFP_KERNEL);
+       if (!sparse)
+               return -ENOMEM;
+
+       sparse->header.id = VFIO_REGION_INFO_CAP_SPARSE_MMAP;
+       sparse->header.version = 1;
+       sparse->nr_areas = nr_areas;
+
+       if (excl_start > 0) {
+               sparse->areas[i].offset = 0;
+               sparse->areas[i].size = excl_start;
+               i++;
+       }
+       if (excl_end < bar_len) {
+               sparse->areas[i].offset = excl_end;
+               sparse->areas[i].size = bar_len - excl_end;
+       }
+
+       ret = vfio_info_add_capability(caps, &sparse->header, size);
+       kfree(sparse);
+       return ret;
+}
+
 int vfio_pci_core_register_dev_region(struct vfio_pci_core_device *vdev,
                                      unsigned int type, unsigned int subtype,
                                      const struct vfio_pci_regops *ops,
@@ -1157,6 +1218,13 @@ int vfio_pci_ioctl_get_region_info(struct vfio_device 
*core_vdev,
                                if (ret)
                                        return ret;
                        }
+                       if (vdev->mmap_exclude_len &&
+                           info->index == vdev->mmap_exclude_bar) {
+                               ret = vfio_pci_mmap_exclude_cap(vdev, 
info->index,
+                                                               caps);
+                               if (ret)
+                                       return ret;
+                       }
                }
 
                break;
@@ -1851,6 +1919,10 @@ int vfio_pci_core_mmap(struct vfio_device *core_vdev, 
struct vm_area_struct *vma
        if (req_start + req_len > phys_len)
                return -EINVAL;
 
+       /* An excluded sub-range is reachable only through its trap, not mmap. 
*/
+       if (vfio_pci_bar_is_excluded(vdev, index, req_start, req_len))
+               return -EINVAL;
+
        /*
         * Ensure the BAR resource region is reserved for use.
         */
diff --git a/drivers/vfio/pci/vfio_pci_dmabuf.c 
b/drivers/vfio/pci/vfio_pci_dmabuf.c
index c16f460c01d6..51983105d38b 100644
--- a/drivers/vfio/pci/vfio_pci_dmabuf.c
+++ b/drivers/vfio/pci/vfio_pci_dmabuf.c
@@ -177,11 +177,24 @@ int vfio_pci_core_get_dmabuf_phys(struct 
vfio_pci_core_device *vdev,
                                  size_t nr_ranges)
 {
        struct pci_dev *pdev = vdev->pdev;
+       unsigned int i;
 
        *provider = pcim_p2pdma_provider(pdev, region_index);
        if (!*provider)
                return -EINVAL;
 
+       /*
+        * A provider (e.g. vfio-cxl) can exclude a BAR sub-range that must be
+        * reached only through its trap. The mmap and read/write paths already
+        * refuse it; reject a DMA-BUF export overlapping it too, so a device fd
+        * holder cannot map the excluded registers to a peer and bypass the 
trap.
+        */
+       for (i = 0; i < nr_ranges; i++)
+               if (vfio_pci_bar_is_excluded(vdev, region_index,
+                                            dma_ranges[i].offset,
+                                            dma_ranges[i].length))
+                       return -EINVAL;
+
        return vfio_pci_core_fill_phys_vec(
                phys_vec, dma_ranges, nr_ranges,
                pci_resource_start(pdev, region_index),
diff --git a/drivers/vfio/pci/vfio_pci_priv.h b/drivers/vfio/pci/vfio_pci_priv.h
index fca9d0dfac90..902d17815ab6 100644
--- a/drivers/vfio/pci/vfio_pci_priv.h
+++ b/drivers/vfio/pci/vfio_pci_priv.h
@@ -44,6 +44,19 @@ ssize_t vfio_pci_config_rw_single(struct 
vfio_pci_core_device *vdev,
 ssize_t vfio_pci_bar_rw(struct vfio_pci_core_device *vdev, char __user *buf,
                        size_t count, loff_t *ppos, bool iswrite);
 
+/*
+ * A provider (e.g. vfio-cxl) can carve a sub-range out of a BAR that must be
+ * reached only through its trap, never the direct BAR.  Returns true when
+ * [start, start + len) on this BAR overlaps that excluded range.
+ */
+static inline bool vfio_pci_bar_is_excluded(struct vfio_pci_core_device *vdev,
+                                           int bar, u64 start, u64 len)
+{
+       return vdev->mmap_exclude_len && bar == vdev->mmap_exclude_bar &&
+              start < vdev->mmap_exclude_start + vdev->mmap_exclude_len &&
+              start + len > vdev->mmap_exclude_start;
+}
+
 #ifdef CONFIG_VFIO_PCI_VGA
 ssize_t vfio_pci_vga_rw(struct vfio_pci_core_device *vdev, char __user *buf,
                        size_t count, loff_t *ppos, bool iswrite);
diff --git a/include/linux/vfio_pci_core.h b/include/linux/vfio_pci_core.h
index 117cd67995d8..43755b91880f 100644
--- a/include/linux/vfio_pci_core.h
+++ b/include/linux/vfio_pci_core.h
@@ -162,6 +162,10 @@ struct vfio_pci_core_device {
        struct notifier_block   nb;
        struct rw_semaphore     memory_lock;
        struct list_head        dmabufs;
+       /* BAR sub-range a provider keeps off mmap, reached only through a trap 
*/
+       int                     mmap_exclude_bar;
+       u64                     mmap_exclude_start;
+       u64                     mmap_exclude_len;
 };
 
 enum vfio_pci_io_width {
@@ -176,6 +180,8 @@ int vfio_pci_core_register_dev_region(struct 
vfio_pci_core_device *vdev,
                                      unsigned int type, unsigned int subtype,
                                      const struct vfio_pci_regops *ops,
                                      size_t size, u32 flags, void *data);
+void vfio_pci_core_set_mmap_exclude(struct vfio_pci_core_device *vdev, int bar,
+                                   u64 start, u64 len);
 void vfio_pci_core_close_device(struct vfio_device *core_vdev);
 int vfio_pci_core_init_dev(struct vfio_device *core_vdev);
 void vfio_pci_core_release_dev(struct vfio_device *core_vdev);
-- 
2.25.1


Reply via email to