1 files changed, 3 insertions, 1 deletions

diff --git a/drivers/infiniband/core/nldev.c b/drivers/infiniband/core/nldev.c
index 3ba24c428c3b..2fae850a3eff 100644
--- a/drivers/infiniband/core/nldev.c
+++ b/drivers/infiniband/core/nldev.c
@@ -214,7 +214,9 @@ static int nldev_port_get_doit(struct sk_buff *skb, struct nlmsghdr *nlh,
 
         err = nlmsg_parse(nlh, 0, tb, RDMA_NLDEV_ATTR_MAX - 1,
                           nldev_policy, extack);
-        if (err || !tb[RDMA_NLDEV_ATTR_PORT_INDEX])
+        if (err ||
+            !tb[RDMA_NLDEV_ATTR_DEV_INDEX] ||
+            !tb[RDMA_NLDEV_ATTR_PORT_INDEX])
                 return -EINVAL;
 
         index = nla_get_u32(tb[RDMA_NLDEV_ATTR_DEV_INDEX]);