igb_uio: fix build on arm64 kernel
[dpdk.git] / lib / librte_eal / linuxapp / igb_uio / igb_uio.c
index bca412b..da71e05 100644 (file)
 #include <linux/pci.h>
 #include <linux/uio_driver.h>
 #include <linux/io.h>
+#include <linux/irq.h>
 #include <linux/msi.h>
 #include <linux/version.h>
+#include <linux/slab.h>
 
 #ifdef CONFIG_XEN_DOM0
 #include <xen/xen.h>
 #endif
 #include <rte_pci_dev_features.h>
 
-/**
- * MSI-X related macros, copy from linux/pci_regs.h in kernel 2.6.39,
- * but none of them in kernel 2.6.35.
- */
-#ifndef PCI_MSIX_ENTRY_SIZE
-#define PCI_MSIX_ENTRY_SIZE             16
-#define PCI_MSIX_ENTRY_LOWER_ADDR       0
-#define PCI_MSIX_ENTRY_UPPER_ADDR       4
-#define PCI_MSIX_ENTRY_DATA             8
-#define PCI_MSIX_ENTRY_VECTOR_CTRL      12
-#define PCI_MSIX_ENTRY_CTRL_MASKBIT     1
-#endif
-
-#ifdef RTE_PCI_CONFIG
-#define PCI_SYS_FILE_BUF_SIZE      10
-#define PCI_DEV_CAP_REG            0xA4
-#define PCI_DEV_CTRL_REG           0xA8
-#define PCI_DEV_CAP_EXT_TAG_MASK   0x20
-#define PCI_DEV_CTRL_EXT_TAG_SHIFT 8
-#define PCI_DEV_CTRL_EXT_TAG_MASK  (1 << PCI_DEV_CTRL_EXT_TAG_SHIFT)
-#endif
-
-#define IGBUIO_NUM_MSI_VECTORS 1
+#include "compat.h"
 
 /**
  * A structure describing the private information for a uio device.
 struct rte_uio_pci_dev {
        struct uio_info info;
        struct pci_dev *pdev;
-       spinlock_t lock; /* spinlock for accessing PCI config space or msix data in multi tasks/isr */
        enum rte_intr_mode mode;
-       struct msix_entry \
-               msix_entries[IGBUIO_NUM_MSI_VECTORS]; /* pointer to the msix vectors to be allocated later */
 };
 
-static char *intr_mode = NULL;
+static char *intr_mode;
 static enum rte_intr_mode igbuio_intr_mode_preferred = RTE_INTR_MODE_MSIX;
 
-static inline struct rte_uio_pci_dev *
-igbuio_get_uio_pci_dev(struct uio_info *info)
-{
-       return container_of(info, struct rte_uio_pci_dev, info);
-}
-
 /* sriov sysfs */
-#if LINUX_VERSION_CODE < KERNEL_VERSION(2, 6, 34)
-static int pci_num_vf(struct pci_dev *dev)
-       struct iov {
-               int pos;
-               int nres;
-               u32 cap;
-               u16 ctrl;
-               u16 total;
-               u16 initial;
-               u16 nr_virtfn;
-       } *iov = (struct iov *)dev->sriov;
-
-       if (!dev->is_physfn)
-               return 0;
-
-       return iov->nr_virtfn;
-}
-#endif
-
 static ssize_t
 show_max_vfs(struct device *dev, struct device_attribute *attr,
             char *buf)
 {
-       return snprintf(buf, 10, "%u\n",
-                       pci_num_vf(container_of(dev, struct pci_dev, dev)));
+       return snprintf(buf, 10, "%u\n", dev_num_vf(dev));
 }
 
 static ssize_t
