PCI: endpoint: Introduce pci_epc_function_is_valid()

Introduce the epc core helper function pci_epc_function_is_valid() to
verify that an epc pointer, a physical function number and a virtual
function number are all valid. This avoids repeating the code pattern:

if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
	return err;

if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
	return err;

in many functions of the endpoint controller core code.

Signed-off-by: Damien Le Moal <dlemoal@kernel.org>
Reviewed-by: Manivannan Sadhasivam <manivannan.sadhasivam@linaro.org>
Reviewed-by: Niklas Cassel <cassel@kernel.org>
Link: https://lore.kernel.org/r/20241012113246.95634-2-dlemoal@kernel.org
Signed-off-by: Manivannan Sadhasivam <manivannan.sadhasivam@linaro.org>

diff --git a/drivers/pci/endpoint/pci-epc-core.c b/drivers/pci/endpoint/pci-epc-core.c
index 17f00710925508e60fbd21116af5b424abdcd3e7..b854f1bab26f8956096ab5359de4410a6a32bcc8 100644 (file)
--- a/drivers/pci/endpoint/pci-epc-core.c
+++ b/drivers/pci/endpoint/pci-epc-core.c
@@ -128,6 +128,18 @@ enum pci_barno pci_epc_get_next_free_bar(const struct pci_epc_features
 }
 EXPORT_SYMBOL_GPL(pci_epc_get_next_free_bar);
 
+static bool pci_epc_function_is_valid(struct pci_epc *epc,
+                                     u8 func_no, u8 vfunc_no)
+{
+       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
+               return false;
+
+       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+               return false;
+
+       return true;
+}
+
 /**
  * pci_epc_get_features() - get the features supported by EPC
  * @epc: the features supported by *this* EPC device will be returned
@@ -145,10 +157,7 @@ const struct pci_epc_features *pci_epc_get_features(struct pci_epc *epc,
 {
        const struct pci_epc_features *epc_features;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
-               return NULL;
-
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return NULL;
 
        if (!epc->ops->get_features)
@@ -218,10 +227,7 @@ int pci_epc_raise_irq(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
 {
        int ret;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
-               return -EINVAL;
-
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return -EINVAL;
 
        if (!epc->ops->raise_irq)
@@ -262,10 +268,7 @@ int pci_epc_map_msi_irq(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
 {
        int ret;
 
-       if (IS_ERR_OR_NULL(epc))
-               return -EINVAL;
-
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return -EINVAL;
 
        if (!epc->ops->map_msi_irq)
@@ -293,10 +296,7 @@ int pci_epc_get_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        int interrupt;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
-               return 0;
-
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return 0;
 
        if (!epc->ops->get_msi)
@@ -329,11 +329,10 @@ int pci_epc_set_msi(struct pci_epc *epc, u8 func_no, u8 vfunc_no, u8 interrupts)
        int ret;
        u8 encode_int;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions ||
-           interrupts < 1 || interrupts > 32)
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return -EINVAL;
 
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (interrupts < 1 || interrupts > 32)
                return -EINVAL;
 
        if (!epc->ops->set_msi)
@@ -361,10 +360,7 @@ int pci_epc_get_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no)
 {
        int interrupt;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
-               return 0;
-
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return 0;
 
        if (!epc->ops->get_msix)
@@ -397,11 +393,10 @@ int pci_epc_set_msix(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
 {
        int ret;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions ||
-           interrupts < 1 || interrupts > 2048)
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return -EINVAL;
 
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (interrupts < 1 || interrupts > 2048)
                return -EINVAL;
 
        if (!epc->ops->set_msix)
@@ -428,10 +423,7 @@ EXPORT_SYMBOL_GPL(pci_epc_set_msix);
 void pci_epc_unmap_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                        phys_addr_t phys_addr)
 {
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
-               return;
-
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return;
 
        if (!epc->ops->unmap_addr)
@@ -459,10 +451,7 @@ int pci_epc_map_addr(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
 {
        int ret;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
-               return -EINVAL;
-
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return -EINVAL;
 
        if (!epc->ops->map_addr)
@@ -489,12 +478,11 @@ EXPORT_SYMBOL_GPL(pci_epc_map_addr);
 void pci_epc_clear_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
                       struct pci_epf_bar *epf_bar)
 {
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions ||
-           (epf_bar->barno == BAR_5 &&
-            epf_bar->flags & PCI_BASE_ADDRESS_MEM_TYPE_64))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return;
 
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (epf_bar->barno == BAR_5 &&
+           epf_bar->flags & PCI_BASE_ADDRESS_MEM_TYPE_64)
                return;
 
        if (!epc->ops->clear_bar)
@@ -521,18 +509,16 @@ int pci_epc_set_bar(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
        int ret;
        int flags = epf_bar->flags;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions ||
-           (epf_bar->barno == BAR_5 &&
-            flags & PCI_BASE_ADDRESS_MEM_TYPE_64) ||
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
+               return -EINVAL;
+
+       if ((epf_bar->barno == BAR_5 && flags & PCI_BASE_ADDRESS_MEM_TYPE_64) ||
            (flags & PCI_BASE_ADDRESS_SPACE_IO &&
             flags & PCI_BASE_ADDRESS_IO_MASK) ||
            (upper_32_bits(epf_bar->size) &&
             !(flags & PCI_BASE_ADDRESS_MEM_TYPE_64)))
                return -EINVAL;
 
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
-               return -EINVAL;
-
        if (!epc->ops->set_bar)
                return 0;
 
@@ -561,10 +547,7 @@ int pci_epc_write_header(struct pci_epc *epc, u8 func_no, u8 vfunc_no,
 {
        int ret;
 
-       if (IS_ERR_OR_NULL(epc) || func_no >= epc->max_functions)
-               return -EINVAL;
-
-       if (vfunc_no > 0 && (!epc->max_vfs || vfunc_no > epc->max_vfs[func_no]))
+       if (!pci_epc_function_is_valid(epc, func_no, vfunc_no))
                return -EINVAL;
 
        /* Only Virtual Function #1 has deviceID */