drivers/target/iscsi/iscsi_target_util.c

   1 /*******************************************************************************
   2  * This file contains the iSCSI Target specific utility functions.
   3  *
   4  * (c) Copyright 2007-2013 Datera, Inc.
   5  *
   6  * Author: Nicholas A. Bellinger <nab@linux-iscsi.org>
   7  *
   8  * This program is free software; you can redistribute it and/or modify
   9  * it under the terms of the GNU General Public License as published by
  10  * the Free Software Foundation; either version 2 of the License, or
  11  * (at your option) any later version.
  12  *
  13  * This program is distributed in the hope that it will be useful,
  14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  16  * GNU General Public License for more details.
  17  ******************************************************************************/
  18
  19 #include <linux/list.h>
  20 #include <linux/sched/signal.h>
  21 #include <net/ipv6.h>         /* ipv6_addr_equal() */
  22 #include <scsi/scsi_tcq.h>
  23 #include <scsi/iscsi_proto.h>
  24 #include <target/target_core_base.h>
  25 #include <target/target_core_fabric.h>
  26 #include <target/iscsi/iscsi_transport.h>
  27
  28 #include <target/iscsi/iscsi_target_core.h>
  29 #include "iscsi_target_parameters.h"
  30 #include "iscsi_target_seq_pdu_list.h"
  31 #include "iscsi_target_datain_values.h"
  32 #include "iscsi_target_erl0.h"
  33 #include "iscsi_target_erl1.h"
  34 #include "iscsi_target_erl2.h"
  35 #include "iscsi_target_tpg.h"
  36 #include "iscsi_target_util.h"
  37 #include "iscsi_target.h"
  38
  39 #define PRINT_BUFF(buff, len)                                   \
  40 {                                                               \
  41         int zzz;                                                \
  42                                                                 \
  43         pr_debug("%d:\n", __LINE__);                            \
  44         for (zzz = 0; zzz < len; zzz++) {                       \
  45                 if (zzz % 16 == 0) {                            \
  46                         if (zzz)                                \
  47                                 pr_debug("\n");                 \
  48                         pr_debug("%4i: ", zzz);                 \
  49                 }                                               \
  50                 pr_debug("%02x ", (unsigned char) (buff)[zzz]); \
  51         }                                                       \
  52         if ((len + 1) % 16)                                     \
  53                 pr_debug("\n");                                 \
  54 }
  55
  56 extern struct list_head g_tiqn_list;
  57 extern spinlock_t tiqn_lock;
  58
  59 /*
  60  *      Called with cmd->r2t_lock held.
  61  */
  62 int iscsit_add_r2t_to_list(
  63         struct iscsi_cmd *cmd,
  64         u32 offset,
  65         u32 xfer_len,
  66         int recovery,
  67         u32 r2t_sn)
  68 {
  69         struct iscsi_r2t *r2t;
  70
  71         r2t = kmem_cache_zalloc(lio_r2t_cache, GFP_ATOMIC);
  72         if (!r2t) {
  73                 pr_err("Unable to allocate memory for struct iscsi_r2t.\n");
  74                 return -1;
  75         }
  76         INIT_LIST_HEAD(&r2t->r2t_list);
  77
  78         r2t->recovery_r2t = recovery;
  79         r2t->r2t_sn = (!r2t_sn) ? cmd->r2t_sn++ : r2t_sn;
  80         r2t->offset = offset;
  81         r2t->xfer_len = xfer_len;
  82         list_add_tail(&r2t->r2t_list, &cmd->cmd_r2t_list);
  83         spin_unlock_bh(&cmd->r2t_lock);
  84
  85         iscsit_add_cmd_to_immediate_queue(cmd, cmd->conn, ISTATE_SEND_R2T);
  86
  87         spin_lock_bh(&cmd->r2t_lock);
  88         return 0;
  89 }
  90
  91 struct iscsi_r2t *iscsit_get_r2t_for_eos(
  92         struct iscsi_cmd *cmd,
  93         u32 offset,
  94         u32 length)
  95 {
  96         struct iscsi_r2t *r2t;
  97
  98         spin_lock_bh(&cmd->r2t_lock);
  99         list_for_each_entry(r2t, &cmd->cmd_r2t_list, r2t_list) {
 100                 if ((r2t->offset <= offset) &&
 101                     (r2t->offset + r2t->xfer_len) >= (offset + length)) {
 102                         spin_unlock_bh(&cmd->r2t_lock);
 103                         return r2t;
 104                 }
 105         }
 106         spin_unlock_bh(&cmd->r2t_lock);
 107
 108         pr_err("Unable to locate R2T for Offset: %u, Length:"
 109                         " %u\n", offset, length);
 110         return NULL;
 111 }
 112
 113 struct iscsi_r2t *iscsit_get_r2t_from_list(struct iscsi_cmd *cmd)
 114 {
 115         struct iscsi_r2t *r2t;
 116
 117         spin_lock_bh(&cmd->r2t_lock);
 118         list_for_each_entry(r2t, &cmd->cmd_r2t_list, r2t_list) {
 119                 if (!r2t->sent_r2t) {
 120                         spin_unlock_bh(&cmd->r2t_lock);
 121                         return r2t;
 122                 }
 123         }
 124         spin_unlock_bh(&cmd->r2t_lock);
 125
 126         pr_err("Unable to locate next R2T to send for ITT:"
 127                         " 0x%08x.\n", cmd->init_task_tag);
 128         return NULL;
 129 }
 130
 131 /*
 132  *      Called with cmd->r2t_lock held.
 133  */
 134 void iscsit_free_r2t(struct iscsi_r2t *r2t, struct iscsi_cmd *cmd)
 135 {
 136         list_del(&r2t->r2t_list);
 137         kmem_cache_free(lio_r2t_cache, r2t);
 138 }
 139
 140 void iscsit_free_r2ts_from_list(struct iscsi_cmd *cmd)
 141 {
 142         struct iscsi_r2t *r2t, *r2t_tmp;
 143
 144         spin_lock_bh(&cmd->r2t_lock);
 145         list_for_each_entry_safe(r2t, r2t_tmp, &cmd->cmd_r2t_list, r2t_list)
 146                 iscsit_free_r2t(r2t, cmd);
 147         spin_unlock_bh(&cmd->r2t_lock);
 148 }
 149
 150 static int iscsit_wait_for_tag(struct se_session *se_sess, int state, int *cpup)
 151 {
 152         int tag = -1;
 153         DEFINE_WAIT(wait);
 154         struct sbq_wait_state *ws;
 155
 156         if (state == TASK_RUNNING)
 157                 return tag;
 158
 159         ws = &se_sess->sess_tag_pool.ws[0];
 160         for (;;) {
 161                 prepare_to_wait_exclusive(&ws->wait, &wait, state);
 162                 if (signal_pending_state(state, current))
 163                         break;
 164                 tag = sbitmap_queue_get(&se_sess->sess_tag_pool, cpup);
 165                 if (tag >= 0)
 166                         break;
 167                 schedule();
 168         }
 169
 170         finish_wait(&ws->wait, &wait);
 171         return tag;
 172 }
 173
 174 /*
 175  * May be called from software interrupt (timer) context for allocating
 176  * iSCSI NopINs.
 177  */
 178 struct iscsi_cmd *iscsit_allocate_cmd(struct iscsi_conn *conn, int state)
 179 {
 180         struct iscsi_cmd *cmd;
 181         struct se_session *se_sess = conn->sess->se_sess;
 182         int size, tag, cpu;
 183
 184         tag = sbitmap_queue_get(&se_sess->sess_tag_pool, &cpu);
 185         if (tag < 0)
 186                 tag = iscsit_wait_for_tag(se_sess, state, &cpu);
 187         if (tag < 0)
 188                 return NULL;
 189
 190         size = sizeof(struct iscsi_cmd) + conn->conn_transport->priv_size;
 191         cmd = (struct iscsi_cmd *)(se_sess->sess_cmd_map + (tag * size));
 192         memset(cmd, 0, size);
 193
 194         cmd->se_cmd.map_tag = tag;
 195         cmd->se_cmd.map_cpu = cpu;
 196         cmd->conn = conn;
 197         cmd->data_direction = DMA_NONE;
 198         INIT_LIST_HEAD(&cmd->i_conn_node);
 199         INIT_LIST_HEAD(&cmd->datain_list);
 200         INIT_LIST_HEAD(&cmd->cmd_r2t_list);
 201         spin_lock_init(&cmd->datain_lock);
 202         spin_lock_init(&cmd->dataout_timeout_lock);
 203         spin_lock_init(&cmd->istate_lock);
 204         spin_lock_init(&cmd->error_lock);
 205         spin_lock_init(&cmd->r2t_lock);
 206         timer_setup(&cmd->dataout_timer, iscsit_handle_dataout_timeout, 0);
 207
 208         return cmd;
 209 }
 210 EXPORT_SYMBOL(iscsit_allocate_cmd);
 211
 212 struct iscsi_seq *iscsit_get_seq_holder_for_datain(
 213         struct iscsi_cmd *cmd,
 214         u32 seq_send_order)
 215 {
 216         u32 i;
 217
 218         for (i = 0; i < cmd->seq_count; i++)
 219                 if (cmd->seq_list[i].seq_send_order == seq_send_order)
 220                         return &cmd->seq_list[i];
 221
 222         return NULL;
 223 }
 224
 225 struct iscsi_seq *iscsit_get_seq_holder_for_r2t(struct iscsi_cmd *cmd)
 226 {
 227         u32 i;
 228
 229         if (!cmd->seq_list) {
 230                 pr_err("struct iscsi_cmd->seq_list is NULL!\n");
 231                 return NULL;
 232         }
 233
 234         for (i = 0; i < cmd->seq_count; i++) {
 235                 if (cmd->seq_list[i].type != SEQTYPE_NORMAL)
 236                         continue;
 237                 if (cmd->seq_list[i].seq_send_order == cmd->seq_send_order) {
 238                         cmd->seq_send_order++;
 239                         return &cmd->seq_list[i];
 240                 }
 241         }
 242
 243         return NULL;
 244 }
 245
 246 struct iscsi_r2t *iscsit_get_holder_for_r2tsn(
 247         struct iscsi_cmd *cmd,
 248         u32 r2t_sn)
 249 {
 250         struct iscsi_r2t *r2t;
 251
 252         spin_lock_bh(&cmd->r2t_lock);
 253         list_for_each_entry(r2t, &cmd->cmd_r2t_list, r2t_list) {
 254                 if (r2t->r2t_sn == r2t_sn) {
 255                         spin_unlock_bh(&cmd->r2t_lock);
 256                         return r2t;
 257                 }
 258         }
 259         spin_unlock_bh(&cmd->r2t_lock);
 260
 261         return NULL;
 262 }
 263
 264 static inline int iscsit_check_received_cmdsn(struct iscsi_session *sess, u32 cmdsn)
 265 {
 266         u32 max_cmdsn;
 267         int ret;
 268
 269         /*
 270          * This is the proper method of checking received CmdSN against
 271          * ExpCmdSN and MaxCmdSN values, as well as accounting for out
 272          * or order CmdSNs due to multiple connection sessions and/or
 273          * CRC failures.
 274          */
 275         max_cmdsn = atomic_read(&sess->max_cmd_sn);
 276         if (iscsi_sna_gt(cmdsn, max_cmdsn)) {
 277                 pr_err("Received CmdSN: 0x%08x is greater than"
 278                        " MaxCmdSN: 0x%08x, ignoring.\n", cmdsn, max_cmdsn);
 279                 ret = CMDSN_MAXCMDSN_OVERRUN;
 280
 281         } else if (cmdsn == sess->exp_cmd_sn) {
 282                 sess->exp_cmd_sn++;
 283                 pr_debug("Received CmdSN matches ExpCmdSN,"
 284                       " incremented ExpCmdSN to: 0x%08x\n",
 285                       sess->exp_cmd_sn);
 286                 ret = CMDSN_NORMAL_OPERATION;
 287
 288         } else if (iscsi_sna_gt(cmdsn, sess->exp_cmd_sn)) {
 289                 pr_debug("Received CmdSN: 0x%08x is greater"
 290                       " than ExpCmdSN: 0x%08x, not acknowledging.\n",
 291                       cmdsn, sess->exp_cmd_sn);
 292                 ret = CMDSN_HIGHER_THAN_EXP;
 293
 294         } else {
 295                 pr_err("Received CmdSN: 0x%08x is less than"
 296                        " ExpCmdSN: 0x%08x, ignoring.\n", cmdsn,
 297                        sess->exp_cmd_sn);
 298                 ret = CMDSN_LOWER_THAN_EXP;
 299         }
 300
 301         return ret;
 302 }
 303
 304 /*
 305  * Commands may be received out of order if MC/S is in use.
 306  * Ensure they are executed in CmdSN order.
 307  */
 308 int iscsit_sequence_cmd(struct iscsi_conn *conn, struct iscsi_cmd *cmd,
 309                         unsigned char *buf, __be32 cmdsn)
 310 {
 311         int ret, cmdsn_ret;
 312         bool reject = false;
 313         u8 reason = ISCSI_REASON_BOOKMARK_NO_RESOURCES;
 314
 315         mutex_lock(&conn->sess->cmdsn_mutex);
 316
 317         cmdsn_ret = iscsit_check_received_cmdsn(conn->sess, be32_to_cpu(cmdsn));
 318         switch (cmdsn_ret) {
 319         case CMDSN_NORMAL_OPERATION:
 320                 ret = iscsit_execute_cmd(cmd, 0);
 321                 if ((ret >= 0) && !list_empty(&conn->sess->sess_ooo_cmdsn_list))
 322                         iscsit_execute_ooo_cmdsns(conn->sess);
 323                 else if (ret < 0) {
 324                         reject = true;
 325                         ret = CMDSN_ERROR_CANNOT_RECOVER;
 326                 }
 327                 break;
 328         case CMDSN_HIGHER_THAN_EXP:
 329                 ret = iscsit_handle_ooo_cmdsn(conn->sess, cmd, be32_to_cpu(cmdsn));
 330                 if (ret < 0) {
 331                         reject = true;
 332                         ret = CMDSN_ERROR_CANNOT_RECOVER;
 333                         break;
 334                 }
 335                 ret = CMDSN_HIGHER_THAN_EXP;
 336                 break;
 337         case CMDSN_LOWER_THAN_EXP:
 338         case CMDSN_MAXCMDSN_OVERRUN:
 339         default:
 340                 cmd->i_state = ISTATE_REMOVE;
 341                 iscsit_add_cmd_to_immediate_queue(cmd, conn, cmd->i_state);
 342                 /*
 343                  * Existing callers for iscsit_sequence_cmd() will silently
 344                  * ignore commands with CMDSN_LOWER_THAN_EXP, so force this
 345                  * return for CMDSN_MAXCMDSN_OVERRUN as well..
 346                  */
 347                 ret = CMDSN_LOWER_THAN_EXP;
 348                 break;
 349         }
 350         mutex_unlock(&conn->sess->cmdsn_mutex);
 351
 352         if (reject)
 353                 iscsit_reject_cmd(cmd, reason, buf);
 354
 355         return ret;
 356 }
 357 EXPORT_SYMBOL(iscsit_sequence_cmd);
 358
 359 int iscsit_check_unsolicited_dataout(struct iscsi_cmd *cmd, unsigned char *buf)
 360 {
 361         struct iscsi_conn *conn = cmd->conn;
 362         struct se_cmd *se_cmd = &cmd->se_cmd;
 363         struct iscsi_data *hdr = (struct iscsi_data *) buf;
 364         u32 payload_length = ntoh24(hdr->dlength);
 365
 366         if (conn->sess->sess_ops->InitialR2T) {
 367                 pr_err("Received unexpected unsolicited data"
 368                         " while InitialR2T=Yes, protocol error.\n");
 369                 transport_send_check_condition_and_sense(se_cmd,
 370                                 TCM_UNEXPECTED_UNSOLICITED_DATA, 0);
 371                 return -1;
 372         }
 373
 374         if ((cmd->first_burst_len + payload_length) >
 375              conn->sess->sess_ops->FirstBurstLength) {
 376                 pr_err("Total %u bytes exceeds FirstBurstLength: %u"
 377                         " for this Unsolicited DataOut Burst.\n",
 378                         (cmd->first_burst_len + payload_length),
 379                                 conn->sess->sess_ops->FirstBurstLength);
 380                 transport_send_check_condition_and_sense(se_cmd,
 381                                 TCM_INCORRECT_AMOUNT_OF_DATA, 0);
 382                 return -1;
 383         }
 384
 385         if (!(hdr->flags & ISCSI_FLAG_CMD_FINAL))
 386                 return 0;
 387
 388         if (((cmd->first_burst_len + payload_length) != cmd->se_cmd.data_length) &&
 389             ((cmd->first_burst_len + payload_length) !=
 390               conn->sess->sess_ops->FirstBurstLength)) {
 391                 pr_err("Unsolicited non-immediate data received %u"
 392                         " does not equal FirstBurstLength: %u, and does"
 393                         " not equal ExpXferLen %u.\n",
 394                         (cmd->first_burst_len + payload_length),
 395                         conn->sess->sess_ops->FirstBurstLength, cmd->se_cmd.data_length);
 396                 transport_send_check_condition_and_sense(se_cmd,
 397                                 TCM_INCORRECT_AMOUNT_OF_DATA, 0);
 398                 return -1;
 399         }
 400         return 0;
 401 }
 402
 403 struct iscsi_cmd *iscsit_find_cmd_from_itt(
 404         struct iscsi_conn *conn,
 405         itt_t init_task_tag)
 406 {
 407         struct iscsi_cmd *cmd;
 408
 409         spin_lock_bh(&conn->cmd_lock);
 410         list_for_each_entry(cmd, &conn->conn_cmd_list, i_conn_node) {
 411                 if (cmd->init_task_tag == init_task_tag) {
 412                         spin_unlock_bh(&conn->cmd_lock);
 413                         return cmd;
 414                 }
 415         }
 416         spin_unlock_bh(&conn->cmd_lock);
 417
 418         pr_err("Unable to locate ITT: 0x%08x on CID: %hu",
 419                         init_task_tag, conn->cid);
 420         return NULL;
 421 }
 422 EXPORT_SYMBOL(iscsit_find_cmd_from_itt);
 423
 424 struct iscsi_cmd *iscsit_find_cmd_from_itt_or_dump(
 425         struct iscsi_conn *conn,
 426         itt_t init_task_tag,
 427         u32 length)
 428 {
 429         struct iscsi_cmd *cmd;
 430
 431         spin_lock_bh(&conn->cmd_lock);
 432         list_for_each_entry(cmd, &conn->conn_cmd_list, i_conn_node) {
 433                 if (cmd->cmd_flags & ICF_GOT_LAST_DATAOUT)
 434                         continue;
 435                 if (cmd->init_task_tag == init_task_tag) {
 436                         spin_unlock_bh(&conn->cmd_lock);
 437                         return cmd;
 438                 }
 439         }
 440         spin_unlock_bh(&conn->cmd_lock);
 441
 442         pr_err("Unable to locate ITT: 0x%08x on CID: %hu,"
 443                         " dumping payload\n", init_task_tag, conn->cid);
 444         if (length)
 445                 iscsit_dump_data_payload(conn, length, 1);
 446
 447         return NULL;
 448 }
 449 EXPORT_SYMBOL(iscsit_find_cmd_from_itt_or_dump);
 450
 451 struct iscsi_cmd *iscsit_find_cmd_from_ttt(
 452         struct iscsi_conn *conn,
 453         u32 targ_xfer_tag)
 454 {
 455         struct iscsi_cmd *cmd = NULL;
 456
 457         spin_lock_bh(&conn->cmd_lock);
 458         list_for_each_entry(cmd, &conn->conn_cmd_list, i_conn_node) {
 459                 if (cmd->targ_xfer_tag == targ_xfer_tag) {
 460                         spin_unlock_bh(&conn->cmd_lock);
 461                         return cmd;
 462                 }
 463         }
 464         spin_unlock_bh(&conn->cmd_lock);
 465
 466         pr_err("Unable to locate TTT: 0x%08x on CID: %hu\n",
 467                         targ_xfer_tag, conn->cid);
 468         return NULL;
 469 }
 470
 471 int iscsit_find_cmd_for_recovery(
 472         struct iscsi_session *sess,
 473         struct iscsi_cmd **cmd_ptr,
 474         struct iscsi_conn_recovery **cr_ptr,
 475         itt_t init_task_tag)
 476 {
 477         struct iscsi_cmd *cmd = NULL;
 478         struct iscsi_conn_recovery *cr;
 479         /*
 480          * Scan through the inactive connection recovery list's command list.
 481          * If init_task_tag matches the command is still alligent.
 482          */
 483         spin_lock(&sess->cr_i_lock);
 484         list_for_each_entry(cr, &sess->cr_inactive_list, cr_list) {
 485                 spin_lock(&cr->conn_recovery_cmd_lock);
 486                 list_for_each_entry(cmd, &cr->conn_recovery_cmd_list, i_conn_node) {
 487                         if (cmd->init_task_tag == init_task_tag) {
 488                                 spin_unlock(&cr->conn_recovery_cmd_lock);
 489                                 spin_unlock(&sess->cr_i_lock);
 490
 491                                 *cr_ptr = cr;
 492                                 *cmd_ptr = cmd;
 493                                 return -2;
 494                         }
 495                 }
 496                 spin_unlock(&cr->conn_recovery_cmd_lock);
 497         }
 498         spin_unlock(&sess->cr_i_lock);
 499         /*
 500          * Scan through the active connection recovery list's command list.
 501          * If init_task_tag matches the command is ready to be reassigned.
 502          */
 503         spin_lock(&sess->cr_a_lock);
 504         list_for_each_entry(cr, &sess->cr_active_list, cr_list) {
 505                 spin_lock(&cr->conn_recovery_cmd_lock);
 506                 list_for_each_entry(cmd, &cr->conn_recovery_cmd_list, i_conn_node) {
 507                         if (cmd->init_task_tag == init_task_tag) {
 508                                 spin_unlock(&cr->conn_recovery_cmd_lock);
 509                                 spin_unlock(&sess->cr_a_lock);
 510
 511                                 *cr_ptr = cr;
 512                                 *cmd_ptr = cmd;
 513                                 return 0;
 514                         }
 515                 }
 516                 spin_unlock(&cr->conn_recovery_cmd_lock);
 517         }
 518         spin_unlock(&sess->cr_a_lock);
 519
 520         return -1;
 521 }
 522
 523 void iscsit_add_cmd_to_immediate_queue(
 524         struct iscsi_cmd *cmd,
 525         struct iscsi_conn *conn,
 526         u8 state)
 527 {
 528         struct iscsi_queue_req *qr;
 529
 530         qr = kmem_cache_zalloc(lio_qr_cache, GFP_ATOMIC);
 531         if (!qr) {
 532                 pr_err("Unable to allocate memory for"
 533                                 " struct iscsi_queue_req\n");
 534                 return;
 535         }
 536         INIT_LIST_HEAD(&qr->qr_list);
 537         qr->cmd = cmd;
 538         qr->state = state;
 539
 540         spin_lock_bh(&conn->immed_queue_lock);
 541         list_add_tail(&qr->qr_list, &conn->immed_queue_list);
 542         atomic_inc(&cmd->immed_queue_count);
 543         atomic_set(&conn->check_immediate_queue, 1);
 544         spin_unlock_bh(&conn->immed_queue_lock);
 545
 546         wake_up(&conn->queues_wq);
 547 }
 548 EXPORT_SYMBOL(iscsit_add_cmd_to_immediate_queue);
 549
 550 struct iscsi_queue_req *iscsit_get_cmd_from_immediate_queue(struct iscsi_conn *conn)
 551 {
 552         struct iscsi_queue_req *qr;
 553
 554         spin_lock_bh(&conn->immed_queue_lock);
 555         if (list_empty(&conn->immed_queue_list)) {
 556                 spin_unlock_bh(&conn->immed_queue_lock);
 557                 return NULL;
 558         }
 559         qr = list_first_entry(&conn->immed_queue_list,
 560                               struct iscsi_queue_req, qr_list);
 561
 562         list_del(&qr->qr_list);
 563         if (qr->cmd)
 564                 atomic_dec(&qr->cmd->immed_queue_count);
 565         spin_unlock_bh(&conn->immed_queue_lock);
 566
 567         return qr;
 568 }
 569
 570 static void iscsit_remove_cmd_from_immediate_queue(
 571         struct iscsi_cmd *cmd,
 572         struct iscsi_conn *conn)
 573 {
 574         struct iscsi_queue_req *qr, *qr_tmp;
 575
 576         spin_lock_bh(&conn->immed_queue_lock);
 577         if (!atomic_read(&cmd->immed_queue_count)) {
 578                 spin_unlock_bh(&conn->immed_queue_lock);
 579                 return;
 580         }
 581
 582         list_for_each_entry_safe(qr, qr_tmp, &conn->immed_queue_list, qr_list) {
 583                 if (qr->cmd != cmd)
 584                         continue;
 585
 586                 atomic_dec(&qr->cmd->immed_queue_count);
 587                 list_del(&qr->qr_list);
 588                 kmem_cache_free(lio_qr_cache, qr);
 589         }
 590         spin_unlock_bh(&conn->immed_queue_lock);
 591
 592         if (atomic_read(&cmd->immed_queue_count)) {
 593                 pr_err("ITT: 0x%08x immed_queue_count: %d\n",
 594                         cmd->init_task_tag,
 595                         atomic_read(&cmd->immed_queue_count));
 596         }
 597 }
 598
 599 int iscsit_add_cmd_to_response_queue(
 600         struct iscsi_cmd *cmd,
 601         struct iscsi_conn *conn,
 602         u8 state)
 603 {
 604         struct iscsi_queue_req *qr;
 605
 606         qr = kmem_cache_zalloc(lio_qr_cache, GFP_ATOMIC);
 607         if (!qr) {
 608                 pr_err("Unable to allocate memory for"
 609                         " struct iscsi_queue_req\n");
 610                 return -ENOMEM;
 611         }
 612         INIT_LIST_HEAD(&qr->qr_list);
 613         qr->cmd = cmd;
 614         qr->state = state;
 615
 616         spin_lock_bh(&conn->response_queue_lock);
 617         list_add_tail(&qr->qr_list, &conn->response_queue_list);
 618         atomic_inc(&cmd->response_queue_count);
 619         spin_unlock_bh(&conn->response_queue_lock);
 620
 621         wake_up(&conn->queues_wq);
 622         return 0;
 623 }
 624
 625 struct iscsi_queue_req *iscsit_get_cmd_from_response_queue(struct iscsi_conn *conn)
 626 {
 627         struct iscsi_queue_req *qr;
 628
 629         spin_lock_bh(&conn->response_queue_lock);
 630         if (list_empty(&conn->response_queue_list)) {
 631                 spin_unlock_bh(&conn->response_queue_lock);
 632                 return NULL;
 633         }
 634
 635         qr = list_first_entry(&conn->response_queue_list,
 636                               struct iscsi_queue_req, qr_list);
 637
 638         list_del(&qr->qr_list);
 639         if (qr->cmd)
 640                 atomic_dec(&qr->cmd->response_queue_count);
 641         spin_unlock_bh(&conn->response_queue_lock);
 642
 643         return qr;
 644 }
 645
 646 static void iscsit_remove_cmd_from_response_queue(
 647         struct iscsi_cmd *cmd,
 648         struct iscsi_conn *conn)
 649 {
 650         struct iscsi_queue_req *qr, *qr_tmp;
 651
 652         spin_lock_bh(&conn->response_queue_lock);
 653         if (!atomic_read(&cmd->response_queue_count)) {
 654                 spin_unlock_bh(&conn->response_queue_lock);
 655                 return;
 656         }
 657
 658         list_for_each_entry_safe(qr, qr_tmp, &conn->response_queue_list,
 659                                 qr_list) {
 660                 if (qr->cmd != cmd)
 661                         continue;
 662
 663                 atomic_dec(&qr->cmd->response_queue_count);
 664                 list_del(&qr->qr_list);
 665                 kmem_cache_free(lio_qr_cache, qr);
 666         }
 667         spin_unlock_bh(&conn->response_queue_lock);
 668
 669         if (atomic_read(&cmd->response_queue_count)) {
 670                 pr_err("ITT: 0x%08x response_queue_count: %d\n",
 671                         cmd->init_task_tag,
 672                         atomic_read(&cmd->response_queue_count));
 673         }
 674 }
 675
 676 bool iscsit_conn_all_queues_empty(struct iscsi_conn *conn)
 677 {
 678         bool empty;
 679
 680         spin_lock_bh(&conn->immed_queue_lock);
 681         empty = list_empty(&conn->immed_queue_list);
 682         spin_unlock_bh(&conn->immed_queue_lock);
 683
 684         if (!empty)
 685                 return empty;
 686
 687         spin_lock_bh(&conn->response_queue_lock);
 688         empty = list_empty(&conn->response_queue_list);
 689         spin_unlock_bh(&conn->response_queue_lock);
 690
 691         return empty;
 692 }
 693
 694 void iscsit_free_queue_reqs_for_conn(struct iscsi_conn *conn)
 695 {
 696         struct iscsi_queue_req *qr, *qr_tmp;
 697
 698         spin_lock_bh(&conn->immed_queue_lock);
 699         list_for_each_entry_safe(qr, qr_tmp, &conn->immed_queue_list, qr_list) {
 700                 list_del(&qr->qr_list);
 701                 if (qr->cmd)
 702                         atomic_dec(&qr->cmd->immed_queue_count);
 703
 704                 kmem_cache_free(lio_qr_cache, qr);
 705         }
 706         spin_unlock_bh(&conn->immed_queue_lock);
 707
 708         spin_lock_bh(&conn->response_queue_lock);
 709         list_for_each_entry_safe(qr, qr_tmp, &conn->response_queue_list,
 710                         qr_list) {
 711                 list_del(&qr->qr_list);
 712                 if (qr->cmd)
 713                         atomic_dec(&qr->cmd->response_queue_count);
 714
 715                 kmem_cache_free(lio_qr_cache, qr);
 716         }
 717         spin_unlock_bh(&conn->response_queue_lock);
 718 }
 719
 720 void iscsit_release_cmd(struct iscsi_cmd *cmd)
 721 {
 722         struct iscsi_session *sess;
 723         struct se_cmd *se_cmd = &cmd->se_cmd;
 724
 725         WARN_ON(!list_empty(&cmd->i_conn_node));
 726
 727         if (cmd->conn)
 728                 sess = cmd->conn->sess;
 729         else
 730                 sess = cmd->sess;
 731
 732         BUG_ON(!sess || !sess->se_sess);
 733
 734         kfree(cmd->buf_ptr);
 735         kfree(cmd->pdu_list);
 736         kfree(cmd->seq_list);
 737         kfree(cmd->tmr_req);
 738         kfree(cmd->iov_data);
 739         kfree(cmd->text_in_ptr);
 740
 741         target_free_tag(sess->se_sess, se_cmd);
 742 }
 743 EXPORT_SYMBOL(iscsit_release_cmd);
 744
 745 void __iscsit_free_cmd(struct iscsi_cmd *cmd, bool check_queues)
 746 {
 747         struct iscsi_conn *conn = cmd->conn;
 748
 749         WARN_ON(!list_empty(&cmd->i_conn_node));
 750
 751         if (cmd->data_direction == DMA_TO_DEVICE) {
 752                 iscsit_stop_dataout_timer(cmd);
 753                 iscsit_free_r2ts_from_list(cmd);
 754         }
 755         if (cmd->data_direction == DMA_FROM_DEVICE)
 756                 iscsit_free_all_datain_reqs(cmd);
 757
 758         if (conn && check_queues) {
 759                 iscsit_remove_cmd_from_immediate_queue(cmd, conn);
 760                 iscsit_remove_cmd_from_response_queue(cmd, conn);
 761         }
 762
 763         if (conn && conn->conn_transport->iscsit_release_cmd)
 764                 conn->conn_transport->iscsit_release_cmd(conn, cmd);
 765 }
 766
 767 void iscsit_free_cmd(struct iscsi_cmd *cmd, bool shutdown)
 768 {
 769         struct se_cmd *se_cmd = cmd->se_cmd.se_tfo ? &cmd->se_cmd : NULL;
 770         int rc;
 771
 772         __iscsit_free_cmd(cmd, shutdown);
 773         if (se_cmd) {
 774                 rc = transport_generic_free_cmd(se_cmd, shutdown);
 775                 if (!rc && shutdown && se_cmd->se_sess) {
 776                         __iscsit_free_cmd(cmd, shutdown);
 777                         target_put_sess_cmd(se_cmd);
 778                 }
 779         } else {
 780                 iscsit_release_cmd(cmd);
 781         }
 782 }
 783 EXPORT_SYMBOL(iscsit_free_cmd);
 784
 785 int iscsit_check_session_usage_count(struct iscsi_session *sess)
 786 {
 787         spin_lock_bh(&sess->session_usage_lock);
 788         if (sess->session_usage_count != 0) {
 789                 sess->session_waiting_on_uc = 1;
 790                 spin_unlock_bh(&sess->session_usage_lock);
 791                 if (in_interrupt())
 792                         return 2;
 793
 794                 wait_for_completion(&sess->session_waiting_on_uc_comp);
 795                 return 1;
 796         }
 797         spin_unlock_bh(&sess->session_usage_lock);
 798
 799         return 0;
 800 }
 801
 802 void iscsit_dec_session_usage_count(struct iscsi_session *sess)
 803 {
 804         spin_lock_bh(&sess->session_usage_lock);
 805         sess->session_usage_count--;
 806
 807         if (!sess->session_usage_count && sess->session_waiting_on_uc)
 808                 complete(&sess->session_waiting_on_uc_comp);
 809
 810         spin_unlock_bh(&sess->session_usage_lock);
 811 }
 812
 813 void iscsit_inc_session_usage_count(struct iscsi_session *sess)
 814 {
 815         spin_lock_bh(&sess->session_usage_lock);
 816         sess->session_usage_count++;
 817         spin_unlock_bh(&sess->session_usage_lock);
 818 }
 819
 820 struct iscsi_conn *iscsit_get_conn_from_cid(struct iscsi_session *sess, u16 cid)
 821 {
 822         struct iscsi_conn *conn;
 823
 824         spin_lock_bh(&sess->conn_lock);
 825         list_for_each_entry(conn, &sess->sess_conn_list, conn_list) {
 826                 if ((conn->cid == cid) &&
 827                     (conn->conn_state == TARG_CONN_STATE_LOGGED_IN)) {
 828                         iscsit_inc_conn_usage_count(conn);
 829                         spin_unlock_bh(&sess->conn_lock);
 830                         return conn;
 831                 }
 832         }
 833         spin_unlock_bh(&sess->conn_lock);
 834
 835         return NULL;
 836 }
 837
 838 struct iscsi_conn *iscsit_get_conn_from_cid_rcfr(struct iscsi_session *sess, u16 cid)
 839 {
 840         struct iscsi_conn *conn;
 841
 842         spin_lock_bh(&sess->conn_lock);
 843         list_for_each_entry(conn, &sess->sess_conn_list, conn_list) {
 844                 if (conn->cid == cid) {
 845                         iscsit_inc_conn_usage_count(conn);
 846                         spin_lock(&conn->state_lock);
 847                         atomic_set(&conn->connection_wait_rcfr, 1);
 848                         spin_unlock(&conn->state_lock);
 849                         spin_unlock_bh(&sess->conn_lock);
 850                         return conn;
 851                 }
 852         }
 853         spin_unlock_bh(&sess->conn_lock);
 854
 855         return NULL;
 856 }
 857
 858 void iscsit_check_conn_usage_count(struct iscsi_conn *conn)
 859 {
 860         spin_lock_bh(&conn->conn_usage_lock);
 861         if (conn->conn_usage_count != 0) {
 862                 conn->conn_waiting_on_uc = 1;
 863                 spin_unlock_bh(&conn->conn_usage_lock);
 864
 865                 wait_for_completion(&conn->conn_waiting_on_uc_comp);
 866                 return;
 867         }
 868         spin_unlock_bh(&conn->conn_usage_lock);
 869 }
 870
 871 void iscsit_dec_conn_usage_count(struct iscsi_conn *conn)
 872 {
 873         spin_lock_bh(&conn->conn_usage_lock);
 874         conn->conn_usage_count--;
 875
 876         if (!conn->conn_usage_count && conn->conn_waiting_on_uc)
 877                 complete(&conn->conn_waiting_on_uc_comp);
 878
 879         spin_unlock_bh(&conn->conn_usage_lock);
 880 }
 881
 882 void iscsit_inc_conn_usage_count(struct iscsi_conn *conn)
 883 {
 884         spin_lock_bh(&conn->conn_usage_lock);
 885         conn->conn_usage_count++;
 886         spin_unlock_bh(&conn->conn_usage_lock);
 887 }
 888
 889 static int iscsit_add_nopin(struct iscsi_conn *conn, int want_response)
 890 {
 891         u8 state;
 892         struct iscsi_cmd *cmd;
 893
 894         cmd = iscsit_allocate_cmd(conn, TASK_RUNNING);
 895         if (!cmd)
 896                 return -1;
 897
 898         cmd->iscsi_opcode = ISCSI_OP_NOOP_IN;
 899         state = (want_response) ? ISTATE_SEND_NOPIN_WANT_RESPONSE :
 900                                 ISTATE_SEND_NOPIN_NO_RESPONSE;
 901         cmd->init_task_tag = RESERVED_ITT;
 902         cmd->targ_xfer_tag = (want_response) ?
 903                              session_get_next_ttt(conn->sess) : 0xFFFFFFFF;
 904         spin_lock_bh(&conn->cmd_lock);
 905         list_add_tail(&cmd->i_conn_node, &conn->conn_cmd_list);
 906         spin_unlock_bh(&conn->cmd_lock);
 907
 908         if (want_response)
 909                 iscsit_start_nopin_response_timer(conn);
 910         iscsit_add_cmd_to_immediate_queue(cmd, conn, state);
 911
 912         return 0;
 913 }
 914
 915 void iscsit_handle_nopin_response_timeout(struct timer_list *t)
 916 {
 917         struct iscsi_conn *conn = from_timer(conn, t, nopin_response_timer);
 918
 919         iscsit_inc_conn_usage_count(conn);
 920
 921         spin_lock_bh(&conn->nopin_timer_lock);
 922         if (conn->nopin_response_timer_flags & ISCSI_TF_STOP) {
 923                 spin_unlock_bh(&conn->nopin_timer_lock);
 924                 iscsit_dec_conn_usage_count(conn);
 925                 return;
 926         }
 927
 928         pr_debug("Did not receive response to NOPIN on CID: %hu on"
 929                 " SID: %u, failing connection.\n", conn->cid,
 930                         conn->sess->sid);
 931         conn->nopin_response_timer_flags &= ~ISCSI_TF_RUNNING;
 932         spin_unlock_bh(&conn->nopin_timer_lock);
 933
 934         {
 935         struct iscsi_portal_group *tpg = conn->sess->tpg;
 936         struct iscsi_tiqn *tiqn = tpg->tpg_tiqn;
 937
 938         if (tiqn) {
 939                 spin_lock_bh(&tiqn->sess_err_stats.lock);
 940                 strcpy(tiqn->sess_err_stats.last_sess_fail_rem_name,
 941                                 conn->sess->sess_ops->InitiatorName);
 942                 tiqn->sess_err_stats.last_sess_failure_type =
 943                                 ISCSI_SESS_ERR_CXN_TIMEOUT;
 944                 tiqn->sess_err_stats.cxn_timeout_errors++;
 945                 atomic_long_inc(&conn->sess->conn_timeout_errors);
 946                 spin_unlock_bh(&tiqn->sess_err_stats.lock);
 947         }
 948         }
 949
 950         iscsit_cause_connection_reinstatement(conn, 0);
 951         iscsit_dec_conn_usage_count(conn);
 952 }
 953
 954 void iscsit_mod_nopin_response_timer(struct iscsi_conn *conn)
 955 {
 956         struct iscsi_session *sess = conn->sess;
 957         struct iscsi_node_attrib *na = iscsit_tpg_get_node_attrib(sess);
 958
 959         spin_lock_bh(&conn->nopin_timer_lock);
 960         if (!(conn->nopin_response_timer_flags & ISCSI_TF_RUNNING)) {
 961                 spin_unlock_bh(&conn->nopin_timer_lock);
 962                 return;
 963         }
 964
 965         mod_timer(&conn->nopin_response_timer,
 966                 (get_jiffies_64() + na->nopin_response_timeout * HZ));
 967         spin_unlock_bh(&conn->nopin_timer_lock);
 968 }
 969
 970 /*
 971  *      Called with conn->nopin_timer_lock held.
 972  */
 973 void iscsit_start_nopin_response_timer(struct iscsi_conn *conn)
 974 {
 975         struct iscsi_session *sess = conn->sess;
 976         struct iscsi_node_attrib *na = iscsit_tpg_get_node_attrib(sess);
 977
 978         spin_lock_bh(&conn->nopin_timer_lock);
 979         if (conn->nopin_response_timer_flags & ISCSI_TF_RUNNING) {
 980                 spin_unlock_bh(&conn->nopin_timer_lock);
 981                 return;
 982         }
 983
 984         conn->nopin_response_timer_flags &= ~ISCSI_TF_STOP;
 985         conn->nopin_response_timer_flags |= ISCSI_TF_RUNNING;
 986         mod_timer(&conn->nopin_response_timer,
 987                   jiffies + na->nopin_response_timeout * HZ);
 988
 989         pr_debug("Started NOPIN Response Timer on CID: %d to %u"
 990                 " seconds\n", conn->cid, na->nopin_response_timeout);
 991         spin_unlock_bh(&conn->nopin_timer_lock);
 992 }
 993
 994 void iscsit_stop_nopin_response_timer(struct iscsi_conn *conn)
 995 {
 996         spin_lock_bh(&conn->nopin_timer_lock);
 997         if (!(conn->nopin_response_timer_flags & ISCSI_TF_RUNNING)) {
 998                 spin_unlock_bh(&conn->nopin_timer_lock);
 999                 return;
1000         }
1001         conn->nopin_response_timer_flags |= ISCSI_TF_STOP;
1002         spin_unlock_bh(&conn->nopin_timer_lock);
1003
1004         del_timer_sync(&conn->nopin_response_timer);
1005
1006         spin_lock_bh(&conn->nopin_timer_lock);
1007         conn->nopin_response_timer_flags &= ~ISCSI_TF_RUNNING;
1008         spin_unlock_bh(&conn->nopin_timer_lock);
1009 }
1010
1011 void iscsit_handle_nopin_timeout(struct timer_list *t)
1012 {
1013         struct iscsi_conn *conn = from_timer(conn, t, nopin_timer);
1014
1015         iscsit_inc_conn_usage_count(conn);
1016
1017         spin_lock_bh(&conn->nopin_timer_lock);
1018         if (conn->nopin_timer_flags & ISCSI_TF_STOP) {
1019                 spin_unlock_bh(&conn->nopin_timer_lock);
1020                 iscsit_dec_conn_usage_count(conn);
1021                 return;
1022         }
1023         conn->nopin_timer_flags &= ~ISCSI_TF_RUNNING;
1024         spin_unlock_bh(&conn->nopin_timer_lock);
1025
1026         iscsit_add_nopin(conn, 1);
1027         iscsit_dec_conn_usage_count(conn);
1028 }
1029
1030 /*
1031  * Called with conn->nopin_timer_lock held.
1032  */
1033 void __iscsit_start_nopin_timer(struct iscsi_conn *conn)
1034 {
1035         struct iscsi_session *sess = conn->sess;
1036         struct iscsi_node_attrib *na = iscsit_tpg_get_node_attrib(sess);
1037         /*
1038         * NOPIN timeout is disabled.
1039          */
1040         if (!na->nopin_timeout)
1041                 return;
1042
1043         if (conn->nopin_timer_flags & ISCSI_TF_RUNNING)
1044                 return;
1045
1046         conn->nopin_timer_flags &= ~ISCSI_TF_STOP;
1047         conn->nopin_timer_flags |= ISCSI_TF_RUNNING;
1048         mod_timer(&conn->nopin_timer, jiffies + na->nopin_timeout * HZ);
1049
1050         pr_debug("Started NOPIN Timer on CID: %d at %u second"
1051                 " interval\n", conn->cid, na->nopin_timeout);
1052 }
1053
1054 void iscsit_start_nopin_timer(struct iscsi_conn *conn)
1055 {
1056         spin_lock_bh(&conn->nopin_timer_lock);
1057         __iscsit_start_nopin_timer(conn);
1058         spin_unlock_bh(&conn->nopin_timer_lock);
1059 }
1060
1061 void iscsit_stop_nopin_timer(struct iscsi_conn *conn)
1062 {
1063         spin_lock_bh(&conn->nopin_timer_lock);
1064         if (!(conn->nopin_timer_flags & ISCSI_TF_RUNNING)) {
1065                 spin_unlock_bh(&conn->nopin_timer_lock);
1066                 return;
1067         }
1068         conn->nopin_timer_flags |= ISCSI_TF_STOP;
1069         spin_unlock_bh(&conn->nopin_timer_lock);
1070
1071         del_timer_sync(&conn->nopin_timer);
1072
1073         spin_lock_bh(&conn->nopin_timer_lock);
1074         conn->nopin_timer_flags &= ~ISCSI_TF_RUNNING;
1075         spin_unlock_bh(&conn->nopin_timer_lock);
1076 }
1077
1078 int iscsit_send_tx_data(
1079         struct iscsi_cmd *cmd,
1080         struct iscsi_conn *conn,
1081         int use_misc)
1082 {
1083         int tx_sent, tx_size;
1084         u32 iov_count;
1085         struct kvec *iov;
1086
1087 send_data:
1088         tx_size = cmd->tx_size;
1089
1090         if (!use_misc) {
1091                 iov = &cmd->iov_data[0];
1092                 iov_count = cmd->iov_data_count;
1093         } else {
1094                 iov = &cmd->iov_misc[0];
1095                 iov_count = cmd->iov_misc_count;
1096         }
1097
1098         tx_sent = tx_data(conn, &iov[0], iov_count, tx_size);
1099         if (tx_size != tx_sent) {
1100                 if (tx_sent == -EAGAIN) {
1101                         pr_err("tx_data() returned -EAGAIN\n");
1102                         goto send_data;
1103                 } else
1104                         return -1;
1105         }
1106         cmd->tx_size = 0;
1107
1108         return 0;
1109 }
1110
1111 int iscsit_fe_sendpage_sg(
1112         struct iscsi_cmd *cmd,
1113         struct iscsi_conn *conn)
1114 {
1115         struct scatterlist *sg = cmd->first_data_sg;
1116         struct kvec iov;
1117         u32 tx_hdr_size, data_len;
1118         u32 offset = cmd->first_data_sg_off;
1119         int tx_sent, iov_off;
1120
1121 send_hdr:
1122         tx_hdr_size = ISCSI_HDR_LEN;
1123         if (conn->conn_ops->HeaderDigest)
1124                 tx_hdr_size += ISCSI_CRC_LEN;
1125
1126         iov.iov_base = cmd->pdu;
1127         iov.iov_len = tx_hdr_size;
1128
1129         tx_sent = tx_data(conn, &iov, 1, tx_hdr_size);
1130         if (tx_hdr_size != tx_sent) {
1131                 if (tx_sent == -EAGAIN) {
1132                         pr_err("tx_data() returned -EAGAIN\n");
1133                         goto send_hdr;
1134                 }
1135                 return -1;
1136         }
1137
1138         data_len = cmd->tx_size - tx_hdr_size - cmd->padding;
1139         /*
1140          * Set iov_off used by padding and data digest tx_data() calls below
1141          * in order to determine proper offset into cmd->iov_data[]
1142          */
1143         if (conn->conn_ops->DataDigest) {
1144                 data_len -= ISCSI_CRC_LEN;
1145                 if (cmd->padding)
1146                         iov_off = (cmd->iov_data_count - 2);
1147                 else
1148                         iov_off = (cmd->iov_data_count - 1);
1149         } else {
1150                 iov_off = (cmd->iov_data_count - 1);
1151         }
1152         /*
1153          * Perform sendpage() for each page in the scatterlist
1154          */
1155         while (data_len) {
1156                 u32 space = (sg->length - offset);
1157                 u32 sub_len = min_t(u32, data_len, space);
1158 send_pg:
1159                 tx_sent = conn->sock->ops->sendpage(conn->sock,
1160                                         sg_page(sg), sg->offset + offset, sub_len, 0);
1161                 if (tx_sent != sub_len) {
1162                         if (tx_sent == -EAGAIN) {
1163                                 pr_err("tcp_sendpage() returned"
1164                                                 " -EAGAIN\n");
1165                                 goto send_pg;
1166                         }
1167
1168                         pr_err("tcp_sendpage() failure: %d\n",
1169                                         tx_sent);
1170                         return -1;
1171                 }
1172
1173                 data_len -= sub_len;
1174                 offset = 0;
1175                 sg = sg_next(sg);
1176         }
1177
1178 send_padding:
1179         if (cmd->padding) {
1180                 struct kvec *iov_p = &cmd->iov_data[iov_off++];
1181
1182                 tx_sent = tx_data(conn, iov_p, 1, cmd->padding);
1183                 if (cmd->padding != tx_sent) {
1184                         if (tx_sent == -EAGAIN) {
1185                                 pr_err("tx_data() returned -EAGAIN\n");
1186                                 goto send_padding;
1187                         }
1188                         return -1;
1189                 }
1190         }
1191
1192 send_datacrc:
1193         if (conn->conn_ops->DataDigest) {
1194                 struct kvec *iov_d = &cmd->iov_data[iov_off];
1195
1196                 tx_sent = tx_data(conn, iov_d, 1, ISCSI_CRC_LEN);
1197                 if (ISCSI_CRC_LEN != tx_sent) {
1198                         if (tx_sent == -EAGAIN) {
1199                                 pr_err("tx_data() returned -EAGAIN\n");
1200                                 goto send_datacrc;
1201                         }
1202                         return -1;
1203                 }
1204         }
1205
1206         return 0;
1207 }
1208
1209 /*
1210  *      This function is used for mainly sending a ISCSI_TARG_LOGIN_RSP PDU
1211  *      back to the Initiator when an expection condition occurs with the
1212  *      errors set in status_class and status_detail.
1213  *
1214  *      Parameters:     iSCSI Connection, Status Class, Status Detail.
1215  *      Returns:        0 on success, -1 on error.
1216  */
1217 int iscsit_tx_login_rsp(struct iscsi_conn *conn, u8 status_class, u8 status_detail)
1218 {
1219         struct iscsi_login_rsp *hdr;
1220         struct iscsi_login *login = conn->conn_login;
1221
1222         login->login_failed = 1;
1223         iscsit_collect_login_stats(conn, status_class, status_detail);
1224
1225         memset(&login->rsp[0], 0, ISCSI_HDR_LEN);
1226
1227         hdr     = (struct iscsi_login_rsp *)&login->rsp[0];
1228         hdr->opcode             = ISCSI_OP_LOGIN_RSP;
1229         hdr->status_class       = status_class;
1230         hdr->status_detail      = status_detail;
1231         hdr->itt                = conn->login_itt;
1232
1233         return conn->conn_transport->iscsit_put_login_tx(conn, login, 0);
1234 }
1235
1236 void iscsit_print_session_params(struct iscsi_session *sess)
1237 {
1238         struct iscsi_conn *conn;
1239
1240         pr_debug("-----------------------------[Session Params for"
1241                 " SID: %u]-----------------------------\n", sess->sid);
1242         spin_lock_bh(&sess->conn_lock);
1243         list_for_each_entry(conn, &sess->sess_conn_list, conn_list)
1244                 iscsi_dump_conn_ops(conn->conn_ops);
1245         spin_unlock_bh(&sess->conn_lock);
1246
1247         iscsi_dump_sess_ops(sess->sess_ops);
1248 }
1249
1250 static int iscsit_do_rx_data(
1251         struct iscsi_conn *conn,
1252         struct iscsi_data_count *count)
1253 {
1254         int data = count->data_length, rx_loop = 0, total_rx = 0;
1255         struct msghdr msg;
1256
1257         if (!conn || !conn->sock || !conn->conn_ops)
1258                 return -1;
1259
1260         memset(&msg, 0, sizeof(struct msghdr));
1261         iov_iter_kvec(&msg.msg_iter, READ | ITER_KVEC,
1262                       count->iov, count->iov_count, data);
1263
1264         while (msg_data_left(&msg)) {
1265                 rx_loop = sock_recvmsg(conn->sock, &msg, MSG_WAITALL);
1266                 if (rx_loop <= 0) {
1267                         pr_debug("rx_loop: %d total_rx: %d\n",
1268                                 rx_loop, total_rx);
1269                         return rx_loop;
1270                 }
1271                 total_rx += rx_loop;
1272                 pr_debug("rx_loop: %d, total_rx: %d, data: %d\n",
1273                                 rx_loop, total_rx, data);
1274         }
1275
1276         return total_rx;
1277 }
1278
1279 int rx_data(
1280         struct iscsi_conn *conn,
1281         struct kvec *iov,
1282         int iov_count,
1283         int data)
1284 {
1285         struct iscsi_data_count c;
1286
1287         if (!conn || !conn->sock || !conn->conn_ops)
1288                 return -1;
1289
1290         memset(&c, 0, sizeof(struct iscsi_data_count));
1291         c.iov = iov;
1292         c.iov_count = iov_count;
1293         c.data_length = data;
1294         c.type = ISCSI_RX_DATA;
1295
1296         return iscsit_do_rx_data(conn, &c);
1297 }
1298
1299 int tx_data(
1300         struct iscsi_conn *conn,
1301         struct kvec *iov,
1302         int iov_count,
1303         int data)
1304 {
1305         struct msghdr msg;
1306         int total_tx = 0;
1307
1308         if (!conn || !conn->sock || !conn->conn_ops)
1309                 return -1;
1310
1311         if (data <= 0) {
1312                 pr_err("Data length is: %d\n", data);
1313                 return -1;
1314         }
1315
1316         memset(&msg, 0, sizeof(struct msghdr));
1317
1318         iov_iter_kvec(&msg.msg_iter, WRITE | ITER_KVEC,
1319                       iov, iov_count, data);
1320
1321         while (msg_data_left(&msg)) {
1322                 int tx_loop = sock_sendmsg(conn->sock, &msg);
1323                 if (tx_loop <= 0) {
1324                         pr_debug("tx_loop: %d total_tx %d\n",
1325                                 tx_loop, total_tx);
1326                         return tx_loop;
1327                 }
1328                 total_tx += tx_loop;
1329                 pr_debug("tx_loop: %d, total_tx: %d, data: %d\n",
1330                                         tx_loop, total_tx, data);
1331         }
1332
1333         return total_tx;
1334 }
1335
1336 void iscsit_collect_login_stats(
1337         struct iscsi_conn *conn,
1338         u8 status_class,
1339         u8 status_detail)
1340 {
1341         struct iscsi_param *intrname = NULL;
1342         struct iscsi_tiqn *tiqn;
1343         struct iscsi_login_stats *ls;
1344
1345         tiqn = iscsit_snmp_get_tiqn(conn);
1346         if (!tiqn)
1347                 return;
1348
1349         ls = &tiqn->login_stats;
1350
1351         spin_lock(&ls->lock);
1352         if (status_class == ISCSI_STATUS_CLS_SUCCESS)
1353                 ls->accepts++;
1354         else if (status_class == ISCSI_STATUS_CLS_REDIRECT) {
1355                 ls->redirects++;
1356                 ls->last_fail_type = ISCSI_LOGIN_FAIL_REDIRECT;
1357         } else if ((status_class == ISCSI_STATUS_CLS_INITIATOR_ERR)  &&
1358                  (status_detail == ISCSI_LOGIN_STATUS_AUTH_FAILED)) {
1359                 ls->authenticate_fails++;
1360                 ls->last_fail_type =  ISCSI_LOGIN_FAIL_AUTHENTICATE;
1361         } else if ((status_class == ISCSI_STATUS_CLS_INITIATOR_ERR)  &&
1362                  (status_detail == ISCSI_LOGIN_STATUS_TGT_FORBIDDEN)) {
1363                 ls->authorize_fails++;
1364                 ls->last_fail_type = ISCSI_LOGIN_FAIL_AUTHORIZE;
1365         } else if ((status_class == ISCSI_STATUS_CLS_INITIATOR_ERR) &&
1366                  (status_detail == ISCSI_LOGIN_STATUS_INIT_ERR)) {
1367                 ls->negotiate_fails++;
1368                 ls->last_fail_type = ISCSI_LOGIN_FAIL_NEGOTIATE;
1369         } else {
1370                 ls->other_fails++;
1371                 ls->last_fail_type = ISCSI_LOGIN_FAIL_OTHER;
1372         }
1373
1374         /* Save initiator name, ip address and time, if it is a failed login */
1375         if (status_class != ISCSI_STATUS_CLS_SUCCESS) {
1376                 if (conn->param_list)
1377                         intrname = iscsi_find_param_from_key(INITIATORNAME,
1378                                                              conn->param_list);
1379                 strlcpy(ls->last_intr_fail_name,
1380                        (intrname ? intrname->value : "Unknown"),
1381                        sizeof(ls->last_intr_fail_name));
1382
1383                 ls->last_intr_fail_ip_family = conn->login_family;
1384
1385                 ls->last_intr_fail_sockaddr = conn->login_sockaddr;
1386                 ls->last_fail_time = get_jiffies_64();
1387         }
1388
1389         spin_unlock(&ls->lock);
1390 }
1391
1392 struct iscsi_tiqn *iscsit_snmp_get_tiqn(struct iscsi_conn *conn)
1393 {
1394         struct iscsi_portal_group *tpg;
1395
1396         if (!conn)
1397                 return NULL;
1398
1399         tpg = conn->tpg;
1400         if (!tpg)
1401                 return NULL;
1402
1403         if (!tpg->tpg_tiqn)
1404                 return NULL;
1405
1406         return tpg->tpg_tiqn;
1407 }