@@ -21,6 +21,15 @@ static void pci_pri_init(struct pci_dev *pdev)
#ifdef CONFIG_PCI_PRI
int pos;
+ /*
+ * As per PCIe r4.0, sec 9.3.7.11, only PF is permitted to
+ * implement PRI and all associated VFs can only use it.
+ * Since PF already initialized the PRI parameters there is
+ * no need to proceed further.
+ */
+ if (pdev->is_virtfn)
+ return;
+
pos = pci_find_ext_capability(pdev, PCI_EXT_CAP_ID_PRI);
if (!pos)
return;
@@ -208,31 +217,55 @@ EXPORT_SYMBOL_GPL(pci_ats_page_aligned);
*/
int pci_enable_pri(struct pci_dev *pdev, u32 reqs)
{
- u16 control, status;
+ u16 status;
u32 max_requests;
+ int ret = 0;
+ struct pci_dev *pf = pci_physfn(pdev);
if (WARN_ON(pdev->pri_enabled))
return -EBUSY;
- if (!pdev->pri_cap)
+ if (!pf->pri_cap)
return -EINVAL;
- pci_read_config_word(pdev, pdev->pri_cap + PCI_PRI_STATUS, &status);
- if (!(status & PCI_PRI_STATUS_STOPPED))
- return -EBUSY;
+ pci_physfn_reslock(pdev);
+
+ if (pdev->is_virtfn && pf->pri_enabled)
+ goto update_status;
- pci_read_config_dword(pdev, pdev->pri_cap + PCI_PRI_MAX_REQ,
- &max_requests);
+ /*
+ * Before updating PRI registers, make sure there is no
+ * outstanding PRI requests.
+ */
+ pci_read_config_word(pf, pf->pri_cap + PCI_PRI_STATUS, &status);
+ if (!(status & PCI_PRI_STATUS_STOPPED)) {
+ ret = -EBUSY;
+ goto done;
+ }
+
+ pci_read_config_dword(pf, pf->pri_cap + PCI_PRI_MAX_REQ, &max_requests);
reqs = min(max_requests, reqs);
- pdev->pri_reqs_alloc = reqs;
- pci_write_config_dword(pdev, pdev->pri_cap + PCI_PRI_ALLOC_REQ, reqs);
+ pf->pri_reqs_alloc = reqs;
+ pci_write_config_dword(pf, pf->pri_cap + PCI_PRI_ALLOC_REQ, reqs);
- control = PCI_PRI_CTRL_ENABLE;
- pci_write_config_word(pdev, pdev->pri_cap + PCI_PRI_CTRL, control);
+ pci_write_config_word(pf, pf->pri_cap + PCI_PRI_CTRL,
+ PCI_PRI_CTRL_ENABLE);
- pdev->pri_enabled = 1;
+ /*
+ * If PRI is not already enabled in PF, increment the PF
+ * pri_ref_cnt to track the usage of PRI interface.
+ */
+ if (pdev->is_virtfn && !pf->pri_enabled) {
+ atomic_inc(&pf->pri_ref_cnt);
+ pf->pri_enabled = 1;
+ }
- return 0;
+update_status:
+ atomic_inc(&pf->pri_ref_cnt);
+ pdev->pri_enabled = 1;
+done:
+ pci_physfn_resunlock(pdev);
+ return ret;
}
EXPORT_SYMBOL_GPL(pci_enable_pri);
@@ -245,18 +278,32 @@ EXPORT_SYMBOL_GPL(pci_enable_pri);
void pci_disable_pri(struct pci_dev *pdev)
{
u16 control;
+ struct pci_dev *pf = pci_physfn(pdev);
if (WARN_ON(!pdev->pri_enabled))
return;
- if (!pdev->pri_cap)
+ if (!pf->pri_cap)
return;
- pci_read_config_word(pdev, pdev->pri_cap + PCI_PRI_CTRL, &control);
+ pci_physfn_reslock(pdev);
+
+ atomic_dec(&pf->pri_ref_cnt);
+
+ /*
+ * If pri_ref_cnt is not zero, then don't modify hardware
+ * registers.
+ */
+ if (atomic_read(&pf->pri_ref_cnt))
+ goto done;
+
+ pci_read_config_word(pf, pf->pri_cap + PCI_PRI_CTRL, &control);
control &= ~PCI_PRI_CTRL_ENABLE;
- pci_write_config_word(pdev, pdev->pri_cap + PCI_PRI_CTRL, control);
+ pci_write_config_word(pf, pf->pri_cap + PCI_PRI_CTRL, control);
+done:
pdev->pri_enabled = 0;
+ pci_physfn_resunlock(pdev);
}
EXPORT_SYMBOL_GPL(pci_disable_pri);
@@ -266,17 +313,29 @@ EXPORT_SYMBOL_GPL(pci_disable_pri);
*/
void pci_restore_pri_state(struct pci_dev *pdev)
{
- u16 control = PCI_PRI_CTRL_ENABLE;
- u32 reqs = pdev->pri_reqs_alloc;
+ u16 control;
+ struct pci_dev *pf = pci_physfn(pdev);
if (!pdev->pri_enabled)
return;
- if (!pdev->pri_cap)
+ if (!pf->pri_cap)
return;
- pci_write_config_dword(pdev, pdev->pri_cap + PCI_PRI_ALLOC_REQ, reqs);
- pci_write_config_word(pdev, pdev->pri_cap + PCI_PRI_CTRL, control);
+ pci_physfn_reslock(pdev);
+
+ /* If PRI is already enabled by other VF's or PF, return */
+ pci_read_config_word(pf, pf->pri_cap + PCI_PRI_CTRL, &control);
+ if (control & PCI_PRI_CTRL_ENABLE)
+ goto done;
+
+ pci_write_config_dword(pf, pf->pri_cap + PCI_PRI_ALLOC_REQ,
+ pf->pri_reqs_alloc);
+ pci_write_config_word(pf, pf->pri_cap + PCI_PRI_CTRL,
+ PCI_PRI_CTRL_ENABLE);
+
+done:
+ pci_physfn_resunlock(pdev);
}
EXPORT_SYMBOL_GPL(pci_restore_pri_state);
@@ -289,17 +348,24 @@ EXPORT_SYMBOL_GPL(pci_restore_pri_state);
*/
int pci_reset_pri(struct pci_dev *pdev)
{
- u16 control;
+ struct pci_dev *pf = pci_physfn(pdev);
if (WARN_ON(pdev->pri_enabled))
return -EBUSY;
- if (!pdev->pri_cap)
+ if (!pf->pri_cap)
return -EINVAL;
- control = PCI_PRI_CTRL_RESET;
- pci_write_config_word(pdev, pdev->pri_cap + PCI_PRI_CTRL, control);
+ pci_physfn_reslock(pdev);
+
+ /* If PRI is already enabled in PF, skip reset and return */
+ if (pf->pri_enabled)
+ goto done;
+ pci_write_config_word(pf, pf->pri_cap + PCI_PRI_CTRL,
+ PCI_PRI_CTRL_RESET);
+done:
+ pci_physfn_resunlock(pdev);
return 0;
}
EXPORT_SYMBOL_GPL(pci_reset_pri);
@@ -427,11 +493,12 @@ EXPORT_SYMBOL_GPL(pci_pasid_features);
int pci_prg_resp_pasid_required(struct pci_dev *pdev)
{
u16 status;
+ struct pci_dev *pf = pci_physfn(pdev);
- if (!pdev->pri_cap)
+ if (!pf->pri_cap)
return 0;
- pci_read_config_word(pdev, pdev->pri_cap + PCI_PRI_STATUS, &status);
+ pci_read_config_word(pf, pf->pri_cap + PCI_PRI_STATUS, &status);
if (status & PCI_PRI_STATUS_PASID)
return 1;
@@ -457,6 +457,7 @@ struct pci_dev {
#ifdef CONFIG_PCI_PRI
u16 pri_cap; /* PRI Capability offset */
u32 pri_reqs_alloc; /* Number of PRI requests allocated */
+ atomic_t pri_ref_cnt; /* Number of PF/VF PRI users */
#endif
#ifdef CONFIG_PCI_PASID
u16 pasid_cap; /* PASID Capability offset */