|
@@ -231,14 +231,20 @@ static int proc_bus_pci_mmap(struct file *file, struct vm_area_struct *vma)
|
|
|
{
|
|
|
struct pci_dev *dev = PDE_DATA(file_inode(file));
|
|
|
struct pci_filp_private *fpriv = file->private_data;
|
|
|
- int i, ret, write_combine;
|
|
|
+ int i, ret, write_combine, res_bit;
|
|
|
|
|
|
if (!capable(CAP_SYS_RAWIO))
|
|
|
return -EPERM;
|
|
|
|
|
|
+ if (fpriv->mmap_state == pci_mmap_io)
|
|
|
+ res_bit = IORESOURCE_IO;
|
|
|
+ else
|
|
|
+ res_bit = IORESOURCE_MEM;
|
|
|
+
|
|
|
/* Make sure the caller is mapping a real resource for this device */
|
|
|
for (i = 0; i < PCI_ROM_RESOURCE; i++) {
|
|
|
- if (pci_mmap_fits(dev, i, vma, PCI_MMAP_PROCFS))
|
|
|
+ if (dev->resource[i].flags & res_bit &&
|
|
|
+ pci_mmap_fits(dev, i, vma, PCI_MMAP_PROCFS))
|
|
|
break;
|
|
|
}
|
|
|
|