@@ -154,10 +154,33 @@ int pci_enable_pri(struct pci_dev *pdev, u32 reqs)
u16 control, status;
u32 max_requests;
int pos;
+ struct pci_dev *pf;
if (WARN_ON(pdev->pri_enabled))
return -EBUSY;
+ /* If PRI Capability is invalid, return error */
+ if (pdev->is_virtfn || pdev->is_physfn) {
+ if (pdev->invalid_cap & PCI_IOV_INVALID_PRI)
+ return -EINVAL;
+ }
+
+ if (pdev->is_virtfn) {
+ pf = pci_physfn(pdev);
+
+ /* If VF config does not match with PF, return error */
+ if (!pf->pri_enabled)
+ return -EINVAL;
+
+ pdev->pri_reqs_alloc = pf->pri_reqs_alloc;
+ pdev->pri_enabled = 1;
+
+ /* Increment PF PRI refcount */
+ atomic_inc(&pf->pri_ref_cnt);
+
+ return 0;
+ }
+
pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_PRI);
if (!pos)
return -EINVAL;
@@ -175,7 +198,6 @@ int pci_enable_pri(struct pci_dev *pdev, u32 reqs)
pci_write_config_word(pdev, pos + PCI_PRI_CTRL, control);
pdev->pri_enabled = 1;
-
return 0;
}
EXPORT_SYMBOL_GPL(pci_enable_pri);
@@ -190,10 +212,27 @@ void pci_disable_pri(struct pci_dev *pdev)
{
u16 control;
int pos;
+ struct pci_dev *pf;
if (WARN_ON(!pdev->pri_enabled))
return;
+ /* All VFs should be disabled before disabling PF */
+ if (atomic_read(&pdev->pri_ref_cnt))
+ return;
+
+ if (pdev->is_virtfn) {
+ /* Since VF shares PRI with PF, use PF config. */
+ pf = pci_physfn(pdev);
+
+ /* Decrement PF PRI refcount */
+ atomic_dec(&pf->pri_ref_cnt);
+
+ pdev->pri_enabled = 0;
+
+ return;
+ }
+
pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_PRI);
if (!pos)
return;
@@ -219,6 +258,9 @@ void pci_restore_pri_state(struct pci_dev *pdev)
if (!pdev->pri_enabled)
return;
+ if (pdev->is_virtfn)
+ return;
+
pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_PRI);
if (!pos)
return;
@@ -243,6 +285,9 @@ int pci_reset_pri(struct pci_dev *pdev)
if (WARN_ON(pdev->pri_enabled))
return -EBUSY;
+ if (pdev->is_virtfn)
+ return 0;
+
pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_PRI);
if (!pos)
return -EINVAL;
@@ -448,6 +448,7 @@ struct pci_dev {
#endif
#ifdef CONFIG_PCI_PRI
u32 pri_reqs_alloc; /* Number of PRI requests allocated */
+ atomic_t pri_ref_cnt; /* Number of VFs with PRI enabled */
#endif
#ifdef CONFIG_PCI_PASID
u16 pasid_features;