1 files changed, 20 insertions, 7 deletions

diff --git a/drivers/infiniband/ulp/srpt/ib_srpt.c b/drivers/infiniband/ulp/srpt/ib_srpt.c
index 0e537d8d0e47..fe09f2788b15 100644
--- a/drivers/infiniband/ulp/srpt/ib_srpt.c
+++ b/drivers/infiniband/ulp/srpt/ib_srpt.c
@@ -1078,6 +1078,7 @@ static void srpt_unmap_sg_to_ib_sge(struct srpt_rdma_ch *ch,
 static int srpt_map_sg_to_ib_sge(struct srpt_rdma_ch *ch,
                                  struct srpt_send_ioctx *ioctx)
 {
+        struct ib_device *dev = ch->sport->sdev->device;
         struct se_cmd *cmd;
         struct scatterlist *sg, *sg_orig;
         int sg_cnt;
@@ -1124,7 +1125,7 @@ static int srpt_map_sg_to_ib_sge(struct srpt_rdma_ch *ch,
 
         db = ioctx->rbufs;
         tsize = cmd->data_length;
-        dma_len = sg_dma_len(&sg[0]);
+        dma_len = ib_sg_dma_len(dev, &sg[0]);
         riu = ioctx->rdma_ius;
 
         /*
@@ -1155,7 +1156,8 @@ static int srpt_map_sg_to_ib_sge(struct srpt_rdma_ch *ch,
                                         ++j;
                                         if (j < count) {
                                                 sg = sg_next(sg);
-                                                dma_len = sg_dma_len(sg);
+                                                dma_len = ib_sg_dma_len(
+                                                                dev, sg);
                                         }
                                 }
                         } else {
@@ -1192,8 +1194,8 @@ static int srpt_map_sg_to_ib_sge(struct srpt_rdma_ch *ch,
         tsize = cmd->data_length;
         riu = ioctx->rdma_ius;
         sg = sg_orig;
-        dma_len = sg_dma_len(&sg[0]);
-        dma_addr = sg_dma_address(&sg[0]);
+        dma_len = ib_sg_dma_len(dev, &sg[0]);
+        dma_addr = ib_sg_dma_address(dev, &sg[0]);
 
         /* this second loop is really mapped sg_addres to rdma_iu->ib_sge */
         for (i = 0, j = 0;
@@ -1216,8 +1218,10 @@ static int srpt_map_sg_to_ib_sge(struct srpt_rdma_ch *ch,
                                         ++j;
                                         if (j < count) {
                                                 sg = sg_next(sg);
-                                                dma_len = sg_dma_len(sg);
-                                                dma_addr = sg_dma_address(sg);
+                                                dma_len = ib_sg_dma_len(
+                                                                dev, sg);
+                                                dma_addr = ib_sg_dma_address(
+                                                                dev, sg);
                                         }
                                 }
                         } else {
@@ -2580,7 +2584,7 @@ static int srpt_cm_req_recv(struct ib_cm_id *cm_id,
                 goto destroy_ib;
         }
 
-        ch->sess = transport_init_session();
+        ch->sess = transport_init_session(TARGET_PROT_NORMAL);
         if (IS_ERR(ch->sess)) {
                 rej->reason = __constant_cpu_to_be32(
                                 SRP_LOGIN_REJ_INSUFFICIENT_RESOURCES);
@@ -3081,6 +3085,14 @@ static void srpt_queue_tm_rsp(struct se_cmd *cmd)
         srpt_queue_response(cmd);
 }
 
+static void srpt_aborted_task(struct se_cmd *cmd)
+{
+        struct srpt_send_ioctx *ioctx = container_of(cmd,
+                                struct srpt_send_ioctx, cmd);
+
+        srpt_unmap_sg_to_ib_sge(ioctx->ch, ioctx);
+}
+
 static int srpt_queue_status(struct se_cmd *cmd)
 {
         struct srpt_send_ioctx *ioctx;
@@ -3928,6 +3940,7 @@ static struct target_core_fabric_ops srpt_template = {
         .queue_data_in                  = srpt_queue_data_in,
         .queue_status                   = srpt_queue_status,
         .queue_tm_rsp                   = srpt_queue_tm_rsp,
+        .aborted_task                   = srpt_aborted_task,
         /*
          * Setup function pointers for generic logic in
          * target_core_fabric_configfs.c