1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
|
#include <errno.h>
#include <fcntl.h>
#include <stdio.h>
#include <linux/vfio.h>
#include <sys/ioctl.h>
#include <sys/mman.h>
#include <string.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdint.h>
const char glob_uuid[] = "d0b24768-9c51-11e7-8de5-c7b0c2769e69";
#define TEST_SIZE 1024 * 1024
int main(int argc, char* argv[])
{
int container, group, device, i, ret;
struct vfio_group_status group_status = { .argsz = sizeof(group_status) };
struct vfio_device_info device_info = { .argsz = sizeof(device_info) };
struct vfio_iommu_type1_info iommu_info = { .argsz = sizeof(iommu_info) };
struct vfio_region_info region_info = { .argsz = sizeof(region_info) };
struct vfio_iommu_type1_dma_map dma_map = { .argsz = sizeof(dma_map) };
struct vfio_iommu_type1_dma_unmap dma_unmap = { .argsz = sizeof(dma_unmap) };
void *map;
size_t sz;
/* Create a new container */
container = open("/dev/vfio/vfio", O_RDWR);
if (ioctl(container, VFIO_GET_API_VERSION) != VFIO_API_VERSION) {
printf("Unknown API version\n");
goto out;
}
if (!ioctl(container, VFIO_CHECK_EXTENSION, VFIO_TYPE1_IOMMU)) {
printf("Doesn't support the IOMMU driver we want\n");
goto out;
}
group = open("/dev/vfio/0", O_RDWR);
/* Test the group is viable and available */
ioctl(group, VFIO_GROUP_GET_STATUS, &group_status);
if (!(group_status.flags & VFIO_GROUP_FLAGS_VIABLE)) {
printf("Group is not viable\n");
goto out;
}
/* Set container */
ioctl(group, VFIO_GROUP_SET_CONTAINER, &container);
ioctl(group, VFIO_GROUP_GET_STATUS, &group_status);
if (!(group_status.flags & VFIO_GROUP_FLAGS_CONTAINER_SET)) {
printf("Container not set\n");
goto out;
}
/* Enable the IOMMU model we want */
ioctl(container, VFIO_SET_IOMMU, VFIO_TYPE1_IOMMU);
/* Get addition IOMMU info */
ioctl(container, VFIO_IOMMU_GET_INFO, &iommu_info);
/* Get a file descriptor for the device */
device = ioctl(group, VFIO_GROUP_GET_DEVICE_FD, glob_uuid);
/* Test and setup the device */
ioctl(device, VFIO_DEVICE_GET_INFO, &device_info);
printf("Device supports %d regions, %d irqs\n",
device_info.num_regions, device_info.num_irqs);
for (i = 0; i < device_info.num_regions; i++) {
region_info.index = i;
if (ioctl(device, VFIO_DEVICE_GET_REGION_INFO, ®ion_info))
continue;
if (!region_info.size) {
//printf("Region:%d unimplemented PCI BAR\n", i);
continue;
}
printf("Region:%d size %lu, offset 0x%lx, flags 0x%x\n", i,
(unsigned long)region_info.size,
(unsigned long)region_info.offset, region_info.flags);
if (region_info.flags & VFIO_REGION_INFO_FLAG_MMAP) {
int j;
map = mmap(NULL, (size_t)region_info.size,
PROT_READ, MAP_SHARED, device,
(off_t)region_info.offset);
if (map == MAP_FAILED) {
printf("mmap failed\n");
continue;
}
sz = region_info.size / sizeof(uint32_t *);
for (j = 0; j < sz; j++) {
printf("%08x\n", ((uint32_t *)map)[j]);
}
printf("\n");
/*
fwrite(map, 1, region_info.size > 16 ? 16 :
region_info.size, stdout);
printf("]\n");
*/
munmap(map, (size_t)region_info.size);
}
}
/* sample DMA mapping */
/* Allocate some space and setup a DMA mapping */
map = mmap(NULL, TEST_SIZE, PROT_READ,
MAP_PRIVATE | MAP_ANONYMOUS, -1, 0);
if (map == MAP_FAILED) {
printf("Failed to allocate memory\n");
goto out;
}
dma_map.vaddr = (unsigned long)map;
dma_map.flags = VFIO_DMA_MAP_FLAG_READ;
dma_map.size = TEST_SIZE;
dma_map.iova = 0;
ret = ioctl(container, VFIO_IOMMU_MAP_DMA, &dma_map);
if (ret) {
printf("Failed to map memory (%s)\n", strerror(errno));
goto out;
}
printf("DMA Mapping: [");
fwrite(map, 1, region_info.size, stdout);
printf("]\n");
/* unmap */
dma_unmap.size = dma_map.size;
dma_unmap.iova = 0;
ret = ioctl(container, VFIO_IOMMU_UNMAP_DMA, &dma_unmap);
if (ret) {
printf("Failed to unmap memory (%s)\n", strerror(errno));
goto out;
}
return 0;
out:
return -1;
}
|
