vfio: Add PCI device driver

Add PCI device support for VFIO. PCI devices expose regions for accessing config space, I/O port space, and MMIO areas of the device. PCI config access is virtualized in the kernel, allowing us to ensure the integrity of the system, by preventing various accesses while reducing duplicate support across various userspace drivers. I/O port supports read/write access while MMIO also supports mmap of sufficiently sized regions. Support for INTx, MSI, and MSI-X interrupts are provided using eventfds to userspace. Signed-off-by: Alex Williamson <alex.williamson@redhat.com>

vfio: Add PCI device driver
Add PCI device support for VFIO. PCI devices expose regions for accessing config space, I/O port space, and MMIO areas of the device. PCI config access is virtualized in the kernel, allowing us to ensure the integrity of the system, by preventing various accesses while reducing duplicate support across various userspace drivers. I/O port supports read/write access while MMIO also supports mmap of sufficiently sized regions. Support for INTx, MSI, and MSI-X interrupts are provided using eventfds to userspace. Signed-off-by: Alex Williamson <alex.williamson@redhat.com>
Alex Williamson
1 parent 73fa0d10d0
Showing 9 changed files with 3259 additions and 0 deletions Side-by-side Diff
drivers/vfio/Kconfig
drivers/vfio/pci/Kconfig
drivers/vfio/pci/Makefile
drivers/vfio/pci/vfio_pci.c
drivers/vfio/pci/vfio_pci_config.c
drivers/vfio/pci/vfio_pci_intrs.c
drivers/vfio/pci/vfio_pci_private.h
drivers/vfio/pci/vfio_pci_rdwr.c
include/linux/vfio.h
@@ -12,4 +12,6 @@
 	  See Documentation/vfio.txt for more details.
  
 	  If you don't know what to do here, say N.
