X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=kernel%2Fdrivers%2Finfiniband%2Fcore%2Fmad.c;fp=kernel%2Fdrivers%2Finfiniband%2Fcore%2Fmad.c;h=8d84c563ba75351f053c223841c144e87afd80ec;hb=52f993b8e89487ec9ee15a7fb4979e0f09a45b27;hp=2281de122038e45a5c375f7d7669111ab2aee2de;hpb=c189ccac5702322ed843fe17057035b7222a59b6;p=kvmfornfv.git

diff --git a/kernel/drivers/infiniband/core/mad.c b/kernel/drivers/infiniband/core/mad.c
index 2281de122..8d84c563b 100644
--- a/kernel/drivers/infiniband/core/mad.c
+++ b/kernel/drivers/infiniband/core/mad.c
@@ -1745,7 +1745,7 @@ find_mad_agent(struct ib_mad_port_private *port_priv,
 			if (!class)
 				goto out;
 			if (convert_mgmt_class(mad_hdr->mgmt_class) >=
-			    IB_MGMT_MAX_METHODS)
+			    ARRAY_SIZE(class->method_table))
 				goto out;
 			method = class->method_table[convert_mgmt_class(
 							mad_hdr->mgmt_class)];