From 860d5904fb71bbf154901030712ed00e4403b79c Mon Sep 17 00:00:00 2001 From: Michael Brown Date: Tue, 22 Mar 2016 16:11:58 +0000 Subject: [PATCH] [arbel] Fix received packet length Signed-off-by: Michael Brown --- src/drivers/infiniband/arbel.c | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/src/drivers/infiniband/arbel.c b/src/drivers/infiniband/arbel.c index 5ab701551..9671174c3 100644 --- a/src/drivers/infiniband/arbel.c +++ b/src/drivers/infiniband/arbel.c @@ -1646,8 +1646,6 @@ static int arbel_complete ( struct ib_device *ibdev, MLX_FILL_1 ( &recv_wqe->data[0], 0, byte_count, 0 ); MLX_FILL_1 ( &recv_wqe->data[0], 1, l_key, ARBEL_INVALID_LKEY ); - assert ( len <= iob_tailroom ( iobuf ) ); - iob_put ( iobuf, len ); memset ( &recv_dest, 0, sizeof ( recv_dest ) ); recv_dest.qpn = qpn; switch ( qp->type ) { @@ -1657,6 +1655,7 @@ static int arbel_complete ( struct ib_device *ibdev, /* Locate corresponding GRH */ assert ( arbel_recv_wq->grh != NULL ); grh = &arbel_recv_wq->grh[wqe_idx]; + len -= sizeof ( *grh ); /* Construct address vector */ source = &recv_source; memset ( source, 0, sizeof ( *source ) ); @@ -1677,6 +1676,8 @@ static int arbel_complete ( struct ib_device *ibdev, assert ( 0 ); return -EINVAL; } + assert ( len <= iob_tailroom ( iobuf ) ); + iob_put ( iobuf, len ); /* Hand off to completion handler */ ib_complete_recv ( ibdev, qp, &recv_dest, source, iobuf, rc ); }