+
+source "drivers/vfio/pci/Kconfig"
+config VFIO_PCI
+	tristate "VFIO support for PCI devices"
+	depends on VFIO && PCI && EVENTFD
+	help
+	  Support for the PCI VFIO bus driver.  This is required to make
+	  use of PCI drivers using the VFIO framework.
+
+	  If you don't know what to do here, say N.
+
+vfio-pci-y := vfio_pci.o vfio_pci_intrs.o vfio_pci_rdwr.o vfio_pci_config.o
+
+obj-$(CONFIG_VFIO_PCI) += vfio-pci.o
+/*
+ * Copyright (C) 2012 Red Hat, Inc.  All rights reserved.
+ *     Author: Alex Williamson <alex.williamson@redhat.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ *
+ * Derived from original vfio:
+ * Copyright 2010 Cisco Systems, Inc.  All rights reserved.
+ * Author: Tom Lyon, pugs@cisco.com
+ */
+
+#include <linux/device.h>
+#include <linux/eventfd.h>
+#include <linux/interrupt.h>
+#include <linux/iommu.h>
+#include <linux/module.h>
+#include <linux/mutex.h>
+#include <linux/notifier.h>
+#include <linux/pci.h>
+#include <linux/pm_runtime.h>
+#include <linux/slab.h>
+#include <linux/types.h>
+#include <linux/uaccess.h>
+#include <linux/vfio.h>
+
+#include "vfio_pci_private.h"
+
+#define DRIVER_VERSION  "0.2"
+#define DRIVER_AUTHOR   "Alex Williamson <alex.williamson@redhat.com>"
+#define DRIVER_DESC     "VFIO PCI - User Level meta-driver"
+
+static bool nointxmask;
+module_param_named(nointxmask, nointxmask, bool, S_IRUGO | S_IWUSR);
+MODULE_PARM_DESC(nointxmask,
+		  "Disable support for PCI 2.3 style INTx masking.  If this resolves problems for specific devices, report lspci -vvvxxx to linux-pci@vger.kernel.org so the device can be fixed automatically via the broken_intx_masking flag.");
+
+static int vfio_pci_enable(struct vfio_pci_device *vdev)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	int ret;
+	u16 cmd;
+	u8 msix_pos;
+
+	vdev->reset_works = (pci_reset_function(pdev) == 0);
+	pci_save_state(pdev);
+	vdev->pci_saved_state = pci_store_saved_state(pdev);
+	if (!vdev->pci_saved_state)
+		pr_debug("%s: Couldn't store %s saved state\n",
+			 __func__, dev_name(&pdev->dev));
+
+	ret = vfio_config_init(vdev);
+	if (ret)
+		goto out;
+
+	if (likely(!nointxmask))
+		vdev->pci_2_3 = pci_intx_mask_supported(pdev);
+
+	pci_read_config_word(pdev, PCI_COMMAND, &cmd);
+	if (vdev->pci_2_3 && (cmd & PCI_COMMAND_INTX_DISABLE)) {
+		cmd &= ~PCI_COMMAND_INTX_DISABLE;
+		pci_write_config_word(pdev, PCI_COMMAND, cmd);
+	}
+
+	msix_pos = pci_find_capability(pdev, PCI_CAP_ID_MSIX);
+	if (msix_pos) {
+		u16 flags;
+		u32 table;
+
+		pci_read_config_word(pdev, msix_pos + PCI_MSIX_FLAGS, &flags);
+		pci_read_config_dword(pdev, msix_pos + PCI_MSIX_TABLE, &table);
+
+		vdev->msix_bar = table & PCI_MSIX_FLAGS_BIRMASK;
+		vdev->msix_offset = table & ~PCI_MSIX_FLAGS_BIRMASK;
+		vdev->msix_size = ((flags & PCI_MSIX_FLAGS_QSIZE) + 1) * 16;
+	} else
+		vdev->msix_bar = 0xFF;
+
+	ret = pci_enable_device(pdev);
+	if (ret)
+		goto out;
+
+	return ret;
+
+out:
+	kfree(vdev->pci_saved_state);
+	vdev->pci_saved_state = NULL;
+	vfio_config_free(vdev);
+	return ret;
+}
+
+static void vfio_pci_disable(struct vfio_pci_device *vdev)
+{
+	int bar;
+
+	pci_disable_device(vdev->pdev);
+
+	vfio_pci_set_irqs_ioctl(vdev, VFIO_IRQ_SET_DATA_NONE |
+				VFIO_IRQ_SET_ACTION_TRIGGER,
+				vdev->irq_type, 0, 0, NULL);
+
+	vdev->virq_disabled = false;
+
+	vfio_config_free(vdev);
+
+	pci_reset_function(vdev->pdev);
+
+	if (pci_load_and_free_saved_state(vdev->pdev,
+					  &vdev->pci_saved_state) == 0)
+		pci_restore_state(vdev->pdev);
+	else
+		pr_info("%s: Couldn't reload %s saved state\n",
+			__func__, dev_name(&vdev->pdev->dev));
+
+	for (bar = PCI_STD_RESOURCES; bar <= PCI_STD_RESOURCE_END; bar++) {
+		if (!vdev->barmap[bar])
+			continue;
+		pci_iounmap(vdev->pdev, vdev->barmap[bar]);
+		pci_release_selected_regions(vdev->pdev, 1 << bar);
+		vdev->barmap[bar] = NULL;
+	}
+}
+
+static void vfio_pci_release(void *device_data)
+{
+	struct vfio_pci_device *vdev = device_data;
+
+	if (atomic_dec_and_test(&vdev->refcnt))
+		vfio_pci_disable(vdev);
+
+	module_put(THIS_MODULE);
+}
+
+static int vfio_pci_open(void *device_data)
+{
+	struct vfio_pci_device *vdev = device_data;
+
+	if (!try_module_get(THIS_MODULE))
+		return -ENODEV;
+
+	if (atomic_inc_return(&vdev->refcnt) == 1) {
+		int ret = vfio_pci_enable(vdev);
+		if (ret) {
+			module_put(THIS_MODULE);
+			return ret;
+		}
+	}
+
+	return 0;
+}
+
+static int vfio_pci_get_irq_count(struct vfio_pci_device *vdev, int irq_type)
+{
+	if (irq_type == VFIO_PCI_INTX_IRQ_INDEX) {
+		u8 pin;
+		pci_read_config_byte(vdev->pdev, PCI_INTERRUPT_PIN, &pin);
+		if (pin)
+			return 1;
+
+	} else if (irq_type == VFIO_PCI_MSI_IRQ_INDEX) {
+		u8 pos;
+		u16 flags;
+
+		pos = pci_find_capability(vdev->pdev, PCI_CAP_ID_MSI);
+		if (pos) {
+			pci_read_config_word(vdev->pdev,
+					     pos + PCI_MSI_FLAGS, &flags);
+
+			return 1 << (flags & PCI_MSI_FLAGS_QMASK);
+		}
+	} else if (irq_type == VFIO_PCI_MSIX_IRQ_INDEX) {
+		u8 pos;
+		u16 flags;
+
+		pos = pci_find_capability(vdev->pdev, PCI_CAP_ID_MSIX);
+		if (pos) {
+			pci_read_config_word(vdev->pdev,
+					     pos + PCI_MSIX_FLAGS, &flags);
+
+			return (flags & PCI_MSIX_FLAGS_QSIZE) + 1;
+		}
+	}
+
+	return 0;
+}
+
+static long vfio_pci_ioctl(void *device_data,
+			   unsigned int cmd, unsigned long arg)
+{
+	struct vfio_pci_device *vdev = device_data;
+	unsigned long minsz;
+
+	if (cmd == VFIO_DEVICE_GET_INFO) {
+		struct vfio_device_info info;
+
+		minsz = offsetofend(struct vfio_device_info, num_irqs);
+
+		if (copy_from_user(&info, (void __user *)arg, minsz))
+			return -EFAULT;
+
+		if (info.argsz < minsz)
+			return -EINVAL;
+
+		info.flags = VFIO_DEVICE_FLAGS_PCI;
+
+		if (vdev->reset_works)
+			info.flags |= VFIO_DEVICE_FLAGS_RESET;
+
+		info.num_regions = VFIO_PCI_NUM_REGIONS;
+		info.num_irqs = VFIO_PCI_NUM_IRQS;
+
+		return copy_to_user((void __user *)arg, &info, minsz);
+
+	} else if (cmd == VFIO_DEVICE_GET_REGION_INFO) {
+		struct pci_dev *pdev = vdev->pdev;
+		struct vfio_region_info info;
+
+		minsz = offsetofend(struct vfio_region_info, offset);
+
+		if (copy_from_user(&info, (void __user *)arg, minsz))
+			return -EFAULT;
+
+		if (info.argsz < minsz)
+			return -EINVAL;
+
+		switch (info.index) {
+		case VFIO_PCI_CONFIG_REGION_INDEX:
+			info.offset = VFIO_PCI_INDEX_TO_OFFSET(info.index);
+			info.size = pdev->cfg_size;
+			info.flags = VFIO_REGION_INFO_FLAG_READ |
+				     VFIO_REGION_INFO_FLAG_WRITE;
+			break;
+		case VFIO_PCI_BAR0_REGION_INDEX ... VFIO_PCI_BAR5_REGION_INDEX:
+			info.offset = VFIO_PCI_INDEX_TO_OFFSET(info.index);
+			info.size = pci_resource_len(pdev, info.index);
+			if (!info.size) {
+				info.flags = 0;
+				break;
+			}
+
+			info.flags = VFIO_REGION_INFO_FLAG_READ |
+				     VFIO_REGION_INFO_FLAG_WRITE;
+			if (pci_resource_flags(pdev, info.index) &
+			    IORESOURCE_MEM && info.size >= PAGE_SIZE)
+				info.flags |= VFIO_REGION_INFO_FLAG_MMAP;
+			break;
+		case VFIO_PCI_ROM_REGION_INDEX:
+		{
+			void __iomem *io;
+			size_t size;
+
+			info.offset = VFIO_PCI_INDEX_TO_OFFSET(info.index);
+			info.flags = 0;
+
+			/* Report the BAR size, not the ROM size */
+			info.size = pci_resource_len(pdev, info.index);
+			if (!info.size)
+				break;
+
+			/* Is it really there? */
+			io = pci_map_rom(pdev, &size);
+			if (!io || !size) {
+				info.size = 0;
+				break;
+			}
+			pci_unmap_rom(pdev, io);
+
+			info.flags = VFIO_REGION_INFO_FLAG_READ;
+			break;
+		}
+		default:
+			return -EINVAL;
+		}
+
+		return copy_to_user((void __user *)arg, &info, minsz);
+
+	} else if (cmd == VFIO_DEVICE_GET_IRQ_INFO) {
+		struct vfio_irq_info info;
+
+		minsz = offsetofend(struct vfio_irq_info, count);
+
+		if (copy_from_user(&info, (void __user *)arg, minsz))
+			return -EFAULT;
+
+		if (info.argsz < minsz || info.index >= VFIO_PCI_NUM_IRQS)
+			return -EINVAL;
+
+		info.flags = VFIO_IRQ_INFO_EVENTFD;
+
+		info.count = vfio_pci_get_irq_count(vdev, info.index);
+
+		if (info.index == VFIO_PCI_INTX_IRQ_INDEX)
+			info.flags |= (VFIO_IRQ_INFO_MASKABLE |
+				       VFIO_IRQ_INFO_AUTOMASKED);
+		else
+			info.flags |= VFIO_IRQ_INFO_NORESIZE;
+
+		return copy_to_user((void __user *)arg, &info, minsz);
+
+	} else if (cmd == VFIO_DEVICE_SET_IRQS) {
+		struct vfio_irq_set hdr;
+		u8 *data = NULL;
+		int ret = 0;
+
+		minsz = offsetofend(struct vfio_irq_set, count);
+
+		if (copy_from_user(&hdr, (void __user *)arg, minsz))
+			return -EFAULT;
+
+		if (hdr.argsz < minsz || hdr.index >= VFIO_PCI_NUM_IRQS ||
+		    hdr.flags & ~(VFIO_IRQ_SET_DATA_TYPE_MASK |
+				  VFIO_IRQ_SET_ACTION_TYPE_MASK))
+			return -EINVAL;
+
+		if (!(hdr.flags & VFIO_IRQ_SET_DATA_NONE)) {
+			size_t size;
+
+			if (hdr.flags & VFIO_IRQ_SET_DATA_BOOL)
+				size = sizeof(uint8_t);
+			else if (hdr.flags & VFIO_IRQ_SET_DATA_EVENTFD)
+				size = sizeof(int32_t);
+			else
+				return -EINVAL;
+
+			if (hdr.argsz - minsz < hdr.count * size ||
+			    hdr.count > vfio_pci_get_irq_count(vdev, hdr.index))
+				return -EINVAL;
+
+			data = kmalloc(hdr.count * size, GFP_KERNEL);
+			if (!data)
+				return -ENOMEM;
+
+			if (copy_from_user(data, (void __user *)(arg + minsz),
+					   hdr.count * size)) {
+				kfree(data);
+				return -EFAULT;
+			}
+		}
+
+		mutex_lock(&vdev->igate);
+
+		ret = vfio_pci_set_irqs_ioctl(vdev, hdr.flags, hdr.index,
+					      hdr.start, hdr.count, data);
+
+		mutex_unlock(&vdev->igate);
+		kfree(data);
+
+		return ret;
+
+	} else if (cmd == VFIO_DEVICE_RESET)
+		return vdev->reset_works ?
+			pci_reset_function(vdev->pdev) : -EINVAL;
+
+	return -ENOTTY;
+}
+
+static ssize_t vfio_pci_read(void *device_data, char __user *buf,
+			     size_t count, loff_t *ppos)
+{
+	unsigned int index = VFIO_PCI_OFFSET_TO_INDEX(*ppos);
+	struct vfio_pci_device *vdev = device_data;
+	struct pci_dev *pdev = vdev->pdev;
+
+	if (index >= VFIO_PCI_NUM_REGIONS)
+		return -EINVAL;
+
+	if (index == VFIO_PCI_CONFIG_REGION_INDEX)
+		return vfio_pci_config_readwrite(vdev, buf, count, ppos, false);
+	else if (index == VFIO_PCI_ROM_REGION_INDEX)
+		return vfio_pci_mem_readwrite(vdev, buf, count, ppos, false);
+	else if (pci_resource_flags(pdev, index) & IORESOURCE_IO)
+		return vfio_pci_io_readwrite(vdev, buf, count, ppos, false);
+	else if (pci_resource_flags(pdev, index) & IORESOURCE_MEM)
+		return vfio_pci_mem_readwrite(vdev, buf, count, ppos, false);
+
+	return -EINVAL;
+}
+
+static ssize_t vfio_pci_write(void *device_data, const char __user *buf,
+			      size_t count, loff_t *ppos)
+{
+	unsigned int index = VFIO_PCI_OFFSET_TO_INDEX(*ppos);
+	struct vfio_pci_device *vdev = device_data;
+	struct pci_dev *pdev = vdev->pdev;
+
+	if (index >= VFIO_PCI_NUM_REGIONS)
+		return -EINVAL;
+
+	if (index == VFIO_PCI_CONFIG_REGION_INDEX)
+		return vfio_pci_config_readwrite(vdev, (char __user *)buf,
+						 count, ppos, true);
+	else if (index == VFIO_PCI_ROM_REGION_INDEX)
+		return -EINVAL;
+	else if (pci_resource_flags(pdev, index) & IORESOURCE_IO)
+		return vfio_pci_io_readwrite(vdev, (char __user *)buf,
+					     count, ppos, true);
+	else if (pci_resource_flags(pdev, index) & IORESOURCE_MEM) {
+		return vfio_pci_mem_readwrite(vdev, (char __user *)buf,
+					      count, ppos, true);
+	}
+
+	return -EINVAL;
+}
+
+static int vfio_pci_mmap(void *device_data, struct vm_area_struct *vma)
+{
+	struct vfio_pci_device *vdev = device_data;
+	struct pci_dev *pdev = vdev->pdev;
+	unsigned int index;
+	u64 phys_len, req_len, pgoff, req_start, phys;
+	int ret;
+
+	index = vma->vm_pgoff >> (VFIO_PCI_OFFSET_SHIFT - PAGE_SHIFT);
+
+	if (vma->vm_end < vma->vm_start)
+		return -EINVAL;
+	if ((vma->vm_flags & VM_SHARED) == 0)
+		return -EINVAL;
+	if (index >= VFIO_PCI_ROM_REGION_INDEX)
+		return -EINVAL;
+	if (!(pci_resource_flags(pdev, index) & IORESOURCE_MEM))
+		return -EINVAL;
+
+	phys_len = pci_resource_len(pdev, index);
+	req_len = vma->vm_end - vma->vm_start;
+	pgoff = vma->vm_pgoff &
+		((1U << (VFIO_PCI_OFFSET_SHIFT - PAGE_SHIFT)) - 1);
+	req_start = pgoff << PAGE_SHIFT;
+
+	if (phys_len < PAGE_SIZE || req_start + req_len > phys_len)
+		return -EINVAL;
+
+	if (index == vdev->msix_bar) {
+		/*
+		 * Disallow mmaps overlapping the MSI-X table; users don't
+		 * get to touch this directly.  We could find somewhere
+		 * else to map the overlap, but page granularity is only
+		 * a recommendation, not a requirement, so the user needs
+		 * to know which bits are real.  Requiring them to mmap
+		 * around the table makes that clear.
+		 */
+
+		/* If neither entirely above nor below, then it overlaps */
+		if (!(req_start >= vdev->msix_offset + vdev->msix_size ||
+		      req_start + req_len <= vdev->msix_offset))
+			return -EINVAL;
+	}
+
+	/*
+	 * Even though we don't make use of the barmap for the mmap,
+	 * we need to request the region and the barmap tracks that.
+	 */
+	if (!vdev->barmap[index]) {
+		ret = pci_request_selected_regions(pdev,
+						   1 << index, "vfio-pci");
+		if (ret)
+			return ret;
+
+		vdev->barmap[index] = pci_iomap(pdev, index, 0);
+	}
+
+	vma->vm_private_data = vdev;
+	vma->vm_flags |= (VM_IO | VM_RESERVED);
+	vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
+
+	phys = (pci_resource_start(pdev, index) >> PAGE_SHIFT) + pgoff;
+
+	return remap_pfn_range(vma, vma->vm_start, phys,
+			       req_len, vma->vm_page_prot);
+}
+
+static const struct vfio_device_ops vfio_pci_ops = {
+	.name		= "vfio-pci",
+	.open		= vfio_pci_open,
+	.release	= vfio_pci_release,
+	.ioctl		= vfio_pci_ioctl,
+	.read		= vfio_pci_read,
+	.write		= vfio_pci_write,
+	.mmap		= vfio_pci_mmap,
+};
+
+static int vfio_pci_probe(struct pci_dev *pdev, const struct pci_device_id *id)
+{
+	u8 type;
+	struct vfio_pci_device *vdev;
+	struct iommu_group *group;
+	int ret;
+
+	pci_read_config_byte(pdev, PCI_HEADER_TYPE, &type);
+	if ((type & PCI_HEADER_TYPE) != PCI_HEADER_TYPE_NORMAL)
+		return -EINVAL;
+
+	group = iommu_group_get(&pdev->dev);
+	if (!group)
+		return -EINVAL;
+
+	vdev = kzalloc(sizeof(*vdev), GFP_KERNEL);
+	if (!vdev) {
+		iommu_group_put(group);
+		return -ENOMEM;
+	}
+
+	vdev->pdev = pdev;
+	vdev->irq_type = VFIO_PCI_NUM_IRQS;
+	mutex_init(&vdev->igate);
+	spin_lock_init(&vdev->irqlock);
+	atomic_set(&vdev->refcnt, 0);
+
+	ret = vfio_add_group_dev(&pdev->dev, &vfio_pci_ops, vdev);
+	if (ret) {
+		iommu_group_put(group);
+		kfree(vdev);
+	}
+
+	return ret;
+}
+
+static void vfio_pci_remove(struct pci_dev *pdev)
+{
+	struct vfio_pci_device *vdev;
+
+	vdev = vfio_del_group_dev(&pdev->dev);
+	if (!vdev)
+		return;
+
+	iommu_group_put(pdev->dev.iommu_group);
+	kfree(vdev);
+}
+
+static struct pci_driver vfio_pci_driver = {
+	.name		= "vfio-pci",
+	.id_table	= NULL, /* only dynamic ids */
+	.probe		= vfio_pci_probe,
+	.remove		= vfio_pci_remove,
+};
+
+static void __exit vfio_pci_cleanup(void)
+{
+	pci_unregister_driver(&vfio_pci_driver);
+	vfio_pci_virqfd_exit();
+	vfio_pci_uninit_perm_bits();
+}
+
+static int __init vfio_pci_init(void)
+{
+	int ret;
+
+	/* Allocate shared config space permision data used by all devices */
+	ret = vfio_pci_init_perm_bits();
+	if (ret)
+		return ret;
+
+	/* Start the virqfd cleanup handler */
+	ret = vfio_pci_virqfd_init();
+	if (ret)
+		goto out_virqfd;
+
+	/* Register and scan for devices */
+	ret = pci_register_driver(&vfio_pci_driver);
+	if (ret)
+		goto out_driver;
+
+	return 0;
+
+out_virqfd:
+	vfio_pci_virqfd_exit();
+out_driver:
+	vfio_pci_uninit_perm_bits();
+	return ret;
+}
+
+module_init(vfio_pci_init);
+module_exit(vfio_pci_cleanup);
+
+MODULE_VERSION(DRIVER_VERSION);
+MODULE_LICENSE("GPL v2");
+MODULE_AUTHOR(DRIVER_AUTHOR);
+MODULE_DESCRIPTION(DRIVER_DESC);
+/*
+ * VFIO PCI config space virtualization
+ *
+ * Copyright (C) 2012 Red Hat, Inc.  All rights reserved.
+ *     Author: Alex Williamson <alex.williamson@redhat.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ *
+ * Derived from original vfio:
+ * Copyright 2010 Cisco Systems, Inc.  All rights reserved.
+ * Author: Tom Lyon, pugs@cisco.com
+ */
+
+/*
+ * This code handles reading and writing of PCI configuration registers.
+ * This is hairy because we want to allow a lot of flexibility to the
+ * user driver, but cannot trust it with all of the config fields.
+ * Tables determine which fields can be read and written, as well as
+ * which fields are 'virtualized' - special actions and translations to
+ * make it appear to the user that he has control, when in fact things
+ * must be negotiated with the underlying OS.
+ */
+
+#include <linux/fs.h>
+#include <linux/pci.h>
+#include <linux/uaccess.h>
+#include <linux/vfio.h>
+
+#include "vfio_pci_private.h"
+
+#define PCI_CFG_SPACE_SIZE	256
+
+/* Useful "pseudo" capabilities */
+#define PCI_CAP_ID_BASIC	0
+#define PCI_CAP_ID_INVALID	0xFF
+
+#define is_bar(offset)	\
+	((offset >= PCI_BASE_ADDRESS_0 && offset < PCI_BASE_ADDRESS_5 + 4) || \
+	 (offset >= PCI_ROM_ADDRESS && offset < PCI_ROM_ADDRESS + 4))
+
+/*
+ * Lengths of PCI Config Capabilities
+ *   0: Removed from the user visible capability list
+ *   FF: Variable length
+ */
+static u8 pci_cap_length[] = {
+	[PCI_CAP_ID_BASIC]	= PCI_STD_HEADER_SIZEOF, /* pci config header */
+	[PCI_CAP_ID_PM]		= PCI_PM_SIZEOF,
+	[PCI_CAP_ID_AGP]	= PCI_AGP_SIZEOF,
+	[PCI_CAP_ID_VPD]	= PCI_CAP_VPD_SIZEOF,
+	[PCI_CAP_ID_SLOTID]	= 0,		/* bridge - don't care */
+	[PCI_CAP_ID_MSI]	= 0xFF,		/* 10, 14, 20, or 24 */
+	[PCI_CAP_ID_CHSWP]	= 0,		/* cpci - not yet */
+	[PCI_CAP_ID_PCIX]	= 0xFF,		/* 8 or 24 */
+	[PCI_CAP_ID_HT]		= 0xFF,		/* hypertransport */
+	[PCI_CAP_ID_VNDR]	= 0xFF,		/* variable */
+	[PCI_CAP_ID_DBG]	= 0,		/* debug - don't care */
+	[PCI_CAP_ID_CCRC]	= 0,		/* cpci - not yet */
+	[PCI_CAP_ID_SHPC]	= 0,		/* hotswap - not yet */
+	[PCI_CAP_ID_SSVID]	= 0,		/* bridge - don't care */
+	[PCI_CAP_ID_AGP3]	= 0,		/* AGP8x - not yet */
+	[PCI_CAP_ID_SECDEV]	= 0,		/* secure device not yet */
+	[PCI_CAP_ID_EXP]	= 0xFF,		/* 20 or 44 */
+	[PCI_CAP_ID_MSIX]	= PCI_CAP_MSIX_SIZEOF,
+	[PCI_CAP_ID_SATA]	= 0xFF,
+	[PCI_CAP_ID_AF]		= PCI_CAP_AF_SIZEOF,
+};
+
+/*
+ * Lengths of PCIe/PCI-X Extended Config Capabilities
+ *   0: Removed or masked from the user visible capabilty list
+ *   FF: Variable length
+ */
+static u16 pci_ext_cap_length[] = {
+	[PCI_EXT_CAP_ID_ERR]	=	PCI_ERR_ROOT_COMMAND,
+	[PCI_EXT_CAP_ID_VC]	=	0xFF,
+	[PCI_EXT_CAP_ID_DSN]	=	PCI_EXT_CAP_DSN_SIZEOF,
+	[PCI_EXT_CAP_ID_PWR]	=	PCI_EXT_CAP_PWR_SIZEOF,
+	[PCI_EXT_CAP_ID_RCLD]	=	0,	/* root only - don't care */
+	[PCI_EXT_CAP_ID_RCILC]	=	0,	/* root only - don't care */
+	[PCI_EXT_CAP_ID_RCEC]	=	0,	/* root only - don't care */
+	[PCI_EXT_CAP_ID_MFVC]	=	0xFF,
+	[PCI_EXT_CAP_ID_VC9]	=	0xFF,	/* same as CAP_ID_VC */
+	[PCI_EXT_CAP_ID_RCRB]	=	0,	/* root only - don't care */
+	[PCI_EXT_CAP_ID_VNDR]	=	0xFF,
+	[PCI_EXT_CAP_ID_CAC]	=	0,	/* obsolete */
+	[PCI_EXT_CAP_ID_ACS]	=	0xFF,
+	[PCI_EXT_CAP_ID_ARI]	=	PCI_EXT_CAP_ARI_SIZEOF,
+	[PCI_EXT_CAP_ID_ATS]	=	PCI_EXT_CAP_ATS_SIZEOF,
+	[PCI_EXT_CAP_ID_SRIOV]	=	PCI_EXT_CAP_SRIOV_SIZEOF,
+	[PCI_EXT_CAP_ID_MRIOV]	=	0,	/* not yet */
+	[PCI_EXT_CAP_ID_MCAST]	=	PCI_EXT_CAP_MCAST_ENDPOINT_SIZEOF,
+	[PCI_EXT_CAP_ID_PRI]	=	PCI_EXT_CAP_PRI_SIZEOF,
+	[PCI_EXT_CAP_ID_AMD_XXX] =	0,	/* not yet */
+	[PCI_EXT_CAP_ID_REBAR]	=	0xFF,
+	[PCI_EXT_CAP_ID_DPA]	=	0xFF,
+	[PCI_EXT_CAP_ID_TPH]	=	0xFF,
+	[PCI_EXT_CAP_ID_LTR]	=	PCI_EXT_CAP_LTR_SIZEOF,
+	[PCI_EXT_CAP_ID_SECPCI]	=	0,	/* not yet */
+	[PCI_EXT_CAP_ID_PMUX]	=	0,	/* not yet */
+	[PCI_EXT_CAP_ID_PASID]	=	0,	/* not yet */
+};
+
+/*
+ * Read/Write Permission Bits - one bit for each bit in capability
+ * Any field can be read if it exists, but what is read depends on
+ * whether the field is 'virtualized', or just pass thru to the
+ * hardware.  Any virtualized field is also virtualized for writes.
+ * Writes are only permitted if they have a 1 bit here.
+ */
+struct perm_bits {
+	u8	*virt;		/* read/write virtual data, not hw */
+	u8	*write;		/* writeable bits */
+	int	(*readfn)(struct vfio_pci_device *vdev, int pos, int count,
+			  struct perm_bits *perm, int offset, __le32 *val);
+	int	(*writefn)(struct vfio_pci_device *vdev, int pos, int count,
+			   struct perm_bits *perm, int offset, __le32 val);
+};
+
+#define	NO_VIRT		0
+#define	ALL_VIRT	0xFFFFFFFFU
+#define	NO_WRITE	0
+#define	ALL_WRITE	0xFFFFFFFFU
+
+static int vfio_user_config_read(struct pci_dev *pdev, int offset,
+				 __le32 *val, int count)
+{
+	int ret = -EINVAL;
+	u32 tmp_val = 0;
+
+	switch (count) {
+	case 1:
+	{
+		u8 tmp;
+		ret = pci_user_read_config_byte(pdev, offset, &tmp);
+		tmp_val = tmp;
+		break;
+	}
+	case 2:
+	{
+		u16 tmp;
+		ret = pci_user_read_config_word(pdev, offset, &tmp);
+		tmp_val = tmp;
+		break;
+	}
+	case 4:
+		ret = pci_user_read_config_dword(pdev, offset, &tmp_val);
+		break;
+	}
+
+	*val = cpu_to_le32(tmp_val);
+
+	return pcibios_err_to_errno(ret);
+}
+
+static int vfio_user_config_write(struct pci_dev *pdev, int offset,
+				  __le32 val, int count)
+{
+	int ret = -EINVAL;
+	u32 tmp_val = le32_to_cpu(val);
+
+	switch (count) {
+	case 1:
+		ret = pci_user_write_config_byte(pdev, offset, tmp_val);
+		break;
+	case 2:
+		ret = pci_user_write_config_word(pdev, offset, tmp_val);
+		break;
+	case 4:
+		ret = pci_user_write_config_dword(pdev, offset, tmp_val);
+		break;
+	}
+
+	return pcibios_err_to_errno(ret);
+}
+
+static int vfio_default_config_read(struct vfio_pci_device *vdev, int pos,
+				    int count, struct perm_bits *perm,
+				    int offset, __le32 *val)
+{
+	__le32 virt = 0;
+
+	memcpy(val, vdev->vconfig + pos, count);
+
+	memcpy(&virt, perm->virt + offset, count);
+
+	/* Any non-virtualized bits? */
+	if (cpu_to_le32(~0U >> (32 - (count * 8))) != virt) {
+		struct pci_dev *pdev = vdev->pdev;
+		__le32 phys_val = 0;
+		int ret;
+
+		ret = vfio_user_config_read(pdev, pos, &phys_val, count);
+		if (ret)
+			return ret;
+
+		*val = (phys_val & ~virt) | (*val & virt);
+	}
+
+	return count;
+}
+
+static int vfio_default_config_write(struct vfio_pci_device *vdev, int pos,
+				     int count, struct perm_bits *perm,
+				     int offset, __le32 val)
+{
+	__le32 virt = 0, write = 0;
+
+	memcpy(&write, perm->write + offset, count);
+
+	if (!write)
+		return count; /* drop, no writable bits */
+
+	memcpy(&virt, perm->virt + offset, count);
+
+	/* Virtualized and writable bits go to vconfig */
+	if (write & virt) {
+		__le32 virt_val = 0;
+
+		memcpy(&virt_val, vdev->vconfig + pos, count);
+
+		virt_val &= ~(write & virt);
+		virt_val |= (val & (write & virt));
+
+		memcpy(vdev->vconfig + pos, &virt_val, count);
+	}
+
+	/* Non-virtualzed and writable bits go to hardware */
+	if (write & ~virt) {
+		struct pci_dev *pdev = vdev->pdev;
+		__le32 phys_val = 0;
+		int ret;
+
+		ret = vfio_user_config_read(pdev, pos, &phys_val, count);
+		if (ret)
+			return ret;
+
+		phys_val &= ~(write & ~virt);
+		phys_val |= (val & (write & ~virt));
+
+		ret = vfio_user_config_write(pdev, pos, phys_val, count);
+		if (ret)
+			return ret;
+	}
+
+	return count;
+}
+
+/* Allow direct read from hardware, except for capability next pointer */
+static int vfio_direct_config_read(struct vfio_pci_device *vdev, int pos,
+				   int count, struct perm_bits *perm,
+				   int offset, __le32 *val)
+{
+	int ret;
+
+	ret = vfio_user_config_read(vdev->pdev, pos, val, count);
+	if (ret)
+		return pcibios_err_to_errno(ret);
+
+	if (pos >= PCI_CFG_SPACE_SIZE) { /* Extended cap header mangling */
+		if (offset < 4)
+			memcpy(val, vdev->vconfig + pos, count);
+	} else if (pos >= PCI_STD_HEADER_SIZEOF) { /* Std cap mangling */
+		if (offset == PCI_CAP_LIST_ID && count > 1)
+			memcpy(val, vdev->vconfig + pos,
+			       min(PCI_CAP_FLAGS, count));
+		else if (offset == PCI_CAP_LIST_NEXT)
+			memcpy(val, vdev->vconfig + pos, 1);
+	}
+
+	return count;
+}
+
+static int vfio_direct_config_write(struct vfio_pci_device *vdev, int pos,
+				    int count, struct perm_bits *perm,
+				    int offset, __le32 val)
+{
+	int ret;
+
+	ret = vfio_user_config_write(vdev->pdev, pos, val, count);
+	if (ret)
+		return ret;
+
+	return count;
+}
+
+/* Default all regions to read-only, no-virtualization */
+static struct perm_bits cap_perms[PCI_CAP_ID_MAX + 1] = {
+	[0 ... PCI_CAP_ID_MAX] = { .readfn = vfio_direct_config_read }
+};
+static struct perm_bits ecap_perms[PCI_EXT_CAP_ID_MAX + 1] = {
+	[0 ... PCI_EXT_CAP_ID_MAX] = { .readfn = vfio_direct_config_read }
+};
+
+static void free_perm_bits(struct perm_bits *perm)
+{
+	kfree(perm->virt);
+	kfree(perm->write);
+	perm->virt = NULL;
+	perm->write = NULL;
+}
+
+static int alloc_perm_bits(struct perm_bits *perm, int size)
+{
+	/*
+	 * Round up all permission bits to the next dword, this lets us
+	 * ignore whether a read/write exceeds the defined capability
+	 * structure.  We can do this because:
+	 *  - Standard config space is already dword aligned
+	 *  - Capabilities are all dword alinged (bits 0:1 of next reserved)
+	 *  - Express capabilities defined as dword aligned
+	 */
+	size = round_up(size, 4);
+
+	/*
+	 * Zero state is
+	 * - All Readable, None Writeable, None Virtualized
+	 */
+	perm->virt = kzalloc(size, GFP_KERNEL);
+	perm->write = kzalloc(size, GFP_KERNEL);
+	if (!perm->virt || !perm->write) {
+		free_perm_bits(perm);
+		return -ENOMEM;
+	}
+
+	perm->readfn = vfio_default_config_read;
+	perm->writefn = vfio_default_config_write;
+
+	return 0;
+}
+
+/*
+ * Helper functions for filling in permission tables
+ */
+static inline void p_setb(struct perm_bits *p, int off, u8 virt, u8 write)
+{
+	p->virt[off] = virt;
+	p->write[off] = write;
+}
+
+/* Handle endian-ness - pci and tables are little-endian */
+static inline void p_setw(struct perm_bits *p, int off, u16 virt, u16 write)
+{
+	*(__le16 *)(&p->virt[off]) = cpu_to_le16(virt);
+	*(__le16 *)(&p->write[off]) = cpu_to_le16(write);
+}
+
+/* Handle endian-ness - pci and tables are little-endian */
+static inline void p_setd(struct perm_bits *p, int off, u32 virt, u32 write)
+{
+	*(__le32 *)(&p->virt[off]) = cpu_to_le32(virt);
+	*(__le32 *)(&p->write[off]) = cpu_to_le32(write);
+}
+
+/*
+ * Restore the *real* BARs after we detect a FLR or backdoor reset.
+ * (backdoor = some device specific technique that we didn't catch)
+ */
+static void vfio_bar_restore(struct vfio_pci_device *vdev)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	u32 *rbar = vdev->rbar;
+	int i;
+
+	if (pdev->is_virtfn)
+		return;
+
+	pr_info("%s: %s reset recovery - restoring bars\n",
+		__func__, dev_name(&pdev->dev));
+
+	for (i = PCI_BASE_ADDRESS_0; i <= PCI_BASE_ADDRESS_5; i += 4, rbar++)
+		pci_user_write_config_dword(pdev, i, *rbar);
+
+	pci_user_write_config_dword(pdev, PCI_ROM_ADDRESS, *rbar);
+}
+
+static __le32 vfio_generate_bar_flags(struct pci_dev *pdev, int bar)
+{
+	unsigned long flags = pci_resource_flags(pdev, bar);
+	u32 val;
+
+	if (flags & IORESOURCE_IO)
+		return cpu_to_le32(PCI_BASE_ADDRESS_SPACE_IO);
+
+	val = PCI_BASE_ADDRESS_SPACE_MEMORY;
+
+	if (flags & IORESOURCE_PREFETCH)
+		val |= PCI_BASE_ADDRESS_MEM_PREFETCH;
+
+	if (flags & IORESOURCE_MEM_64)
+		val |= PCI_BASE_ADDRESS_MEM_TYPE_64;
+
+	return cpu_to_le32(val);
+}
+
+/*
+ * Pretend we're hardware and tweak the values of the *virtual* PCI BARs
+ * to reflect the hardware capabilities.  This implements BAR sizing.
+ */
+static void vfio_bar_fixup(struct vfio_pci_device *vdev)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	int i;
+	__le32 *bar;
+	u64 mask;
+
+	bar = (__le32 *)&vdev->vconfig[PCI_BASE_ADDRESS_0];
+
+	for (i = PCI_STD_RESOURCES; i <= PCI_STD_RESOURCE_END; i++, bar++) {
+		if (!pci_resource_start(pdev, i)) {
+			*bar = 0; /* Unmapped by host = unimplemented to user */
+			continue;
+		}
+
+		mask = ~(pci_resource_len(pdev, i) - 1);
+
+		*bar &= cpu_to_le32((u32)mask);
+		*bar |= vfio_generate_bar_flags(pdev, i);
+
+		if (*bar & cpu_to_le32(PCI_BASE_ADDRESS_MEM_TYPE_64)) {
+			bar++;
+			*bar &= cpu_to_le32((u32)(mask >> 32));
+			i++;
+		}
+	}
+
+	bar = (__le32 *)&vdev->vconfig[PCI_ROM_ADDRESS];
+
+	/*
+	 * NB. we expose the actual BAR size here, regardless of whether
+	 * we can read it.  When we report the REGION_INFO for the ROM
+	 * we report what PCI tells us is the actual ROM size.
+	 */
+	if (pci_resource_start(pdev, PCI_ROM_RESOURCE)) {
+		mask = ~(pci_resource_len(pdev, PCI_ROM_RESOURCE) - 1);
+		mask |= PCI_ROM_ADDRESS_ENABLE;
+		*bar &= cpu_to_le32((u32)mask);
+	} else
+		*bar = 0;
+
+	vdev->bardirty = false;
+}
+
+static int vfio_basic_config_read(struct vfio_pci_device *vdev, int pos,
+				  int count, struct perm_bits *perm,
+				  int offset, __le32 *val)
+{
+	if (is_bar(offset)) /* pos == offset for basic config */
+		vfio_bar_fixup(vdev);
+
+	count = vfio_default_config_read(vdev, pos, count, perm, offset, val);
+
+	/* Mask in virtual memory enable for SR-IOV devices */
+	if (offset == PCI_COMMAND && vdev->pdev->is_virtfn) {
+		u16 cmd = le16_to_cpu(*(__le16 *)&vdev->vconfig[PCI_COMMAND]);
+		u32 tmp_val = le32_to_cpu(*val);
+
+		tmp_val |= cmd & PCI_COMMAND_MEMORY;
+		*val = cpu_to_le32(tmp_val);
+	}
+
+	return count;
+}
+
+static int vfio_basic_config_write(struct vfio_pci_device *vdev, int pos,
+				   int count, struct perm_bits *perm,
+				   int offset, __le32 val)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	__le16 *virt_cmd;
+	u16 new_cmd = 0;
+	int ret;
+
+	virt_cmd = (__le16 *)&vdev->vconfig[PCI_COMMAND];
+
+	if (offset == PCI_COMMAND) {
+		bool phys_mem, virt_mem, new_mem, phys_io, virt_io, new_io;
+		u16 phys_cmd;
+
+		ret = pci_user_read_config_word(pdev, PCI_COMMAND, &phys_cmd);
+		if (ret)
+			return ret;
+
+		new_cmd = le32_to_cpu(val);
+
+		phys_mem = !!(phys_cmd & PCI_COMMAND_MEMORY);
+		virt_mem = !!(le16_to_cpu(*virt_cmd) & PCI_COMMAND_MEMORY);
+		new_mem = !!(new_cmd & PCI_COMMAND_MEMORY);
+
+		phys_io = !!(phys_cmd & PCI_COMMAND_IO);
+		virt_io = !!(le16_to_cpu(*virt_cmd) & PCI_COMMAND_IO);
+		new_io = !!(new_cmd & PCI_COMMAND_IO);
+
+		/*
+		 * If the user is writing mem/io enable (new_mem/io) and we
+		 * think it's already enabled (virt_mem/io), but the hardware
+		 * shows it disabled (phys_mem/io, then the device has
+		 * undergone some kind of backdoor reset and needs to be
+		 * restored before we allow it to enable the bars.
+		 * SR-IOV devices will trigger this, but we catch them later
+		 */
+		if ((new_mem && virt_mem && !phys_mem) ||
+		    (new_io && virt_io && !phys_io))
+			vfio_bar_restore(vdev);
+	}
+
+	count = vfio_default_config_write(vdev, pos, count, perm, offset, val);
+	if (count < 0)
+		return count;
+
+	/*
+	 * Save current memory/io enable bits in vconfig to allow for
+	 * the test above next time.
+	 */
+	if (offset == PCI_COMMAND) {
+		u16 mask = PCI_COMMAND_MEMORY | PCI_COMMAND_IO;
+
+		*virt_cmd &= cpu_to_le16(~mask);
+		*virt_cmd |= cpu_to_le16(new_cmd & mask);
+	}
+
+	/* Emulate INTx disable */
+	if (offset >= PCI_COMMAND && offset <= PCI_COMMAND + 1) {
+		bool virt_intx_disable;
+
+		virt_intx_disable = !!(le16_to_cpu(*virt_cmd) &
+				       PCI_COMMAND_INTX_DISABLE);
+
+		if (virt_intx_disable && !vdev->virq_disabled) {
+			vdev->virq_disabled = true;
+			vfio_pci_intx_mask(vdev);
+		} else if (!virt_intx_disable && vdev->virq_disabled) {
+			vdev->virq_disabled = false;
+			vfio_pci_intx_unmask(vdev);
+		}
+	}
+
+	if (is_bar(offset))
+		vdev->bardirty = true;
+
+	return count;
+}
+
+/* Permissions for the Basic PCI Header */
+static int __init init_pci_cap_basic_perm(struct perm_bits *perm)
+{
+	if (alloc_perm_bits(perm, PCI_STD_HEADER_SIZEOF))
+		return -ENOMEM;
+
+	perm->readfn = vfio_basic_config_read;
+	perm->writefn = vfio_basic_config_write;
+
+	/* Virtualized for SR-IOV functions, which just have FFFF */
+	p_setw(perm, PCI_VENDOR_ID, (u16)ALL_VIRT, NO_WRITE);
+	p_setw(perm, PCI_DEVICE_ID, (u16)ALL_VIRT, NO_WRITE);
+
+	/*
+	 * Virtualize INTx disable, we use it internally for interrupt
+	 * control and can emulate it for non-PCI 2.3 devices.
+	 */
+	p_setw(perm, PCI_COMMAND, PCI_COMMAND_INTX_DISABLE, (u16)ALL_WRITE);
+
+	/* Virtualize capability list, we might want to skip/disable */
+	p_setw(perm, PCI_STATUS, PCI_STATUS_CAP_LIST, NO_WRITE);
+
+	/* No harm to write */
+	p_setb(perm, PCI_CACHE_LINE_SIZE, NO_VIRT, (u8)ALL_WRITE);
+	p_setb(perm, PCI_LATENCY_TIMER, NO_VIRT, (u8)ALL_WRITE);
+	p_setb(perm, PCI_BIST, NO_VIRT, (u8)ALL_WRITE);
+
+	/* Virtualize all bars, can't touch the real ones */
+	p_setd(perm, PCI_BASE_ADDRESS_0, ALL_VIRT, ALL_WRITE);
+	p_setd(perm, PCI_BASE_ADDRESS_1, ALL_VIRT, ALL_WRITE);
+	p_setd(perm, PCI_BASE_ADDRESS_2, ALL_VIRT, ALL_WRITE);
+	p_setd(perm, PCI_BASE_ADDRESS_3, ALL_VIRT, ALL_WRITE);
+	p_setd(perm, PCI_BASE_ADDRESS_4, ALL_VIRT, ALL_WRITE);
+	p_setd(perm, PCI_BASE_ADDRESS_5, ALL_VIRT, ALL_WRITE);
+	p_setd(perm, PCI_ROM_ADDRESS, ALL_VIRT, ALL_WRITE);
+
+	/* Allow us to adjust capability chain */
+	p_setb(perm, PCI_CAPABILITY_LIST, (u8)ALL_VIRT, NO_WRITE);
+
+	/* Sometimes used by sw, just virtualize */
+	p_setb(perm, PCI_INTERRUPT_LINE, (u8)ALL_VIRT, (u8)ALL_WRITE);
+	return 0;
+}
+
+/* Permissions for the Power Management capability */
+static int __init init_pci_cap_pm_perm(struct perm_bits *perm)
+{
+	if (alloc_perm_bits(perm, pci_cap_length[PCI_CAP_ID_PM]))
+		return -ENOMEM;
+
+	/*
+	 * We always virtualize the next field so we can remove
+	 * capabilities from the chain if we want to.
+	 */
+	p_setb(perm, PCI_CAP_LIST_NEXT, (u8)ALL_VIRT, NO_WRITE);
+
+	/*
+	 * Power management is defined *per function*,
+	 * so we let the user write this
+	 */
+	p_setd(perm, PCI_PM_CTRL, NO_VIRT, ALL_WRITE);
+	return 0;
+}
+
+/* Permissions for PCI-X capability */
+static int __init init_pci_cap_pcix_perm(struct perm_bits *perm)
+{
+	/* Alloc 24, but only 8 are used in v0 */
+	if (alloc_perm_bits(perm, PCI_CAP_PCIX_SIZEOF_V2))
+		return -ENOMEM;
+
+	p_setb(perm, PCI_CAP_LIST_NEXT, (u8)ALL_VIRT, NO_WRITE);
+
+	p_setw(perm, PCI_X_CMD, NO_VIRT, (u16)ALL_WRITE);
+	p_setd(perm, PCI_X_ECC_CSR, NO_VIRT, ALL_WRITE);
+	return 0;
+}
+
+/* Permissions for PCI Express capability */
+static int __init init_pci_cap_exp_perm(struct perm_bits *perm)
+{
+	/* Alloc larger of two possible sizes */
+	if (alloc_perm_bits(perm, PCI_CAP_EXP_ENDPOINT_SIZEOF_V2))
+		return -ENOMEM;
+
+	p_setb(perm, PCI_CAP_LIST_NEXT, (u8)ALL_VIRT, NO_WRITE);
+
+	/*
+	 * Allow writes to device control fields (includes FLR!)
+	 * but not to devctl_phantom which could confuse IOMMU
+	 * or to the ARI bit in devctl2 which is set at probe time
+	 */
+	p_setw(perm, PCI_EXP_DEVCTL, NO_VIRT, ~PCI_EXP_DEVCTL_PHANTOM);
+	p_setw(perm, PCI_EXP_DEVCTL2, NO_VIRT, ~PCI_EXP_DEVCTL2_ARI);
+	return 0;
+}
+
+/* Permissions for Advanced Function capability */
+static int __init init_pci_cap_af_perm(struct perm_bits *perm)
+{
+	if (alloc_perm_bits(perm, pci_cap_length[PCI_CAP_ID_AF]))
+		return -ENOMEM;
+
+	p_setb(perm, PCI_CAP_LIST_NEXT, (u8)ALL_VIRT, NO_WRITE);
+	p_setb(perm, PCI_AF_CTRL, NO_VIRT, PCI_AF_CTRL_FLR);
+	return 0;
+}
+
+/* Permissions for Advanced Error Reporting extended capability */
+static int __init init_pci_ext_cap_err_perm(struct perm_bits *perm)
+{
+	u32 mask;
+
+	if (alloc_perm_bits(perm, pci_ext_cap_length[PCI_EXT_CAP_ID_ERR]))
+		return -ENOMEM;
+
+	/*
+	 * Virtualize the first dword of all express capabilities
+	 * because it includes the next pointer.  This lets us later
+	 * remove capabilities from the chain if we need to.
+	 */
+	p_setd(perm, 0, ALL_VIRT, NO_WRITE);
+
+	/* Writable bits mask */
+	mask =	PCI_ERR_UNC_TRAIN |		/* Training */
+		PCI_ERR_UNC_DLP |		/* Data Link Protocol */
+		PCI_ERR_UNC_SURPDN |		/* Surprise Down */
+		PCI_ERR_UNC_POISON_TLP |	/* Poisoned TLP */
+		PCI_ERR_UNC_FCP |		/* Flow Control Protocol */
+		PCI_ERR_UNC_COMP_TIME |		/* Completion Timeout */
+		PCI_ERR_UNC_COMP_ABORT |	/* Completer Abort */
+		PCI_ERR_UNC_UNX_COMP |		/* Unexpected Completion */
+		PCI_ERR_UNC_RX_OVER |		/* Receiver Overflow */
+		PCI_ERR_UNC_MALF_TLP |		/* Malformed TLP */
+		PCI_ERR_UNC_ECRC |		/* ECRC Error Status */
+		PCI_ERR_UNC_UNSUP |		/* Unsupported Request */
+		PCI_ERR_UNC_ACSV |		/* ACS Violation */
+		PCI_ERR_UNC_INTN |		/* internal error */
+		PCI_ERR_UNC_MCBTLP |		/* MC blocked TLP */
+		PCI_ERR_UNC_ATOMEG |		/* Atomic egress blocked */
+		PCI_ERR_UNC_TLPPRE;		/* TLP prefix blocked */
+	p_setd(perm, PCI_ERR_UNCOR_STATUS, NO_VIRT, mask);
+	p_setd(perm, PCI_ERR_UNCOR_MASK, NO_VIRT, mask);
+	p_setd(perm, PCI_ERR_UNCOR_SEVER, NO_VIRT, mask);
+
+	mask =	PCI_ERR_COR_RCVR |		/* Receiver Error Status */
+		PCI_ERR_COR_BAD_TLP |		/* Bad TLP Status */
+		PCI_ERR_COR_BAD_DLLP |		/* Bad DLLP Status */
+		PCI_ERR_COR_REP_ROLL |		/* REPLAY_NUM Rollover */
+		PCI_ERR_COR_REP_TIMER |		/* Replay Timer Timeout */
+		PCI_ERR_COR_ADV_NFAT |		/* Advisory Non-Fatal */
+		PCI_ERR_COR_INTERNAL |		/* Corrected Internal */
+		PCI_ERR_COR_LOG_OVER;		/* Header Log Overflow */
+	p_setd(perm, PCI_ERR_COR_STATUS, NO_VIRT, mask);
+	p_setd(perm, PCI_ERR_COR_MASK, NO_VIRT, mask);
+
+	mask =	PCI_ERR_CAP_ECRC_GENE |		/* ECRC Generation Enable */
+		PCI_ERR_CAP_ECRC_CHKE;		/* ECRC Check Enable */
+	p_setd(perm, PCI_ERR_CAP, NO_VIRT, mask);
+	return 0;
+}
+
+/* Permissions for Power Budgeting extended capability */
+static int __init init_pci_ext_cap_pwr_perm(struct perm_bits *perm)
+{
+	if (alloc_perm_bits(perm, pci_ext_cap_length[PCI_EXT_CAP_ID_PWR]))
+		return -ENOMEM;
+
+	p_setd(perm, 0, ALL_VIRT, NO_WRITE);
+
+	/* Writing the data selector is OK, the info is still read-only */
+	p_setb(perm, PCI_PWR_DATA, NO_VIRT, (u8)ALL_WRITE);
+	return 0;
+}
+
+/*
+ * Initialize the shared permission tables
+ */
+void vfio_pci_uninit_perm_bits(void)
+{
+	free_perm_bits(&cap_perms[PCI_CAP_ID_BASIC]);
+
+	free_perm_bits(&cap_perms[PCI_CAP_ID_PM]);
+	free_perm_bits(&cap_perms[PCI_CAP_ID_PCIX]);
+	free_perm_bits(&cap_perms[PCI_CAP_ID_EXP]);
+	free_perm_bits(&cap_perms[PCI_CAP_ID_AF]);
+
+	free_perm_bits(&ecap_perms[PCI_EXT_CAP_ID_ERR]);
+	free_perm_bits(&ecap_perms[PCI_EXT_CAP_ID_PWR]);
+}
+
+int __init vfio_pci_init_perm_bits(void)
+{
+	int ret;
+
+	/* Basic config space */
+	ret = init_pci_cap_basic_perm(&cap_perms[PCI_CAP_ID_BASIC]);
+
+	/* Capabilities */
+	ret |= init_pci_cap_pm_perm(&cap_perms[PCI_CAP_ID_PM]);
+	cap_perms[PCI_CAP_ID_VPD].writefn = vfio_direct_config_write;
+	ret |= init_pci_cap_pcix_perm(&cap_perms[PCI_CAP_ID_PCIX]);
+	cap_perms[PCI_CAP_ID_VNDR].writefn = vfio_direct_config_write;
+	ret |= init_pci_cap_exp_perm(&cap_perms[PCI_CAP_ID_EXP]);
+	ret |= init_pci_cap_af_perm(&cap_perms[PCI_CAP_ID_AF]);
+
+	/* Extended capabilities */
+	ret |= init_pci_ext_cap_err_perm(&ecap_perms[PCI_EXT_CAP_ID_ERR]);
+	ret |= init_pci_ext_cap_pwr_perm(&ecap_perms[PCI_EXT_CAP_ID_PWR]);
+	ecap_perms[PCI_EXT_CAP_ID_VNDR].writefn = vfio_direct_config_write;
+
+	if (ret)
+		vfio_pci_uninit_perm_bits();
+
+	return ret;
+}
+
+static int vfio_find_cap_start(struct vfio_pci_device *vdev, int pos)
+{
+	u8 cap;
+	int base = (pos >= PCI_CFG_SPACE_SIZE) ? PCI_CFG_SPACE_SIZE :
+						 PCI_STD_HEADER_SIZEOF;
+	base /= 4;
+	pos /= 4;
+
+	cap = vdev->pci_config_map[pos];
+
+	if (cap == PCI_CAP_ID_BASIC)
+		return 0;
+
+	/* XXX Can we have to abutting capabilities of the same type? */
+	while (pos - 1 >= base && vdev->pci_config_map[pos - 1] == cap)
+		pos--;
+
+	return pos * 4;
+}
+
+static int vfio_msi_config_read(struct vfio_pci_device *vdev, int pos,
+				int count, struct perm_bits *perm,
+				int offset, __le32 *val)
+{
+	/* Update max available queue size from msi_qmax */
+	if (offset <= PCI_MSI_FLAGS && offset + count >= PCI_MSI_FLAGS) {
+		__le16 *flags;
+		int start;
+
+		start = vfio_find_cap_start(vdev, pos);
+
+		flags = (__le16 *)&vdev->vconfig[start];
+
+		*flags &= cpu_to_le16(~PCI_MSI_FLAGS_QMASK);
+		*flags |= cpu_to_le16(vdev->msi_qmax << 1);
+	}
+
+	return vfio_default_config_read(vdev, pos, count, perm, offset, val);
+}
+
+static int vfio_msi_config_write(struct vfio_pci_device *vdev, int pos,
+				 int count, struct perm_bits *perm,
+				 int offset, __le32 val)
+{
+	count = vfio_default_config_write(vdev, pos, count, perm, offset, val);
+	if (count < 0)
+		return count;
+
+	/* Fixup and write configured queue size and enable to hardware */
+	if (offset <= PCI_MSI_FLAGS && offset + count >= PCI_MSI_FLAGS) {
+		__le16 *pflags;
+		u16 flags;
+		int start, ret;
+
+		start = vfio_find_cap_start(vdev, pos);
+
+		pflags = (__le16 *)&vdev->vconfig[start + PCI_MSI_FLAGS];
+
+		flags = le16_to_cpu(*pflags);
+
+		/* MSI is enabled via ioctl */
+		if  (!is_msi(vdev))
+			flags &= ~PCI_MSI_FLAGS_ENABLE;
+
+		/* Check queue size */
+		if ((flags & PCI_MSI_FLAGS_QSIZE) >> 4 > vdev->msi_qmax) {
+			flags &= ~PCI_MSI_FLAGS_QSIZE;
+			flags |= vdev->msi_qmax << 4;
+		}
+
+		/* Write back to virt and to hardware */
+		*pflags = cpu_to_le16(flags);
+		ret = pci_user_write_config_word(vdev->pdev,
+						 start + PCI_MSI_FLAGS,
+						 flags);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+	}
+
+	return count;
+}
+
+/*
+ * MSI determination is per-device, so this routine gets used beyond
+ * initialization time. Don't add __init
+ */
+static int init_pci_cap_msi_perm(struct perm_bits *perm, int len, u16 flags)
+{
+	if (alloc_perm_bits(perm, len))
+		return -ENOMEM;
+
+	perm->readfn = vfio_msi_config_read;
+	perm->writefn = vfio_msi_config_write;
+
+	p_setb(perm, PCI_CAP_LIST_NEXT, (u8)ALL_VIRT, NO_WRITE);
+
+	/*
+	 * The upper byte of the control register is reserved,
+	 * just setup the lower byte.
+	 */
+	p_setb(perm, PCI_MSI_FLAGS, (u8)ALL_VIRT, (u8)ALL_WRITE);
+	p_setd(perm, PCI_MSI_ADDRESS_LO, ALL_VIRT, ALL_WRITE);
+	if (flags & PCI_MSI_FLAGS_64BIT) {
+		p_setd(perm, PCI_MSI_ADDRESS_HI, ALL_VIRT, ALL_WRITE);
+		p_setw(perm, PCI_MSI_DATA_64, (u16)ALL_VIRT, (u16)ALL_WRITE);
+		if (flags & PCI_MSI_FLAGS_MASKBIT) {
+			p_setd(perm, PCI_MSI_MASK_64, NO_VIRT, ALL_WRITE);
+			p_setd(perm, PCI_MSI_PENDING_64, NO_VIRT, ALL_WRITE);
+		}
+	} else {
+		p_setw(perm, PCI_MSI_DATA_32, (u16)ALL_VIRT, (u16)ALL_WRITE);
+		if (flags & PCI_MSI_FLAGS_MASKBIT) {
+			p_setd(perm, PCI_MSI_MASK_32, NO_VIRT, ALL_WRITE);
+			p_setd(perm, PCI_MSI_PENDING_32, NO_VIRT, ALL_WRITE);
+		}
+	}
+	return 0;
+}
+
+/* Determine MSI CAP field length; initialize msi_perms on 1st call per vdev */
+static int vfio_msi_cap_len(struct vfio_pci_device *vdev, u8 pos)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	int len, ret;
+	u16 flags;
+
+	ret = pci_read_config_word(pdev, pos + PCI_MSI_FLAGS, &flags);
+	if (ret)
+		return pcibios_err_to_errno(ret);
+
+	len = 10; /* Minimum size */
+	if (flags & PCI_MSI_FLAGS_64BIT)
+		len += 4;
+	if (flags & PCI_MSI_FLAGS_MASKBIT)
+		len += 10;
+
+	if (vdev->msi_perm)
+		return len;
+
+	vdev->msi_perm = kmalloc(sizeof(struct perm_bits), GFP_KERNEL);
+	if (!vdev->msi_perm)
+		return -ENOMEM;
+
+	ret = init_pci_cap_msi_perm(vdev->msi_perm, len, flags);
+	if (ret)
+		return ret;
+
+	return len;
+}
+
+/* Determine extended capability length for VC (2 & 9) and MFVC */
+static int vfio_vc_cap_len(struct vfio_pci_device *vdev, u16 pos)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	u32 tmp;
+	int ret, evcc, phases, vc_arb;
+	int len = PCI_CAP_VC_BASE_SIZEOF;
+
+	ret = pci_read_config_dword(pdev, pos + PCI_VC_PORT_REG1, &tmp);
+	if (ret)
+		return pcibios_err_to_errno(ret);
+
+	evcc = tmp & PCI_VC_REG1_EVCC; /* extended vc count */
+	ret = pci_read_config_dword(pdev, pos + PCI_VC_PORT_REG2, &tmp);
+	if (ret)
+		return pcibios_err_to_errno(ret);
+
+	if (tmp & PCI_VC_REG2_128_PHASE)
+		phases = 128;
+	else if (tmp & PCI_VC_REG2_64_PHASE)
+		phases = 64;
+	else if (tmp & PCI_VC_REG2_32_PHASE)
+		phases = 32;
+	else
+		phases = 0;
+
+	vc_arb = phases * 4;
+
+	/*
+	 * Port arbitration tables are root & switch only;
+	 * function arbitration tables are function 0 only.
+	 * In either case, we'll never let user write them so
+	 * we don't care how big they are
+	 */
+	len += (1 + evcc) * PCI_CAP_VC_PER_VC_SIZEOF;
+	if (vc_arb) {
+		len = round_up(len, 16);
+		len += vc_arb / 8;
+	}
+	return len;
+}
+
+static int vfio_cap_len(struct vfio_pci_device *vdev, u8 cap, u8 pos)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	u16 word;
+	u8 byte;
+	int ret;
+
+	switch (cap) {
+	case PCI_CAP_ID_MSI:
+		return vfio_msi_cap_len(vdev, pos);
+	case PCI_CAP_ID_PCIX:
+		ret = pci_read_config_word(pdev, pos + PCI_X_CMD, &word);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		if (PCI_X_CMD_VERSION(word)) {
+			vdev->extended_caps = true;
+			return PCI_CAP_PCIX_SIZEOF_V2;
+		} else
+			return PCI_CAP_PCIX_SIZEOF_V0;
+	case PCI_CAP_ID_VNDR:
+		/* length follows next field */
+		ret = pci_read_config_byte(pdev, pos + PCI_CAP_FLAGS, &byte);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		return byte;
+	case PCI_CAP_ID_EXP:
+		/* length based on version */
+		ret = pci_read_config_word(pdev, pos + PCI_EXP_FLAGS, &word);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		if ((word & PCI_EXP_FLAGS_VERS) == 1)
+			return PCI_CAP_EXP_ENDPOINT_SIZEOF_V1;
+		else {
+			vdev->extended_caps = true;
+			return PCI_CAP_EXP_ENDPOINT_SIZEOF_V2;
+		}
+	case PCI_CAP_ID_HT:
+		ret = pci_read_config_byte(pdev, pos + 3, &byte);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		return (byte & HT_3BIT_CAP_MASK) ?
+			HT_CAP_SIZEOF_SHORT : HT_CAP_SIZEOF_LONG;
+	case PCI_CAP_ID_SATA:
+		ret = pci_read_config_byte(pdev, pos + PCI_SATA_REGS, &byte);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		byte &= PCI_SATA_REGS_MASK;
+		if (byte == PCI_SATA_REGS_INLINE)
+			return PCI_SATA_SIZEOF_LONG;
+		else
+			return PCI_SATA_SIZEOF_SHORT;
+	default:
+		pr_warn("%s: %s unknown length for pci cap 0x%x@0x%x\n",
+			dev_name(&pdev->dev), __func__, cap, pos);
+	}
+
+	return 0;
+}
+
+static int vfio_ext_cap_len(struct vfio_pci_device *vdev, u16 ecap, u16 epos)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	u8 byte;
+	u32 dword;
+	int ret;
+
+	switch (ecap) {
+	case PCI_EXT_CAP_ID_VNDR:
+		ret = pci_read_config_dword(pdev, epos + PCI_VSEC_HDR, &dword);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		return dword >> PCI_VSEC_HDR_LEN_SHIFT;
+	case PCI_EXT_CAP_ID_VC:
+	case PCI_EXT_CAP_ID_VC9:
+	case PCI_EXT_CAP_ID_MFVC:
+		return vfio_vc_cap_len(vdev, epos);
+	case PCI_EXT_CAP_ID_ACS:
+		ret = pci_read_config_byte(pdev, epos + PCI_ACS_CAP, &byte);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		if (byte & PCI_ACS_EC) {
+			int bits;
+
+			ret = pci_read_config_byte(pdev,
+						   epos + PCI_ACS_EGRESS_BITS,
+						   &byte);
+			if (ret)
+				return pcibios_err_to_errno(ret);
+
+			bits = byte ? round_up(byte, 32) : 256;
+			return 8 + (bits / 8);
+		}
+		return 8;
+
+	case PCI_EXT_CAP_ID_REBAR:
+		ret = pci_read_config_byte(pdev, epos + PCI_REBAR_CTRL, &byte);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		byte &= PCI_REBAR_CTRL_NBAR_MASK;
+		byte >>= PCI_REBAR_CTRL_NBAR_SHIFT;
+
+		return 4 + (byte * 8);
+	case PCI_EXT_CAP_ID_DPA:
+		ret = pci_read_config_byte(pdev, epos + PCI_DPA_CAP, &byte);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		byte &= PCI_DPA_CAP_SUBSTATE_MASK;
+		byte = round_up(byte + 1, 4);
+		return PCI_DPA_BASE_SIZEOF + byte;
+	case PCI_EXT_CAP_ID_TPH:
+		ret = pci_read_config_dword(pdev, epos + PCI_TPH_CAP, &dword);
+		if (ret)
+			return pcibios_err_to_errno(ret);
+
+		if ((dword & PCI_TPH_CAP_LOC_MASK) == PCI_TPH_LOC_CAP) {
+			int sts;
+
+			sts = byte & PCI_TPH_CAP_ST_MASK;
+			sts >>= PCI_TPH_CAP_ST_SHIFT;
+			return PCI_TPH_BASE_SIZEOF + round_up(sts * 2, 4);
+		}
+		return PCI_TPH_BASE_SIZEOF;
+	default:
+		pr_warn("%s: %s unknown length for pci ecap 0x%x@0x%x\n",
+			dev_name(&pdev->dev), __func__, ecap, epos);
+	}
+
+	return 0;
+}
+
+static int vfio_fill_vconfig_bytes(struct vfio_pci_device *vdev,
+				   int offset, int size)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	int ret = 0;
+
+	/*
+	 * We try to read physical config space in the largest chunks
+	 * we can, assuming that all of the fields support dword access.
+	 * pci_save_state() makes this same assumption and seems to do ok.
+	 */
+	while (size) {
+		int filled;
+
+		if (size >= 4 && !(offset % 4)) {
+			__le32 *dwordp = (__le32 *)&vdev->vconfig[offset];
+			u32 dword;
+
+			ret = pci_read_config_dword(pdev, offset, &dword);
+			if (ret)
+				return ret;
+			*dwordp = cpu_to_le32(dword);
+			filled = 4;
+		} else if (size >= 2 && !(offset % 2)) {
+			__le16 *wordp = (__le16 *)&vdev->vconfig[offset];
+			u16 word;
+
+			ret = pci_read_config_word(pdev, offset, &word);
+			if (ret)
+				return ret;
+			*wordp = cpu_to_le16(word);
+			filled = 2;
+		} else {
+			u8 *byte = &vdev->vconfig[offset];
+			ret = pci_read_config_byte(pdev, offset, byte);
+			if (ret)
+				return ret;
+			filled = 1;
+		}
+
+		offset += filled;
+		size -= filled;
+	}
+
+	return ret;
+}
+
+static int vfio_cap_init(struct vfio_pci_device *vdev)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	u8 *map = vdev->pci_config_map;
+	u16 status;
+	u8 pos, *prev, cap;
+	int loops, ret, caps = 0;
+
+	/* Any capabilities? */
+	ret = pci_read_config_word(pdev, PCI_STATUS, &status);
+	if (ret)
+		return ret;
+
+	if (!(status & PCI_STATUS_CAP_LIST))
+		return 0; /* Done */
+
+	ret = pci_read_config_byte(pdev, PCI_CAPABILITY_LIST, &pos);
+	if (ret)
+		return ret;
+
+	/* Mark the previous position in case we want to skip a capability */
+	prev = &vdev->vconfig[PCI_CAPABILITY_LIST];
+
+	/* We can bound our loop, capabilities are dword aligned */
+	loops = (PCI_CFG_SPACE_SIZE - PCI_STD_HEADER_SIZEOF) / PCI_CAP_SIZEOF;
+	while (pos && loops--) {
+		u8 next;
+		int i, len = 0;
+
+		ret = pci_read_config_byte(pdev, pos, &cap);
+		if (ret)
+			return ret;
+
+		ret = pci_read_config_byte(pdev,
+					   pos + PCI_CAP_LIST_NEXT, &next);
+		if (ret)
+			return ret;
+
+		if (cap <= PCI_CAP_ID_MAX) {
+			len = pci_cap_length[cap];
+			if (len == 0xFF) { /* Variable length */
+				len = vfio_cap_len(vdev, cap, pos);
+				if (len < 0)
+					return len;
+			}
+		}
+
+		if (!len) {
+			pr_info("%s: %s hiding cap 0x%x\n",
+				__func__, dev_name(&pdev->dev), cap);
+			*prev = next;
+			pos = next;
+			continue;
+		}
+
+		/* Sanity check, do we overlap other capabilities? */
+		for (i = 0; i < len; i += 4) {
+			if (likely(map[(pos + i) / 4] == PCI_CAP_ID_INVALID))
+				continue;
+
+			pr_warn("%s: %s pci config conflict @0x%x, was cap 0x%x now cap 0x%x\n",
+				__func__, dev_name(&pdev->dev),
+				pos + i, map[pos + i], cap);
+		}
+
+		memset(map + (pos / 4), cap, len / 4);
+		ret = vfio_fill_vconfig_bytes(vdev, pos, len);
+		if (ret)
+			return ret;
+
+		prev = &vdev->vconfig[pos + PCI_CAP_LIST_NEXT];
+		pos = next;
+		caps++;
+	}
+
+	/* If we didn't fill any capabilities, clear the status flag */
+	if (!caps) {
+		__le16 *vstatus = (__le16 *)&vdev->vconfig[PCI_STATUS];
+		*vstatus &= ~cpu_to_le16(PCI_STATUS_CAP_LIST);
+	}
+
+	return 0;
+}
+
+static int vfio_ecap_init(struct vfio_pci_device *vdev)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	u8 *map = vdev->pci_config_map;
+	u16 epos;
+	__le32 *prev = NULL;
+	int loops, ret, ecaps = 0;
+
+	if (!vdev->extended_caps)
+		return 0;
+
+	epos = PCI_CFG_SPACE_SIZE;
+
+	loops = (pdev->cfg_size - PCI_CFG_SPACE_SIZE) / PCI_CAP_SIZEOF;
+
+	while (loops-- && epos >= PCI_CFG_SPACE_SIZE) {
+		u32 header;
+		u16 ecap;
+		int i, len = 0;
+		bool hidden = false;
+
+		ret = pci_read_config_dword(pdev, epos, &header);
+		if (ret)
+			return ret;
+
+		ecap = PCI_EXT_CAP_ID(header);
+
+		if (ecap <= PCI_EXT_CAP_ID_MAX) {
+			len = pci_ext_cap_length[ecap];
+			if (len == 0xFF) {
+				len = vfio_ext_cap_len(vdev, ecap, epos);
+				if (len < 0)
+					return ret;
+			}
+		}
+
+		if (!len) {
+			pr_info("%s: %s hiding ecap 0x%x@0x%x\n",
+				__func__, dev_name(&pdev->dev), ecap, epos);
+
+			/* If not the first in the chain, we can skip over it */
+			if (prev) {
+				u32 val = epos = PCI_EXT_CAP_NEXT(header);
+				*prev &= cpu_to_le32(~(0xffcU << 20));
+				*prev |= cpu_to_le32(val << 20);
+				continue;
+			}
+
+			/*
+			 * Otherwise, fill in a placeholder, the direct
+			 * readfn will virtualize this automatically
+			 */
+			len = PCI_CAP_SIZEOF;
+			hidden = true;
+		}
+
+		for (i = 0; i < len; i += 4) {
+			if (likely(map[(epos + i) / 4] == PCI_CAP_ID_INVALID))
+				continue;
+
+			pr_warn("%s: %s pci config conflict @0x%x, was ecap 0x%x now ecap 0x%x\n",
+				__func__, dev_name(&pdev->dev),
+				epos + i, map[epos + i], ecap);
+		}
+
+		/*
+		 * Even though ecap is 2 bytes, we're currently a long way
+		 * from exceeding 1 byte capabilities.  If we ever make it
+		 * up to 0xFF we'll need to up this to a two-byte, byte map.
+		 */
+		BUILD_BUG_ON(PCI_EXT_CAP_ID_MAX >= PCI_CAP_ID_INVALID);
+
+		memset(map + (epos / 4), ecap, len / 4);
+		ret = vfio_fill_vconfig_bytes(vdev, epos, len);
+		if (ret)
+			return ret;
+
+		/*
+		 * If we're just using this capability to anchor the list,
+		 * hide the real ID.  Only count real ecaps.  XXX PCI spec
+		 * indicates to use cap id = 0, version = 0, next = 0 if
+		 * ecaps are absent, hope users check all the way to next.
+		 */
+		if (hidden)
+			*(__le32 *)&vdev->vconfig[epos] &=
+				cpu_to_le32((0xffcU << 20));
+		else
+			ecaps++;
+
+		prev = (__le32 *)&vdev->vconfig[epos];
+		epos = PCI_EXT_CAP_NEXT(header);
+	}
+
+	if (!ecaps)
+		*(u32 *)&vdev->vconfig[PCI_CFG_SPACE_SIZE] = 0;
+
+	return 0;
+}
+
+/*
+ * For each device we allocate a pci_config_map that indicates the
+ * capability occupying each dword and thus the struct perm_bits we
+ * use for read and write.  We also allocate a virtualized config
+ * space which tracks reads and writes to bits that we emulate for
+ * the user.  Initial values filled from device.
+ *
+ * Using shared stuct perm_bits between all vfio-pci devices saves
+ * us from allocating cfg_size buffers for virt and write for every
+ * device.  We could remove vconfig and allocate individual buffers
+ * for each area requring emulated bits, but the array of pointers
+ * would be comparable in size (at least for standard config space).
+ */
+int vfio_config_init(struct vfio_pci_device *vdev)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	u8 *map, *vconfig;
+	int ret;
+
+	/*
+	 * Config space, caps and ecaps are all dword aligned, so we can
+	 * use one byte per dword to record the type.
+	 */
+	map = kmalloc(pdev->cfg_size / 4, GFP_KERNEL);
+	if (!map)
+		return -ENOMEM;
+
+	vconfig = kmalloc(pdev->cfg_size, GFP_KERNEL);
+	if (!vconfig) {
+		kfree(map);
+		return -ENOMEM;
+	}
+
+	vdev->pci_config_map = map;
+	vdev->vconfig = vconfig;
+
+	memset(map, PCI_CAP_ID_BASIC, PCI_STD_HEADER_SIZEOF / 4);
+	memset(map + (PCI_STD_HEADER_SIZEOF / 4), PCI_CAP_ID_INVALID,
+	       (pdev->cfg_size - PCI_STD_HEADER_SIZEOF) / 4);
+
+	ret = vfio_fill_vconfig_bytes(vdev, 0, PCI_STD_HEADER_SIZEOF);
+	if (ret)
+		goto out;
+
+	vdev->bardirty = true;
+
+	/*
+	 * XXX can we just pci_load_saved_state/pci_restore_state?
+	 * may need to rebuild vconfig after that
+	 */
+
+	/* For restore after reset */
+	vdev->rbar[0] = le32_to_cpu(*(__le32 *)&vconfig[PCI_BASE_ADDRESS_0]);
+	vdev->rbar[1] = le32_to_cpu(*(__le32 *)&vconfig[PCI_BASE_ADDRESS_1]);
+	vdev->rbar[2] = le32_to_cpu(*(__le32 *)&vconfig[PCI_BASE_ADDRESS_2]);
+	vdev->rbar[3] = le32_to_cpu(*(__le32 *)&vconfig[PCI_BASE_ADDRESS_3]);
+	vdev->rbar[4] = le32_to_cpu(*(__le32 *)&vconfig[PCI_BASE_ADDRESS_4]);
+	vdev->rbar[5] = le32_to_cpu(*(__le32 *)&vconfig[PCI_BASE_ADDRESS_5]);
+	vdev->rbar[6] = le32_to_cpu(*(__le32 *)&vconfig[PCI_ROM_ADDRESS]);
+
+	if (pdev->is_virtfn) {
+		*(__le16 *)&vconfig[PCI_VENDOR_ID] = cpu_to_le16(pdev->vendor);
+		*(__le16 *)&vconfig[PCI_DEVICE_ID] = cpu_to_le16(pdev->device);
+	}
+
+	ret = vfio_cap_init(vdev);
+	if (ret)
+		goto out;
+
+	ret = vfio_ecap_init(vdev);
+	if (ret)
+		goto out;
+
+	return 0;
+
+out:
+	kfree(map);
+	vdev->pci_config_map = NULL;
+	kfree(vconfig);
+	vdev->vconfig = NULL;
+	return pcibios_err_to_errno(ret);
+}
+
+void vfio_config_free(struct vfio_pci_device *vdev)
+{
+	kfree(vdev->vconfig);
+	vdev->vconfig = NULL;
+	kfree(vdev->pci_config_map);
+	vdev->pci_config_map = NULL;
+	kfree(vdev->msi_perm);
+	vdev->msi_perm = NULL;
+}
+
+static ssize_t vfio_config_do_rw(struct vfio_pci_device *vdev, char __user *buf,
+				 size_t count, loff_t *ppos, bool iswrite)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	struct perm_bits *perm;
+	__le32 val = 0;
+	int cap_start = 0, offset;
+	u8 cap_id;
+	ssize_t ret = count;
+
+	if (*ppos < 0 || *ppos + count > pdev->cfg_size)
+		return -EFAULT;
+
+	/*
+	 * gcc can't seem to figure out we're a static function, only called
+	 * with count of 1/2/4 and hits copy_from_user_overflow without this.
+	 */
+	if (count > sizeof(val))
+		return -EINVAL;
+
+	cap_id = vdev->pci_config_map[*ppos / 4];
+
+	if (cap_id == PCI_CAP_ID_INVALID) {
+		if (iswrite)
+			return ret; /* drop */
+
+		/*
+		 * Per PCI spec 3.0, section 6.1, reads from reserved and
+		 * unimplemented registers return 0
+		 */
+		if (copy_to_user(buf, &val, count))
+			return -EFAULT;
+
+		return ret;
+	}
+
+	/*
+	 * All capabilities are minimum 4 bytes and aligned on dword
+	 * boundaries.  Since we don't support unaligned accesses, we're
+	 * only ever accessing a single capability.
+	 */
+	if (*ppos >= PCI_CFG_SPACE_SIZE) {
+		WARN_ON(cap_id > PCI_EXT_CAP_ID_MAX);
+
+		perm = &ecap_perms[cap_id];
+		cap_start = vfio_find_cap_start(vdev, *ppos);
+
+	} else {
+		WARN_ON(cap_id > PCI_CAP_ID_MAX);
+
+		perm = &cap_perms[cap_id];
+
+		if (cap_id == PCI_CAP_ID_MSI)
+			perm = vdev->msi_perm;
+
+		if (cap_id > PCI_CAP_ID_BASIC)
+			cap_start = vfio_find_cap_start(vdev, *ppos);
+	}
+
+	WARN_ON(!cap_start && cap_id != PCI_CAP_ID_BASIC);
+	WARN_ON(cap_start > *ppos);
+
+	offset = *ppos - cap_start;
+
+	if (iswrite) {
+		if (!perm->writefn)
+			return ret;
+
+		if (copy_from_user(&val, buf, count))
+			return -EFAULT;
+
+		ret = perm->writefn(vdev, *ppos, count, perm, offset, val);
+	} else {
+		if (perm->readfn) {
+			ret = perm->readfn(vdev, *ppos, count,
+					   perm, offset, &val);
+			if (ret < 0)
+				return ret;
+		}
+
+		if (copy_to_user(buf, &val, count))
+			return -EFAULT;
+	}
+
+	return ret;
+}
+
+ssize_t vfio_pci_config_readwrite(struct vfio_pci_device *vdev,
+				  char __user *buf, size_t count,
+				  loff_t *ppos, bool iswrite)
+{
+	size_t done = 0;
+	int ret = 0;
+	loff_t pos = *ppos;
+
+	pos &= VFIO_PCI_OFFSET_MASK;
+
+	/*
+	 * We want to both keep the access size the caller users as well as
+	 * support reading large chunks of config space in a single call.
+	 * PCI doesn't support unaligned accesses, so we can safely break
+	 * those apart.
+	 */
+	while (count) {
+		if (count >= 4 && !(pos % 4))
+			ret = vfio_config_do_rw(vdev, buf, 4, &pos, iswrite);
+		else if (count >= 2 && !(pos % 2))
+			ret = vfio_config_do_rw(vdev, buf, 2, &pos, iswrite);
+		else
+			ret = vfio_config_do_rw(vdev, buf, 1, &pos, iswrite);
+
+		if (ret < 0)
+			return ret;
+
+		count -= ret;
+		done += ret;
+		buf += ret;
+		pos += ret;
+	}
+
+	*ppos += done;
+
+	return done;
+}
+/*
+ * VFIO PCI interrupt handling
+ *
+ * Copyright (C) 2012 Red Hat, Inc.  All rights reserved.
+ *     Author: Alex Williamson <alex.williamson@redhat.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ *
+ * Derived from original vfio:
+ * Copyright 2010 Cisco Systems, Inc.  All rights reserved.
+ * Author: Tom Lyon, pugs@cisco.com
+ */
+
+#include <linux/device.h>
+#include <linux/interrupt.h>
+#include <linux/eventfd.h>
+#include <linux/pci.h>
+#include <linux/file.h>
+#include <linux/poll.h>
+#include <linux/vfio.h>
+#include <linux/wait.h>
+#include <linux/workqueue.h>
+
+#include "vfio_pci_private.h"
+
+/*
+ * IRQfd - generic
+ */
+struct virqfd {
+	struct vfio_pci_device	*vdev;
+	struct eventfd_ctx	*eventfd;
+	int			(*handler)(struct vfio_pci_device *, void *);
+	void			(*thread)(struct vfio_pci_device *, void *);
+	void			*data;
+	struct work_struct	inject;
+	wait_queue_t		wait;
+	poll_table		pt;
+	struct work_struct	shutdown;
+	struct virqfd		**pvirqfd;
+};
+
+static struct workqueue_struct *vfio_irqfd_cleanup_wq;
+
+int __init vfio_pci_virqfd_init(void)
+{
+	vfio_irqfd_cleanup_wq =
+		create_singlethread_workqueue("vfio-irqfd-cleanup");
+	if (!vfio_irqfd_cleanup_wq)
+		return -ENOMEM;
+
+	return 0;
+}
+
+void vfio_pci_virqfd_exit(void)
+{
+	destroy_workqueue(vfio_irqfd_cleanup_wq);
+}
+
+static void virqfd_deactivate(struct virqfd *virqfd)
+{
+	queue_work(vfio_irqfd_cleanup_wq, &virqfd->shutdown);
+}
+
+static int virqfd_wakeup(wait_queue_t *wait, unsigned mode, int sync, void *key)
+{
+	struct virqfd *virqfd = container_of(wait, struct virqfd, wait);
+	unsigned long flags = (unsigned long)key;
+
+	if (flags & POLLIN) {
+		/* An event has been signaled, call function */
+		if ((!virqfd->handler ||
+		     virqfd->handler(virqfd->vdev, virqfd->data)) &&
+		    virqfd->thread)
+			schedule_work(&virqfd->inject);
+	}
+
+	if (flags & POLLHUP)
+		/* The eventfd is closing, detach from VFIO */
+		virqfd_deactivate(virqfd);
+
+	return 0;
+}
+
+static void virqfd_ptable_queue_proc(struct file *file,
+				     wait_queue_head_t *wqh, poll_table *pt)
+{
+	struct virqfd *virqfd = container_of(pt, struct virqfd, pt);
+	add_wait_queue(wqh, &virqfd->wait);
+}
+
+static void virqfd_shutdown(struct work_struct *work)
+{
+	struct virqfd *virqfd = container_of(work, struct virqfd, shutdown);
+	struct virqfd **pvirqfd = virqfd->pvirqfd;
+	u64 cnt;
+
+	eventfd_ctx_remove_wait_queue(virqfd->eventfd, &virqfd->wait, &cnt);
+	flush_work(&virqfd->inject);
+	eventfd_ctx_put(virqfd->eventfd);
+
+	kfree(virqfd);
+	*pvirqfd = NULL;
+}
+
+static void virqfd_inject(struct work_struct *work)
+{
+	struct virqfd *virqfd = container_of(work, struct virqfd, inject);
+	if (virqfd->thread)
+		virqfd->thread(virqfd->vdev, virqfd->data);
+}
+
+static int virqfd_enable(struct vfio_pci_device *vdev,
+			 int (*handler)(struct vfio_pci_device *, void *),
+			 void (*thread)(struct vfio_pci_device *, void *),
+			 void *data, struct virqfd **pvirqfd, int fd)
+{
+	struct file *file = NULL;
+	struct eventfd_ctx *ctx = NULL;
+	struct virqfd *virqfd;
+	int ret = 0;
+	unsigned int events;
+
+	if (*pvirqfd)
+		return -EBUSY;
+
+	virqfd = kzalloc(sizeof(*virqfd), GFP_KERNEL);
+	if (!virqfd)
+		return -ENOMEM;
+
+	virqfd->pvirqfd = pvirqfd;
+	*pvirqfd = virqfd;
+	virqfd->vdev = vdev;
+	virqfd->handler = handler;
+	virqfd->thread = thread;
+	virqfd->data = data;
+
+	INIT_WORK(&virqfd->shutdown, virqfd_shutdown);
+	INIT_WORK(&virqfd->inject, virqfd_inject);
+
+	file = eventfd_fget(fd);
+	if (IS_ERR(file)) {
+		ret = PTR_ERR(file);
+		goto fail;
+	}
+
+	ctx = eventfd_ctx_fileget(file);
+	if (IS_ERR(ctx)) {
+		ret = PTR_ERR(ctx);
+		goto fail;
+	}
+
+	virqfd->eventfd = ctx;
+
+	/*
+	 * Install our own custom wake-up handling so we are notified via
+	 * a callback whenever someone signals the underlying eventfd.
+	 */
+	init_waitqueue_func_entry(&virqfd->wait, virqfd_wakeup);
+	init_poll_funcptr(&virqfd->pt, virqfd_ptable_queue_proc);
+
+	events = file->f_op->poll(file, &virqfd->pt);
+
+	/*
+	 * Check if there was an event already pending on the eventfd
+	 * before we registered and trigger it as if we didn't miss it.
+	 */
+	if (events & POLLIN) {
+		if ((!handler || handler(vdev, data)) && thread)
+			schedule_work(&virqfd->inject);
+	}
+
+	/*
+	 * Do not drop the file until the irqfd is fully initialized,
+	 * otherwise we might race against the POLLHUP.
+	 */
+	fput(file);
+
+	return 0;
+
+fail:
+	if (ctx && !IS_ERR(ctx))
+		eventfd_ctx_put(ctx);
+
+	if (file && !IS_ERR(file))
+		fput(file);
+
+	kfree(virqfd);
+	*pvirqfd = NULL;
+
+	return ret;
+}
+
+static void virqfd_disable(struct virqfd *virqfd)
+{
+	if (!virqfd)
+		return;
+
+	virqfd_deactivate(virqfd);
+
+	/* Block until we know all outstanding shutdown jobs have completed. */
+	flush_workqueue(vfio_irqfd_cleanup_wq);
+}
+
+/*
+ * INTx
+ */
+static void vfio_send_intx_eventfd(struct vfio_pci_device *vdev, void *unused)
+{
+	if (likely(is_intx(vdev) && !vdev->virq_disabled))
+		eventfd_signal(vdev->ctx[0].trigger, 1);
+}
+
+void vfio_pci_intx_mask(struct vfio_pci_device *vdev)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	unsigned long flags;
+
+	spin_lock_irqsave(&vdev->irqlock, flags);
+
+	/*
+	 * Masking can come from interrupt, ioctl, or config space
+	 * via INTx disable.  The latter means this can get called
+	 * even when not using intx delivery.  In this case, just
+	 * try to have the physical bit follow the virtual bit.
+	 */
+	if (unlikely(!is_intx(vdev))) {
+		if (vdev->pci_2_3)
+			pci_intx(pdev, 0);
+	} else if (!vdev->ctx[0].masked) {
+		/*
+		 * Can't use check_and_mask here because we always want to
+		 * mask, not just when something is pending.
+		 */
+		if (vdev->pci_2_3)
+			pci_intx(pdev, 0);
+		else
+			disable_irq_nosync(pdev->irq);
+
+		vdev->ctx[0].masked = true;
+	}
+
+	spin_unlock_irqrestore(&vdev->irqlock, flags);
+}
+
+/*
+ * If this is triggered by an eventfd, we can't call eventfd_signal
+ * or else we'll deadlock on the eventfd wait queue.  Return >0 when
+ * a signal is necessary, which can then be handled via a work queue
+ * or directly depending on the caller.
+ */
+int vfio_pci_intx_unmask_handler(struct vfio_pci_device *vdev, void *unused)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	unsigned long flags;
+	int ret = 0;
+
+	spin_lock_irqsave(&vdev->irqlock, flags);
+
+	/*
+	 * Unmasking comes from ioctl or config, so again, have the
+	 * physical bit follow the virtual even when not using INTx.
+	 */
+	if (unlikely(!is_intx(vdev))) {
+		if (vdev->pci_2_3)
+			pci_intx(pdev, 1);
+	} else if (vdev->ctx[0].masked && !vdev->virq_disabled) {
+		/*
+		 * A pending interrupt here would immediately trigger,
+		 * but we can avoid that overhead by just re-sending
+		 * the interrupt to the user.
+		 */
+		if (vdev->pci_2_3) {
+			if (!pci_check_and_unmask_intx(pdev))
+				ret = 1;
+		} else
+			enable_irq(pdev->irq);
+
+		vdev->ctx[0].masked = (ret > 0);
+	}
+
+	spin_unlock_irqrestore(&vdev->irqlock, flags);
+
+	return ret;
+}
+
+void vfio_pci_intx_unmask(struct vfio_pci_device *vdev)
+{
+	if (vfio_pci_intx_unmask_handler(vdev, NULL) > 0)
+		vfio_send_intx_eventfd(vdev, NULL);
+}
+
+static irqreturn_t vfio_intx_handler(int irq, void *dev_id)
+{
+	struct vfio_pci_device *vdev = dev_id;
+	unsigned long flags;
+	int ret = IRQ_NONE;
+
+	spin_lock_irqsave(&vdev->irqlock, flags);
+
+	if (!vdev->pci_2_3) {
+		disable_irq_nosync(vdev->pdev->irq);
+		vdev->ctx[0].masked = true;
+		ret = IRQ_HANDLED;
+	} else if (!vdev->ctx[0].masked &&  /* may be shared */
+		   pci_check_and_mask_intx(vdev->pdev)) {
+		vdev->ctx[0].masked = true;
+		ret = IRQ_HANDLED;
+	}
+
+	spin_unlock_irqrestore(&vdev->irqlock, flags);
+
+	if (ret == IRQ_HANDLED)
+		vfio_send_intx_eventfd(vdev, NULL);
+
+	return ret;
+}
+
+static int vfio_intx_enable(struct vfio_pci_device *vdev)
+{
+	if (!is_irq_none(vdev))
+		return -EINVAL;
+
+	if (!vdev->pdev->irq)
+		return -ENODEV;
+
+	vdev->ctx = kzalloc(sizeof(struct vfio_pci_irq_ctx), GFP_KERNEL);
+	if (!vdev->ctx)
+		return -ENOMEM;
+
+	vdev->num_ctx = 1;
+	vdev->irq_type = VFIO_PCI_INTX_IRQ_INDEX;
+
+	return 0;
+}
+
+static int vfio_intx_set_signal(struct vfio_pci_device *vdev, int fd)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	unsigned long irqflags = IRQF_SHARED;
+	struct eventfd_ctx *trigger;
+	unsigned long flags;
+	int ret;
+
+	if (vdev->ctx[0].trigger) {
+		free_irq(pdev->irq, vdev);
+		kfree(vdev->ctx[0].name);
+		eventfd_ctx_put(vdev->ctx[0].trigger);
+		vdev->ctx[0].trigger = NULL;
+	}
+
+	if (fd < 0) /* Disable only */
+		return 0;
+
+	vdev->ctx[0].name = kasprintf(GFP_KERNEL, "vfio-intx(%s)",
+				      pci_name(pdev));
+	if (!vdev->ctx[0].name)
+		return -ENOMEM;
+
+	trigger = eventfd_ctx_fdget(fd);
+	if (IS_ERR(trigger)) {
+		kfree(vdev->ctx[0].name);
+		return PTR_ERR(trigger);
+	}
+
+	if (!vdev->pci_2_3)
+		irqflags = 0;
+
+	ret = request_irq(pdev->irq, vfio_intx_handler,
+			  irqflags, vdev->ctx[0].name, vdev);
+	if (ret) {
+		kfree(vdev->ctx[0].name);
+		eventfd_ctx_put(trigger);
+		return ret;
+	}
+
+	vdev->ctx[0].trigger = trigger;
+
+	/*
+	 * INTx disable will stick across the new irq setup,
+	 * disable_irq won't.
+	 */
+	spin_lock_irqsave(&vdev->irqlock, flags);
+	if (!vdev->pci_2_3 && (vdev->ctx[0].masked || vdev->virq_disabled))
+		disable_irq_nosync(pdev->irq);
+	spin_unlock_irqrestore(&vdev->irqlock, flags);
+
+	return 0;
+}
+
+static void vfio_intx_disable(struct vfio_pci_device *vdev)
+{
+	vfio_intx_set_signal(vdev, -1);
+	virqfd_disable(vdev->ctx[0].unmask);
+	virqfd_disable(vdev->ctx[0].mask);
+	vdev->irq_type = VFIO_PCI_NUM_IRQS;
+	vdev->num_ctx = 0;
+	kfree(vdev->ctx);
+}
+
+/*
+ * MSI/MSI-X
+ */
+static irqreturn_t vfio_msihandler(int irq, void *arg)
+{
+	struct eventfd_ctx *trigger = arg;
+
+	eventfd_signal(trigger, 1);
+	return IRQ_HANDLED;
+}
+
+static int vfio_msi_enable(struct vfio_pci_device *vdev, int nvec, bool msix)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	int ret;
+
+	if (!is_irq_none(vdev))
+		return -EINVAL;
+
+	vdev->ctx = kzalloc(nvec * sizeof(struct vfio_pci_irq_ctx), GFP_KERNEL);
+	if (!vdev->ctx)
+		return -ENOMEM;
+
+	if (msix) {
+		int i;
+
+		vdev->msix = kzalloc(nvec * sizeof(struct msix_entry),
+				     GFP_KERNEL);
+		if (!vdev->msix) {
+			kfree(vdev->ctx);
+			return -ENOMEM;
+		}
+
+		for (i = 0; i < nvec; i++)
+			vdev->msix[i].entry = i;
+
+		ret = pci_enable_msix(pdev, vdev->msix, nvec);
+		if (ret) {
+			kfree(vdev->msix);
+			kfree(vdev->ctx);
+			return ret;
+		}
+	} else {
+		ret = pci_enable_msi_block(pdev, nvec);
+		if (ret) {
+			kfree(vdev->ctx);
+			return ret;
+		}
+	}
+
+	vdev->num_ctx = nvec;
+	vdev->irq_type = msix ? VFIO_PCI_MSIX_IRQ_INDEX :
+				VFIO_PCI_MSI_IRQ_INDEX;
+
+	if (!msix) {
+		/*
+		 * Compute the virtual hardware field for max msi vectors -
+		 * it is the log base 2 of the number of vectors.
+		 */
+		vdev->msi_qmax = fls(nvec * 2 - 1) - 1;
+	}
+
+	return 0;
+}
+
+static int vfio_msi_set_vector_signal(struct vfio_pci_device *vdev,
+				      int vector, int fd, bool msix)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	int irq = msix ? vdev->msix[vector].vector : pdev->irq + vector;
+	char *name = msix ? "vfio-msix" : "vfio-msi";
+	struct eventfd_ctx *trigger;
+	int ret;
+
+	if (vector >= vdev->num_ctx)
+		return -EINVAL;
+
+	if (vdev->ctx[vector].trigger) {
+		free_irq(irq, vdev->ctx[vector].trigger);
+		kfree(vdev->ctx[vector].name);
+		eventfd_ctx_put(vdev->ctx[vector].trigger);
+		vdev->ctx[vector].trigger = NULL;
+	}
+
+	if (fd < 0)
+		return 0;
+
+	vdev->ctx[vector].name = kasprintf(GFP_KERNEL, "%s[%d](%s)",
+					   name, vector, pci_name(pdev));
+	if (!vdev->ctx[vector].name)
+		return -ENOMEM;
+
+	trigger = eventfd_ctx_fdget(fd);
+	if (IS_ERR(trigger)) {
+		kfree(vdev->ctx[vector].name);
+		return PTR_ERR(trigger);
+	}
+
+	ret = request_irq(irq, vfio_msihandler, 0,
+			  vdev->ctx[vector].name, trigger);
+	if (ret) {
+		kfree(vdev->ctx[vector].name);
+		eventfd_ctx_put(trigger);
+		return ret;
+	}
+
+	vdev->ctx[vector].trigger = trigger;
+
+	return 0;
+}
+
+static int vfio_msi_set_block(struct vfio_pci_device *vdev, unsigned start,
+			      unsigned count, int32_t *fds, bool msix)
+{
+	int i, j, ret = 0;
+
+	if (start + count > vdev->num_ctx)
+		return -EINVAL;
+
+	for (i = 0, j = start; i < count && !ret; i++, j++) {
+		int fd = fds ? fds[i] : -1;
+		ret = vfio_msi_set_vector_signal(vdev, j, fd, msix);
+	}
+
+	if (ret) {
+		for (--j; j >= start; j--)
+			vfio_msi_set_vector_signal(vdev, j, -1, msix);
+	}
+
+	return ret;
+}
+
+static void vfio_msi_disable(struct vfio_pci_device *vdev, bool msix)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	int i;
+
+	vfio_msi_set_block(vdev, 0, vdev->num_ctx, NULL, msix);
+
+	for (i = 0; i < vdev->num_ctx; i++) {
+		virqfd_disable(vdev->ctx[i].unmask);
+		virqfd_disable(vdev->ctx[i].mask);
+	}
+
+	if (msix) {
+		pci_disable_msix(vdev->pdev);
+		kfree(vdev->msix);
+	} else
+		pci_disable_msi(pdev);
+
+	vdev->irq_type = VFIO_PCI_NUM_IRQS;
+	vdev->num_ctx = 0;
+	kfree(vdev->ctx);
+}
+
+/*
+ * IOCTL support
+ */
+static int vfio_pci_set_intx_unmask(struct vfio_pci_device *vdev,
+				    unsigned index, unsigned start,
+				    unsigned count, uint32_t flags, void *data)
+{
+	if (!is_intx(vdev) || start != 0 || count != 1)
+		return -EINVAL;
+
+	if (flags & VFIO_IRQ_SET_DATA_NONE) {
+		vfio_pci_intx_unmask(vdev);
+	} else if (flags & VFIO_IRQ_SET_DATA_BOOL) {
+		uint8_t unmask = *(uint8_t *)data;
+		if (unmask)
+			vfio_pci_intx_unmask(vdev);
+	} else if (flags & VFIO_IRQ_SET_DATA_EVENTFD) {
+		int32_t fd = *(int32_t *)data;
+		if (fd >= 0)
+			return virqfd_enable(vdev, vfio_pci_intx_unmask_handler,
+					     vfio_send_intx_eventfd, NULL,
+					     &vdev->ctx[0].unmask, fd);
+
+		virqfd_disable(vdev->ctx[0].unmask);
+	}
+
+	return 0;
+}
+
+static int vfio_pci_set_intx_mask(struct vfio_pci_device *vdev,
+				  unsigned index, unsigned start,
+				  unsigned count, uint32_t flags, void *data)
+{
+	if (!is_intx(vdev) || start != 0 || count != 1)
+		return -EINVAL;
+
+	if (flags & VFIO_IRQ_SET_DATA_NONE) {
+		vfio_pci_intx_mask(vdev);
+	} else if (flags & VFIO_IRQ_SET_DATA_BOOL) {
+		uint8_t mask = *(uint8_t *)data;
+		if (mask)
+			vfio_pci_intx_mask(vdev);
+	} else if (flags & VFIO_IRQ_SET_DATA_EVENTFD) {
+		return -ENOTTY; /* XXX implement me */
+	}
+
+	return 0;
+}
+
+static int vfio_pci_set_intx_trigger(struct vfio_pci_device *vdev,
+				     unsigned index, unsigned start,
+				     unsigned count, uint32_t flags, void *data)
+{
+	if (is_intx(vdev) && !count && (flags & VFIO_IRQ_SET_DATA_NONE)) {
+		vfio_intx_disable(vdev);
+		return 0;
+	}
+
+	if (!(is_intx(vdev) || is_irq_none(vdev)) || start != 0 || count != 1)
+		return -EINVAL;
+
+	if (flags & VFIO_IRQ_SET_DATA_EVENTFD) {
+		int32_t fd = *(int32_t *)data;
+		int ret;
+
+		if (is_intx(vdev))
+			return vfio_intx_set_signal(vdev, fd);
+
+		ret = vfio_intx_enable(vdev);
+		if (ret)
+			return ret;
+
+		ret = vfio_intx_set_signal(vdev, fd);
+		if (ret)
+			vfio_intx_disable(vdev);
+
+		return ret;
+	}
+
+	if (!is_intx(vdev))
+		return -EINVAL;
+
+	if (flags & VFIO_IRQ_SET_DATA_NONE) {
+		vfio_send_intx_eventfd(vdev, NULL);
+	} else if (flags & VFIO_IRQ_SET_DATA_BOOL) {
+		uint8_t trigger = *(uint8_t *)data;
+		if (trigger)
+			vfio_send_intx_eventfd(vdev, NULL);
+	}
+	return 0;
+}
+
+static int vfio_pci_set_msi_trigger(struct vfio_pci_device *vdev,
+				    unsigned index, unsigned start,
+				    unsigned count, uint32_t flags, void *data)
+{
+	int i;
+	bool msix = (index == VFIO_PCI_MSIX_IRQ_INDEX) ? true : false;
+
+	if (irq_is(vdev, index) && !count && (flags & VFIO_IRQ_SET_DATA_NONE)) {
+		vfio_msi_disable(vdev, msix);
+		return 0;
+	}
+
+	if (!(irq_is(vdev, index) || is_irq_none(vdev)))
+		return -EINVAL;
+
+	if (flags & VFIO_IRQ_SET_DATA_EVENTFD) {
+		int32_t *fds = data;
+		int ret;
+
+		if (vdev->irq_type == index)
+			return vfio_msi_set_block(vdev, start, count,
+						  fds, msix);
+
+		ret = vfio_msi_enable(vdev, start + count, msix);
+		if (ret)
+			return ret;
+
+		ret = vfio_msi_set_block(vdev, start, count, fds, msix);
+		if (ret)
+			vfio_msi_disable(vdev, msix);
+
+		return ret;
+	}
+
+	if (!irq_is(vdev, index) || start + count > vdev->num_ctx)
+		return -EINVAL;
+
+	for (i = start; i < start + count; i++) {
+		if (!vdev->ctx[i].trigger)
+			continue;
+		if (flags & VFIO_IRQ_SET_DATA_NONE) {
+			eventfd_signal(vdev->ctx[i].trigger, 1);
+		} else if (flags & VFIO_IRQ_SET_DATA_BOOL) {
+			uint8_t *bools = data;
+			if (bools[i - start])
+				eventfd_signal(vdev->ctx[i].trigger, 1);
+		}
+	}
+	return 0;
+}
+
+int vfio_pci_set_irqs_ioctl(struct vfio_pci_device *vdev, uint32_t flags,
+			    unsigned index, unsigned start, unsigned count,
+			    void *data)
+{
+	int (*func)(struct vfio_pci_device *vdev, unsigned index,
+		    unsigned start, unsigned count, uint32_t flags,
+		    void *data) = NULL;
+
+	switch (index) {
+	case VFIO_PCI_INTX_IRQ_INDEX:
+		switch (flags & VFIO_IRQ_SET_ACTION_TYPE_MASK) {
+		case VFIO_IRQ_SET_ACTION_MASK:
+			func = vfio_pci_set_intx_mask;
+			break;
+		case VFIO_IRQ_SET_ACTION_UNMASK:
+			func = vfio_pci_set_intx_unmask;
+			break;
+		case VFIO_IRQ_SET_ACTION_TRIGGER:
+			func = vfio_pci_set_intx_trigger;
+			break;
+		}
+		break;
+	case VFIO_PCI_MSI_IRQ_INDEX:
+	case VFIO_PCI_MSIX_IRQ_INDEX:
+		switch (flags & VFIO_IRQ_SET_ACTION_TYPE_MASK) {
+		case VFIO_IRQ_SET_ACTION_MASK:
+		case VFIO_IRQ_SET_ACTION_UNMASK:
+			/* XXX Need masking support exported */
+			break;
+		case VFIO_IRQ_SET_ACTION_TRIGGER:
+			func = vfio_pci_set_msi_trigger;
+			break;
+		}
+		break;
+	}
+
+	if (!func)
+		return -ENOTTY;
+
+	return func(vdev, index, start, count, flags, data);
+}
+/*
+ * Copyright (C) 2012 Red Hat, Inc.  All rights reserved.
+ *     Author: Alex Williamson <alex.williamson@redhat.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ *
+ * Derived from original vfio:
+ * Copyright 2010 Cisco Systems, Inc.  All rights reserved.
+ * Author: Tom Lyon, pugs@cisco.com
+ */
+
+#include <linux/mutex.h>
+#include <linux/pci.h>
+
+#ifndef VFIO_PCI_PRIVATE_H
+#define VFIO_PCI_PRIVATE_H
+
+#define VFIO_PCI_OFFSET_SHIFT   40
+
+#define VFIO_PCI_OFFSET_TO_INDEX(off)	(off >> VFIO_PCI_OFFSET_SHIFT)
+#define VFIO_PCI_INDEX_TO_OFFSET(index)	((u64)(index) << VFIO_PCI_OFFSET_SHIFT)
+#define VFIO_PCI_OFFSET_MASK	(((u64)(1) << VFIO_PCI_OFFSET_SHIFT) - 1)
+
+struct vfio_pci_irq_ctx {
+	struct eventfd_ctx	*trigger;
+	struct virqfd		*unmask;
+	struct virqfd		*mask;
+	char			*name;
+	bool			masked;
+};
+
+struct vfio_pci_device {
+	struct pci_dev		*pdev;
+	void __iomem		*barmap[PCI_STD_RESOURCE_END + 1];
+	u8			*pci_config_map;
+	u8			*vconfig;
+	struct perm_bits	*msi_perm;
+	spinlock_t		irqlock;
+	struct mutex		igate;
+	struct msix_entry	*msix;
+	struct vfio_pci_irq_ctx	*ctx;
+	int			num_ctx;
+	int			irq_type;
+	u8			msi_qmax;
+	u8			msix_bar;
+	u16			msix_size;
+	u32			msix_offset;
+	u32			rbar[7];
+	bool			pci_2_3;
+	bool			virq_disabled;
+	bool			reset_works;
+	bool			extended_caps;
+	bool			bardirty;
+	struct pci_saved_state	*pci_saved_state;
+	atomic_t		refcnt;
+};
+
+#define is_intx(vdev) (vdev->irq_type == VFIO_PCI_INTX_IRQ_INDEX)
+#define is_msi(vdev) (vdev->irq_type == VFIO_PCI_MSI_IRQ_INDEX)
+#define is_msix(vdev) (vdev->irq_type == VFIO_PCI_MSIX_IRQ_INDEX)
+#define is_irq_none(vdev) (!(is_intx(vdev) || is_msi(vdev) || is_msix(vdev)))
+#define irq_is(vdev, type) (vdev->irq_type == type)
+
+extern void vfio_pci_intx_mask(struct vfio_pci_device *vdev);
+extern void vfio_pci_intx_unmask(struct vfio_pci_device *vdev);
+
+extern int vfio_pci_set_irqs_ioctl(struct vfio_pci_device *vdev,
+				   uint32_t flags, unsigned index,
+				   unsigned start, unsigned count, void *data);
+
+extern ssize_t vfio_pci_config_readwrite(struct vfio_pci_device *vdev,
+					 char __user *buf, size_t count,
+					 loff_t *ppos, bool iswrite);
+extern ssize_t vfio_pci_mem_readwrite(struct vfio_pci_device *vdev,
+				      char __user *buf, size_t count,
+				      loff_t *ppos, bool iswrite);
+extern ssize_t vfio_pci_io_readwrite(struct vfio_pci_device *vdev,
+				     char __user *buf, size_t count,
+				     loff_t *ppos, bool iswrite);
+
+extern int vfio_pci_init_perm_bits(void);
+extern void vfio_pci_uninit_perm_bits(void);
+
+extern int vfio_pci_virqfd_init(void);
+extern void vfio_pci_virqfd_exit(void);
+
+extern int vfio_config_init(struct vfio_pci_device *vdev);
+extern void vfio_config_free(struct vfio_pci_device *vdev);
+#endif /* VFIO_PCI_PRIVATE_H */
+/*
+ * VFIO PCI I/O Port & MMIO access
+ *
+ * Copyright (C) 2012 Red Hat, Inc.  All rights reserved.
+ *     Author: Alex Williamson <alex.williamson@redhat.com>
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License version 2 as
+ * published by the Free Software Foundation.
+ *
+ * Derived from original vfio:
+ * Copyright 2010 Cisco Systems, Inc.  All rights reserved.
+ * Author: Tom Lyon, pugs@cisco.com
+ */
+
+#include <linux/fs.h>
+#include <linux/pci.h>
+#include <linux/uaccess.h>
+#include <linux/io.h>
+
+#include "vfio_pci_private.h"
+
+/* I/O Port BAR access */
+ssize_t vfio_pci_io_readwrite(struct vfio_pci_device *vdev, char __user *buf,
+			      size_t count, loff_t *ppos, bool iswrite)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	loff_t pos = *ppos & VFIO_PCI_OFFSET_MASK;
+	int bar = VFIO_PCI_OFFSET_TO_INDEX(*ppos);
+	void __iomem *io;
+	size_t done = 0;
+
+	if (!pci_resource_start(pdev, bar))
+		return -EINVAL;
+
+	if (pos + count > pci_resource_len(pdev, bar))
+		return -EINVAL;
+
+	if (!vdev->barmap[bar]) {
+		int ret;
+
+		ret = pci_request_selected_regions(pdev, 1 << bar, "vfio");
+		if (ret)
+			return ret;
+
+		vdev->barmap[bar] = pci_iomap(pdev, bar, 0);
+
+		if (!vdev->barmap[bar]) {
+			pci_release_selected_regions(pdev, 1 << bar);
+			return -EINVAL;
+		}
+	}
+
+	io = vdev->barmap[bar];
+
+	while (count) {
+		int filled;
+
+		if (count >= 3 && !(pos % 4)) {
+			__le32 val;
+
+			if (iswrite) {
+				if (copy_from_user(&val, buf, 4))
+					return -EFAULT;
+
+				iowrite32(le32_to_cpu(val), io + pos);
+			} else {
+				val = cpu_to_le32(ioread32(io + pos));
+
+				if (copy_to_user(buf, &val, 4))
+					return -EFAULT;
+			}
+
+			filled = 4;
+
+		} else if ((pos % 2) == 0 && count >= 2) {
+			__le16 val;
+
+			if (iswrite) {
+				if (copy_from_user(&val, buf, 2))
+					return -EFAULT;
+
+				iowrite16(le16_to_cpu(val), io + pos);
+			} else {
+				val = cpu_to_le16(ioread16(io + pos));
+
+				if (copy_to_user(buf, &val, 2))
+					return -EFAULT;
+			}
+
+			filled = 2;
+		} else {
+			u8 val;
+
+			if (iswrite) {
+				if (copy_from_user(&val, buf, 1))
+					return -EFAULT;
+
+				iowrite8(val, io + pos);
+			} else {
+				val = ioread8(io + pos);
+
+				if (copy_to_user(buf, &val, 1))
+					return -EFAULT;
+			}
+
+			filled = 1;
+		}
+
+		count -= filled;
+		done += filled;
+		buf += filled;
+		pos += filled;
+	}
+
+	*ppos += done;
+
+	return done;
+}
+
+/*
+ * MMIO BAR access
+ * We handle two excluded ranges here as well, if the user tries to read
+ * the ROM beyond what PCI tells us is available or the MSI-X table region,
+ * we return 0xFF and writes are dropped.
+ */
+ssize_t vfio_pci_mem_readwrite(struct vfio_pci_device *vdev, char __user *buf,
+			       size_t count, loff_t *ppos, bool iswrite)
+{
+	struct pci_dev *pdev = vdev->pdev;
+	loff_t pos = *ppos & VFIO_PCI_OFFSET_MASK;
+	int bar = VFIO_PCI_OFFSET_TO_INDEX(*ppos);
+	void __iomem *io;
+	resource_size_t end;
+	size_t done = 0;
+	size_t x_start = 0, x_end = 0; /* excluded range */
+
+	if (!pci_resource_start(pdev, bar))
+		return -EINVAL;
+
+	end = pci_resource_len(pdev, bar);
+
+	if (pos > end)
+		return -EINVAL;
+
+	if (pos == end)
+		return 0;
+
+	if (pos + count > end)
+		count = end - pos;
+
+	if (bar == PCI_ROM_RESOURCE) {
+		io = pci_map_rom(pdev, &x_start);
+		x_end = end;
+	} else {
+		if (!vdev->barmap[bar]) {
+			int ret;
+
+			ret = pci_request_selected_regions(pdev, 1 << bar,
+							   "vfio");
+			if (ret)
+				return ret;
+
+			vdev->barmap[bar] = pci_iomap(pdev, bar, 0);
+
+			if (!vdev->barmap[bar]) {
+				pci_release_selected_regions(pdev, 1 << bar);
+				return -EINVAL;
+			}
+		}
+
+		io = vdev->barmap[bar];
+
+		if (bar == vdev->msix_bar) {
+			x_start = vdev->msix_offset;
+			x_end = vdev->msix_offset + vdev->msix_size;
+		}
+	}
+
+	if (!io)
+		return -EINVAL;
+
+	while (count) {
+		size_t fillable, filled;
+
+		if (pos < x_start)
+			fillable = x_start - pos;
+		else if (pos >= x_end)
+			fillable = end - pos;
+		else
+			fillable = 0;
+
+		if (fillable >= 4 && !(pos % 4) && (count >= 4)) {
+			__le32 val;
+
+			if (iswrite) {
+				if (copy_from_user(&val, buf, 4))
+					goto out;
+
+				iowrite32(le32_to_cpu(val), io + pos);
+			} else {
+				val = cpu_to_le32(ioread32(io + pos));
+
+				if (copy_to_user(buf, &val, 4))
+					goto out;
+			}
+
+			filled = 4;
+		} else if (fillable >= 2 && !(pos % 2) && (count >= 2)) {
+			__le16 val;
+
+			if (iswrite) {
+				if (copy_from_user(&val, buf, 2))
+					goto out;
+
+				iowrite16(le16_to_cpu(val), io + pos);
+			} else {
+				val = cpu_to_le16(ioread16(io + pos));
+
+				if (copy_to_user(buf, &val, 2))
+					goto out;
+			}
+
+			filled = 2;
+		} else if (fillable) {
+			u8 val;
+
+			if (iswrite) {
+				if (copy_from_user(&val, buf, 1))
+					goto out;
+
+				iowrite8(val, io + pos);
+			} else {
+				val = ioread8(io + pos);
+
+				if (copy_to_user(buf, &val, 1))
+					goto out;
+			}
+
+			filled = 1;
+		} else {
+			/* Drop writes, fill reads with FF */
+			if (!iswrite) {
+				char val = 0xFF;
+				size_t i;
+
+				for (i = 0; i < x_end - pos; i++) {
+					if (put_user(val, buf + i))
+						goto out;
+				}
+			}
+
+			filled = x_end - pos;
+		}
+
+		count -= filled;
+		done += filled;
+		buf += filled;
+		pos += filled;
+	}
+
+	*ppos += done;
+
+out:
+	if (bar == PCI_ROM_RESOURCE)
+		pci_unmap_rom(pdev, io);
+
+	return count ? -EFAULT : done;
+}
@@ -223,6 +223,7 @@
 	__u32	argsz;
 	__u32	flags;
 #define VFIO_DEVICE_FLAGS_RESET	(1 << 0)	/* Device supports reset */
