diff options
Diffstat (limited to 'samples/test_mmap.c')
-rw-r--r-- | samples/test_mmap.c | 199 |
1 files changed, 0 insertions, 199 deletions
diff --git a/samples/test_mmap.c b/samples/test_mmap.c deleted file mode 100644 index 02c32f1..0000000 --- a/samples/test_mmap.c +++ /dev/null @@ -1,199 +0,0 @@ -/* - * Userspace mediated device sample application - * - * Copyright (c) 2019, Nutanix Inc. All rights reserved. - * Author: Thanos Makatos <thanos@nutanix.com> - * Swapnil Ingle <swapnil.ingle@nutanix.com> - * Felipe Franciosi <felipe@nutanix.com> - * - * Redistribution and use in source and binary forms, with or without - * modification, are permitted provided that the following conditions are met: - * * Redistributions of source code must retain the above copyright - * notice, this list of conditions and the following disclaimer. - * * Redistributions in binary form must reproduce the above copyright - * notice, this list of conditions and the following disclaimer in the - * documentation and/or other materials provided with the distribution. - * * Neither the name of Nutanix nor the names of its contributors may be - * used to endorse or promote products derived from this software without - * specific prior written permission. - * - * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" - * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE - * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE - * ARE DISCLAIMED. IN NO EVENT SHALL <COPYRIGHT HOLDER> BE LIABLE FOR ANY - * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES - * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR - * SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER - * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT - * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY - * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH - * DAMAGE. - * - */ - -#define _GNU_SOURCE -#include <stdio.h> -#include <errno.h> -#include <unistd.h> -#include <stdlib.h> -#include <sys/types.h> -#include <sys/stat.h> -#include <fcntl.h> -#include <sys/mman.h> -#include <stdbool.h> -#include <string.h> -#include <linux/vfio.h> -#include <limits.h> -#include <assert.h> -#include <sys/ioctl.h> -#include <inttypes.h> - -#define VFIO_PATH "/dev/vfio/" -#define VFIO_CTR_PATH VFIO_PATH "vfio" -#define SYSFS_PCI_DEV_PATH "/sys/bus/pci/devices/" -#define SYSFS_IOMMU_GROUP "/iommu_group" - -static int -pci_group_id(const char *bdf) -{ - char *dev_path; - char group_path[PATH_MAX]; - int group_id; - - assert(bdf); - - asprintf(&dev_path, SYSFS_PCI_DEV_PATH "%s" SYSFS_IOMMU_GROUP, bdf); - memset(group_path, 0, sizeof(group_path)); - readlink(dev_path, group_path, sizeof(group_path)); - free(dev_path); - sscanf(basename(group_path), "%d", &group_id); - return group_id; -} - -static inline void* -test_map_dma(const int fd, const unsigned long size, const unsigned long iova) -{ - int err; - struct vfio_iommu_type1_dma_map dma_map = { - .argsz = sizeof(dma_map), - .size = size, - .iova = iova, - .flags = VFIO_DMA_MAP_FLAG_READ | VFIO_DMA_MAP_FLAG_WRITE, - }; - - /* Allocate some space and setup a DMA mapping */ - /* FIXME it *must* be MAP_SHARED */ - dma_map.vaddr = (unsigned long long)mmap(0, size, PROT_READ | PROT_WRITE, - MAP_SHARED | MAP_ANONYMOUS, 0, 0); - if (dma_map.vaddr == (unsigned long)MAP_FAILED) { - perror("failed to map DMA"); - return NULL; - } - printf("%llx\n", dma_map.vaddr); - strcpy((char*)dma_map.vaddr, "foo"); - - fprintf(stderr, "attempting to MAP_DMA IOVA=%llx\n", dma_map.iova); - - err = ioctl(fd, VFIO_IOMMU_MAP_DMA, &dma_map); - if (err) { - fprintf(stderr, "failed to MAP_DMA: %d (errno=%d)", err, errno); - return NULL; - } - printf("[%s]\n", (char*)dma_map.vaddr); - - return (void*)dma_map.vaddr; -} - -static inline void -test_unmap_dma(const int fd, const unsigned long size, const unsigned long iova) -{ - int err; - struct vfio_iommu_type1_dma_unmap dma_unmap = { - .argsz = sizeof dma_unmap, - .size = size, - .iova = iova, - .flags = 0 - }; - - err = ioctl(fd, VFIO_IOMMU_UNMAP_DMA, &dma_unmap); - if (err) { - perror("UNMAP_DMA\n"); - return; - } - printf("unmapped IOVA=%llx\n", dma_unmap.iova); -} - -int main(int argc, char * argv[]) -{ - int err, vfio_ctr_fd, vfio_grp_fd, vfio_dev_fd; - char *grp_path; -#ifdef DEBUG - struct vfio_group_status grp_status; -#endif - struct vfio_iommu_type1_info iommu_info; - void *dma_map_addr = NULL; - - if (argc != 2) { - printf("Usage: %s <device bdf in full>\n", argv[0]); - printf(" ex: %s 0000:82:00.0\n", argv[0]); - return EXIT_FAILURE; - } - - vfio_ctr_fd = open(VFIO_CTR_PATH, O_RDWR); - assert(vfio_ctr_fd >= 0); - -#ifdef DEBUG - err = ioctl(vfio_ctr_fd, VFIO_GET_API_VERSION); - assert(err == VFIO_API_VERSION); - err = ioctl(vfio_ctr_fd, VFIO_CHECK_EXTENSION, VFIO_TYPE1_IOMMU); - assert(err == 1); -#endif - - // Open the VFIO entry for this device's IOMMU GROUP. - err = asprintf(&grp_path, VFIO_PATH "%d", pci_group_id(argv[1])); - assert(err > 0); - vfio_grp_fd = open(grp_path, O_RDWR); - assert(vfio_grp_fd >= 0); - free(grp_path); - -#ifdef DEBUG - // Ensure group is viable. - memset(&grp_status, 0, sizeof(grp_status)); - grp_status.argsz = sizeof(grp_status); - err = ioctl(vfio_grp_fd, VFIO_GROUP_GET_STATUS, &grp_status); - assert(!err); - assert((grp_status.flags & VFIO_GROUP_FLAGS_VIABLE) == 1); -#endif - - // Add the group to the container. - err = ioctl(vfio_grp_fd, VFIO_GROUP_SET_CONTAINER, &vfio_ctr_fd); - assert(!err); - - // Enable IOMMU type 1 on container. - err = ioctl(vfio_ctr_fd, VFIO_SET_IOMMU, VFIO_TYPE1v2_IOMMU); - assert(!err); - - // Fetch IOMMU information from VFIO. - memset(&iommu_info, 0, sizeof(iommu_info)); - iommu_info.argsz = sizeof(iommu_info); - err = ioctl(vfio_ctr_fd, VFIO_IOMMU_GET_INFO, &iommu_info); - assert(!err); - - // Get a device fd from VFIO. - vfio_dev_fd = ioctl(vfio_grp_fd, VFIO_GROUP_GET_DEVICE_FD, argv[1]); - assert(vfio_dev_fd >= 0); - - void *p; - p = mmap(NULL, sysconf(_SC_PAGESIZE), PROT_READ | PROT_WRITE, - MAP_SHARED, vfio_dev_fd, 0); - assert(p != MAP_FAILED); - printf("%p\n", p); - printf("%s\n", (char*)p); - - dma_map_addr = test_map_dma(vfio_ctr_fd, 4096, 0xdeadbeef000); - if (!dma_map_addr) - exit(EXIT_FAILURE); - test_unmap_dma(vfio_ctr_fd, 4096, 0xdeadbeef000); - - return 0; -} |