From: Zhu Yanjun <yanjun.zhu@xxxxxxxxx>
Use the standard method to produce udp source port based on the
commit 2b880b2e5e03 ("RDMA/mlx5: Define RoCEv2 udp source port
when set path").
Signed-off-by: Zhu Yanjun <yanjun.zhu@xxxxxxxxx>
---
drivers/infiniband/sw/rxe/rxe_verbs.c | 18 ++++++++++++++++++
1 file changed, 18 insertions(+)
diff --git a/drivers/infiniband/sw/rxe/rxe_verbs.c b/drivers/infiniband/sw/rxe/rxe_verbs.c
index 0aa0d7e52773..9a0748ad6417 100644
--- a/drivers/infiniband/sw/rxe/rxe_verbs.c
+++ b/drivers/infiniband/sw/rxe/rxe_verbs.c
@@ -451,6 +451,14 @@ static int rxe_create_qp(struct ib_qp *ibqp, struct ib_qp_init_attr *init,
return err;
}
+static u16 rxe_get_udp_sport(u32 fl, u32 lqpn, u32 rqpn)
+{
+ if (!fl)
+ fl = rdma_calc_flow_label(lqpn, rqpn);
+
+ return rdma_flow_label_to_udp_sport(fl);
+}
+
static int rxe_modify_qp(struct ib_qp *ibqp, struct ib_qp_attr *attr,
int mask, struct ib_udata *udata)
{
@@ -469,6 +477,16 @@ static int rxe_modify_qp(struct ib_qp *ibqp, struct ib_qp_attr *attr,
if (err)
goto err1;
+ if (mask & IB_QP_AV) {
+ if (attr->ah_attr.ah_flags & IB_AH_GRH) {
+ u32 fl = attr->ah_attr.grh.flow_label;
+ u32 lqp = qp->ibqp.qp_num;
+ u32 rqp = qp->attr.dest_qp_num;
+
+ qp->src_port = rxe_get_udp_sport(fl, lqp, rqp);
+ }
+ }
+
return 0;
err1:
--
2.27.0
