From: Gioh Kim <gi-oh.kim@xxxxxxxxxxxxxxx>
If rtrs_iu_alloc failed to allocate buffer or map dma,
there are some allocated addresses and some NULL addresses
in the array. rtrs_iu_free should check data before free.
Signed-off-by: Gioh Kim <gi-oh.kim@xxxxxxxxxxxxxxx>
---
drivers/infiniband/ulp/rtrs/rtrs.c | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/drivers/infiniband/ulp/rtrs/rtrs.c b/drivers/infiniband/ulp/rtrs/rtrs.c
index 5163e662f86f..ca2d2d3e4192 100644
--- a/drivers/infiniband/ulp/rtrs/rtrs.c
+++ b/drivers/infiniband/ulp/rtrs/rtrs.c
@@ -61,8 +61,12 @@ void rtrs_iu_free(struct rtrs_iu *ius, struct ib_device *ibdev, u32 queue_size)
for (i = 0; i < queue_size; i++) {
iu = &ius[i];
- ib_dma_unmap_single(ibdev, iu->dma_addr, iu->size, iu->direction);
- kfree(iu->buf);
+ if (iu->dma_addr) {
+ ib_dma_unmap_single(ibdev, iu->dma_addr, iu->size, iu->direction);
+ }
+ if (iu->buf) {
+ kfree(iu->buf);
+ }
}
kfree(ius);
}
--
2.20.1
