ib_srpt: Fix srpt_handle_cmd send_ioctx->ioctx_kref leak on exception

This patch addresses a bug in srpt_handle_cmd() failure handling where
send_ioctx->kref is being leaked with the local extra reference after init,
causing the expected kref_put() in srpt_handle_send_comp() to not be the final
call to invoke srpt_put_send_ioctx_kref() -> transport_generic_free_cmd() and
perform se_cmd descriptor memory release.

It also fixes a SCF_SCSI_RESERVATION_CONFLICT handling bug where this code
is incorrectly falling through to transport_handle_cdb_direct() after
invoking srpt_queue_status() to send SAM_STAT_RESERVATION_CONFLICT status.

Note this patch is for >= v3.3 mainline code, and current lio-core.git
code has already been converted to target_submit_cmd() + se_cmd->cmd_kref usage,
and internal ioctx->kref usage has been removed.  I'm including this patch
now into target-pending/for-next with a CC' for v3.3 stable.

Cc: Bart Van Assche <bvanassche@acm.org>
Cc: Roland Dreier <roland@purestorage.com>
Cc: stable@vger.kernel.org
Signed-off-by: Nicholas Bellinger <nab@linux-iscsi.org>

diff --git a/drivers/infiniband/ulp/srpt/ib_srpt.c b/drivers/infiniband/ulp/srpt/ib_srpt.c
index 6735b63..c3cb907 100644 (file)
--- a/drivers/infiniband/ulp/srpt/ib_srpt.c
+++ b/drivers/infiniband/ulp/srpt/ib_srpt.c
@@ -1757,6 +1757,7 @@ static int srpt_handle_cmd(struct srpt_rdma_ch *ch,
                       srp_cmd->tag);
                cmd->se_cmd_flags |= SCF_SCSI_CDB_EXCEPTION;
                cmd->scsi_sense_reason = TCM_INVALID_CDB_FIELD;
+               kref_put(&send_ioctx->kref, srpt_put_send_ioctx_kref);
                goto send_sense;
        }
 
@@ -1764,15 +1765,19 @@ static int srpt_handle_cmd(struct srpt_rdma_ch *ch,
        cmd->data_direction = dir;
        unpacked_lun = srpt_unpack_lun((uint8_t *)&srp_cmd->lun,
                                       sizeof(srp_cmd->lun));
-       if (transport_lookup_cmd_lun(cmd, unpacked_lun) < 0)
+       if (transport_lookup_cmd_lun(cmd, unpacked_lun) < 0) {
+               kref_put(&send_ioctx->kref, srpt_put_send_ioctx_kref);
                goto send_sense;
+       }
        ret = transport_generic_allocate_tasks(cmd, srp_cmd->cdb);
-       if (cmd->se_cmd_flags & SCF_SCSI_RESERVATION_CONFLICT)
-               srpt_queue_status(cmd);
-       else if (cmd->se_cmd_flags & SCF_SCSI_CDB_EXCEPTION)
-               goto send_sense;
-       else
-               WARN_ON_ONCE(ret);
+       if (ret < 0) {
+               kref_put(&send_ioctx->kref, srpt_put_send_ioctx_kref);
+               if (cmd->se_cmd_flags & SCF_SCSI_RESERVATION_CONFLICT) {
+                       srpt_queue_status(cmd);
+                       return 0;
+               } else
+                       goto send_sense;
+       }
 
        transport_handle_cdb_direct(cmd);
        return 0;