1 // SPDX-License-Identifier: GPL-2.0 2 #include <linux/limits.h> 3 #include <linux/sizes.h> 4 #include <linux/vfio.h> 5 #include <linux/iommufd.h> 6 7 #include <stdint.h> 8 #include <stdio.h> 9 #include <sys/ioctl.h> 10 #include <unistd.h> 11 12 #include <libvfio.h> 13 #include "kselftest_harness.h" 14 15 static const char iommu_dev_path[] = "/dev/iommu"; 16 static const char *cdev_path; 17 18 static int vfio_device_bind_iommufd_ioctl(int cdev_fd, int iommufd) 19 { 20 struct vfio_device_bind_iommufd bind_args = { 21 .argsz = sizeof(bind_args), 22 .iommufd = iommufd, 23 }; 24 25 return ioctl(cdev_fd, VFIO_DEVICE_BIND_IOMMUFD, &bind_args); 26 } 27 28 static int vfio_device_get_info_ioctl(int cdev_fd) 29 { 30 struct vfio_device_info info_args = { .argsz = sizeof(info_args) }; 31 32 return ioctl(cdev_fd, VFIO_DEVICE_GET_INFO, &info_args); 33 } 34 35 static int vfio_device_ioas_alloc_ioctl(int iommufd, struct iommu_ioas_alloc *alloc_args) 36 { 37 *alloc_args = (struct iommu_ioas_alloc){ 38 .size = sizeof(struct iommu_ioas_alloc), 39 }; 40 41 return ioctl(iommufd, IOMMU_IOAS_ALLOC, alloc_args); 42 } 43 44 static int vfio_device_attach_iommufd_pt_ioctl(int cdev_fd, u32 pt_id) 45 { 46 struct vfio_device_attach_iommufd_pt attach_args = { 47 .argsz = sizeof(attach_args), 48 .pt_id = pt_id, 49 }; 50 51 return ioctl(cdev_fd, VFIO_DEVICE_ATTACH_IOMMUFD_PT, &attach_args); 52 } 53 54 static int vfio_device_detach_iommufd_pt_ioctl(int cdev_fd) 55 { 56 struct vfio_device_detach_iommufd_pt detach_args = { 57 .argsz = sizeof(detach_args), 58 }; 59 60 return ioctl(cdev_fd, VFIO_DEVICE_DETACH_IOMMUFD_PT, &detach_args); 61 } 62 63 FIXTURE(vfio_cdev) { 64 int cdev_fd; 65 int iommufd; 66 }; 67 68 FIXTURE_SETUP(vfio_cdev) 69 { 70 ASSERT_LE(0, (self->cdev_fd = open(cdev_path, O_RDWR, 0))); 71 ASSERT_LE(0, (self->iommufd = open(iommu_dev_path, O_RDWR, 0))); 72 } 73 74 FIXTURE_TEARDOWN(vfio_cdev) 75 { 76 ASSERT_EQ(0, close(self->cdev_fd)); 77 ASSERT_EQ(0, close(self->iommufd)); 78 } 79 80 TEST_F(vfio_cdev, bind) 81 { 82 ASSERT_EQ(0, vfio_device_bind_iommufd_ioctl(self->cdev_fd, self->iommufd)); 83 ASSERT_EQ(0, vfio_device_get_info_ioctl(self->cdev_fd)); 84 } 85 86 TEST_F(vfio_cdev, get_info_without_bind_fails) 87 { 88 ASSERT_NE(0, vfio_device_get_info_ioctl(self->cdev_fd)); 89 } 90 91 TEST_F(vfio_cdev, bind_bad_iommufd_fails) 92 { 93 ASSERT_NE(0, vfio_device_bind_iommufd_ioctl(self->cdev_fd, -2)); 94 } 95 96 TEST_F(vfio_cdev, repeated_bind_fails) 97 { 98 ASSERT_EQ(0, vfio_device_bind_iommufd_ioctl(self->cdev_fd, self->iommufd)); 99 ASSERT_NE(0, vfio_device_bind_iommufd_ioctl(self->cdev_fd, self->iommufd)); 100 } 101 102 TEST_F(vfio_cdev, attach_detatch_pt) 103 { 104 struct iommu_ioas_alloc alloc_args; 105 106 ASSERT_EQ(0, vfio_device_bind_iommufd_ioctl(self->cdev_fd, self->iommufd)); 107 ASSERT_EQ(0, vfio_device_ioas_alloc_ioctl(self->iommufd, &alloc_args)); 108 ASSERT_EQ(0, vfio_device_attach_iommufd_pt_ioctl(self->cdev_fd, alloc_args.out_ioas_id)); 109 ASSERT_EQ(0, vfio_device_detach_iommufd_pt_ioctl(self->cdev_fd)); 110 } 111 112 TEST_F(vfio_cdev, attach_invalid_pt_fails) 113 { 114 ASSERT_EQ(0, vfio_device_bind_iommufd_ioctl(self->cdev_fd, self->iommufd)); 115 ASSERT_NE(0, vfio_device_attach_iommufd_pt_ioctl(self->cdev_fd, UINT32_MAX)); 116 } 117 118 int main(int argc, char *argv[]) 119 { 120 const char *device_bdf = vfio_selftests_get_bdf(&argc, argv); 121 122 cdev_path = vfio_pci_get_cdev_path(device_bdf); 123 printf("Using cdev device %s\n", cdev_path); 124 125 return test_harness_run(argc, argv); 126 } 127