+cc linux-pci as I forgot to 

On 2020/9/7 19:16, Yicong Yang wrote:
> Previosly we use pci_probe_reset_function() to probe whehter a function
> can be reset and use __pci_reset_function_locked() to perform a function
> reset. These two functions have lots of common lines.
>
> Factor the two functions and reduce the redundancy.
>
> Signed-off-by: Yicong Yang <yangyic...@hisilicon.com>
> ---
>  drivers/pci/pci.c   | 61 
> ++++++++++++++++-------------------------------------
>  drivers/pci/pci.h   |  2 +-
>  drivers/pci/probe.c |  2 +-
>  3 files changed, 20 insertions(+), 45 deletions(-)
>
> diff --git a/drivers/pci/pci.c b/drivers/pci/pci.c
> index e39c549..e3e5f0f 100644
> --- a/drivers/pci/pci.c
> +++ b/drivers/pci/pci.c
> @@ -5006,9 +5006,11 @@ static void pci_dev_restore(struct pci_dev *dev)
>  }
>  
>  /**
> - * __pci_reset_function_locked - reset a PCI device function while holding
> - * the @dev mutex lock.
> + * pci_probe_reset_function - check whether the device can be safely reset
> + *                            or reset a PCI device function while holding
> + *                            the @dev mutex lock.
>   * @dev: PCI device to reset
> + * @probe: Probe or not whether the device can be reset.
>   *
>   * Some devices allow an individual function to be reset without affecting
>   * other functions in the same device.  The PCI device must be responsive
> @@ -5022,10 +5024,10 @@ static void pci_dev_restore(struct pci_dev *dev)
>   * device including MSI, bus mastering, BARs, decoding IO and memory spaces,
>   * etc.
>   *
> - * Returns 0 if the device function was successfully reset or negative if the
> - * device doesn't support resetting a single function.
> + * Returns 0 if the device function can be reset or was successfully reset.
> + * negative if the device doesn't support resetting a single function.
>   */
> -int __pci_reset_function_locked(struct pci_dev *dev)
> +int pci_probe_reset_function(struct pci_dev *dev, int probe)
>  {
>       int rc;
>  
> @@ -5039,61 +5041,34 @@ int __pci_reset_function_locked(struct pci_dev *dev)
>        * other error, we're also finished: this indicates that further
>        * reset mechanisms might be broken on the device.
>        */
> -     rc = pci_dev_specific_reset(dev, 0);
> +     rc = pci_dev_specific_reset(dev, probe);
>       if (rc != -ENOTTY)
>               return rc;
>       if (pcie_has_flr(dev)) {
> +             if (probe)
> +                     return 0;
>               rc = pcie_flr(dev);
>               if (rc != -ENOTTY)
>                       return rc;
>       }
> -     rc = pci_af_flr(dev, 0);
> +     rc = pci_af_flr(dev, probe);
>       if (rc != -ENOTTY)
>               return rc;
> -     rc = pci_pm_reset(dev, 0);
> +     rc = pci_pm_reset(dev, probe);
>       if (rc != -ENOTTY)
>               return rc;
> -     rc = pci_dev_reset_slot_function(dev, 0);
> +     rc = pci_dev_reset_slot_function(dev, probe);
>       if (rc != -ENOTTY)
>               return rc;
> -     return pci_parent_bus_reset(dev, 0);
> +
> +     return pci_parent_bus_reset(dev, probe);
>  }
> -EXPORT_SYMBOL_GPL(__pci_reset_function_locked);
>  
> -/**
> - * pci_probe_reset_function - check whether the device can be safely reset
> - * @dev: PCI device to reset
> - *
> - * Some devices allow an individual function to be reset without affecting
> - * other functions in the same device.  The PCI device must be responsive
> - * to PCI config space in order to use this function.
> - *
> - * Returns 0 if the device function can be reset or negative if the
> - * device doesn't support resetting a single function.
> - */
> -int pci_probe_reset_function(struct pci_dev *dev)
> +int __pci_reset_function_locked(struct pci_dev *dev)
>  {
> -     int rc;
> -
> -     might_sleep();
> -
> -     rc = pci_dev_specific_reset(dev, 1);
> -     if (rc != -ENOTTY)
> -             return rc;
> -     if (pcie_has_flr(dev))
> -             return 0;
> -     rc = pci_af_flr(dev, 1);
> -     if (rc != -ENOTTY)
> -             return rc;
> -     rc = pci_pm_reset(dev, 1);
> -     if (rc != -ENOTTY)
> -             return rc;
> -     rc = pci_dev_reset_slot_function(dev, 1);
> -     if (rc != -ENOTTY)
> -             return rc;
> -
> -     return pci_parent_bus_reset(dev, 1);
> +     return pci_probe_reset_function(dev, 0);
>  }
> +EXPORT_SYMBOL_GPL(__pci_reset_function_locked);
>  
>  /**
>   * pci_reset_function - quiesce and reset a PCI device function
> diff --git a/drivers/pci/pci.h b/drivers/pci/pci.h
> index fa12f7c..73740dd 100644
> --- a/drivers/pci/pci.h
> +++ b/drivers/pci/pci.h
> @@ -39,7 +39,7 @@ enum pci_mmap_api {
>  int pci_mmap_fits(struct pci_dev *pdev, int resno, struct vm_area_struct 
> *vmai,
>                 enum pci_mmap_api mmap_api);
>  
> -int pci_probe_reset_function(struct pci_dev *dev);
> +int pci_probe_reset_function(struct pci_dev *dev, int probe);
>  int pci_bridge_secondary_bus_reset(struct pci_dev *dev);
>  int pci_bus_error_reset(struct pci_dev *dev);
>  
> diff --git a/drivers/pci/probe.c b/drivers/pci/probe.c
> index 03d3712..793cc8a 100644
> --- a/drivers/pci/probe.c
> +++ b/drivers/pci/probe.c
> @@ -2403,7 +2403,7 @@ static void pci_init_capabilities(struct pci_dev *dev)
>  
>       pcie_report_downtraining(dev);
>  
> -     if (pci_probe_reset_function(dev) == 0)
> +     if (pci_probe_reset_function(dev, 1) == 0)
>               dev->reset_fn = 1;
>  }
>  

Reply via email to