From: Manish Honap <[email protected]> Exercise the vfio-cxl contract on a bound CXL Type-2 device: the two VFIO regions and the geometry capability, the HDM memory mmap (including a 2 MB huge fault), the dword-aligned trapped decoder block, and the lock-on-commit FSM. The decoder writes land in the per-open shadow and each test reopens the device, so the FSM tests repeat cleanly.
Cover the HDM memory two ways: a host-CPU load/store of the mmap, and the path a VMM actually uses, mmap plus a stage-2 IOAS map for the device's ATS access. The mmap flag is required for the IOAS path, so assert it is advertised rather than skipping when it is absent. Signed-off-by: Manish Honap <[email protected]> --- MAINTAINERS | 1 + tools/testing/selftests/vfio/Makefile | 1 + .../selftests/vfio/lib/vfio_pci_device.c | 57 +- .../selftests/vfio/vfio_cxl_type2_test.c | 799 ++++++++++++++++++ 4 files changed, 855 insertions(+), 3 deletions(-) create mode 100644 tools/testing/selftests/vfio/vfio_cxl_type2_test.c diff --git a/MAINTAINERS b/MAINTAINERS index b9361a8d618e..192b1681b3bd 100644 --- a/MAINTAINERS +++ b/MAINTAINERS @@ -28319,6 +28319,7 @@ L: [email protected] S: Supported F: Documentation/driver-api/vfio-pci-cxl.rst F: drivers/vfio/pci/cxl/ +F: tools/testing/selftests/vfio/vfio_cxl_type2_test.c VFIO DRIVER M: Alex Williamson <[email protected]> diff --git a/tools/testing/selftests/vfio/Makefile b/tools/testing/selftests/vfio/Makefile index 2c32c48db509..08f88e88cb4d 100644 --- a/tools/testing/selftests/vfio/Makefile +++ b/tools/testing/selftests/vfio/Makefile @@ -13,6 +13,7 @@ TEST_GEN_PROGS += vfio_pci_device_test TEST_GEN_PROGS += vfio_pci_device_init_perf_test TEST_GEN_PROGS += vfio_pci_driver_test TEST_GEN_PROGS += vfio_pci_sriov_uapi_test +TEST_GEN_PROGS += vfio_cxl_type2_test TEST_FILES += scripts/cleanup.sh TEST_FILES += scripts/lib.sh diff --git a/tools/testing/selftests/vfio/lib/vfio_pci_device.c b/tools/testing/selftests/vfio/lib/vfio_pci_device.c index 94dc5fcecbeb..ab49b41653c4 100644 --- a/tools/testing/selftests/vfio/lib/vfio_pci_device.c +++ b/tools/testing/selftests/vfio/lib/vfio_pci_device.c @@ -160,9 +160,31 @@ static void vfio_pci_region_get(struct vfio_pci_device *device, int index, ioctl_assert(device->fd, VFIO_DEVICE_GET_REGION_INFO, info); } +/* Return the sparse-mmap capability in @info, or NULL if the region has none. */ +static struct vfio_region_info_cap_sparse_mmap * +vfio_pci_sparse_mmap_cap(struct vfio_region_info *info) +{ + struct vfio_info_cap_header *hdr; + u32 offset; + + if (!(info->flags & VFIO_REGION_INFO_FLAG_CAPS)) + return NULL; + + for (offset = info->cap_offset; offset; offset = hdr->next) { + hdr = (void *)info + offset; + if (hdr->id == VFIO_REGION_INFO_CAP_SPARSE_MMAP) + return (struct vfio_region_info_cap_sparse_mmap *)hdr; + } + + return NULL; +} + static void vfio_pci_bar_map(struct vfio_pci_device *device, int index) { struct vfio_pci_bar *bar = &device->bars[index]; + struct vfio_region_info_cap_sparse_mmap *sparse; + u8 infobuf[1024] = {}; + struct vfio_region_info *info = (void *)infobuf; size_t align, size; int prot = 0; void *vaddr; @@ -190,9 +212,38 @@ static void vfio_pci_bar_map(struct vfio_pci_device *device, int index) align = min_t(size_t, size, SZ_1G); vaddr = mmap_reserve(size, align, 0); - bar->vaddr = mmap(vaddr, size, prot, MAP_SHARED | MAP_FIXED, - device->fd, bar->info.offset); - VFIO_ASSERT_NE(bar->vaddr, MAP_FAILED); + + /* + * A BAR that is only partially mmappable, such as a CXL Type-2 component + * BAR with the HDM decoder block trapped, advertises the mmappable + * ranges through a sparse-mmap capability. Map each area within the + * reservation and leave the excluded ranges unmapped; mapping the whole + * BAR would be rejected. + */ + info->argsz = sizeof(infobuf); + info->index = index; + ioctl_assert(device->fd, VFIO_DEVICE_GET_REGION_INFO, info); + sparse = vfio_pci_sparse_mmap_cap(info); + if (sparse) { + u32 i; + + bar->vaddr = vaddr; + for (i = 0; i < sparse->nr_areas; i++) { + void *p; + + if (!sparse->areas[i].size) + continue; + p = mmap(vaddr + sparse->areas[i].offset, + sparse->areas[i].size, prot, + MAP_SHARED | MAP_FIXED, device->fd, + bar->info.offset + sparse->areas[i].offset); + VFIO_ASSERT_NE(p, MAP_FAILED); + } + } else { + bar->vaddr = mmap(vaddr, size, prot, MAP_SHARED | MAP_FIXED, + device->fd, bar->info.offset); + VFIO_ASSERT_NE(bar->vaddr, MAP_FAILED); + } madvise(bar->vaddr, size, MADV_HUGEPAGE); } diff --git a/tools/testing/selftests/vfio/vfio_cxl_type2_test.c b/tools/testing/selftests/vfio/vfio_cxl_type2_test.c new file mode 100644 index 000000000000..8c23ddd014ca --- /dev/null +++ b/tools/testing/selftests/vfio/vfio_cxl_type2_test.c @@ -0,0 +1,799 @@ +// SPDX-License-Identifier: GPL-2.0-only +/* + * vfio_cxl_type2_test - corner-case tests for the vfio-cxl kernel contract. + * + * Exercises the user-visible surface the vfio-cxl module adds to a CXL Type-2 + * device: the two VFIO regions (HDM memory and the trapped HDM decoder block), + * the component-register geometry capability, and the lock-on-commit decoder + * FSM the kernel runs on the trapped block. + * + * Unlike a plain vfio-pci device the guest programs its own endpoint decoder, + * so the trapped block enforces the commit handshake and freezes a locked + * decoder. These tests drive that FSM directly. Writes to the decoder block + * land in the per-open kernel shadow only, never on the physical decoder, and + * each test reopens the device (fresh shadow), so the FSM tests are safe to + * repeat and do not leak state between tests. + * + * Usage: ./vfio_cxl_type2_test <BDF> (or export VFIO_SELFTESTS_BDF=<BDF>). + * The device must be bound to vfio-pci with the vfio-cxl module available. + * + * Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. + */ + +#include <fcntl.h> +#include <stdint.h> +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include <unistd.h> + +#include <sys/ioctl.h> +#include <sys/mman.h> + +#include <linux/pci_regs.h> +#include <linux/sizes.h> +#include <linux/vfio.h> + +#include <cxl/cxl_regs.h> + +#include <libvfio.h> + +#include "kselftest_harness.h" + +#define PCI_DVSEC_VENDOR_ID_CXL 0x1e98 +#define PCI_DVSEC_ID_CXL_DEVICE 0x0000 + +/* CXL r3.1 8.1.9.1: Register Block Identifier for the component registers. */ +#define CXL_REGLOC_RBI_COMPONENT 1 + +/* + * Register Locator DVSEC block-1 field masks. The uapi pci_regs.h names expand + * to __GENMASK(), which is not a macro in this userspace include path, so use + * explicit values. + */ +#define REG_LOCATOR_BIR_MASK 0x00000007 +#define REG_LOCATOR_BLOCK_ID_MASK 0x0000ff00 +#define REG_LOCATOR_BLOCK_OFF_LOW_MASK 0xffff0000 + +/* + * vfio-pci's region-offset packing is kernel-internal (vfio_pci_core.h), not + * UAPI. Define it locally; the guards let a future kernel hoist it to UAPI. + */ +#ifndef VFIO_PCI_OFFSET_SHIFT +#define VFIO_PCI_OFFSET_SHIFT 40 +#endif +#ifndef VFIO_PCI_INDEX_TO_OFFSET +#define VFIO_PCI_INDEX_TO_OFFSET(i) ((uint64_t)(i) << VFIO_PCI_OFFSET_SHIFT) +#endif + +static const char *device_bdf; + +/* Locate a region-info capability by id inside a GET_REGION_INFO buffer. */ +static const struct vfio_info_cap_header * +find_region_cap(const void *buf, size_t bufsz, uint16_t id) +{ + const struct vfio_region_info *ri = buf; + const struct vfio_info_cap_header *cap; + size_t off = ri->cap_offset; + + while (off && off + sizeof(*cap) <= bufsz) { + cap = (const void *)((const char *)buf + off); + if (cap->id == id) + return cap; + off = cap->next; + } + return NULL; +} + +/* + * Find a CXL region by scanning every region's VFIO_REGION_INFO_CAP_TYPE for + * the CXL type and the requested subtype. Returns the region index or -1. + * @buf is a caller scratch buffer left holding the matched region's info + * (with caps). + */ +static int find_cxl_region(int fd, uint32_t nregions, uint32_t subtype, + void *buf, size_t bufsz) +{ + uint32_t i; + + for (i = 0; i < nregions; i++) { + struct vfio_region_info *ri = buf; + const struct vfio_region_info_cap_type *t; + const struct vfio_info_cap_header *hdr; + + memset(buf, 0, bufsz); + ri->argsz = bufsz; + ri->index = i; + if (ioctl(fd, VFIO_DEVICE_GET_REGION_INFO, ri)) + continue; + if (!(ri->flags & VFIO_REGION_INFO_FLAG_CAPS)) + continue; + + hdr = find_region_cap(buf, bufsz, VFIO_REGION_INFO_CAP_TYPE); + if (!hdr) + continue; + t = (const void *)hdr; + if (t->type == VFIO_REGION_TYPE_CXL && t->subtype == subtype) + return i; + } + return -1; +} + +/* Walk the PCI extended capability list for the CXL Device DVSEC. */ +static uint16_t find_cxl_dvsec(struct vfio_pci_device *dev) +{ + uint16_t pos = PCI_CFG_SPACE_SIZE; + int iter = 0; + + while (pos && iter++ < 64) { + uint32_t hdr = vfio_pci_config_readl(dev, pos); + uint16_t cap_id = hdr & 0xffff; + uint16_t next = (hdr >> 20) & 0xffc; + uint32_t h1, h2; + + if (cap_id == PCI_EXT_CAP_ID_DVSEC) { + h1 = vfio_pci_config_readl(dev, pos + 4); + h2 = vfio_pci_config_readl(dev, pos + 8); + if ((h1 & 0xffff) == PCI_DVSEC_VENDOR_ID_CXL && + (h2 & 0xffff) == PCI_DVSEC_ID_CXL_DEVICE) + return pos; + } + pos = next; + } + return 0; +} + +FIXTURE(vfio_cxl) { + struct iommu *iommu; + struct vfio_pci_device *dev; + + int mem_idx; + uint64_t mem_size; + uint32_t mem_flags; + int comp_idx; + uint64_t comp_size; + uint32_t comp_bar; + uint64_t comp_offset; /* HDM block offset within comp_bar */ + uint64_t comp_off; /* mmap/rw base offset of the comp region */ + uint16_t dvsec; +}; + +FIXTURE_SETUP(vfio_cxl) +{ + uint8_t infobuf[512] = {}; + struct vfio_device_info *info = (void *)infobuf; + const struct vfio_region_info_cap_cxl_comp_regs *geo; + const struct vfio_info_cap_header *hdr; + uint8_t rbuf[1024]; + + self->iommu = iommu_init(default_iommu_mode); + self->dev = vfio_pci_device_init(device_bdf, self->iommu); + + info->argsz = sizeof(infobuf); + ASSERT_EQ(0, ioctl(self->dev->fd, VFIO_DEVICE_GET_INFO, info)); + + if (!(info->flags & VFIO_DEVICE_FLAGS_CXL)) + SKIP(return, "not a CXL Type-2 device"); + + self->mem_idx = find_cxl_region(self->dev->fd, info->num_regions, + VFIO_REGION_SUBTYPE_CXL_MEM, + rbuf, sizeof(rbuf)); + ASSERT_GE(self->mem_idx, 0); + self->mem_size = ((struct vfio_region_info *)rbuf)->size; + self->mem_flags = ((struct vfio_region_info *)rbuf)->flags; + + self->comp_idx = find_cxl_region(self->dev->fd, info->num_regions, + VFIO_REGION_SUBTYPE_CXL_COMP_REGS, + rbuf, sizeof(rbuf)); + ASSERT_GE(self->comp_idx, 0); + self->comp_size = ((struct vfio_region_info *)rbuf)->size; + + /* The geometry cap rides on the component-register region. */ + hdr = find_region_cap(rbuf, sizeof(rbuf), + VFIO_REGION_INFO_CAP_CXL_COMP_REGS); + ASSERT_NE(NULL, hdr); + geo = (const void *)hdr; + self->comp_bar = geo->bar; + self->comp_offset = geo->offset; + + self->comp_off = VFIO_PCI_INDEX_TO_OFFSET(self->comp_idx); + self->dvsec = find_cxl_dvsec(self->dev); +} + +FIXTURE_TEARDOWN(vfio_cxl) +{ + vfio_pci_device_cleanup(self->dev); + iommu_cleanup(self->iommu); +} + +/* GET_INFO advertises the flag and both CXL regions with a sane geometry cap. */ +TEST_F(vfio_cxl, device_is_cxl) +{ + ASSERT_NE(self->mem_idx, self->comp_idx); + ASSERT_GT(self->mem_size, 0); + ASSERT_GT(self->comp_size, 0); + ASSERT_LT(self->comp_bar, PCI_STD_NUM_BARS); + /* The HDM memory must advertise mmap; a VMM needs it for stage-2. */ + ASSERT_NE(0, self->mem_flags & VFIO_REGION_INFO_FLAG_MMAP); +} + +/* + * The component BAR carries the physical HDM decoder block, which vfio traps + * and excludes from mmap so the guest cannot reprogram it directly. Mapping the + * whole BAR must fail; mapping the ranges around the excluded block, as the + * sparse-mmap capability advertises, must succeed. + */ +TEST_F(vfio_cxl, comp_bar_sparse_mmap) +{ + size_t page_size = getpagesize(); + uint8_t rbuf[1024] = {}; + struct vfio_region_info *ri = (void *)rbuf; + const struct vfio_region_info_cap_sparse_mmap *sm; + const struct vfio_info_cap_header *hdr; + uint64_t bar_off, decoder_page; + void *map; + uint32_t i; + + /* Region info for the component BAR, with capabilities. */ + ri->argsz = sizeof(rbuf); + ri->index = self->comp_bar; + ASSERT_EQ(0, ioctl(self->dev->fd, VFIO_DEVICE_GET_REGION_INFO, ri)); + ASSERT_NE(0, ri->flags & VFIO_REGION_INFO_FLAG_MMAP); + bar_off = ri->offset; + + /* The trapped decoder block splits the BAR, so it must be sparse. */ + hdr = find_region_cap(rbuf, sizeof(rbuf), + VFIO_REGION_INFO_CAP_SPARSE_MMAP); + ASSERT_NE(NULL, hdr); + sm = (const void *)hdr; + ASSERT_GT(sm->nr_areas, 0); + + /* Mapping the whole BAR must fail: it covers the excluded block. */ + map = mmap(NULL, ri->size, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, bar_off); + ASSERT_EQ(MAP_FAILED, map); + + /* Every advertised area is page aligned and must map. */ + for (i = 0; i < sm->nr_areas; i++) { + uint64_t ao = sm->areas[i].offset; + uint64_t as = sm->areas[i].size; + + if (!as) + continue; + ASSERT_EQ(0, ao & (page_size - 1)); + ASSERT_EQ(0, as & (page_size - 1)); + + map = mmap(NULL, as, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, bar_off + ao); + ASSERT_NE(MAP_FAILED, map); + ASSERT_EQ(0, munmap(map, as)); + } + + /* The page holding the decoder block must never be mmappable. */ + decoder_page = self->comp_offset & ~(uint64_t)(page_size - 1); + map = mmap(NULL, page_size, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, bar_off + decoder_page); + ASSERT_EQ(MAP_FAILED, map); +} + +/* mmap one page of the HDM memory, write a pattern, read it back. */ +TEST_F(vfio_cxl, hdm_mem_mmap_rw) +{ + uint64_t off = VFIO_PCI_INDEX_TO_OFFSET(self->mem_idx); + uint32_t pattern = 0xdeadbeefU, readback = 0; + void *map; + + if (self->mem_size < SZ_4K) + SKIP(return, "HDM memory < 4K"); + + map = mmap(NULL, SZ_4K, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, off); + ASSERT_NE(MAP_FAILED, map); + + memcpy(map, &pattern, sizeof(pattern)); + memcpy(&readback, map, sizeof(readback)); + ASSERT_EQ(pattern, readback); + + ASSERT_EQ(0, munmap(map, SZ_4K)); +} + +/* + * A 2 MB-aligned window should map as a huge (PMD) fault. The kernel falls back + * to base pages when it cannot, so only correctness (write/read) is asserted. + */ +TEST_F(vfio_cxl, hdm_mem_huge_mmap) +{ + uint64_t off = VFIO_PCI_INDEX_TO_OFFSET(self->mem_idx); + uint32_t pattern = 0x5a5a5a5aU, readback = 0; + void *map, *last; + + if (self->mem_size < SZ_2M) + SKIP(return, "HDM memory < 2M"); + + map = mmap(NULL, SZ_2M, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, off); + ASSERT_NE(MAP_FAILED, map); + + /* Touch the last dword so a 2 MB PMD fault covers the whole window. */ + last = (char *)map + SZ_2M - sizeof(pattern); + memcpy(last, &pattern, sizeof(pattern)); + memcpy(&readback, last, sizeof(readback)); + ASSERT_EQ(pattern, readback); + + ASSERT_EQ(0, munmap(map, SZ_2M)); +} + +/* + * A guest driver disables and re-enables PCI Memory-Space during init and + * reset. The committed HDM decoder stays valid across that toggle, so once + * Memory-Space is re-enabled the coherent HDM memory must be reachable again + * without a reset. This is the regression test for the hdm_valid access gate + * being cleared by a Memory-Space disable and never restored, which left a + * later valid mmap fault wrongly SIGBUS-ing. + * + * The region is exercised only through the mmap path (as a VMM does) and only + * while Memory-Space is enabled. An access with Memory-Space disabled aborts + * on the fabric as a fatal host error, so the test never attempts one: the + * toggle in between is pure config-space writes. + */ +TEST_F(vfio_cxl, hdm_mem_survives_mem_space_toggle) +{ + uint64_t off = VFIO_PCI_INDEX_TO_OFFSET(self->mem_idx); + uint32_t pattern = 0x12345678U, readback = 0; + uint16_t cmd; + void *map; + + if (self->mem_size < SZ_4K) + SKIP(return, "HDM memory < 4K"); + + /* Seed a known pattern through the mmap path with Memory-Space on. */ + cmd = vfio_pci_config_readw(self->dev, PCI_COMMAND); + vfio_pci_config_writew(self->dev, PCI_COMMAND, + cmd | PCI_COMMAND_MEMORY); + map = mmap(NULL, SZ_4K, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, off); + ASSERT_NE(MAP_FAILED, map); + memcpy(map, &pattern, sizeof(pattern)); + ASSERT_EQ(0, munmap(map, SZ_4K)); + + /* + * Toggle Memory-Space off and back on with no HDM access in between, + * as a guest driver does during init/reset. + */ + vfio_pci_config_writew(self->dev, PCI_COMMAND, + cmd & ~PCI_COMMAND_MEMORY); + vfio_pci_config_writew(self->dev, PCI_COMMAND, + cmd | PCI_COMMAND_MEMORY); + + /* + * The committed decoder stayed valid across the toggle, so a fresh mmap + * fault succeeds and the seeded pattern reads back, without a reset. + * Before the fix the gate was cleared by the disable and never restored, + * so the fault wrongly SIGBUS-ed. + */ + map = mmap(NULL, SZ_4K, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, off); + ASSERT_NE(MAP_FAILED, map); + memcpy(&readback, map, sizeof(readback)); + ASSERT_EQ(pattern, readback); + ASSERT_EQ(0, munmap(map, SZ_4K)); + + /* Restore PCI_COMMAND. */ + vfio_pci_config_writew(self->dev, PCI_COMMAND, cmd); +} + +/* + * Mirror how a VMM uses the region: mmap the HDM memory and map it into the + * IOAS (stage-2) so the device can reach it over ATS. The mmap flag is required + * for that path, so its absence is a failure, not a skip. The host CPU does not + * dereference the mapping; the guest reaches it through stage-2. + */ +TEST_F(vfio_cxl, hdm_mem_ioas_map) +{ + uint64_t off = VFIO_PCI_INDEX_TO_OFFSET(self->mem_idx); + struct iova_allocator *iova_alloc; + struct dma_region region; + void *map; + + ASSERT_NE(0, self->mem_flags & VFIO_REGION_INFO_FLAG_MMAP); + + /* iova_allocator_alloc() requires a power-of-2 size. */ + if (self->mem_size < SZ_2M) + SKIP(return, "HDM memory < 2M"); + + map = mmap(NULL, SZ_2M, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, off); + ASSERT_NE(MAP_FAILED, map); + + iova_alloc = iova_allocator_init(self->iommu); + region.vaddr = map; + region.size = SZ_2M; + region.iova = iova_allocator_alloc(iova_alloc, SZ_2M); + + iommu_map(self->iommu, ®ion); + iommu_unmap(self->iommu, ®ion); + + iova_allocator_cleanup(iova_alloc); + ASSERT_EQ(0, munmap(map, SZ_2M)); +} + +/* The trapped block starts at the HDM decoder registers; CTRL 0 reads back. */ +TEST_F(vfio_cxl, comp_regs_hdm_read) +{ + uint64_t ctrl = self->comp_off + CXL_HDM_DECODER0_CTRL_OFFSET(0); + uint32_t val = 0; + + ASSERT_GE(self->comp_size, CXL_HDM_DECODER0_CTRL_OFFSET(0) + 4); + ASSERT_EQ((ssize_t)sizeof(val), + pread(self->dev->fd, &val, sizeof(val), ctrl)); +} + +/* The decoder registers only take aligned dword accesses. */ +TEST_F(vfio_cxl, comp_regs_reject_unaligned) +{ + uint32_t val = 0; + uint16_t half = 0; + + /* Unaligned offset. */ + ASSERT_EQ(-1, pread(self->dev->fd, &val, sizeof(val), + self->comp_off + 1)); + /* Non-dword size. */ + ASSERT_EQ(-1, pread(self->dev->fd, &half, sizeof(half), + self->comp_off)); +} + +/* Accesses past the region end are rejected. */ +TEST_F(vfio_cxl, comp_regs_reject_out_of_range) +{ + uint32_t val = 0; + + ASSERT_EQ(-1, pread(self->dev->fd, &val, sizeof(val), + self->comp_off + self->comp_size)); +} + +/* + * Commit handshake. This series supports only a firmware committed+locked + * decoder, which is the state the device boots in: the host resolved the HPA + * before the guest saw the device, so the guest view is frozen until a reset + * and a decommit request is ignored. If a decoder is ever seen uncommitted, the + * shadow FSM instead lets a commit request reach COMMITTED and a clear tear it + * back down; assert whichever contract applies to the decoder's actual state. + */ +TEST_F(vfio_cxl, hdm_commit_fsm) +{ + uint64_t ctrl = self->comp_off + CXL_HDM_DECODER0_CTRL_OFFSET(0); + uint32_t v, orig; + + ASSERT_EQ((ssize_t)sizeof(orig), + pread(self->dev->fd, &orig, sizeof(orig), ctrl)); + + if ((orig & CXL_HDM_DECODER0_CTRL_COMMITTED) && + (orig & CXL_HDM_DECODER0_CTRL_LOCK)) { + /* Committed+locked: a decommit request must be ignored. */ + v = orig & ~CXL_HDM_DECODER0_CTRL_COMMIT; + ASSERT_EQ((ssize_t)sizeof(v), + pwrite(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_EQ((ssize_t)sizeof(v), + pread(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_TRUE(v & CXL_HDM_DECODER0_CTRL_COMMITTED); + return; + } + + /* Uncommitted: the commit handshake round-trips through the shadow. */ + v = orig | CXL_HDM_DECODER0_CTRL_COMMIT; + ASSERT_EQ((ssize_t)sizeof(v), + pwrite(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_EQ((ssize_t)sizeof(v), + pread(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_TRUE(v & CXL_HDM_DECODER0_CTRL_COMMITTED); + + v &= ~CXL_HDM_DECODER0_CTRL_COMMIT; + ASSERT_EQ((ssize_t)sizeof(v), + pwrite(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_EQ((ssize_t)sizeof(v), + pread(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_FALSE(v & CXL_HDM_DECODER0_CTRL_COMMITTED); +} + +/* + * Lock on commit: a decoder committed with LOCK set is frozen until reset, so + * a later attempt to clear COMMIT is ignored. The freeze lives in the per-open + * shadow, so the next test's reopen starts clean. + */ +TEST_F(vfio_cxl, hdm_lock_on_commit) +{ + uint64_t ctrl = self->comp_off + CXL_HDM_DECODER0_CTRL_OFFSET(0); + uint32_t v; + + v = CXL_HDM_DECODER0_CTRL_COMMIT | CXL_HDM_DECODER0_CTRL_LOCK; + ASSERT_EQ((ssize_t)sizeof(v), + pwrite(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_EQ((ssize_t)sizeof(v), + pread(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_TRUE(v & CXL_HDM_DECODER0_CTRL_COMMITTED); + ASSERT_TRUE(v & CXL_HDM_DECODER0_CTRL_LOCK); + + /* Attempt to decommit the locked decoder; it must stay committed. */ + v = 0; + ASSERT_EQ((ssize_t)sizeof(v), + pwrite(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_EQ((ssize_t)sizeof(v), + pread(self->dev->fd, &v, sizeof(v), ctrl)); + ASSERT_TRUE(v & CXL_HDM_DECODER0_CTRL_COMMITTED); + ASSERT_TRUE(v & CXL_HDM_DECODER0_CTRL_LOCK); +} + +/* + * A base written to the trapped block round-trips through the shadow only while + * the decoder is uncommitted. In the supported production state the decoder is + * already committed, so its base is read-only and the write is ignored; assert + * whichever contract applies to the decoder's actual state. + */ +TEST_F(vfio_cxl, hdm_base_shadow_roundtrip) +{ + uint64_t lo_off = self->comp_off + CXL_HDM_DECODER0_BASE_LOW_OFFSET(0); + uint64_t ctrl_off = self->comp_off + CXL_HDM_DECODER0_CTRL_OFFSET(0); + uint32_t v = 0x30000000U; /* 256 MB-aligned low base bits */ + uint32_t rb = 0, orig = 0, ctrl = 0; + + ASSERT_GE(self->comp_size, CXL_HDM_DECODER0_BASE_LOW_OFFSET(0) + 4); + ASSERT_EQ((ssize_t)sizeof(ctrl), + pread(self->dev->fd, &ctrl, sizeof(ctrl), ctrl_off)); + ASSERT_EQ((ssize_t)sizeof(orig), + pread(self->dev->fd, &orig, sizeof(orig), lo_off)); + + ASSERT_EQ((ssize_t)sizeof(v), + pwrite(self->dev->fd, &v, sizeof(v), lo_off)); + ASSERT_EQ((ssize_t)sizeof(rb), + pread(self->dev->fd, &rb, sizeof(rb), lo_off)); + + if (ctrl & CXL_HDM_DECODER0_CTRL_COMMITTED) { + /* Committed decoder holds its base read-only; write ignored. */ + ASSERT_EQ(orig, rb); + } else { + /* Uncommitted decoder accepts the base into the shadow. */ + ASSERT_EQ(v, rb); + } +} + +/* + * The CXL Device DVSEC body is virtualized by the kernel; a config read of it + * is served from the shadow (no SIGBUS / -EIO). The value itself is + * firmware-dependent, so only success is asserted. + */ +TEST_F(vfio_cxl, dvsec_body_read) +{ + uint32_t v; + + if (!self->dvsec) + SKIP(return, "CXL Device DVSEC not found"); + + v = vfio_pci_config_readl(self->dev, self->dvsec + PCI_DVSEC_HEADER1); + ASSERT_NE(0xffffffffU, v); +} + +/* + * Guest-initiated CXL reset: a 0->1 write of Initiate_CXL_Reset in the DVSEC + * asks the kernel to run the reset sequence. The bit self-clears and STATUS2 + * reports the outcome. This is the path a guest drives through QEMU, so the + * kernel must complete it and report RESET_COMPLETE, not RESET_ERROR. + */ +TEST_F(vfio_cxl, guest_cxl_reset) +{ + uint16_t cap, ctrl2, status2; + + if (!self->dvsec) + SKIP(return, "CXL Device DVSEC not found"); + + cap = vfio_pci_config_readw(self->dev, self->dvsec + PCI_DVSEC_CXL_CAP); + if (!(cap & PCI_DVSEC_CXL_RST_CAPABLE)) + SKIP(return, "device does not support CXL reset"); + + /* Request the reset through the control register. */ + ctrl2 = vfio_pci_config_readw(self->dev, + self->dvsec + PCI_DVSEC_CXL_CTRL2); + vfio_pci_config_writew(self->dev, self->dvsec + PCI_DVSEC_CXL_CTRL2, + ctrl2 | PCI_DVSEC_CXL_INIT_CXL_RST); + + /* Initiate_CXL_Reset self-clears once the sequence has run. */ + ctrl2 = vfio_pci_config_readw(self->dev, + self->dvsec + PCI_DVSEC_CXL_CTRL2); + ASSERT_FALSE(ctrl2 & PCI_DVSEC_CXL_INIT_CXL_RST); + + /* STATUS2 reports the outcome; a completed reset sets RESET_COMPLETE. */ + status2 = vfio_pci_config_readw(self->dev, + self->dvsec + PCI_DVSEC_CXL_STATUS2); + ASSERT_TRUE(status2 & PCI_DVSEC_CXL_RST_DONE); + ASSERT_FALSE(status2 & PCI_DVSEC_CXL_RST_ERR); +} + +/* + * The component BAR is reachable by fd read everywhere except the trapped + * decoder block, which is served only through the comp-regs region. A VMM + * relies on this split when it forwards accesses that land in the excluded + * mmap page: non-decoder bytes go to the BAR, the decoder goes to the trap. + */ +TEST_F(vfio_cxl, comp_bar_rdwr_split) +{ + uint64_t bar_off = VFIO_PCI_INDEX_TO_OFFSET(self->comp_bar); + uint32_t val; + + /* A non-decoder dword of the BAR reads back through the fd. */ + ASSERT_EQ((ssize_t)sizeof(val), + pread(self->dev->fd, &val, sizeof(val), bar_off)); + + /* The trapped decoder block is off-limits to raw BAR fd access. */ + ASSERT_EQ(-1, pread(self->dev->fd, &val, sizeof(val), + bar_off + self->comp_offset)); +} + +/* + * Mirror the guest's HDM discovery: a guest does NOT use VFIO's geometry cap. + * It finds the CXL Register Locator DVSEC in config space, takes the component + * BAR and offset from it, walks the component-register capability array over + * the BAR to locate the HDM decoder, then reads decoder 0's base/size/ctrl and + * derives the memory range from a committed decoder. + */ +TEST_F(vfio_cxl, guest_hdm_discovery) +{ + uint64_t bar_off = VFIO_PCI_INDEX_TO_OFFSET(self->comp_bar); + uint32_t reg_lo, reg_hi, cap_array, cap_count, hdr; + uint32_t bl, bh, sl, sh, ctrl; + uint64_t block_off, cm, hdm_off = 0, base, size; + uint16_t pos = PCI_CFG_SPACE_SIZE, regloc = 0, block1; + int iter = 0, bar, i; + + /* 1. Find the CXL Register Locator DVSEC in config space. */ + while (pos && iter++ < 64) { + uint32_t h = vfio_pci_config_readl(self->dev, pos); + + if ((h & 0xffff) == PCI_EXT_CAP_ID_DVSEC) { + uint32_t h1 = vfio_pci_config_readl(self->dev, pos + 4); + uint32_t h2 = vfio_pci_config_readl(self->dev, pos + 8); + + if ((h1 & 0xffff) == PCI_DVSEC_VENDOR_ID_CXL && + (h2 & 0xffff) == PCI_DVSEC_CXL_REG_LOCATOR) { + regloc = pos; + break; + } + } + pos = (h >> 20) & 0xffc; + } + ASSERT_NE(0, regloc); + + /* 2. Take the component register block BAR and offset from block 1. */ + block1 = regloc + PCI_DVSEC_CXL_REG_LOCATOR_BLOCK1; + reg_lo = vfio_pci_config_readl(self->dev, block1); + reg_hi = vfio_pci_config_readl(self->dev, block1 + 4); + + ASSERT_EQ(CXL_REGLOC_RBI_COMPONENT, + (reg_lo & REG_LOCATOR_BLOCK_ID_MASK) >> 8); + bar = reg_lo & REG_LOCATOR_BIR_MASK; + block_off = ((uint64_t)reg_hi << 32) | + (reg_lo & REG_LOCATOR_BLOCK_OFF_LOW_MASK); + + /* The DVSEC must name the same BAR the geometry cap reported. */ + ASSERT_EQ(self->comp_bar, bar); + + /* 3. Walk the CM capability array over the BAR to find the HDM cap. */ + cm = block_off + CXL_CM_OFFSET; + ASSERT_EQ((ssize_t)sizeof(cap_array), + pread(self->dev->fd, &cap_array, sizeof(cap_array), + bar_off + cm + CXL_CM_CAP_HDR_OFFSET)); + ASSERT_EQ(CM_CAP_HDR_CAP_ID, cap_array & CXL_CM_CAP_HDR_ID_MASK); + + cap_count = (cap_array & CXL_CM_CAP_HDR_ARRAY_SIZE_MASK) >> 24; + for (i = 1; i <= (int)cap_count; i++) { + ASSERT_EQ((ssize_t)sizeof(hdr), + pread(self->dev->fd, &hdr, sizeof(hdr), + bar_off + cm + i * 4)); + if ((hdr & CXL_CM_CAP_HDR_ID_MASK) == CXL_CM_CAP_CAP_ID_HDM) { + hdm_off = cm + ((hdr & CXL_CM_CAP_PTR_MASK) >> 20); + break; + } + } + ASSERT_NE(0, hdm_off); + + /* The guest's manual walk must land on the decoder the kernel traps. */ + ASSERT_EQ(self->comp_offset, hdm_off); + + /* 4. Read decoder 0 through the trapped region and derive base/size. */ + ASSERT_EQ((ssize_t)sizeof(bl), + pread(self->dev->fd, &bl, sizeof(bl), + self->comp_off + CXL_HDM_DECODER0_BASE_LOW_OFFSET(0))); + ASSERT_EQ((ssize_t)sizeof(bh), + pread(self->dev->fd, &bh, sizeof(bh), + self->comp_off + CXL_HDM_DECODER0_BASE_HIGH_OFFSET(0))); + ASSERT_EQ((ssize_t)sizeof(sl), + pread(self->dev->fd, &sl, sizeof(sl), + self->comp_off + CXL_HDM_DECODER0_SIZE_LOW_OFFSET(0))); + ASSERT_EQ((ssize_t)sizeof(sh), + pread(self->dev->fd, &sh, sizeof(sh), + self->comp_off + CXL_HDM_DECODER0_SIZE_HIGH_OFFSET(0))); + ASSERT_EQ((ssize_t)sizeof(ctrl), + pread(self->dev->fd, &ctrl, sizeof(ctrl), + self->comp_off + CXL_HDM_DECODER0_CTRL_OFFSET(0))); + + base = ((uint64_t)bh << 32) | bl; + size = ((uint64_t)sh << 32) | sl; + + /* + * A guest only accepts a committed decoder. When firmware left decoder 0 + * committed the derived range must be non-empty; the base is read to + * mirror the driver even though its value is firmware-defined. + */ + if (!(ctrl & CXL_HDM_DECODER0_CTRL_COMMITTED)) + SKIP(return, "HDM decoder 0 not committed by firmware"); + + ASSERT_GT(size, 0); + ASSERT_LT(base, base + size); +} + +/* + * Tie the committed decoder's advertised geometry to the HDM memory region a + * VMM hands the guest. Read decoder 0's base and size through the trapped + * region, confirm the mmap-able region covers exactly that range, then map the + * advertised base and touch it. A region that advertises the decoder but maps + * PROT_NONE (missing READ/WRITE flags) or a size that disagrees with the + * decoder passes discovery yet faults the guest on first access; catch that + * here instead of on hardware. + */ +TEST_F(vfio_cxl, hdm_mem_touch_committed_base) +{ + uint64_t mem_off = VFIO_PCI_INDEX_TO_OFFSET(self->mem_idx); + uint32_t pattern = 0xc0ffee11U, readback = 0; + uint32_t bl, bh, sl, sh, ctrl; + uint64_t base, size; + void *map; + + ASSERT_EQ((ssize_t)sizeof(ctrl), + pread(self->dev->fd, &ctrl, sizeof(ctrl), + self->comp_off + CXL_HDM_DECODER0_CTRL_OFFSET(0))); + if (!(ctrl & CXL_HDM_DECODER0_CTRL_COMMITTED)) + SKIP(return, "HDM decoder 0 not committed by firmware"); + + ASSERT_EQ((ssize_t)sizeof(bl), + pread(self->dev->fd, &bl, sizeof(bl), + self->comp_off + CXL_HDM_DECODER0_BASE_LOW_OFFSET(0))); + ASSERT_EQ((ssize_t)sizeof(bh), + pread(self->dev->fd, &bh, sizeof(bh), + self->comp_off + CXL_HDM_DECODER0_BASE_HIGH_OFFSET(0))); + ASSERT_EQ((ssize_t)sizeof(sl), + pread(self->dev->fd, &sl, sizeof(sl), + self->comp_off + CXL_HDM_DECODER0_SIZE_LOW_OFFSET(0))); + ASSERT_EQ((ssize_t)sizeof(sh), + pread(self->dev->fd, &sh, sizeof(sh), + self->comp_off + CXL_HDM_DECODER0_SIZE_HIGH_OFFSET(0))); + + base = ((uint64_t)bh << 32) | bl; + size = ((uint64_t)sh << 32) | sl; + + ASSERT_GT(size, 0); + ASSERT_LT(base, base + size); + /* The mmap-able HDM region must cover exactly the committed decoder. */ + ASSERT_EQ(size, self->mem_size); + + if (self->mem_size < SZ_4K) + SKIP(return, "HDM memory < 4K"); + + /* + * Region offset 0 is the decoder's advertised base. Map it read/write and + * touch it: a PROT_NONE mapping (region missing READ/WRITE flags) faults + * here rather than round-tripping the pattern. + */ + map = mmap(NULL, SZ_4K, PROT_READ | PROT_WRITE, MAP_SHARED, + self->dev->fd, mem_off); + ASSERT_NE(MAP_FAILED, map); + + memcpy(map, &pattern, sizeof(pattern)); + memcpy(&readback, map, sizeof(readback)); + ASSERT_EQ(pattern, readback); + + ASSERT_EQ(0, munmap(map, SZ_4K)); +} + +int main(int argc, char *argv[]) +{ + device_bdf = vfio_selftests_get_bdf(&argc, argv); + return test_harness_run(argc, argv); +} -- 2.25.1