@@ -116,9 +67,9 @@ store_max_vfs(struct device *dev, struct device_attribute *attr,
 {
        int err = 0;
        unsigned long max_vfs;
-       struct pci_dev *pdev = container_of(dev, struct pci_dev, dev);
+       struct pci_dev *pdev = to_pci_dev(dev);
 
-       if (0 != strict_strtoul(buf, 0, &max_vfs))
+       if (0 != kstrtoul(buf, 0, &max_vfs))
                return -EINVAL;
 
        if (0 == max_vfs)
@@ -131,105 +82,10 @@ store_max_vfs(struct device *dev, struct device_attribute *attr,
        return err ? err : count;
 }
 
-#ifdef RTE_PCI_CONFIG
-static ssize_t
-show_extended_tag(struct device *dev, struct device_attribute *attr, char *buf)
-{
-       struct pci_dev *pci_dev = container_of(dev, struct pci_dev, dev);
-       uint32_t val = 0;
-
-       pci_read_config_dword(pci_dev, PCI_DEV_CAP_REG, &val);
-       if (!(val & PCI_DEV_CAP_EXT_TAG_MASK)) /* Not supported */
-               return snprintf(buf, PCI_SYS_FILE_BUF_SIZE, "%s\n", "invalid");
-
-       val = 0;
-       pci_bus_read_config_dword(pci_dev->bus, pci_dev->devfn,
-                                       PCI_DEV_CTRL_REG, &val);
-
-       return snprintf(buf, PCI_SYS_FILE_BUF_SIZE, "%s\n",
-               (val & PCI_DEV_CTRL_EXT_TAG_MASK) ? "on" : "off");
-}
-
-static ssize_t
-store_extended_tag(struct device *dev,
-                  struct device_attribute *attr,
-                  const char *buf,
-                  size_t count)
-{
-       struct pci_dev *pci_dev = container_of(dev, struct pci_dev, dev);
-       uint32_t val = 0, enable;
-
-       if (strncmp(buf, "on", 2) == 0)
-               enable = 1;
-       else if (strncmp(buf, "off", 3) == 0)
-               enable = 0;
-       else
-               return -EINVAL;
-
-       pci_bus_read_config_dword(pci_dev->bus, pci_dev->devfn,
-                                       PCI_DEV_CAP_REG, &val);
-       if (!(val & PCI_DEV_CAP_EXT_TAG_MASK)) /* Not supported */
-               return -EPERM;
-
-       val = 0;
-       pci_bus_read_config_dword(pci_dev->bus, pci_dev->devfn,
-                                       PCI_DEV_CTRL_REG, &val);
-       if (enable)
-               val |= PCI_DEV_CTRL_EXT_TAG_MASK;
-       else
-               val &= ~PCI_DEV_CTRL_EXT_TAG_MASK;
-       pci_bus_write_config_dword(pci_dev->bus, pci_dev->devfn,
-                                       PCI_DEV_CTRL_REG, val);
-
-       return count;
-}
-
-static ssize_t
-show_max_read_request_size(struct device *dev,
-                          struct device_attribute *attr,
-                          char *buf)
-{
-       struct pci_dev *pci_dev = container_of(dev, struct pci_dev, dev);
-       int val = pcie_get_readrq(pci_dev);
-
-       return snprintf(buf, PCI_SYS_FILE_BUF_SIZE, "%d\n", val);
-}
-
-static ssize_t
-store_max_read_request_size(struct device *dev,
-                           struct device_attribute *attr,
-                           const char *buf,
-                           size_t count)
-{
-       struct pci_dev *pci_dev = container_of(dev, struct pci_dev, dev);
-       unsigned long size = 0;
-       int ret;
-
-       if (strict_strtoul(buf, 0, &size) != 0)
-               return -EINVAL;
-
-       ret = pcie_set_readrq(pci_dev, (int)size);
-       if (ret < 0)
-               return ret;
-
-       return count;
-}
-#endif
-
 static DEVICE_ATTR(max_vfs, S_IRUGO | S_IWUSR, show_max_vfs, store_max_vfs);
-#ifdef RTE_PCI_CONFIG
-static DEVICE_ATTR(extended_tag, S_IRUGO | S_IWUSR, show_extended_tag,
-       store_extended_tag);
-static DEVICE_ATTR(max_read_request_size, S_IRUGO | S_IWUSR,
-       show_max_read_request_size, store_max_read_request_size);
-#endif
 
 static struct attribute *dev_attrs[] = {
        &dev_attr_max_vfs.attr,
-#ifdef RTE_PCI_CONFIG
-       &dev_attr_extended_tag.attr,
-       &dev_attr_max_read_request_size.attr,
-#endif
        NULL,
 };
 
@@ -237,93 +93,6 @@ static const struct attribute_group dev_attr_grp = {
        .attrs = dev_attrs,
 };
 
-static inline int
-pci_lock(struct pci_dev * pdev)
-{
-       /* Some function names changes between 3.2.0 and 3.3.0... */
-#if LINUX_VERSION_CODE < KERNEL_VERSION(3, 3, 0)
-       pci_block_user_cfg_access(pdev);
-       return 1;
-#else
-       return pci_cfg_access_trylock(pdev);
-#endif
-}
-
-static inline void
-pci_unlock(struct pci_dev * pdev)
-{
-       /* Some function names changes between 3.2.0 and 3.3.0... */
-#if LINUX_VERSION_CODE < KERNEL_VERSION(3, 3, 0)
-       pci_unblock_user_cfg_access(pdev);
-#else
-       pci_cfg_access_unlock(pdev);
-#endif
-}
-
-/**
- * It masks the msix on/off of generating MSI-X messages.
- */
-static int
-igbuio_msix_mask_irq(struct msi_desc *desc, int32_t state)
-{
-       uint32_t mask_bits = desc->masked;
-       unsigned offset = desc->msi_attrib.entry_nr * PCI_MSIX_ENTRY_SIZE +
-                                               PCI_MSIX_ENTRY_VECTOR_CTRL;
-
-       if (state != 0)
-               mask_bits &= ~PCI_MSIX_ENTRY_CTRL_MASKBIT;
-       else
-               mask_bits |= PCI_MSIX_ENTRY_CTRL_MASKBIT;
-
-       if (mask_bits != desc->masked) {
-               writel(mask_bits, desc->mask_base + offset);
-               readl(desc->mask_base);
-               desc->masked = mask_bits;
-       }
-
-       return 0;
-}
-
-/**
- * This function sets/clears the masks for generating LSC interrupts.
- *
- * @param info
- *   The pointer to struct uio_info.
- * @param on
- *   The on/off flag of masking LSC.
- * @return
- *   -On success, zero value.
- *   -On failure, a negative value.
- */
-static int
-igbuio_set_interrupt_mask(struct rte_uio_pci_dev *udev, int32_t state)
-{
-       struct pci_dev *pdev = udev->pdev;
-
-       if (udev->mode == RTE_INTR_MODE_MSIX) {
-               struct msi_desc *desc;
-
-               list_for_each_entry(desc, &pdev->msi_list, list) {
-                       igbuio_msix_mask_irq(desc, state);
-               }
-       } else if (udev->mode == RTE_INTR_MODE_LEGACY) {
-               uint32_t status;
-               uint16_t old, new;
-
-               pci_read_config_dword(pdev, PCI_COMMAND, &status);
-               old = status;
-               if (state != 0)
-                       new = old & (~PCI_COMMAND_INTX_DISABLE);
-               else
-                       new = old | PCI_COMMAND_INTX_DISABLE;
-
-               if (old != new)
-                       pci_write_config_word(pdev, PCI_COMMAND, new);
-       }
-
-       return 0;
-}
-
 /**
  * This is the irqcontrol callback to be registered to uio_info.
  * It can be used to disable/enable interrupt from user space processes.
@@ -340,20 +109,35 @@ igbuio_set_interrupt_mask(struct rte_uio_pci_dev *udev, int32_t state)
 static int
 igbuio_pci_irqcontrol(struct uio_info *info, s32 irq_state)
 {
-       unsigned long flags;
-       struct rte_uio_pci_dev *udev = igbuio_get_uio_pci_dev(info);
+       struct rte_uio_pci_dev *udev = info->priv;
        struct pci_dev *pdev = udev->pdev;
 
-       spin_lock_irqsave(&udev->lock, flags);
-       if (!pci_lock(pdev)) {
-               spin_unlock_irqrestore(&udev->lock, flags);
-               return -1;
+#ifdef HAVE_IRQ_DATA
+       struct irq_data *irq = irq_get_irq_data(udev->info.irq);
+#else
+       unsigned int irq = udev->info.irq;
+#endif
+
+       pci_cfg_access_lock(pdev);
+
+       if (udev->mode == RTE_INTR_MODE_MSIX || udev->mode == RTE_INTR_MODE_MSI) {
+#ifdef HAVE_PCI_MSI_MASK_IRQ
+               if (irq_state == 1)
+                       pci_msi_unmask_irq(irq);
+               else
+                       pci_msi_mask_irq(irq);
+#else
+               if (irq_state == 1)
+                       unmask_msi_irq(irq);
+               else
+                       mask_msi_irq(irq);
+#endif
        }
 
-       igbuio_set_interrupt_mask(udev, irq_state);
+       if (udev->mode == RTE_INTR_MODE_LEGACY)
+               pci_intx(pdev, !!irq_state);
 
-       pci_unlock(pdev);
-       spin_unlock_irqrestore(&udev->lock, flags);
+       pci_cfg_access_unlock(pdev);
 
        return 0;
 }
@@ -365,37 +149,46 @@ igbuio_pci_irqcontrol(struct uio_info *info, s32 irq_state)
 static irqreturn_t
 igbuio_pci_irqhandler(int irq, struct uio_info *info)
 {
-       irqreturn_t ret = IRQ_NONE;
-       unsigned long flags;
-       struct rte_uio_pci_dev *udev = igbuio_get_uio_pci_dev(info);
-       struct pci_dev *pdev = udev->pdev;
-       uint32_t cmd_status_dword;
-       uint16_t status;
-
-       spin_lock_irqsave(&udev->lock, flags);
-       /* block userspace PCI config reads/writes */
-       if (!pci_lock(pdev))
-               goto spin_unlock;
-
-       /* for legacy mode, interrupt maybe shared */
-       if (udev->mode == RTE_INTR_MODE_LEGACY) {
-               pci_read_config_dword(pdev, PCI_COMMAND, &cmd_status_dword);
-               status = cmd_status_dword >> 16;
-               /* interrupt is not ours, goes to out */
-               if (!(status & PCI_STATUS_INTERRUPT))
-                       goto done;
-       }
+       struct rte_uio_pci_dev *udev = info->priv;
+
+       /* Legacy mode need to mask in hardware */
+       if (udev->mode == RTE_INTR_MODE_LEGACY &&
+           !pci_check_and_mask_intx(udev->pdev))
+               return IRQ_NONE;
+
+       /* Message signal mode, no share IRQ and automasked */
+       return IRQ_HANDLED;
+}
+
+/**
+ * This gets called while opening uio device file.
+ */
+static int
+igbuio_pci_open(struct uio_info *info, struct inode *inode)
+{
+       struct rte_uio_pci_dev *udev = info->priv;
+       struct pci_dev *dev = udev->pdev;
+
+       pci_reset_function(dev);
+
+       /* set bus master, which was cleared by the reset function */
+       pci_set_master(dev);
+
+       return 0;
+}
+
+static int
+igbuio_pci_release(struct uio_info *info, struct inode *inode)
+{
+       struct rte_uio_pci_dev *udev = info->priv;
+       struct pci_dev *dev = udev->pdev;
+
+       /* stop the device from further DMA */
+       pci_clear_master(dev);
 
-       igbuio_set_interrupt_mask(udev, 0);
-       ret = IRQ_HANDLED;
-done:
-       /* unblock userspace PCI config reads/writes */
-       pci_unlock(pdev);
-spin_unlock:
-       spin_unlock_irqrestore(&udev->lock, flags);
-       pr_info("irq 0x%x %s\n", irq, (ret == IRQ_HANDLED) ? "handled" : "not handled");
+       pci_reset_function(dev);
 
-       return ret;
+       return 0;
 }
 
 #ifdef CONFIG_XEN_DOM0
@@ -406,7 +199,9 @@ igbuio_dom0_mmap_phys(struct uio_info *info, struct vm_area_struct *vma)
 
        idx = (int)vma->vm_pgoff;
        vma->vm_page_prot = pgprot_noncached(vma->vm_page_prot);
+#ifdef HAVE_PTE_MASK_PAGE_IOMAP
        vma->vm_page_prot.pgprot |= _PAGE_IOMAP;
+#endif
 
        return remap_pfn_range(vma,
                        vma->vm_start,
@@ -450,7 +245,7 @@ igbuio_pci_setup_iomem(struct pci_dev *dev, struct uio_info *info,
        unsigned long addr, len;
        void *internal_addr;
 
-       if (sizeof(info->mem) / sizeof(info->mem[0]) <= n)
+       if (n >= ARRAY_SIZE(info->mem))
                return -EINVAL;
 
        addr = pci_resource_start(dev, pci_bar);
@@ -475,7 +270,7 @@ igbuio_pci_setup_ioport(struct pci_dev *dev, struct uio_info *info,
 {
        unsigned long addr, len;
 
-       if (sizeof(info->port) / sizeof(info->port[0]) <= n)
+       if (n >= ARRAY_SIZE(info->port))
                return -EINVAL;
 
        addr = pci_resource_start(dev, pci_bar);
@@ -503,6 +298,94 @@ igbuio_pci_release_iomem(struct uio_info *info)
        }
 }
 
+static int
+igbuio_pci_enable_interrupts(struct rte_uio_pci_dev *udev)
+{
+       int err = 0;
+#ifndef HAVE_ALLOC_IRQ_VECTORS
+       struct msix_entry msix_entry;
+#endif
+
+       switch (igbuio_intr_mode_preferred) {
+       case RTE_INTR_MODE_MSIX:
+               /* Only 1 msi-x vector needed */
+#ifndef HAVE_ALLOC_IRQ_VECTORS
+               msix_entry.entry = 0;
+               if (pci_enable_msix(udev->pdev, &msix_entry, 1) == 0) {
+                       dev_dbg(&udev->pdev->dev, "using MSI-X");
+                       udev->info.irq_flags = IRQF_NO_THREAD;
+                       udev->info.irq = msix_entry.vector;
+                       udev->mode = RTE_INTR_MODE_MSIX;
+                       break;
+               }
+#else
+               if (pci_alloc_irq_vectors(udev->pdev, 1, 1, PCI_IRQ_MSIX) == 1) {
+                       dev_dbg(&udev->pdev->dev, "using MSI-X");
+                       udev->info.irq_flags = IRQF_NO_THREAD;
+                       udev->info.irq = pci_irq_vector(udev->pdev, 0);
+                       udev->mode = RTE_INTR_MODE_MSIX;
+                       break;
+               }
+#endif
+       /* fall back to MSI */
+       case RTE_INTR_MODE_MSI:
+#ifndef HAVE_ALLOC_IRQ_VECTORS
+               if (pci_enable_msi(udev->pdev) == 0) {
+                       dev_dbg(&udev->pdev->dev, "using MSI");
+                       udev->info.irq_flags = IRQF_NO_THREAD;
+                       udev->info.irq = udev->pdev->irq;
+                       udev->mode = RTE_INTR_MODE_MSI;
+                       break;
+               }
+#else
+               if (pci_alloc_irq_vectors(udev->pdev, 1, 1, PCI_IRQ_MSI) == 1) {
+                       dev_dbg(&udev->pdev->dev, "using MSI");
+                       udev->info.irq_flags = IRQF_NO_THREAD;
+                       udev->info.irq = pci_irq_vector(udev->pdev, 0);
+                       udev->mode = RTE_INTR_MODE_MSI;
+                       break;
+               }
+#endif
+       /* fall back to INTX */
+       case RTE_INTR_MODE_LEGACY:
+               if (pci_intx_mask_supported(udev->pdev)) {
+                       dev_dbg(&udev->pdev->dev, "using INTX");
+                       udev->info.irq_flags = IRQF_SHARED | IRQF_NO_THREAD;
+                       udev->info.irq = udev->pdev->irq;
+                       udev->mode = RTE_INTR_MODE_LEGACY;
+                       break;
+               }
+               dev_notice(&udev->pdev->dev, "PCI INTX mask not supported\n");
+       /* fall back to no IRQ */
+       case RTE_INTR_MODE_NONE:
+               udev->mode = RTE_INTR_MODE_NONE;
+               udev->info.irq = UIO_IRQ_NONE;
+               break;
+
+       default:
+               dev_err(&udev->pdev->dev, "invalid IRQ mode %u",
+                       igbuio_intr_mode_preferred);
+               err = -EINVAL;
+       }
+
+       return err;
+}
+
+static void
+igbuio_pci_disable_interrupts(struct rte_uio_pci_dev *udev)
+{
+#ifndef HAVE_ALLOC_IRQ_VECTORS
+       if (udev->mode == RTE_INTR_MODE_MSIX)
+               pci_disable_msix(udev->pdev);
+       if (udev->mode == RTE_INTR_MODE_MSI)
+               pci_disable_msi(udev->pdev);
+#else
+       if (udev->mode == RTE_INTR_MODE_MSIX ||
+           udev->mode == RTE_INTR_MODE_MSI)
+               pci_free_irq_vectors(udev->pdev);
+#endif
+}
+
 static int
 igbuio_setup_bars(struct pci_dev *dev, struct uio_info *info)
 {
@@ -520,7 +403,7 @@ igbuio_setup_bars(struct pci_dev *dev, struct uio_info *info)
        iom = 0;
        iop = 0;
 
-       for (i = 0; i != sizeof(bar_names) / sizeof(bar_names[0]); i++) {
+       for (i = 0; i < ARRAY_SIZE(bar_names); i++) {
                if (pci_resource_len(dev, i) != 0 &&
                                pci_resource_start(dev, i) != 0) {
                        flags = pci_resource_flags(dev, i);
@@ -540,7 +423,7 @@ igbuio_setup_bars(struct pci_dev *dev, struct uio_info *info)
                }
        }
 
-       return (iom != 0) ? ret : -ENOENT;
+       return (iom != 0 || iop != 0) ? ret : -ENOENT;
 }
 
 #if LINUX_VERSION_CODE < KERNEL_VERSION(3, 8, 0)
@@ -551,6 +434,8 @@ static int
 igbuio_pci_probe(struct pci_dev *dev, const struct pci_device_id *id)
 {
        struct rte_uio_pci_dev *udev;
+       dma_addr_t map_dma_addr;
+       void *map_addr;
        int err;
 
        udev = kzalloc(sizeof(struct rte_uio_pci_dev), GFP_KERNEL);
@@ -567,16 +452,6 @@ igbuio_pci_probe(struct pci_dev *dev, const struct pci_device_id *id)
                goto fail_free;
        }
 
-       /*
-        * reserve device's PCI memory regions for use by this
-        * module
-        */
-       err = pci_request_regions(dev, "igb_uio");
-       if (err != 0) {
-               dev_err(&dev->dev, "Cannot request regions\n");
-               goto fail_disable;
-       }
-
        /* enable bus mastering on the device */
        pci_set_master(dev);
 
@@ -603,6 +478,8 @@ igbuio_pci_probe(struct pci_dev *dev, const struct pci_device_id *id)
        udev->info.version = "0.1";
        udev->info.handler = igbuio_pci_irqhandler;
        udev->info.irqcontrol = igbuio_pci_irqcontrol;
+       udev->info.open = igbuio_pci_open;
+       udev->info.release = igbuio_pci_release;
 #ifdef CONFIG_XEN_DOM0
        /* check if the driver run on Xen Dom0 */
        if (xen_initial_domain())
@@ -610,63 +487,54 @@ igbuio_pci_probe(struct pci_dev *dev, const struct pci_device_id *id)
 #endif
        udev->info.priv = udev;
        udev->pdev = dev;
-       udev->mode = RTE_INTR_MODE_LEGACY;
-       spin_lock_init(&udev->lock);
 
-       /* check if it need to try msix first */
-       if (igbuio_intr_mode_preferred == RTE_INTR_MODE_MSIX) {
-               int vector;
-
-               for (vector = 0; vector < IGBUIO_NUM_MSI_VECTORS; vector ++)
-                       udev->msix_entries[vector].entry = vector;
-
-               if (pci_enable_msix(udev->pdev, udev->msix_entries, IGBUIO_NUM_MSI_VECTORS) == 0) {
-                       udev->mode = RTE_INTR_MODE_MSIX;
-               }
-               else {
-                       pci_disable_msix(udev->pdev);
-                       pr_info("fail to enable pci msix, or not enough msix entries\n");
-               }
-       }
-       switch (udev->mode) {
-       case RTE_INTR_MODE_MSIX:
-               udev->info.irq_flags = 0;
-               udev->info.irq = udev->msix_entries[0].vector;
-               break;
-       case RTE_INTR_MODE_MSI:
-               break;
-       case RTE_INTR_MODE_LEGACY:
-               udev->info.irq_flags = IRQF_SHARED;
-               udev->info.irq = dev->irq;
-               break;
-       default:
-               break;
-       }
-
-       pci_set_drvdata(dev, udev);
-       igbuio_pci_irqcontrol(&udev->info, 0);
+       err = igbuio_pci_enable_interrupts(udev);
+       if (err != 0)
+               goto fail_release_iomem;
 
        err = sysfs_create_group(&dev->dev.kobj, &dev_attr_grp);
        if (err != 0)
-               goto fail_release_iomem;
+               goto fail_disable_interrupts;
 
        /* register uio driver */
        err = uio_register_device(&dev->dev, &udev->info);
        if (err != 0)
                goto fail_remove_group;
 
-       pr_info("uio device registered with irq %lx\n", udev->info.irq);
+       pci_set_drvdata(dev, udev);
+
+       dev_info(&dev->dev, "uio device registered with irq %lx\n",
+                udev->info.irq);
+
+       /*
+        * Doing a harmless dma mapping for attaching the device to
+        * the iommu identity mapping if kernel boots with iommu=pt.
+        * Note this is not a problem if no IOMMU at all.
+        */
+       map_addr = dma_alloc_coherent(&dev->dev, 1024, &map_dma_addr,
+                       GFP_KERNEL);
+       if (map_addr)
+               memset(map_addr, 0, 1024);
+
+       if (!map_addr)
+               dev_info(&dev->dev, "dma mapping failed\n");
+       else {
+               dev_info(&dev->dev, "mapping 1K dma=%#llx host=%p\n",
+                        (unsigned long long)map_dma_addr, map_addr);
+
+               dma_free_coherent(&dev->dev, 1024, map_addr, map_dma_addr);
+               dev_info(&dev->dev, "unmapping 1K dma=%#llx host=%p\n",
+                        (unsigned long long)map_dma_addr, map_addr);
+       }
 
        return 0;
 
 fail_remove_group:
        sysfs_remove_group(&dev->dev.kobj, &dev_attr_grp);
+fail_disable_interrupts:
+       igbuio_pci_disable_interrupts(udev);
 fail_release_iomem:
        igbuio_pci_release_iomem(&udev->info);
-       if (udev->mode == RTE_INTR_MODE_MSIX)
-               pci_disable_msix(udev->pdev);
-       pci_release_regions(dev);
-fail_disable:
        pci_disable_device(dev);
 fail_free:
        kfree(udev);
@@ -677,23 +545,15 @@ fail_free:
 static void
 igbuio_pci_remove(struct pci_dev *dev)
 {
-       struct uio_info *info = pci_get_drvdata(dev);
-
-       if (info->priv == NULL) {
-               pr_notice("Not igbuio device\n");
-               return;
-       }
+       struct rte_uio_pci_dev *udev = pci_get_drvdata(dev);
 
        sysfs_remove_group(&dev->dev.kobj, &dev_attr_grp);
-       uio_unregister_device(info);
-       igbuio_pci_release_iomem(info);
-       if (((struct rte_uio_pci_dev *)info->priv)->mode ==
-                       RTE_INTR_MODE_MSIX)
-               pci_disable_msix(dev);
-       pci_release_regions(dev);
+       uio_unregister_device(&udev->info);
+       igbuio_pci_disable_interrupts(udev);
+       igbuio_pci_release_iomem(&udev->info);
        pci_disable_device(dev);
        pci_set_drvdata(dev, NULL);
-       kfree(info);
+       kfree(udev);
 }
 
 static int
@@ -707,6 +567,9 @@ igbuio_config_intr_mode(char *intr_str)
        if (!strcmp(intr_str, RTE_INTR_MODE_MSIX_NAME)) {
                igbuio_intr_mode_preferred = RTE_INTR_MODE_MSIX;
                pr_info("Use MSIX interrupt\n");
+       } else if (!strcmp(intr_str, RTE_INTR_MODE_MSI_NAME)) {
+               igbuio_intr_mode_preferred = RTE_INTR_MODE_MSI;
+               pr_info("Use MSI interrupt\n");
        } else if (!strcmp(intr_str, RTE_INTR_MODE_LEGACY_NAME)) {
                igbuio_intr_mode_preferred = RTE_INTR_MODE_LEGACY;
                pr_info("Use legacy interrupt\n");
@@ -746,10 +609,11 @@ igbuio_pci_exit_module(void)
 module_init(igbuio_pci_init_module);
 module_exit(igbuio_pci_exit_module);
 
-module_param(intr_mode, charp, S_IRUGO | S_IWUSR);
+module_param(intr_mode, charp, S_IRUGO);
 MODULE_PARM_DESC(intr_mode,
 "igb_uio interrupt mode (default=msix):\n"
 "    " RTE_INTR_MODE_MSIX_NAME "       Use MSIX interrupt\n"
+"    " RTE_INTR_MODE_MSI_NAME "        Use MSI interrupt\n"
 "    " RTE_INTR_MODE_LEGACY_NAME "     Use Legacy interrupt\n"
 "\n");