RDMA/irdma: Add missing comp_mask check in alloc_ucontext

irdma has a comp_mask field that was never checked for validity, check
it.

Signed-off-by: Jason Gunthorpe <jgg@nvidia.com>
Reviewed-by: Jacob Moroni <jmoroni@google.com>
Signed-off-by: Leon Romanovsky <leonro@nvidia.com>

diff --git a/drivers/infiniband/hw/irdma/verbs.c b/drivers/infiniband/hw/irdma/verbs.c
index 8f32eda2165ea6af88b199f8e00d8f9fa5315622..17086048d2d7fc2fae29378d514abdc5b4c710bd 100644 (file)
--- a/drivers/infiniband/hw/irdma/verbs.c
+++ b/drivers/infiniband/hw/irdma/verbs.c
@@ -296,7 +296,9 @@ static int irdma_alloc_ucontext(struct ib_ucontext *uctx,
        if (udata->outlen < IRDMA_ALLOC_UCTX_MIN_RESP_LEN)
                return -EINVAL;
 
-       ret = ib_copy_validate_udata_in(udata, req, rsvd8);
+       ret = ib_copy_validate_udata_in_cm(udata, req, rsvd8,
+                                          IRDMA_ALLOC_UCTX_USE_RAW_ATTR |
+                                                  IRDMA_SUPPORT_WQE_FORMAT_V2);
        if (ret)
                return ret;