+#define VFIO_DEVICE_FLAGS_PCI	(1 << 1)	/* vfio-pci device */
 	__u32	num_regions;	/* Max region index + 1 */
 	__u32	num_irqs;	/* Max IRQ index + 1 */
 };
@@ -363,6 +364,31 @@
  * Reset a device.
  */
 #define VFIO_DEVICE_RESET		_IO(VFIO_TYPE, VFIO_BASE + 11)
+
+/*
+ * The VFIO-PCI bus driver makes use of the following fixed region and
+ * IRQ index mapping.  Unimplemented regions return a size of zero.
+ * Unimplemented IRQ types return a count of zero.
+ */
+
+enum {
+	VFIO_PCI_BAR0_REGION_INDEX,
+	VFIO_PCI_BAR1_REGION_INDEX,
+	VFIO_PCI_BAR2_REGION_INDEX,
+	VFIO_PCI_BAR3_REGION_INDEX,
+	VFIO_PCI_BAR4_REGION_INDEX,
+	VFIO_PCI_BAR5_REGION_INDEX,
+	VFIO_PCI_ROM_REGION_INDEX,
+	VFIO_PCI_CONFIG_REGION_INDEX,
+	VFIO_PCI_NUM_REGIONS
+};
+
+enum {
+	VFIO_PCI_INTX_IRQ_INDEX,
+	VFIO_PCI_MSI_IRQ_INDEX,
+	VFIO_PCI_MSIX_IRQ_INDEX,
+	VFIO_PCI_NUM_IRQS
+};
  
 /* -------- API for Type1 VFIO IOMMU -------- */
