IB/core: Add inline function to validate port

Signed-off-by: Yuval Shaia <yuval.shaia@oracle.com>
Reviewed-by: Leon Romanovsky <leonro@mellanox.com>
Signed-off-by: Doug Ledford <dledford@redhat.com>

5 files changed, 20 insertions, 18 deletions

diff --git a/drivers/infiniband/core/cache.c b/drivers/infiniband/core/cache.c
index 2e52021aa999..b1371eb9f46c 100644
--- a/drivers/infiniband/core/cache.c
+++ b/drivers/infiniband/core/cache.c
@@ -504,8 +504,7 @@ int ib_find_cached_gid_by_port(struct ib_device *ib_dev,
         struct ib_gid_attr val = {.ndev = ndev, .gid_type = gid_type};
         unsigned long flags;
 
-        if (port < rdma_start_port(ib_dev) ||
-            port > rdma_end_port(ib_dev))
+        if (!rdma_is_port_valid(ib_dev, port))
                 return -ENOENT;
 
         table = ib_dev->cache.ports[port - rdma_start_port(ib_dev)].gid;
@@ -562,8 +561,7 @@ static int ib_cache_gid_find_by_filter(struct ib_device *ib_dev,
         bool found = false;
 
 
-        if (port < rdma_start_port(ib_dev) ||
-            port > rdma_end_port(ib_dev) ||
+        if (!rdma_is_port_valid(ib_dev, port) ||
             !rdma_protocol_roce(ib_dev, port))
                 return -EPROTONOSUPPORT;
 
@@ -845,7 +843,7 @@ int ib_get_cached_gid(struct ib_device *device,
         unsigned long flags;
         struct ib_gid_table *table;
 
-        if (port_num < rdma_start_port(device) || port_num > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, port_num))
                 return -EINVAL;
 
         table = device->cache.ports[port_num - rdma_start_port(device)].gid;
@@ -895,7 +893,7 @@ int ib_get_cached_pkey(struct ib_device *device,
         unsigned long flags;
         int ret = 0;
 
-        if (port_num < rdma_start_port(device) || port_num > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, port_num))
                 return -EINVAL;
 
         read_lock_irqsave(&device->cache.lock, flags);
@@ -924,7 +922,7 @@ int ib_find_cached_pkey(struct ib_device *device,
         int ret = -ENOENT;
         int partial_ix = -1;
 
-        if (port_num < rdma_start_port(device) || port_num > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, port_num))
                 return -EINVAL;
 
         read_lock_irqsave(&device->cache.lock, flags);
@@ -964,7 +962,7 @@ int ib_find_exact_cached_pkey(struct ib_device *device,
         int i;
         int ret = -ENOENT;
 
-        if (port_num < rdma_start_port(device) || port_num > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, port_num))
                 return -EINVAL;
 
         read_lock_irqsave(&device->cache.lock, flags);
@@ -993,7 +991,7 @@ int ib_get_cached_lmc(struct ib_device *device,
         unsigned long flags;
         int ret = 0;
 
-        if (port_num < rdma_start_port(device) || port_num > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, port_num))
                 return -EINVAL;
 
         read_lock_irqsave(&device->cache.lock, flags);
@@ -1038,7 +1036,7 @@ static void ib_cache_update(struct ib_device *device,
         bool                       use_roce_gid_table =
                                         rdma_cap_roce_gid_table(device, port);
 
-        if (port < rdma_start_port(device) || port > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, port))
                 return;
 
         table = device->cache.ports[port - rdma_start_port(device)].gid;
diff --git a/drivers/infiniband/core/cma.c b/drivers/infiniband/core/cma.c
index 7cb01a9463da..9c93e2fa969b 100644
--- a/drivers/infiniband/core/cma.c
+++ b/drivers/infiniband/core/cma.c
@@ -269,8 +269,7 @@ struct cma_device *cma_enum_devices_by_ibdev(cma_device_filter	filter,
 int cma_get_default_gid_type(struct cma_device *cma_dev,
                              unsigned int port)
 {
-        if (port < rdma_start_port(cma_dev->device) ||
-            port > rdma_end_port(cma_dev->device))
+        if (!rdma_is_port_valid(cma_dev->device, port))
                 return -EINVAL;
 
         return cma_dev->default_gid_type[port - rdma_start_port(cma_dev->device)];
@@ -282,8 +281,7 @@ int cma_set_default_gid_type(struct cma_device *cma_dev,
 {
         unsigned long supported_gids;
 
-        if (port < rdma_start_port(cma_dev->device) ||
-            port > rdma_end_port(cma_dev->device))
+        if (!rdma_is_port_valid(cma_dev->device, port))
                 return -EINVAL;
 
         supported_gids = roce_gid_type_mask_support(cma_dev->device, port);
diff --git a/drivers/infiniband/core/device.c b/drivers/infiniband/core/device.c
index 571974cd3919..f2e48655a906 100644
--- a/drivers/infiniband/core/device.c
+++ b/drivers/infiniband/core/device.c
@@ -659,7 +659,7 @@ int ib_query_port(struct ib_device *device,
         union ib_gid gid;
         int err;
 
-        if (port_num < rdma_start_port(device) || port_num > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, port_num))
                 return -EINVAL;
 
         memset(port_attr, 0, sizeof(*port_attr));
@@ -825,7 +825,7 @@ int ib_modify_port(struct ib_device *device,
         if (!device->modify_port)
                 return -ENOSYS;
 
-        if (port_num < rdma_start_port(device) || port_num > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, port_num))
                 return -EINVAL;
 
         return device->modify_port(device, port_num, port_modify_mask,
diff --git a/drivers/infiniband/core/verbs.c b/drivers/infiniband/core/verbs.c
index 71580cc28c9e..9b77fbc86903 100644
--- a/drivers/infiniband/core/verbs.c
+++ b/drivers/infiniband/core/verbs.c
@@ -1205,8 +1205,7 @@ int ib_resolve_eth_dmac(struct ib_device *device,
 {
         int           ret = 0;
 
-        if (ah_attr->port_num < rdma_start_port(device) ||
-            ah_attr->port_num > rdma_end_port(device))
+        if (!rdma_is_port_valid(device, ah_attr->port_num))
                 return -EINVAL;
 
         if (!rdma_cap_eth_ah(device, ah_attr->port_num))
diff --git a/include/rdma/ib_verbs.h b/include/rdma/ib_verbs.h
index e55afec6bb84..b1ac9735fbbe 100644
--- a/include/rdma/ib_verbs.h
+++ b/include/rdma/ib_verbs.h
@@ -2280,6 +2280,13 @@ static inline u8 rdma_end_port(const struct ib_device *device)
         return rdma_cap_ib_switch(device) ? 0 : device->phys_port_cnt;
 }
 
+static inline int rdma_is_port_valid(const struct ib_device *device,
+                                     unsigned int port)
+{
+        return (port >= rdma_start_port(device) &&
+                port <= rdma_end_port(device));
+}
+
 static inline bool rdma_protocol_ib(const struct ib_device *device, u8 port_num)
 {
         return device->port_immutable[port_num].core_cap_flags & RDMA_CORE_CAP_PROT_IB;