1 /* SPDX-License-Identifier: GPL-2.0-only */
2 #ifndef SELFTESTS_VFIO_LIB_INCLUDE_LIBVFIO_VFIO_PCI_DEVICE_H
3 #define SELFTESTS_VFIO_LIB_INCLUDE_LIBVFIO_VFIO_PCI_DEVICE_H
4
5 #include <fcntl.h>
6 #include <linux/vfio.h>
7 #include <linux/pci_regs.h>
8
9 #include <libvfio/assert.h>
10 #include <libvfio/iommu.h>
11 #include <libvfio/vfio_pci_driver.h>
12
13 struct vfio_pci_bar {
14 struct vfio_region_info info;
15 void *vaddr;
16 };
17
18 struct vfio_pci_device {
19 const char *bdf;
20 int fd;
21 int group_fd;
22
23 struct iommu *iommu;
24
25 struct vfio_device_info info;
26 struct vfio_region_info config_space;
27 struct vfio_pci_bar bars[PCI_STD_NUM_BARS];
28
29 struct vfio_irq_info msi_info;
30 struct vfio_irq_info msix_info;
31
32 /* eventfds for MSI and MSI-x interrupts */
33 int msi_eventfds[PCI_MSIX_FLAGS_QSIZE + 1];
34
35 struct vfio_pci_driver driver;
36 };
37
38 #define dev_info(_dev, _fmt, ...) printf("%s: " _fmt, (_dev)->bdf, ##__VA_ARGS__)
39 #define dev_err(_dev, _fmt, ...) fprintf(stderr, "%s: " _fmt, (_dev)->bdf, ##__VA_ARGS__)
40
41 struct vfio_pci_device *vfio_pci_device_init(const char *bdf, struct iommu *iommu);
42 void vfio_pci_device_cleanup(struct vfio_pci_device *device);
43
44 void vfio_pci_device_reset(struct vfio_pci_device *device);
45
46 void vfio_pci_config_access(struct vfio_pci_device *device, bool write,
47 size_t config, size_t size, void *data);
48
49 #define vfio_pci_config_read(_device, _offset, _type) ({ \
50 _type __data; \
51 vfio_pci_config_access((_device), false, _offset, sizeof(__data), &__data); \
52 __data; \
53 })
54
55 #define vfio_pci_config_readb(_d, _o) vfio_pci_config_read(_d, _o, u8)
56 #define vfio_pci_config_readw(_d, _o) vfio_pci_config_read(_d, _o, u16)
57 #define vfio_pci_config_readl(_d, _o) vfio_pci_config_read(_d, _o, u32)
58
59 #define vfio_pci_config_write(_device, _offset, _value, _type) do { \
60 _type __data = (_value); \
61 vfio_pci_config_access((_device), true, _offset, sizeof(_type), &__data); \
62 } while (0)
63
64 #define vfio_pci_config_writeb(_d, _o, _v) vfio_pci_config_write(_d, _o, _v, u8)
65 #define vfio_pci_config_writew(_d, _o, _v) vfio_pci_config_write(_d, _o, _v, u16)
66 #define vfio_pci_config_writel(_d, _o, _v) vfio_pci_config_write(_d, _o, _v, u32)
67
68 void vfio_pci_irq_enable(struct vfio_pci_device *device, u32 index,
69 u32 vector, int count);
70 void vfio_pci_irq_disable(struct vfio_pci_device *device, u32 index);
71 void vfio_pci_irq_trigger(struct vfio_pci_device *device, u32 index, u32 vector);
72
fcntl_set_nonblock(int fd)73 static inline void fcntl_set_nonblock(int fd)
74 {
75 int r;
76
77 r = fcntl(fd, F_GETFL, 0);
78 VFIO_ASSERT_NE(r, -1, "F_GETFL failed for fd %d\n", fd);
79
80 r = fcntl(fd, F_SETFL, r | O_NONBLOCK);
81 VFIO_ASSERT_NE(r, -1, "F_SETFL O_NONBLOCK failed for fd %d\n", fd);
82 }
83
vfio_pci_msi_enable(struct vfio_pci_device * device,u32 vector,int count)84 static inline void vfio_pci_msi_enable(struct vfio_pci_device *device,
85 u32 vector, int count)
86 {
87 vfio_pci_irq_enable(device, VFIO_PCI_MSI_IRQ_INDEX, vector, count);
88 }
89
vfio_pci_msi_disable(struct vfio_pci_device * device)90 static inline void vfio_pci_msi_disable(struct vfio_pci_device *device)
91 {
92 vfio_pci_irq_disable(device, VFIO_PCI_MSI_IRQ_INDEX);
93 }
94
vfio_pci_msix_enable(struct vfio_pci_device * device,u32 vector,int count)95 static inline void vfio_pci_msix_enable(struct vfio_pci_device *device,
96 u32 vector, int count)
97 {
98 vfio_pci_irq_enable(device, VFIO_PCI_MSIX_IRQ_INDEX, vector, count);
99 }
100
vfio_pci_msix_disable(struct vfio_pci_device * device)101 static inline void vfio_pci_msix_disable(struct vfio_pci_device *device)
102 {
103 vfio_pci_irq_disable(device, VFIO_PCI_MSIX_IRQ_INDEX);
104 }
105
__to_iova(struct vfio_pci_device * device,void * vaddr,iova_t * iova)106 static inline int __to_iova(struct vfio_pci_device *device, void *vaddr, iova_t *iova)
107 {
108 return __iommu_hva2iova(device->iommu, vaddr, iova);
109 }
110
to_iova(struct vfio_pci_device * device,void * vaddr)111 static inline iova_t to_iova(struct vfio_pci_device *device, void *vaddr)
112 {
113 return iommu_hva2iova(device->iommu, vaddr);
114 }
115
vfio_pci_device_match(struct vfio_pci_device * device,u16 vendor_id,u16 device_id)116 static inline bool vfio_pci_device_match(struct vfio_pci_device *device,
117 u16 vendor_id, u16 device_id)
118 {
119 return (vendor_id == vfio_pci_config_readw(device, PCI_VENDOR_ID)) &&
120 (device_id == vfio_pci_config_readw(device, PCI_DEVICE_ID));
121 }
122
123 const char *vfio_pci_get_cdev_path(const char *bdf);
124
125 #endif /* SELFTESTS_VFIO_LIB_INCLUDE_LIBVFIO_VFIO_PCI_DEVICE_H */
126