...	...	@@ -12,4 +12,6 @@
12	12	See Documentation/vfio.txt for more details.
13	13
14	14	If you don't know what to do here, say N.
	15	+
	16	+source "drivers/vfio/pci/Kconfig"
	1	+config VFIO_PCI
	2	+ tristate "VFIO support for PCI devices"
	3	+ depends on VFIO && PCI && EVENTFD
	4	+ help
	5	+ Support for the PCI VFIO bus driver. This is required to make
	6	+ use of PCI drivers using the VFIO framework.
	7	+
	8	+ If you don't know what to do here, say N.
	1	+
	2	+vfio-pci-y := vfio_pci.o vfio_pci_intrs.o vfio_pci_rdwr.o vfio_pci_config.o
	3	+
	4	+obj-$(CONFIG_VFIO_PCI) += vfio-pci.o
	1	+/*
	2	+ * Copyright (C) 2012 Red Hat, Inc. All rights reserved.
	3	+ * Author: Alex Williamson <alex.williamson@redhat.com>
	4	+ *
	5	+ * This program is free software; you can redistribute it and/or modify
	6	+ * it under the terms of the GNU General Public License version 2 as
	7	+ * published by the Free Software Foundation.
	8	+ *
	9	+ * Derived from original vfio:
	10	+ * Copyright 2010 Cisco Systems, Inc. All rights reserved.
	11	+ * Author: Tom Lyon, pugs@cisco.com
	12	+ */
	13	+
	14	+#include <linux/device.h>
	15	+#include <linux/eventfd.h>
	16	+#include <linux/interrupt.h>
	17	+#include <linux/iommu.h>
	18	+#include <linux/module.h>
	19	+#include <linux/mutex.h>
	20	+#include <linux/notifier.h>
	21	+#include <linux/pci.h>
	22	+#include <linux/pm_runtime.h>
	23	+#include <linux/slab.h>
	24	+#include <linux/types.h>
	25	+#include <linux/uaccess.h>
	26	+#include <linux/vfio.h>
	27	+
	28	+#include "vfio_pci_private.h"
	29	+
	30	+#define DRIVER_VERSION "0.2"
	31	+#define DRIVER_AUTHOR "Alex Williamson <alex.williamson@redhat.com>"
	32	+#define DRIVER_DESC "VFIO PCI - User Level meta-driver"
	33	+
	34	+static bool nointxmask;
	35	+module_param_named(nointxmask, nointxmask, bool, S_IRUGO \| S_IWUSR);
	36	+MODULE_PARM_DESC(nointxmask,
	37	+ "Disable support for PCI 2.3 style INTx masking. If this resolves problems for specific devices, report lspci -vvvxxx to linux-pci@vger.kernel.org so the device can be fixed automatically via the broken_intx_masking flag.");
	38	+
	39	+static int vfio_pci_enable(struct vfio_pci_device *vdev)
	40	+{
	41	+ struct pci_dev *pdev = vdev->pdev;
	42	+ int ret;
	43	+ u16 cmd;
	44	+ u8 msix_pos;
	45	+
	46	+ vdev->reset_works = (pci_reset_function(pdev) == 0);
	47	+ pci_save_state(pdev);
	48	+ vdev->pci_saved_state = pci_store_saved_state(pdev);
	49	+ if (!vdev->pci_saved_state)
	50	+ pr_debug("%s: Couldn't store %s saved state\n",
	51	+ __func__, dev_name(&pdev->dev));
	52	+
	53	+ ret = vfio_config_init(vdev);
	54	+ if (ret)
	55	+ goto out;
	56	+
	57	+ if (likely(!nointxmask))
	58	+ vdev->pci_2_3 = pci_intx_mask_supported(pdev);
	59	+
	60	+ pci_read_config_word(pdev, PCI_COMMAND, &cmd);
	61	+ if (vdev->pci_2_3 && (cmd & PCI_COMMAND_INTX_DISABLE)) {
	62	+ cmd &= ~PCI_COMMAND_INTX_DISABLE;
	63	+ pci_write_config_word(pdev, PCI_COMMAND, cmd);
	64	+ }
	65	+
	66	+ msix_pos = pci_find_capability(pdev, PCI_CAP_ID_MSIX);
	67	+ if (msix_pos) {
	68	+ u16 flags;
	69	+ u32 table;
	70	+
	71	+ pci_read_config_word(pdev, msix_pos + PCI_MSIX_FLAGS, &flags);
	72	+ pci_read_config_dword(pdev, msix_pos + PCI_MSIX_TABLE, &table);
	73	+
	74	+ vdev->msix_bar = table & PCI_MSIX_FLAGS_BIRMASK;
	75	+ vdev->msix_offset = table & ~PCI_MSIX_FLAGS_BIRMASK;
	76	+ vdev->msix_size = ((flags & PCI_MSIX_FLAGS_QSIZE) + 1) * 16;
	77	+ } else
	78	+ vdev->msix_bar = 0xFF;
	79	+
	80	+ ret = pci_enable_device(pdev);
	81	+ if (ret)
	82	+ goto out;
	83	+
	84	+ return ret;
	85	+
	86	+out:
	87	+ kfree(vdev->pci_saved_state);
	88	+ vdev->pci_saved_state = NULL;
	89	+ vfio_config_free(vdev);
	90	+ return ret;
	91	+}
	92	+
	93	+static void vfio_pci_disable(struct vfio_pci_device *vdev)
	94	+{
	95	+ int bar;
	96	+
	97	+ pci_disable_device(vdev->pdev);
	98	+
	99	+ vfio_pci_set_irqs_ioctl(vdev, VFIO_IRQ_SET_DATA_NONE \|
	100	+ VFIO_IRQ_SET_ACTION_TRIGGER,
	101	+ vdev->irq_type, 0, 0, NULL);
	102	+
	103	+ vdev->virq_disabled = false;
	104	+
	105	+ vfio_config_free(vdev);
	106	+
	107	+ pci_reset_function(vdev->pdev);
	108	+
	109	+ if (pci_load_and_free_saved_state(vdev->pdev,
	110	+ &vdev->pci_saved_state) == 0)
	111	+ pci_restore_state(vdev->pdev);
	112	+ else
	113	+ pr_info("%s: Couldn't reload %s saved state\n",
	114	+ __func__, dev_name(&vdev->pdev->dev));
	115	+
	116	+ for (bar = PCI_STD_RESOURCES; bar <= PCI_STD_RESOURCE_END; bar++) {
	117	+ if (!vdev->barmap[bar])
	118	+ continue;
	119	+ pci_iounmap(vdev->pdev, vdev->barmap[bar]);
	120	+ pci_release_selected_regions(vdev->pdev, 1 << bar);
	121	+ vdev->barmap[bar] = NULL;
	122	+ }
	123	+}
	124	+
	125	+static void vfio_pci_release(void *device_data)
	126	+{
	127	+ struct vfio_pci_device *vdev = device_data;
	128	+
	129	+ if (atomic_dec_and_test(&vdev->refcnt))
	130	+ vfio_pci_disable(vdev);
	131	+
	132	+ module_put(THIS_MODULE);
	133	+}
	134	+
	135	+static int vfio_pci_open(void *device_data)
	136	+{
	137	+ struct vfio_pci_device *vdev = device_data;
	138	+
	139	+ if (!try_module_get(THIS_MODULE))
	140	+ return -ENODEV;
	141	+
	142	+ if (atomic_inc_return(&vdev->refcnt) == 1) {
	143	+ int ret = vfio_pci_enable(vdev);
	144	+ if (ret) {
	145	+ module_put(THIS_MODULE);
	146	+ return ret;
	147	+ }
	148	+ }
	149	+
	150	+ return 0;
	151	+}
	152	+
	153	+static int vfio_pci_get_irq_count(struct vfio_pci_device *vdev, int irq_type)
	154	+{
	155	+ if (irq_type == VFIO_PCI_INTX_IRQ_INDEX) {
	156	+ u8 pin;
	157	+ pci_read_config_byte(vdev->pdev, PCI_INTERRUPT_PIN, &pin);
	158	+ if (pin)
	159	+ return 1;
	160	+
	161	+ } else if (irq_type == VFIO_PCI_MSI_IRQ_INDEX) {
	162	+ u8 pos;
	163	+ u16 flags;
	164	+
	165	+ pos = pci_find_capability(vdev->pdev, PCI_CAP_ID_MSI);
	166	+ if (pos) {
	167	+ pci_read_config_word(vdev->pdev,
	168	+ pos + PCI_MSI_FLAGS, &flags);
	169	+
	170	+ return 1 << (flags & PCI_MSI_FLAGS_QMASK);
	171	+ }
	172	+ } else if (irq_type == VFIO_PCI_MSIX_IRQ_INDEX) {
	173	+ u8 pos;
	174	+ u16 flags;
	175	+
	176	+ pos = pci_find_capability(vdev->pdev, PCI_CAP_ID_MSIX);
	177	+ if (pos) {
	178	+ pci_read_config_word(vdev->pdev,
	179	+ pos + PCI_MSIX_FLAGS, &flags);
	180	+
	181	+ return (flags & PCI_MSIX_FLAGS_QSIZE) + 1;
	182	+ }
	183	+ }
	184	+
	185	+ return 0;
	186	+}
	187	+
	188	+static long vfio_pci_ioctl(void *device_data,
	189	+ unsigned int cmd, unsigned long arg)
	190	+{
	191	+ struct vfio_pci_device *vdev = device_data;
	192	+ unsigned long minsz;
	193	+
	194	+ if (cmd == VFIO_DEVICE_GET_INFO) {
	195	+ struct vfio_device_info info;
	196	+
	197	+ minsz = offsetofend(struct vfio_device_info, num_irqs);
	198	+
	199	+ if (copy_from_user(&info, (void __user *)arg, minsz))
	200	+ return -EFAULT;
	201	+
	202	+ if (info.argsz < minsz)
	203	+ return -EINVAL;
	204	+
	205	+ info.flags = VFIO_DEVICE_FLAGS_PCI;
	206	+
	207	+ if (vdev->reset_works)
	208	+ info.flags \|= VFIO_DEVICE_FLAGS_RESET;
	209	+
	210	+ info.num_regions = VFIO_PCI_NUM_REGIONS;
	211	+ info.num_irqs = VFIO_PCI_NUM_IRQS;
	212	+
	213	+ return copy_to_user((void __user *)arg, &info, minsz);
	214	+
	215	+ } else if (cmd == VFIO_DEVICE_GET_REGION_INFO) {
	216	+ struct pci_dev *pdev = vdev->pdev;
	217	+ struct vfio_region_info info;
	218	+
	219	+ minsz = offsetofend(struct vfio_region_info, offset);
	220	+
	221	+ if (copy_from_user(&info, (void __user *)arg, minsz))
	222	+ return -EFAULT;
	223	+
	224	+ if (info.argsz < minsz)
	225	+ return -EINVAL;
	226	+
	227	+ switch (info.index) {
	228	+ case VFIO_PCI_CONFIG_REGION_INDEX:
	229	+ info.offset = VFIO_PCI_INDEX_TO_OFFSET(info.index);
	230	+ info.size = pdev->cfg_size;
	231	+ info.flags = VFIO_REGION_INFO_FLAG_READ \|
	232	+ VFIO_REGION_INFO_FLAG_WRITE;
	233	+ break;
	234	+ case VFIO_PCI_BAR0_REGION_INDEX ... VFIO_PCI_BAR5_REGION_INDEX:
	235	+ info.offset = VFIO_PCI_INDEX_TO_OFFSET(info.index);
	236	+ info.size = pci_resource_len(pdev, info.index);
	237	+ if (!info.size) {
	238	+ info.flags = 0;
	239	+ break;
	240	+ }
	241	+
	242	+ info.flags = VFIO_REGION_INFO_FLAG_READ \|
	243	+ VFIO_REGION_INFO_FLAG_WRITE;
	244	+ if (pci_resource_flags(pdev, info.index) &
	245	+ IORESOURCE_MEM && info.size >= PAGE_SIZE)
	246	+ info.flags \|= VFIO_REGION_INFO_FLAG_MMAP;
	247	+ break;
	248	+ case VFIO_PCI_ROM_REGION_INDEX:
	249	+ {
	250	+ void __iomem *io;
	251	+ size_t size;
	252	+
	253	+ info.offset = VFIO_PCI_INDEX_TO_OFFSET(info.index);
	254	+ info.flags = 0;
	255	+
	256	+ /* Report the BAR size, not the ROM size */
	257	+ info.size = pci_resource_len(pdev, info.index);
	258	+ if (!info.size)
	259	+ break;
	260	+
	261	+ /* Is it really there? */
	262	+ io = pci_map_rom(pdev, &size);
	263	+ if (!io \|\| !size) {
	264	+ info.size = 0;
	265	+ break;
	266	+ }
	267	+ pci_unmap_rom(pdev, io);
	268	+
	269	+ info.flags = VFIO_REGION_INFO_FLAG_READ;
	270	+ break;
	271	+ }
	272	+ default:
	273	+ return -EINVAL;
	274	+ }
	275	+
	276	+ return copy_to_user((void __user *)arg, &info, minsz);
	277	+
	278	+ } else if (cmd == VFIO_DEVICE_GET_IRQ_INFO) {
	279	+ struct vfio_irq_info info;
	280	+
	281	+ minsz = offsetofend(struct vfio_irq_info, count);
	282	+
	283	+ if (copy_from_user(&info, (void __user *)arg, minsz))
	284	+ return -EFAULT;
	285	+
	286	+ if (info.argsz < minsz \|\| info.index >= VFIO_PCI_NUM_IRQS)
	287	+ return -EINVAL;
	288	+
	289	+ info.flags = VFIO_IRQ_INFO_EVENTFD;
	290	+
	291	+ info.count = vfio_pci_get_irq_count(vdev, info.index);
	292	+
	293	+ if (info.index == VFIO_PCI_INTX_IRQ_INDEX)
	294	+ info.flags \|= (VFIO_IRQ_INFO_MASKABLE \|
	295	+ VFIO_IRQ_INFO_AUTOMASKED);
	296	+ else
	297	+ info.flags \|= VFIO_IRQ_INFO_NORESIZE;
	298	+
	299	+ return copy_to_user((void __user *)arg, &info, minsz);
	300	+
	301	+ } else if (cmd == VFIO_DEVICE_SET_IRQS) {
	302	+ struct vfio_irq_set hdr;
	303	+ u8 *data = NULL;
	304	+ int ret = 0;
	305	+
	306	+ minsz = offsetofend(struct vfio_irq_set, count);
	307	+
	308	+ if (copy_from_user(&hdr, (void __user *)arg, minsz))
	309	+ return -EFAULT;
	310	+
	311	+ if (hdr.argsz < minsz \|\| hdr.index >= VFIO_PCI_NUM_IRQS \|\|
	312	+ hdr.flags & ~(VFIO_IRQ_SET_DATA_TYPE_MASK \|
	313	+ VFIO_IRQ_SET_ACTION_TYPE_MASK))
	314	+ return -EINVAL;
	315	+
	316	+ if (!(hdr.flags & VFIO_IRQ_SET_DATA_NONE)) {
	317	+ size_t size;
	318	+
	319	+ if (hdr.flags & VFIO_IRQ_SET_DATA_BOOL)
	320	+ size = sizeof(uint8_t);
	321	+ else if (hdr.flags & VFIO_IRQ_SET_DATA_EVENTFD)
	322	+ size = sizeof(int32_t);
	323	+ else
	324	+ return -EINVAL;
	325	+
	326	+ if (hdr.argsz - minsz < hdr.count * size \|\|
	327	+ hdr.count > vfio_pci_get_irq_count(vdev, hdr.index))
	328	+ return -EINVAL;
	329	+
	330	+ data = kmalloc(hdr.count * size, GFP_KERNEL);
	331	+ if (!data)
	332	+ return -ENOMEM;
	333	+
	334	+ if (copy_from_user(data, (void __user *)(arg + minsz),
	335	+ hdr.count * size)) {
	336	+ kfree(data);
	337	+ return -EFAULT;
	338	+ }
	339	+ }
	340	+
	341	+ mutex_lock(&vdev->igate);
	342	+
	343	+ ret = vfio_pci_set_irqs_ioctl(vdev, hdr.flags, hdr.index,
	344	+ hdr.start, hdr.count, data);
	345	+
	346	+ mutex_unlock(&vdev->igate);
	347	+ kfree(data);
	348	+
	349	+ return ret;
	350	+
	351	+ } else if (cmd == VFIO_DEVICE_RESET)
	352	+ return vdev->reset_works ?
	353	+ pci_reset_function(vdev->pdev) : -EINVAL;
	354	+
	355	+ return -ENOTTY;
	356	+}
	357	+
	358	+static ssize_t vfio_pci_read(void device_data, char __user buf,
	359	+ size_t count, loff_t *ppos)
	360	+{
	361	+ unsigned int index = VFIO_PCI_OFFSET_TO_INDEX(*ppos);
	362	+ struct vfio_pci_device *vdev = device_data;
	363	+ struct pci_dev *pdev = vdev->pdev;
	364	+
	365	+ if (index >= VFIO_PCI_NUM_REGIONS)
	366	+ return -EINVAL;
	367	+
	368	+ if (index == VFIO_PCI_CONFIG_REGION_INDEX)
	369	+ return vfio_pci_config_readwrite(vdev, buf, count, ppos, false);
	370	+ else if (index == VFIO_PCI_ROM_REGION_INDEX)
	371	+ return vfio_pci_mem_readwrite(vdev, buf, count, ppos, false);
	372	+ else if (pci_resource_flags(pdev, index) & IORESOURCE_IO)
	373	+ return vfio_pci_io_readwrite(vdev, buf, count, ppos, false);
	374	+ else if (pci_resource_flags(pdev, index) & IORESOURCE_MEM)
	375	+ return vfio_pci_mem_readwrite(vdev, buf, count, ppos, false);
	376	+
	377	+ return -EINVAL;
	378	+}
	379	+
	380	+static ssize_t vfio_pci_write(void device_data, const char __user buf,
	381	+ size_t count, loff_t *ppos)
	382	+{
	383	+ unsigned int index = VFIO_PCI_OFFSET_TO_INDEX(*ppos);
	384	+ struct vfio_pci_device *vdev = device_data;
	385	+ struct pci_dev *pdev = vdev->pdev;
	386	+
	387	+ if (index >= VFIO_PCI_NUM_REGIONS)
	388	+ return -EINVAL;
	389	+
	390	+ if (index == VFIO_PCI_CONFIG_REGION_INDEX)
	391	+ return vfio_pci_config_readwrite(vdev, (char __user *)buf,
	392	+ count, ppos, true);
	393	+ else if (index == VFIO_PCI_ROM_REGION_INDEX)
	394	+ return -EINVAL;
	395	+ else if (pci_resource_flags(pdev, index) & IORESOURCE_IO)
	396	+ return vfio_pci_io_readwrite(vdev, (char __user *)buf,
	397	+ count, ppos, true);
	398	+ else if (pci_resource_flags(pdev, index) & IORESOURCE_MEM) {
	399	+ return vfio_pci_mem_readwrite(vdev, (char __user *)buf,
	400	+ count, ppos, true);
	401	+ }
	402	+
	403	+ return -EINVAL;
	404	+}
	405	+
	406	+static int vfio_pci_mmap(void device_data, struct vm_area_struct vma)
	407	+{
	408	+ struct vfio_pci_device *vdev = device_data;
	409	+ struct pci_dev *pdev = vdev->pdev;
	410	+ unsigned int index;
	411	+ u64 phys_len, req_len, pgoff, req_start, phys;
	412	+ int ret;
	413	+
	414	+ index = vma->vm_pgoff >> (VFIO_PCI_OFFSET_SHIFT - PAGE_SHIFT);
	415	+
	416	+ if (vma->vm_end < vma->vm_start)
	417	+ return -EINVAL;
	418	+ if ((vma->vm_flags & VM_SHARED) == 0)
	419	+ return -EINVAL;
	420	+ if (index >= VFIO_PCI_ROM_REGION_INDEX)
	421	+ return -EINVAL;
	422	+ if (!(pci_resource_flags(pdev, index) & IORESOURCE_MEM))
	423	+ return -EINVAL;
	424	+
	425	+ phys_len = pci_resource_len(pdev, index);
	426	+ req_len = vma->vm_end - vma->vm_start;
	427	+ pgoff = vma->vm_pgoff &
	428	+ ((1U << (VFIO_PCI_OFFSET_SHIFT - PAGE_SHIFT)) - 1);
	429	+ req_start = pgoff << PAGE_SHIFT;
	430	+
	431	+ if (phys_len < PAGE_SIZE \|\| req_start + req_len > phys_len)
	432	+ return -EINVAL;
	433	+
	434	+ if (index == vdev->msix_bar) {
	435	+ /*
	436	+ * Disallow mmaps overlapping the MSI-X table; users don't
	437	+ * get to touch this directly. We could find somewhere
	438	+ * else to map the overlap, but page granularity is only
	439	+ * a recommendation, not a requirement, so the user needs
	440	+ * to know which bits are real. Requiring them to mmap
	441	+ * around the table makes that clear.
	442	+ */
	443	+
	444	+ /* If neither entirely above nor below, then it overlaps */
	445	+ if (!(req_start >= vdev->msix_offset + vdev->msix_size \|\|
	446	+ req_start + req_len <= vdev->msix_offset))
	447	+ return -EINVAL;
	448	+ }
	449	+
	450	+ /*
	451	+ * Even though we don't make use of the barmap for the mmap,
	452	+ * we need to request the region and the barmap tracks that.
	453	+ */
	454	+ if (!vdev->barmap[index]) {
	455	+ ret = pci_request_selected_regions(pdev,
	456	+ 1 << index, "vfio-pci");
	457	+ if (ret)
	458	+ return ret;
	459	+
	460	+ vdev->barmap[index] = pci_iomap(pdev, index, 0);
	461	+ }
	462	+
	463	+ vma->vm_private_data = vdev;
	464	+ vma->vm_flags \|= (VM_IO \| VM_RESERVED);
	465	+ vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
	466	+
	467	+ phys = (pci_resource_start(pdev, index) >> PAGE_SHIFT) + pgoff;
	468	+
	469	+ return remap_pfn_range(vma, vma->vm_start, phys,
	470	+ req_len, vma->vm_page_prot);
	471	+}
	472	+
	473	+static const struct vfio_device_ops vfio_pci_ops = {
	474	+ .name = "vfio-pci",
	475	+ .open = vfio_pci_open,
	476	+ .release = vfio_pci_release,
	477	+ .ioctl = vfio_pci_ioctl,
	478	+ .read = vfio_pci_read,
	479	+ .write = vfio_pci_write,
	480	+ .mmap = vfio_pci_mmap,
	481	+};
	482	+
	483	+static int vfio_pci_probe(struct pci_dev pdev, const struct pci_device_id id)
	484	+{
	485	+ u8 type;
	486	+ struct vfio_pci_device *vdev;
	487	+ struct iommu_group *group;
	488	+ int ret;
	489	+
	490	+ pci_read_config_byte(pdev, PCI_HEADER_TYPE, &type);
	491	+ if ((type & PCI_HEADER_TYPE) != PCI_HEADER_TYPE_NORMAL)
	492	+ return -EINVAL;
	493	+
	494	+ group = iommu_group_get(&pdev->dev);
	495	+ if (!group)
	496	+ return -EINVAL;
	497	+
	498	+ vdev = kzalloc(sizeof(*vdev), GFP_KERNEL);
	499	+ if (!vdev) {
	500	+ iommu_group_put(group);
	501	+ return -ENOMEM;
	502	+ }
	503	+
	504	+ vdev->pdev = pdev;
	505	+ vdev->irq_type = VFIO_PCI_NUM_IRQS;
	506	+ mutex_init(&vdev->igate);
	507	+ spin_lock_init(&vdev->irqlock);
	508	+ atomic_set(&vdev->refcnt, 0);
	509	+
	510	+ ret = vfio_add_group_dev(&pdev->dev, &vfio_pci_ops, vdev);
	511	+ if (ret) {
	512	+ iommu_group_put(group);
	513	+ kfree(vdev);
	514	+ }
	515	+
	516	+ return ret;
	517	+}
	518	+
	519	+static void vfio_pci_remove(struct pci_dev *pdev)
	520	+{
	521	+ struct vfio_pci_device *vdev;
	522	+
	523	+ vdev = vfio_del_group_dev(&pdev->dev);
	524	+ if (!vdev)
	525	+ return;
	526	+
	527	+ iommu_group_put(pdev->dev.iommu_group);
	528	+ kfree(vdev);
	529	+}
	530	+
	531	+static struct pci_driver vfio_pci_driver = {
	532	+ .name = "vfio-pci",
	533	+ .id_table = NULL, /* only dynamic ids */
	534	+ .probe = vfio_pci_probe,
	535	+ .remove = vfio_pci_remove,
	536	+};
	537	+
	538	+static void __exit vfio_pci_cleanup(void)
	539	+{
	540	+ pci_unregister_driver(&vfio_pci_driver);
	541	+ vfio_pci_virqfd_exit();
	542	+ vfio_pci_uninit_perm_bits();
	543	+}
	544	+
	545	+static int __init vfio_pci_init(void)
	546	+{
	547	+ int ret;
	548	+
	549	+ /* Allocate shared config space permision data used by all devices */
	550	+ ret = vfio_pci_init_perm_bits();
	551	+ if (ret)
	552	+ return ret;
	553	+
	554	+ /* Start the virqfd cleanup handler */
	555	+ ret = vfio_pci_virqfd_init();
	556	+ if (ret)
	557	+ goto out_virqfd;
	558	+
	559	+ /* Register and scan for devices */
	560	+ ret = pci_register_driver(&vfio_pci_driver);
	561	+ if (ret)
	562	+ goto out_driver;
	563	+
	564	+ return 0;
	565	+
	566	+out_virqfd:
	567	+ vfio_pci_virqfd_exit();
	568	+out_driver:
	569	+ vfio_pci_uninit_perm_bits();
	570	+ return ret;
	571	+}
	572	+
	573	+module_init(vfio_pci_init);
	574	+module_exit(vfio_pci_cleanup);
	575	+
	576	+MODULE_VERSION(DRIVER_VERSION);
	577	+MODULE_LICENSE("GPL v2");
	578	+MODULE_AUTHOR(DRIVER_AUTHOR);
	579	+MODULE_DESCRIPTION(DRIVER_DESC);
	1	+/*
	2	+ * VFIO PCI interrupt handling
	3	+ *
	4	+ * Copyright (C) 2012 Red Hat, Inc. All rights reserved.
	5	+ * Author: Alex Williamson <alex.williamson@redhat.com>
	6	+ *
	7	+ * This program is free software; you can redistribute it and/or modify
	8	+ * it under the terms of the GNU General Public License version 2 as
	9	+ * published by the Free Software Foundation.
	10	+ *
	11	+ * Derived from original vfio:
	12	+ * Copyright 2010 Cisco Systems, Inc. All rights reserved.
	13	+ * Author: Tom Lyon, pugs@cisco.com
	14	+ */
	15	+
	16	+#include <linux/device.h>
	17	+#include <linux/interrupt.h>
	18	+#include <linux/eventfd.h>
	19	+#include <linux/pci.h>
	20	+#include <linux/file.h>
	21	+#include <linux/poll.h>
	22	+#include <linux/vfio.h>
	23	+#include <linux/wait.h>
	24	+#include <linux/workqueue.h>
	25	+
	26	+#include "vfio_pci_private.h"
	27	+
	28	+/*
	29	+ * IRQfd - generic
	30	+ */
	31	+struct virqfd {
	32	+ struct vfio_pci_device *vdev;
	33	+ struct eventfd_ctx *eventfd;
	34	+ int (handler)(struct vfio_pci_device , void *);
	35	+ void (thread)(struct vfio_pci_device , void *);
	36	+ void *data;
	37	+ struct work_struct inject;
	38	+ wait_queue_t wait;
	39	+ poll_table pt;
	40	+ struct work_struct shutdown;
	41	+ struct virqfd **pvirqfd;
	42	+};
	43	+
	44	+static struct workqueue_struct *vfio_irqfd_cleanup_wq;
	45	+
	46	+int __init vfio_pci_virqfd_init(void)
	47	+{
	48	+ vfio_irqfd_cleanup_wq =
	49	+ create_singlethread_workqueue("vfio-irqfd-cleanup");
	50	+ if (!vfio_irqfd_cleanup_wq)
	51	+ return -ENOMEM;
	52	+
	53	+ return 0;
	54	+}
	55	+
	56	+void vfio_pci_virqfd_exit(void)
	57	+{
	58	+ destroy_workqueue(vfio_irqfd_cleanup_wq);
	59	+}
	60	+
	61	+static void virqfd_deactivate(struct virqfd *virqfd)
	62	+{
	63	+ queue_work(vfio_irqfd_cleanup_wq, &virqfd->shutdown);
	64	+}
	65	+
	66	+static int virqfd_wakeup(wait_queue_t wait, unsigned mode, int sync, void key)
	67	+{
	68	+ struct virqfd *virqfd = container_of(wait, struct virqfd, wait);
	69	+ unsigned long flags = (unsigned long)key;
	70	+
	71	+ if (flags & POLLIN) {
	72	+ /* An event has been signaled, call function */
	73	+ if ((!virqfd->handler \|\|
	74	+ virqfd->handler(virqfd->vdev, virqfd->data)) &&
	75	+ virqfd->thread)
	76	+ schedule_work(&virqfd->inject);
	77	+ }
	78	+
	79	+ if (flags & POLLHUP)
	80	+ /* The eventfd is closing, detach from VFIO */
	81	+ virqfd_deactivate(virqfd);
	82	+
	83	+ return 0;
	84	+}
	85	+
	86	+static void virqfd_ptable_queue_proc(struct file *file,
	87	+ wait_queue_head_t wqh, poll_table pt)
	88	+{
	89	+ struct virqfd *virqfd = container_of(pt, struct virqfd, pt);
	90	+ add_wait_queue(wqh, &virqfd->wait);
	91	+}
	92	+
	93	+static void virqfd_shutdown(struct work_struct *work)
	94	+{
	95	+ struct virqfd *virqfd = container_of(work, struct virqfd, shutdown);
	96	+ struct virqfd **pvirqfd = virqfd->pvirqfd;
	97	+ u64 cnt;
	98	+
	99	+ eventfd_ctx_remove_wait_queue(virqfd->eventfd, &virqfd->wait, &cnt);
	100	+ flush_work(&virqfd->inject);
	101	+ eventfd_ctx_put(virqfd->eventfd);
	102	+
	103	+ kfree(virqfd);
	104	+ *pvirqfd = NULL;
	105	+}
	106	+
	107	+static void virqfd_inject(struct work_struct *work)
	108	+{
	109	+ struct virqfd *virqfd = container_of(work, struct virqfd, inject);
	110	+ if (virqfd->thread)
	111	+ virqfd->thread(virqfd->vdev, virqfd->data);
	112	+}
	113	+
	114	+static int virqfd_enable(struct vfio_pci_device *vdev,
	115	+ int (handler)(struct vfio_pci_device , void *),
	116	+ void (thread)(struct vfio_pci_device , void *),
	117	+ void data, struct virqfd *pvirqfd, int fd)
	118	+{
	119	+ struct file *file = NULL;
	120	+ struct eventfd_ctx *ctx = NULL;
	121	+ struct virqfd *virqfd;
	122	+ int ret = 0;
	123	+ unsigned int events;
	124	+
	125	+ if (*pvirqfd)
	126	+ return -EBUSY;
	127	+
	128	+ virqfd = kzalloc(sizeof(*virqfd), GFP_KERNEL);
	129	+ if (!virqfd)
	130	+ return -ENOMEM;
	131	+
	132	+ virqfd->pvirqfd = pvirqfd;
	133	+ *pvirqfd = virqfd;
	134	+ virqfd->vdev = vdev;
	135	+ virqfd->handler = handler;
	136	+ virqfd->thread = thread;
	137	+ virqfd->data = data;
	138	+
	139	+ INIT_WORK(&virqfd->shutdown, virqfd_shutdown);
	140	+ INIT_WORK(&virqfd->inject, virqfd_inject);
	141	+
	142	+ file = eventfd_fget(fd);
	143	+ if (IS_ERR(file)) {
	144	+ ret = PTR_ERR(file);
	145	+ goto fail;
	146	+ }
	147	+
	148	+ ctx = eventfd_ctx_fileget(file);
	149	+ if (IS_ERR(ctx)) {
	150	+ ret = PTR_ERR(ctx);
	151	+ goto fail;
	152	+ }
	153	+
	154	+ virqfd->eventfd = ctx;
	155	+
	156	+ /*
	157	+ * Install our own custom wake-up handling so we are notified via
	158	+ * a callback whenever someone signals the underlying eventfd.
	159	+ */
	160	+ init_waitqueue_func_entry(&virqfd->wait, virqfd_wakeup);
	161	+ init_poll_funcptr(&virqfd->pt, virqfd_ptable_queue_proc);
	162	+
	163	+ events = file->f_op->poll(file, &virqfd->pt);
	164	+
	165	+ /*
	166	+ * Check if there was an event already pending on the eventfd
	167	+ * before we registered and trigger it as if we didn't miss it.
	168	+ */
	169	+ if (events & POLLIN) {
	170	+ if ((!handler \|\| handler(vdev, data)) && thread)
	171	+ schedule_work(&virqfd->inject);
	172	+ }
	173	+
	174	+ /*
	175	+ * Do not drop the file until the irqfd is fully initialized,
	176	+ * otherwise we might race against the POLLHUP.
	177	+ */
	178	+ fput(file);
	179	+
	180	+ return 0;
	181	+
	182	+fail:
	183	+ if (ctx && !IS_ERR(ctx))
	184	+ eventfd_ctx_put(ctx);
	185	+
	186	+ if (file && !IS_ERR(file))
	187	+ fput(file);
	188	+
	189	+ kfree(virqfd);
	190	+ *pvirqfd = NULL;
	191	+
	192	+ return ret;
	193	+}
	194	+
	195	+static void virqfd_disable(struct virqfd *virqfd)
	196	+{
	197	+ if (!virqfd)
	198	+ return;
	199	+
	200	+ virqfd_deactivate(virqfd);
	201	+
	202	+ /* Block until we know all outstanding shutdown jobs have completed. */
	203	+ flush_workqueue(vfio_irqfd_cleanup_wq);
	204	+}
	205	+
	206	+/*
	207	+ * INTx
	208	+ */
	209	+static void vfio_send_intx_eventfd(struct vfio_pci_device vdev, void unused)
	210	+{
	211	+ if (likely(is_intx(vdev) && !vdev->virq_disabled))
	212	+ eventfd_signal(vdev->ctx[0].trigger, 1);
	213	+}
	214	+
	215	+void vfio_pci_intx_mask(struct vfio_pci_device *vdev)
	216	+{
	217	+ struct pci_dev *pdev = vdev->pdev;
	218	+ unsigned long flags;
	219	+
	220	+ spin_lock_irqsave(&vdev->irqlock, flags);
	221	+
	222	+ /*
	223	+ * Masking can come from interrupt, ioctl, or config space
	224	+ * via INTx disable. The latter means this can get called
	225	+ * even when not using intx delivery. In this case, just
	226	+ * try to have the physical bit follow the virtual bit.
	227	+ */
	228	+ if (unlikely(!is_intx(vdev))) {
	229	+ if (vdev->pci_2_3)
	230	+ pci_intx(pdev, 0);
	231	+ } else if (!vdev->ctx[0].masked) {
	232	+ /*
	233	+ * Can't use check_and_mask here because we always want to
	234	+ * mask, not just when something is pending.
	235	+ */
	236	+ if (vdev->pci_2_3)
	237	+ pci_intx(pdev, 0);
	238	+ else
	239	+ disable_irq_nosync(pdev->irq);
	240	+
	241	+ vdev->ctx[0].masked = true;
	242	+ }
	243	+
	244	+ spin_unlock_irqrestore(&vdev->irqlock, flags);
	245	+}
	246	+
	247	+/*
	248	+ * If this is triggered by an eventfd, we can't call eventfd_signal
	249	+ * or else we'll deadlock on the eventfd wait queue. Return >0 when
	250	+ * a signal is necessary, which can then be handled via a work queue
	251	+ * or directly depending on the caller.
	252	+ */
	253	+int vfio_pci_intx_unmask_handler(struct vfio_pci_device vdev, void unused)
	254	+{
	255	+ struct pci_dev *pdev = vdev->pdev;
	256	+ unsigned long flags;
	257	+ int ret = 0;
	258	+
	259	+ spin_lock_irqsave(&vdev->irqlock, flags);
	260	+
	261	+ /*
	262	+ * Unmasking comes from ioctl or config, so again, have the
	263	+ * physical bit follow the virtual even when not using INTx.
	264	+ */
	265	+ if (unlikely(!is_intx(vdev))) {
	266	+ if (vdev->pci_2_3)
	267	+ pci_intx(pdev, 1);
	268	+ } else if (vdev->ctx[0].masked && !vdev->virq_disabled) {
	269	+ /*
	270	+ * A pending interrupt here would immediately trigger,
	271	+ * but we can avoid that overhead by just re-sending
	272	+ * the interrupt to the user.
	273	+ */
	274	+ if (vdev->pci_2_3) {
	275	+ if (!pci_check_and_unmask_intx(pdev))
	276	+ ret = 1;
	277	+ } else
	278	+ enable_irq(pdev->irq);
	279	+
	280	+ vdev->ctx[0].masked = (ret > 0);
	281	+ }
	282	+
	283	+ spin_unlock_irqrestore(&vdev->irqlock, flags);
	284	+
	285	+ return ret;
	286	+}
	287	+
	288	+void vfio_pci_intx_unmask(struct vfio_pci_device *vdev)
	289	+{
	290	+ if (vfio_pci_intx_unmask_handler(vdev, NULL) > 0)
	291	+ vfio_send_intx_eventfd(vdev, NULL);
	292	+}
	293	+
	294	+static irqreturn_t vfio_intx_handler(int irq, void *dev_id)
	295	+{
	296	+ struct vfio_pci_device *vdev = dev_id;
	297	+ unsigned long flags;
	298	+ int ret = IRQ_NONE;
	299	+
	300	+ spin_lock_irqsave(&vdev->irqlock, flags);
	301	+
	302	+ if (!vdev->pci_2_3) {
	303	+ disable_irq_nosync(vdev->pdev->irq);
	304	+ vdev->ctx[0].masked = true;
	305	+ ret = IRQ_HANDLED;
	306	+ } else if (!vdev->ctx[0].masked && /* may be shared */
	307	+ pci_check_and_mask_intx(vdev->pdev)) {
	308	+ vdev->ctx[0].masked = true;
	309	+ ret = IRQ_HANDLED;
	310	+ }
	311	+
	312	+ spin_unlock_irqrestore(&vdev->irqlock, flags);
	313	+
	314	+ if (ret == IRQ_HANDLED)
	315	+ vfio_send_intx_eventfd(vdev, NULL);
	316	+
	317	+ return ret;
	318	+}
	319	+
	320	+static int vfio_intx_enable(struct vfio_pci_device *vdev)
	321	+{
	322	+ if (!is_irq_none(vdev))
	323	+ return -EINVAL;
	324	+
	325	+ if (!vdev->pdev->irq)
	326	+ return -ENODEV;
	327	+
	328	+ vdev->ctx = kzalloc(sizeof(struct vfio_pci_irq_ctx), GFP_KERNEL);
	329	+ if (!vdev->ctx)
	330	+ return -ENOMEM;
	331	+
	332	+ vdev->num_ctx = 1;
	333	+ vdev->irq_type = VFIO_PCI_INTX_IRQ_INDEX;
	334	+
	335	+ return 0;
	336	+}
	337	+
	338	+static int vfio_intx_set_signal(struct vfio_pci_device *vdev, int fd)
	339	+{
	340	+ struct pci_dev *pdev = vdev->pdev;
	341	+ unsigned long irqflags = IRQF_SHARED;
	342	+ struct eventfd_ctx *trigger;
	343	+ unsigned long flags;
	344	+ int ret;
	345	+
	346	+ if (vdev->ctx[0].trigger) {
	347	+ free_irq(pdev->irq, vdev);
	348	+ kfree(vdev->ctx[0].name);
	349	+ eventfd_ctx_put(vdev->ctx[0].trigger);
	350	+ vdev->ctx[0].trigger = NULL;
	351	+ }
	352	+
	353	+ if (fd < 0) /* Disable only */
	354	+ return 0;
	355	+
	356	+ vdev->ctx[0].name = kasprintf(GFP_KERNEL, "vfio-intx(%s)",
	357	+ pci_name(pdev));
	358	+ if (!vdev->ctx[0].name)
	359	+ return -ENOMEM;
	360	+
	361	+ trigger = eventfd_ctx_fdget(fd);
	362	+ if (IS_ERR(trigger)) {
	363	+ kfree(vdev->ctx[0].name);
	364	+ return PTR_ERR(trigger);
	365	+ }
	366	+
	367	+ if (!vdev->pci_2_3)
	368	+ irqflags = 0;
	369	+
	370	+ ret = request_irq(pdev->irq, vfio_intx_handler,
	371	+ irqflags, vdev->ctx[0].name, vdev);
	372	+ if (ret) {
	373	+ kfree(vdev->ctx[0].name);
	374	+ eventfd_ctx_put(trigger);
	375	+ return ret;
	376	+ }
	377	+
	378	+ vdev->ctx[0].trigger = trigger;
	379	+
	380	+ /*
	381	+ * INTx disable will stick across the new irq setup,
	382	+ * disable_irq won't.
	383	+ */
	384	+ spin_lock_irqsave(&vdev->irqlock, flags);
	385	+ if (!vdev->pci_2_3 && (vdev->ctx[0].masked \|\| vdev->virq_disabled))
	386	+ disable_irq_nosync(pdev->irq);
	387	+ spin_unlock_irqrestore(&vdev->irqlock, flags);
	388	+
	389	+ return 0;
	390	+}
	391	+
	392	+static void vfio_intx_disable(struct vfio_pci_device *vdev)
	393	+{
	394	+ vfio_intx_set_signal(vdev, -1);
	395	+ virqfd_disable(vdev->ctx[0].unmask);
	396	+ virqfd_disable(vdev->ctx[0].mask);
	397	+ vdev->irq_type = VFIO_PCI_NUM_IRQS;
	398	+ vdev->num_ctx = 0;
	399	+ kfree(vdev->ctx);
	400	+}
	401	+
	402	+/*
	403	+ * MSI/MSI-X
	404	+ */
	405	+static irqreturn_t vfio_msihandler(int irq, void *arg)
	406	+{
	407	+ struct eventfd_ctx *trigger = arg;
	408	+
	409	+ eventfd_signal(trigger, 1);
	410	+ return IRQ_HANDLED;
	411	+}
	412	+
	413	+static int vfio_msi_enable(struct vfio_pci_device *vdev, int nvec, bool msix)
	414	+{
	415	+ struct pci_dev *pdev = vdev->pdev;
	416	+ int ret;
	417	+
	418	+ if (!is_irq_none(vdev))
	419	+ return -EINVAL;
	420	+
	421	+ vdev->ctx = kzalloc(nvec * sizeof(struct vfio_pci_irq_ctx), GFP_KERNEL);
	422	+ if (!vdev->ctx)
	423	+ return -ENOMEM;
	424	+
	425	+ if (msix) {
	426	+ int i;
	427	+
	428	+ vdev->msix = kzalloc(nvec * sizeof(struct msix_entry),
	429	+ GFP_KERNEL);
	430	+ if (!vdev->msix) {
	431	+ kfree(vdev->ctx);
	432	+ return -ENOMEM;
	433	+ }
	434	+
	435	+ for (i = 0; i < nvec; i++)
	436	+ vdev->msix[i].entry = i;
	437	+
	438	+ ret = pci_enable_msix(pdev, vdev->msix, nvec);
	439	+ if (ret) {
	440	+ kfree(vdev->msix);
	441	+ kfree(vdev->ctx);
	442	+ return ret;
	443	+ }
	444	+ } else {
	445	+ ret = pci_enable_msi_block(pdev, nvec);
	446	+ if (ret) {
	447	+ kfree(vdev->ctx);
	448	+ return ret;
	449	+ }
	450	+ }
	451	+
	452	+ vdev->num_ctx = nvec;
	453	+ vdev->irq_type = msix ? VFIO_PCI_MSIX_IRQ_INDEX :
	454	+ VFIO_PCI_MSI_IRQ_INDEX;
	455	+
	456	+ if (!msix) {
	457	+ /*
	458	+ * Compute the virtual hardware field for max msi vectors -
	459	+ * it is the log base 2 of the number of vectors.
	460	+ */
	461	+ vdev->msi_qmax = fls(nvec * 2 - 1) - 1;
	462	+ }
	463	+
	464	+ return 0;
	465	+}
	466	+
	467	+static int vfio_msi_set_vector_signal(struct vfio_pci_device *vdev,
	468	+ int vector, int fd, bool msix)
	469	+{
	470	+ struct pci_dev *pdev = vdev->pdev;
	471	+ int irq = msix ? vdev->msix[vector].vector : pdev->irq + vector;
	472	+ char *name = msix ? "vfio-msix" : "vfio-msi";
	473	+ struct eventfd_ctx *trigger;
	474	+ int ret;
	475	+
	476	+ if (vector >= vdev->num_ctx)
	477	+ return -EINVAL;
	478	+
	479	+ if (vdev->ctx[vector].trigger) {
	480	+ free_irq(irq, vdev->ctx[vector].trigger);
	481	+ kfree(vdev->ctx[vector].name);
	482	+ eventfd_ctx_put(vdev->ctx[vector].trigger);
	483	+ vdev->ctx[vector].trigger = NULL;
	484	+ }
	485	+
	486	+ if (fd < 0)
	487	+ return 0;
	488	+
	489	+ vdev->ctx[vector].name = kasprintf(GFP_KERNEL, "%s[%d](%s)",
	490	+ name, vector, pci_name(pdev));
	491	+ if (!vdev->ctx[vector].name)
	492	+ return -ENOMEM;
	493	+
	494	+ trigger = eventfd_ctx_fdget(fd);
	495	+ if (IS_ERR(trigger)) {
	496	+ kfree(vdev->ctx[vector].name);
	497	+ return PTR_ERR(trigger);
	498	+ }
	499	+
	500	+ ret = request_irq(irq, vfio_msihandler, 0,
	501	+ vdev->ctx[vector].name, trigger);
	502	+ if (ret) {
	503	+ kfree(vdev->ctx[vector].name);
	504	+ eventfd_ctx_put(trigger);
	505	+ return ret;
	506	+ }
	507	+
	508	+ vdev->ctx[vector].trigger = trigger;
	509	+
	510	+ return 0;
	511	+}
	512	+
	513	+static int vfio_msi_set_block(struct vfio_pci_device *vdev, unsigned start,
	514	+ unsigned count, int32_t *fds, bool msix)
	515	+{
	516	+ int i, j, ret = 0;
	517	+
	518	+ if (start + count > vdev->num_ctx)
	519	+ return -EINVAL;
	520	+
	521	+ for (i = 0, j = start; i < count && !ret; i++, j++) {
	522	+ int fd = fds ? fds[i] : -1;
	523	+ ret = vfio_msi_set_vector_signal(vdev, j, fd, msix);
	524	+ }
	525	+
	526	+ if (ret) {
	527	+ for (--j; j >= start; j--)
	528	+ vfio_msi_set_vector_signal(vdev, j, -1, msix);
	529	+ }
	530	+
	531	+ return ret;
	532	+}
	533	+
	534	+static void vfio_msi_disable(struct vfio_pci_device *vdev, bool msix)
	535	+{
	536	+ struct pci_dev *pdev = vdev->pdev;
	537	+ int i;
	538	+
	539	+ vfio_msi_set_block(vdev, 0, vdev->num_ctx, NULL, msix);
	540	+
	541	+ for (i = 0; i < vdev->num_ctx; i++) {
	542	+ virqfd_disable(vdev->ctx[i].unmask);
	543	+ virqfd_disable(vdev->ctx[i].mask);
	544	+ }
	545	+
	546	+ if (msix) {
	547	+ pci_disable_msix(vdev->pdev);
	548	+ kfree(vdev->msix);
	549	+ } else
	550	+ pci_disable_msi(pdev);
	551	+
	552	+ vdev->irq_type = VFIO_PCI_NUM_IRQS;
	553	+ vdev->num_ctx = 0;
	554	+ kfree(vdev->ctx);
	555	+}
	556	+
	557	+/*
	558	+ * IOCTL support
	559	+ */
	560	+static int vfio_pci_set_intx_unmask(struct vfio_pci_device *vdev,
	561	+ unsigned index, unsigned start,
	562	+ unsigned count, uint32_t flags, void *data)
	563	+{
	564	+ if (!is_intx(vdev) \|\| start != 0 \|\| count != 1)
	565	+ return -EINVAL;
	566	+
	567	+ if (flags & VFIO_IRQ_SET_DATA_NONE) {
	568	+ vfio_pci_intx_unmask(vdev);
	569	+ } else if (flags & VFIO_IRQ_SET_DATA_BOOL) {
	570	+ uint8_t unmask = (uint8_t )data;
	571	+ if (unmask)
	572	+ vfio_pci_intx_unmask(vdev);
	573	+ } else if (flags & VFIO_IRQ_SET_DATA_EVENTFD) {
	574	+ int32_t fd = (int32_t )data;
	575	+ if (fd >= 0)
	576	+ return virqfd_enable(vdev, vfio_pci_intx_unmask_handler,
	577	+ vfio_send_intx_eventfd, NULL,
	578	+ &vdev->ctx[0].unmask, fd);
	579	+
	580	+ virqfd_disable(vdev->ctx[0].unmask);
	581	+ }
	582	+
	583	+ return 0;
	584	+}
	585	+
	586	+static int vfio_pci_set_intx_mask(struct vfio_pci_device *vdev,
	587	+ unsigned index, unsigned start,
	588	+ unsigned count, uint32_t flags, void *data)
	589	+{
	590	+ if (!is_intx(vdev) \|\| start != 0 \|\| count != 1)
	591	+ return -EINVAL;
	592	+
	593	+ if (flags & VFIO_IRQ_SET_DATA_NONE) {
	594	+ vfio_pci_intx_mask(vdev);
	595	+ } else if (flags & VFIO_IRQ_SET_DATA_BOOL) {
	596	+ uint8_t mask = (uint8_t )data;
	597	+ if (mask)
	598	+ vfio_pci_intx_mask(vdev);
	599	+ } else if (flags & VFIO_IRQ_SET_DATA_EVENTFD) {
	600	+ return -ENOTTY; /* XXX implement me */
	601	+ }
	602	+
	603	+ return 0;
	604	+}
	605	+
	606	+static int vfio_pci_set_intx_trigger(struct vfio_pci_device *vdev,
	607	+ unsigned index, unsigned start,
	608	+ unsigned count, uint32_t flags, void *data)
	609	+{
	610	+ if (is_intx(vdev) && !count && (flags & VFIO_IRQ_SET_DATA_NONE)) {
	611	+ vfio_intx_disable(vdev);
	612	+ return 0;
	613	+ }
	614	+
	615	+ if (!(is_intx(vdev) \|\| is_irq_none(vdev)) \|\| start != 0 \|\| count != 1)
	616	+ return -EINVAL;
	617	+
	618	+ if (flags & VFIO_IRQ_SET_DATA_EVENTFD) {
	619	+ int32_t fd = (int32_t )data;
	620	+ int ret;
	621	+
	622	+ if (is_intx(vdev))
	623	+ return vfio_intx_set_signal(vdev, fd);
	624	+
	625	+ ret = vfio_intx_enable(vdev);
	626	+ if (ret)
	627	+ return ret;
	628	+
	629	+ ret = vfio_intx_set_signal(vdev, fd);
	630	+ if (ret)
	631	+ vfio_intx_disable(vdev);
	632	+
	633	+ return ret;
	634	+ }
	635	+
	636	+ if (!is_intx(vdev))
	637	+ return -EINVAL;
	638	+
	639	+ if (flags & VFIO_IRQ_SET_DATA_NONE) {
	640	+ vfio_send_intx_eventfd(vdev, NULL);
	641	+ } else if (flags & VFIO_IRQ_SET_DATA_BOOL) {
	642	+ uint8_t trigger = (uint8_t )data;
	643	+ if (trigger)
	644	+ vfio_send_intx_eventfd(vdev, NULL);
	645	+ }
	646	+ return 0;
	647	+}
	648	+
	649	+static int vfio_pci_set_msi_trigger(struct vfio_pci_device *vdev,
	650	+ unsigned index, unsigned start,
	651	+ unsigned count, uint32_t flags, void *data)
	652	+{
	653	+ int i;
	654	+ bool msix = (index == VFIO_PCI_MSIX_IRQ_INDEX) ? true : false;
	655	+
	656	+ if (irq_is(vdev, index) && !count && (flags & VFIO_IRQ_SET_DATA_NONE)) {
	657	+ vfio_msi_disable(vdev, msix);
	658	+ return 0;
	659	+ }
	660	+
	661	+ if (!(irq_is(vdev, index) \|\| is_irq_none(vdev)))
	662	+ return -EINVAL;
	663	+
	664	+ if (flags & VFIO_IRQ_SET_DATA_EVENTFD) {
	665	+ int32_t *fds = data;
	666	+ int ret;
	667	+
	668	+ if (vdev->irq_type == index)
	669	+ return vfio_msi_set_block(vdev, start, count,
	670	+ fds, msix);
	671	+
	672	+ ret = vfio_msi_enable(vdev, start + count, msix);
	673	+ if (ret)
	674	+ return ret;
	675	+
	676	+ ret = vfio_msi_set_block(vdev, start, count, fds, msix);
	677	+ if (ret)
	678	+ vfio_msi_disable(vdev, msix);
	679	+
	680	+ return ret;
	681	+ }
	682	+
	683	+ if (!irq_is(vdev, index) \|\| start + count > vdev->num_ctx)
	684	+ return -EINVAL;
	685	+
	686	+ for (i = start; i < start + count; i++) {
	687	+ if (!vdev->ctx[i].trigger)
	688	+ continue;
	689	+ if (flags & VFIO_IRQ_SET_DATA_NONE) {
	690	+ eventfd_signal(vdev->ctx[i].trigger, 1);
	691	+ } else if (flags & VFIO_IRQ_SET_DATA_BOOL) {
	692	+ uint8_t *bools = data;
	693	+ if (bools[i - start])
	694	+ eventfd_signal(vdev->ctx[i].trigger, 1);
	695	+ }
	696	+ }
	697	+ return 0;
	698	+}
	699	+
	700	+int vfio_pci_set_irqs_ioctl(struct vfio_pci_device *vdev, uint32_t flags,
	701	+ unsigned index, unsigned start, unsigned count,
	702	+ void *data)
	703	+{
	704	+ int (func)(struct vfio_pci_device vdev, unsigned index,
	705	+ unsigned start, unsigned count, uint32_t flags,
	706	+ void *data) = NULL;
	707	+
	708	+ switch (index) {
	709	+ case VFIO_PCI_INTX_IRQ_INDEX:
	710	+ switch (flags & VFIO_IRQ_SET_ACTION_TYPE_MASK) {
	711	+ case VFIO_IRQ_SET_ACTION_MASK:
	712	+ func = vfio_pci_set_intx_mask;
	713	+ break;
	714	+ case VFIO_IRQ_SET_ACTION_UNMASK:
	715	+ func = vfio_pci_set_intx_unmask;
	716	+ break;
	717	+ case VFIO_IRQ_SET_ACTION_TRIGGER:
	718	+ func = vfio_pci_set_intx_trigger;
	719	+ break;
	720	+ }
	721	+ break;
	722	+ case VFIO_PCI_MSI_IRQ_INDEX:
	723	+ case VFIO_PCI_MSIX_IRQ_INDEX:
	724	+ switch (flags & VFIO_IRQ_SET_ACTION_TYPE_MASK) {
	725	+ case VFIO_IRQ_SET_ACTION_MASK:
	726	+ case VFIO_IRQ_SET_ACTION_UNMASK:
	727	+ /* XXX Need masking support exported */
	728	+ break;
	729	+ case VFIO_IRQ_SET_ACTION_TRIGGER:
	730	+ func = vfio_pci_set_msi_trigger;
	731	+ break;
	732	+ }
	733	+ break;
	734	+ }
	735	+
	736	+ if (!func)
	737	+ return -ENOTTY;
	738	+
	739	+ return func(vdev, index, start, count, flags, data);
	740	+}
	1	+/*
	2	+ * VFIO PCI I/O Port & MMIO access
	3	+ *
	4	+ * Copyright (C) 2012 Red Hat, Inc. All rights reserved.
	5	+ * Author: Alex Williamson <alex.williamson@redhat.com>
	6	+ *
	7	+ * This program is free software; you can redistribute it and/or modify
	8	+ * it under the terms of the GNU General Public License version 2 as
	9	+ * published by the Free Software Foundation.
	10	+ *
	11	+ * Derived from original vfio:
	12	+ * Copyright 2010 Cisco Systems, Inc. All rights reserved.
	13	+ * Author: Tom Lyon, pugs@cisco.com
	14	+ */
	15	+
	16	+#include <linux/fs.h>
	17	+#include <linux/pci.h>
	18	+#include <linux/uaccess.h>
	19	+#include <linux/io.h>
	20	+
	21	+#include "vfio_pci_private.h"
	22	+
	23	+/* I/O Port BAR access */
	24	+ssize_t vfio_pci_io_readwrite(struct vfio_pci_device vdev, char __user buf,
	25	+ size_t count, loff_t *ppos, bool iswrite)
	26	+{
	27	+ struct pci_dev *pdev = vdev->pdev;
	28	+ loff_t pos = *ppos & VFIO_PCI_OFFSET_MASK;
	29	+ int bar = VFIO_PCI_OFFSET_TO_INDEX(*ppos);
	30	+ void __iomem *io;
	31	+ size_t done = 0;
	32	+
	33	+ if (!pci_resource_start(pdev, bar))
	34	+ return -EINVAL;
	35	+
	36	+ if (pos + count > pci_resource_len(pdev, bar))
	37	+ return -EINVAL;
	38	+
	39	+ if (!vdev->barmap[bar]) {
	40	+ int ret;
	41	+
	42	+ ret = pci_request_selected_regions(pdev, 1 << bar, "vfio");
	43	+ if (ret)
	44	+ return ret;
	45	+
	46	+ vdev->barmap[bar] = pci_iomap(pdev, bar, 0);
	47	+
	48	+ if (!vdev->barmap[bar]) {
	49	+ pci_release_selected_regions(pdev, 1 << bar);
	50	+ return -EINVAL;
	51	+ }
	52	+ }
	53	+
	54	+ io = vdev->barmap[bar];
	55	+
	56	+ while (count) {
	57	+ int filled;
	58	+
	59	+ if (count >= 3 && !(pos % 4)) {
	60	+ __le32 val;
	61	+
	62	+ if (iswrite) {
	63	+ if (copy_from_user(&val, buf, 4))
	64	+ return -EFAULT;
	65	+
	66	+ iowrite32(le32_to_cpu(val), io + pos);
	67	+ } else {
	68	+ val = cpu_to_le32(ioread32(io + pos));
	69	+
	70	+ if (copy_to_user(buf, &val, 4))
	71	+ return -EFAULT;
	72	+ }
	73	+
	74	+ filled = 4;
	75	+
	76	+ } else if ((pos % 2) == 0 && count >= 2) {
	77	+ __le16 val;
	78	+
	79	+ if (iswrite) {
	80	+ if (copy_from_user(&val, buf, 2))
	81	+ return -EFAULT;
	82	+
	83	+ iowrite16(le16_to_cpu(val), io + pos);
	84	+ } else {
	85	+ val = cpu_to_le16(ioread16(io + pos));
	86	+
	87	+ if (copy_to_user(buf, &val, 2))
	88	+ return -EFAULT;
	89	+ }
	90	+
	91	+ filled = 2;
	92	+ } else {
	93	+ u8 val;
	94	+
	95	+ if (iswrite) {
	96	+ if (copy_from_user(&val, buf, 1))
	97	+ return -EFAULT;
	98	+
	99	+ iowrite8(val, io + pos);
	100	+ } else {
	101	+ val = ioread8(io + pos);
	102	+
	103	+ if (copy_to_user(buf, &val, 1))
	104	+ return -EFAULT;
	105	+ }
	106	+
	107	+ filled = 1;
	108	+ }
	109	+
	110	+ count -= filled;
	111	+ done += filled;
	112	+ buf += filled;
	113	+ pos += filled;
	114	+ }
	115	+
	116	+ *ppos += done;
	117	+
	118	+ return done;
	119	+}
	120	+
	121	+/*
	122	+ * MMIO BAR access
	123	+ * We handle two excluded ranges here as well, if the user tries to read
	124	+ * the ROM beyond what PCI tells us is available or the MSI-X table region,
	125	+ * we return 0xFF and writes are dropped.
	126	+ */
	127	+ssize_t vfio_pci_mem_readwrite(struct vfio_pci_device vdev, char __user buf,
	128	+ size_t count, loff_t *ppos, bool iswrite)
	129	+{
	130	+ struct pci_dev *pdev = vdev->pdev;
	131	+ loff_t pos = *ppos & VFIO_PCI_OFFSET_MASK;
	132	+ int bar = VFIO_PCI_OFFSET_TO_INDEX(*ppos);
	133	+ void __iomem *io;
	134	+ resource_size_t end;
	135	+ size_t done = 0;
	136	+ size_t x_start = 0, x_end = 0; /* excluded range */
	137	+
	138	+ if (!pci_resource_start(pdev, bar))
	139	+ return -EINVAL;
	140	+
	141	+ end = pci_resource_len(pdev, bar);
	142	+
	143	+ if (pos > end)
	144	+ return -EINVAL;
	145	+
	146	+ if (pos == end)
	147	+ return 0;
	148	+
	149	+ if (pos + count > end)
	150	+ count = end - pos;
	151	+
	152	+ if (bar == PCI_ROM_RESOURCE) {
	153	+ io = pci_map_rom(pdev, &x_start);
	154	+ x_end = end;
	155	+ } else {
	156	+ if (!vdev->barmap[bar]) {
	157	+ int ret;
	158	+
	159	+ ret = pci_request_selected_regions(pdev, 1 << bar,
	160	+ "vfio");
	161	+ if (ret)
	162	+ return ret;
	163	+
	164	+ vdev->barmap[bar] = pci_iomap(pdev, bar, 0);
	165	+
	166	+ if (!vdev->barmap[bar]) {
	167	+ pci_release_selected_regions(pdev, 1 << bar);
	168	+ return -EINVAL;
	169	+ }
	170	+ }
	171	+
	172	+ io = vdev->barmap[bar];
	173	+
	174	+ if (bar == vdev->msix_bar) {
	175	+ x_start = vdev->msix_offset;
	176	+ x_end = vdev->msix_offset + vdev->msix_size;
	177	+ }
	178	+ }
	179	+
	180	+ if (!io)
	181	+ return -EINVAL;
	182	+
	183	+ while (count) {
	184	+ size_t fillable, filled;
	185	+
	186	+ if (pos < x_start)
	187	+ fillable = x_start - pos;
	188	+ else if (pos >= x_end)
	189	+ fillable = end - pos;
	190	+ else
	191	+ fillable = 0;
	192	+
	193	+ if (fillable >= 4 && !(pos % 4) && (count >= 4)) {
	194	+ __le32 val;
	195	+
	196	+ if (iswrite) {
	197	+ if (copy_from_user(&val, buf, 4))
	198	+ goto out;
	199	+
	200	+ iowrite32(le32_to_cpu(val), io + pos);
	201	+ } else {
	202	+ val = cpu_to_le32(ioread32(io + pos));
	203	+
	204	+ if (copy_to_user(buf, &val, 4))
	205	+ goto out;
	206	+ }
	207	+
	208	+ filled = 4;
	209	+ } else if (fillable >= 2 && !(pos % 2) && (count >= 2)) {
	210	+ __le16 val;
	211	+
	212	+ if (iswrite) {
	213	+ if (copy_from_user(&val, buf, 2))
	214	+ goto out;
	215	+
	216	+ iowrite16(le16_to_cpu(val), io + pos);
	217	+ } else {
	218	+ val = cpu_to_le16(ioread16(io + pos));
	219	+
	220	+ if (copy_to_user(buf, &val, 2))
	221	+ goto out;
	222	+ }
	223	+
	224	+ filled = 2;
	225	+ } else if (fillable) {
	226	+ u8 val;
	227	+
	228	+ if (iswrite) {
	229	+ if (copy_from_user(&val, buf, 1))
	230	+ goto out;
	231	+
	232	+ iowrite8(val, io + pos);
	233	+ } else {
	234	+ val = ioread8(io + pos);
	235	+
	236	+ if (copy_to_user(buf, &val, 1))
	237	+ goto out;
	238	+ }
	239	+
	240	+ filled = 1;
	241	+ } else {
	242	+ /* Drop writes, fill reads with FF */
	243	+ if (!iswrite) {
	244	+ char val = 0xFF;
	245	+ size_t i;
	246	+
	247	+ for (i = 0; i < x_end - pos; i++) {
	248	+ if (put_user(val, buf + i))
	249	+ goto out;
	250	+ }
	251	+ }
	252	+
	253	+ filled = x_end - pos;
	254	+ }
	255	+
	256	+ count -= filled;
	257	+ done += filled;
	258	+ buf += filled;
	259	+ pos += filled;
	260	+ }
	261	+
	262	+ *ppos += done;
	263	+
	264	+out:
	265	+ if (bar == PCI_ROM_RESOURCE)
	266	+ pci_unmap_rom(pdev, io);
	267	+
	268	+ return count ? -EFAULT : done;
	269	+}
...	...	@@ -223,6 +223,7 @@
223	223	__u32 argsz;
224	224	__u32 flags;
225	225	#define VFIO_DEVICE_FLAGS_RESET (1 << 0) /* Device supports reset */
	226	+#define VFIO_DEVICE_FLAGS_PCI (1 << 1) /* vfio-pci device */
226	227	__u32 num_regions; /* Max region index + 1 */
227	228	__u32 num_irqs; /* Max IRQ index + 1 */
228	229	};
...	...	@@ -363,6 +364,31 @@
363	364	* Reset a device.
364	365	*/
365	366	#define VFIO_DEVICE_RESET _IO(VFIO_TYPE, VFIO_BASE + 11)
	367	+
	368	+/*
	369	+ * The VFIO-PCI bus driver makes use of the following fixed region and
	370	+ * IRQ index mapping. Unimplemented regions return a size of zero.
	371	+ * Unimplemented IRQ types return a count of zero.
	372	+ */
	373	+
	374	+enum {
	375	+ VFIO_PCI_BAR0_REGION_INDEX,
	376	+ VFIO_PCI_BAR1_REGION_INDEX,
	377	+ VFIO_PCI_BAR2_REGION_INDEX,
	378	+ VFIO_PCI_BAR3_REGION_INDEX,
	379	+ VFIO_PCI_BAR4_REGION_INDEX,
	380	+ VFIO_PCI_BAR5_REGION_INDEX,
	381	+ VFIO_PCI_ROM_REGION_INDEX,
	382	+ VFIO_PCI_CONFIG_REGION_INDEX,
	383	+ VFIO_PCI_NUM_REGIONS
	384	+};
	385	+
	386	+enum {
	387	+ VFIO_PCI_INTX_IRQ_INDEX,
	388	+ VFIO_PCI_MSI_IRQ_INDEX,
	389	+ VFIO_PCI_MSIX_IRQ_INDEX,
	390	+ VFIO_PCI_NUM_IRQS
	391	+};
366	392
367	393	/* -------- API for Type1 VFIO IOMMU -------- */
368	394