Merge drm/drm-next into drm-intel-next-queued
[linux-2.6-microblaze.git] / 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 int iscsit_add_r2t_to_list(
60         struct iscsi_cmd *cmd,
61         u32 offset,
62         u32 xfer_len,
63         int recovery,
64         u32 r2t_sn)
65 {
66         struct iscsi_r2t *r2t;
67
68         lockdep_assert_held(&cmd->r2t_lock);
69
70         r2t = kmem_cache_zalloc(lio_r2t_cache, GFP_ATOMIC);
71         if (!r2t) {
72                 pr_err("Unable to allocate memory for struct iscsi_r2t.\n");
73                 return -1;
74         }
75         INIT_LIST_HEAD(&r2t->r2t_list);
76
77         r2t->recovery_r2t = recovery;
78         r2t->r2t_sn = (!r2t_sn) ? cmd->r2t_sn++ : r2t_sn;
79         r2t->offset = offset;
80         r2t->xfer_len = xfer_len;
81         list_add_tail(&r2t->r2t_list, &cmd->cmd_r2t_list);
82         spin_unlock_bh(&cmd->r2t_lock);
83
84         iscsit_add_cmd_to_immediate_queue(cmd, cmd->conn, ISTATE_SEND_R2T);
85
86         spin_lock_bh(&cmd->r2t_lock);
87         return 0;
88 }
89
90 struct iscsi_r2t *iscsit_get_r2t_for_eos(
91         struct iscsi_cmd *cmd,
92         u32 offset,
93         u32 length)
94 {
95         struct iscsi_r2t *r2t;
96
97         spin_lock_bh(&cmd->r2t_lock);
98         list_for_each_entry(r2t, &cmd->cmd_r2t_list, r2t_list) {
99                 if ((r2t->offset <= offset) &&
100                     (r2t->offset + r2t->xfer_len) >= (offset + length)) {
101                         spin_unlock_bh(&cmd->r2t_lock);
102                         return r2t;
103                 }
104         }
105         spin_unlock_bh(&cmd->r2t_lock);
106
107         pr_err("Unable to locate R2T for Offset: %u, Length:"
108                         " %u\n", offset, length);
109         return NULL;
110 }
111
112 struct iscsi_r2t *iscsit_get_r2t_from_list(struct iscsi_cmd *cmd)
113 {
114         struct iscsi_r2t *r2t;
115
116         spin_lock_bh(&cmd->r2t_lock);
117         list_for_each_entry(r2t, &cmd->cmd_r2t_list, r2t_list) {
118                 if (!r2t->sent_r2t) {
119                         spin_unlock_bh(&cmd->r2t_lock);
120                         return r2t;
121                 }
122         }
123         spin_unlock_bh(&cmd->r2t_lock);
124
125         pr_err("Unable to locate next R2T to send for ITT:"
126                         " 0x%08x.\n", cmd->init_task_tag);
127         return NULL;
128 }
129
130 void iscsit_free_r2t(struct iscsi_r2t *r2t, struct iscsi_cmd *cmd)
131 {
132         lockdep_assert_held(&cmd->r2t_lock);
133
134         list_del(&r2t->r2t_list);
135         kmem_cache_free(lio_r2t_cache, r2t);
136 }
137
138 void iscsit_free_r2ts_from_list(struct iscsi_cmd *cmd)
139 {
140         struct iscsi_r2t *r2t, *r2t_tmp;
141
142         spin_lock_bh(&cmd->r2t_lock);
143         list_for_each_entry_safe(r2t, r2t_tmp, &cmd->cmd_r2t_list, r2t_list)
144                 iscsit_free_r2t(r2t, cmd);
145         spin_unlock_bh(&cmd->r2t_lock);
146 }
147
148 static int iscsit_wait_for_tag(struct se_session *se_sess, int state, int *cpup)
149 {
150         int tag = -1;
151         DEFINE_SBQ_WAIT(wait);
152         struct sbq_wait_state *ws;
153         struct sbitmap_queue *sbq;
154
155         if (state == TASK_RUNNING)
156                 return tag;
157
158         sbq = &se_sess->sess_tag_pool;
159         ws = &sbq->ws[0];
160         for (;;) {
161                 sbitmap_prepare_to_wait(sbq, ws, &wait, state);
162                 if (signal_pending_state(state, current))
163                         break;
164                 tag = sbitmap_queue_get(sbq, cpup);
165                 if (tag >= 0)
166                         break;
167                 schedule();
168         }
169
170         sbitmap_finish_wait(sbq, ws, &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_unmap_cmd)
764                 conn->conn_transport->iscsit_unmap_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         struct iscsi_session *sess = conn->sess;
919
920         iscsit_inc_conn_usage_count(conn);
921
922         spin_lock_bh(&conn->nopin_timer_lock);
923         if (conn->nopin_response_timer_flags & ISCSI_TF_STOP) {
924                 spin_unlock_bh(&conn->nopin_timer_lock);
925                 iscsit_dec_conn_usage_count(conn);
926                 return;
927         }
928
929         pr_err("Did not receive response to NOPIN on CID: %hu, failing"
930                 " connection for I_T Nexus %s,i,0x%6phN,%s,t,0x%02x\n",
931                 conn->cid, sess->sess_ops->InitiatorName, sess->isid,
932                 sess->tpg->tpg_tiqn->tiqn, (u32)sess->tpg->tpgt);
933         conn->nopin_response_timer_flags &= ~ISCSI_TF_RUNNING;
934         spin_unlock_bh(&conn->nopin_timer_lock);
935
936         iscsit_fill_cxn_timeout_err_stats(sess);
937         iscsit_cause_connection_reinstatement(conn, 0);
938         iscsit_dec_conn_usage_count(conn);
939 }
940
941 void iscsit_mod_nopin_response_timer(struct iscsi_conn *conn)
942 {
943         struct iscsi_session *sess = conn->sess;
944         struct iscsi_node_attrib *na = iscsit_tpg_get_node_attrib(sess);
945
946         spin_lock_bh(&conn->nopin_timer_lock);
947         if (!(conn->nopin_response_timer_flags & ISCSI_TF_RUNNING)) {
948                 spin_unlock_bh(&conn->nopin_timer_lock);
949                 return;
950         }
951
952         mod_timer(&conn->nopin_response_timer,
953                 (get_jiffies_64() + na->nopin_response_timeout * HZ));
954         spin_unlock_bh(&conn->nopin_timer_lock);
955 }
956
957 void iscsit_start_nopin_response_timer(struct iscsi_conn *conn)
958 {
959         struct iscsi_session *sess = conn->sess;
960         struct iscsi_node_attrib *na = iscsit_tpg_get_node_attrib(sess);
961
962         spin_lock_bh(&conn->nopin_timer_lock);
963         if (conn->nopin_response_timer_flags & ISCSI_TF_RUNNING) {
964                 spin_unlock_bh(&conn->nopin_timer_lock);
965                 return;
966         }
967
968         conn->nopin_response_timer_flags &= ~ISCSI_TF_STOP;
969         conn->nopin_response_timer_flags |= ISCSI_TF_RUNNING;
970         mod_timer(&conn->nopin_response_timer,
971                   jiffies + na->nopin_response_timeout * HZ);
972
973         pr_debug("Started NOPIN Response Timer on CID: %d to %u"
974                 " seconds\n", conn->cid, na->nopin_response_timeout);
975         spin_unlock_bh(&conn->nopin_timer_lock);
976 }
977
978 void iscsit_stop_nopin_response_timer(struct iscsi_conn *conn)
979 {
980         spin_lock_bh(&conn->nopin_timer_lock);
981         if (!(conn->nopin_response_timer_flags & ISCSI_TF_RUNNING)) {
982                 spin_unlock_bh(&conn->nopin_timer_lock);
983                 return;
984         }
985         conn->nopin_response_timer_flags |= ISCSI_TF_STOP;
986         spin_unlock_bh(&conn->nopin_timer_lock);
987
988         del_timer_sync(&conn->nopin_response_timer);
989
990         spin_lock_bh(&conn->nopin_timer_lock);
991         conn->nopin_response_timer_flags &= ~ISCSI_TF_RUNNING;
992         spin_unlock_bh(&conn->nopin_timer_lock);
993 }
994
995 void iscsit_handle_nopin_timeout(struct timer_list *t)
996 {
997         struct iscsi_conn *conn = from_timer(conn, t, nopin_timer);
998
999         iscsit_inc_conn_usage_count(conn);
1000
1001         spin_lock_bh(&conn->nopin_timer_lock);
1002         if (conn->nopin_timer_flags & ISCSI_TF_STOP) {
1003                 spin_unlock_bh(&conn->nopin_timer_lock);
1004                 iscsit_dec_conn_usage_count(conn);
1005                 return;
1006         }
1007         conn->nopin_timer_flags &= ~ISCSI_TF_RUNNING;
1008         spin_unlock_bh(&conn->nopin_timer_lock);
1009
1010         iscsit_add_nopin(conn, 1);
1011         iscsit_dec_conn_usage_count(conn);
1012 }
1013
1014 void __iscsit_start_nopin_timer(struct iscsi_conn *conn)
1015 {
1016         struct iscsi_session *sess = conn->sess;
1017         struct iscsi_node_attrib *na = iscsit_tpg_get_node_attrib(sess);
1018
1019         lockdep_assert_held(&conn->nopin_timer_lock);
1020
1021         /*
1022         * NOPIN timeout is disabled.
1023          */
1024         if (!na->nopin_timeout)
1025                 return;
1026
1027         if (conn->nopin_timer_flags & ISCSI_TF_RUNNING)
1028                 return;
1029
1030         conn->nopin_timer_flags &= ~ISCSI_TF_STOP;
1031         conn->nopin_timer_flags |= ISCSI_TF_RUNNING;
1032         mod_timer(&conn->nopin_timer, jiffies + na->nopin_timeout * HZ);
1033
1034         pr_debug("Started NOPIN Timer on CID: %d at %u second"
1035                 " interval\n", conn->cid, na->nopin_timeout);
1036 }
1037
1038 void iscsit_start_nopin_timer(struct iscsi_conn *conn)
1039 {
1040         spin_lock_bh(&conn->nopin_timer_lock);
1041         __iscsit_start_nopin_timer(conn);
1042         spin_unlock_bh(&conn->nopin_timer_lock);
1043 }
1044
1045 void iscsit_stop_nopin_timer(struct iscsi_conn *conn)
1046 {
1047         spin_lock_bh(&conn->nopin_timer_lock);
1048         if (!(conn->nopin_timer_flags & ISCSI_TF_RUNNING)) {
1049                 spin_unlock_bh(&conn->nopin_timer_lock);
1050                 return;
1051         }
1052         conn->nopin_timer_flags |= ISCSI_TF_STOP;
1053         spin_unlock_bh(&conn->nopin_timer_lock);
1054
1055         del_timer_sync(&conn->nopin_timer);
1056
1057         spin_lock_bh(&conn->nopin_timer_lock);
1058         conn->nopin_timer_flags &= ~ISCSI_TF_RUNNING;
1059         spin_unlock_bh(&conn->nopin_timer_lock);
1060 }
1061
1062 int iscsit_send_tx_data(
1063         struct iscsi_cmd *cmd,
1064         struct iscsi_conn *conn,
1065         int use_misc)
1066 {
1067         int tx_sent, tx_size;
1068         u32 iov_count;
1069         struct kvec *iov;
1070
1071 send_data:
1072         tx_size = cmd->tx_size;
1073
1074         if (!use_misc) {
1075                 iov = &cmd->iov_data[0];
1076                 iov_count = cmd->iov_data_count;
1077         } else {
1078                 iov = &cmd->iov_misc[0];
1079                 iov_count = cmd->iov_misc_count;
1080         }
1081
1082         tx_sent = tx_data(conn, &iov[0], iov_count, tx_size);
1083         if (tx_size != tx_sent) {
1084                 if (tx_sent == -EAGAIN) {
1085                         pr_err("tx_data() returned -EAGAIN\n");
1086                         goto send_data;
1087                 } else
1088                         return -1;
1089         }
1090         cmd->tx_size = 0;
1091
1092         return 0;
1093 }
1094
1095 int iscsit_fe_sendpage_sg(
1096         struct iscsi_cmd *cmd,
1097         struct iscsi_conn *conn)
1098 {
1099         struct scatterlist *sg = cmd->first_data_sg;
1100         struct kvec iov;
1101         u32 tx_hdr_size, data_len;
1102         u32 offset = cmd->first_data_sg_off;
1103         int tx_sent, iov_off;
1104
1105 send_hdr:
1106         tx_hdr_size = ISCSI_HDR_LEN;
1107         if (conn->conn_ops->HeaderDigest)
1108                 tx_hdr_size += ISCSI_CRC_LEN;
1109
1110         iov.iov_base = cmd->pdu;
1111         iov.iov_len = tx_hdr_size;
1112
1113         tx_sent = tx_data(conn, &iov, 1, tx_hdr_size);
1114         if (tx_hdr_size != tx_sent) {
1115                 if (tx_sent == -EAGAIN) {
1116                         pr_err("tx_data() returned -EAGAIN\n");
1117                         goto send_hdr;
1118                 }
1119                 return -1;
1120         }
1121
1122         data_len = cmd->tx_size - tx_hdr_size - cmd->padding;
1123         /*
1124          * Set iov_off used by padding and data digest tx_data() calls below
1125          * in order to determine proper offset into cmd->iov_data[]
1126          */
1127         if (conn->conn_ops->DataDigest) {
1128                 data_len -= ISCSI_CRC_LEN;
1129                 if (cmd->padding)
1130                         iov_off = (cmd->iov_data_count - 2);
1131                 else
1132                         iov_off = (cmd->iov_data_count - 1);
1133         } else {
1134                 iov_off = (cmd->iov_data_count - 1);
1135         }
1136         /*
1137          * Perform sendpage() for each page in the scatterlist
1138          */
1139         while (data_len) {
1140                 u32 space = (sg->length - offset);
1141                 u32 sub_len = min_t(u32, data_len, space);
1142 send_pg:
1143                 tx_sent = conn->sock->ops->sendpage(conn->sock,
1144                                         sg_page(sg), sg->offset + offset, sub_len, 0);
1145                 if (tx_sent != sub_len) {
1146                         if (tx_sent == -EAGAIN) {
1147                                 pr_err("tcp_sendpage() returned"
1148                                                 " -EAGAIN\n");
1149                                 goto send_pg;
1150                         }
1151
1152                         pr_err("tcp_sendpage() failure: %d\n",
1153                                         tx_sent);
1154                         return -1;
1155                 }
1156
1157                 data_len -= sub_len;
1158                 offset = 0;
1159                 sg = sg_next(sg);
1160         }
1161
1162 send_padding:
1163         if (cmd->padding) {
1164                 struct kvec *iov_p = &cmd->iov_data[iov_off++];
1165
1166                 tx_sent = tx_data(conn, iov_p, 1, cmd->padding);
1167                 if (cmd->padding != tx_sent) {
1168                         if (tx_sent == -EAGAIN) {
1169                                 pr_err("tx_data() returned -EAGAIN\n");
1170                                 goto send_padding;
1171                         }
1172                         return -1;
1173                 }
1174         }
1175
1176 send_datacrc:
1177         if (conn->conn_ops->DataDigest) {
1178                 struct kvec *iov_d = &cmd->iov_data[iov_off];
1179
1180                 tx_sent = tx_data(conn, iov_d, 1, ISCSI_CRC_LEN);
1181                 if (ISCSI_CRC_LEN != tx_sent) {
1182                         if (tx_sent == -EAGAIN) {
1183                                 pr_err("tx_data() returned -EAGAIN\n");
1184                                 goto send_datacrc;
1185                         }
1186                         return -1;
1187                 }
1188         }
1189
1190         return 0;
1191 }
1192
1193 /*
1194  *      This function is used for mainly sending a ISCSI_TARG_LOGIN_RSP PDU
1195  *      back to the Initiator when an expection condition occurs with the
1196  *      errors set in status_class and status_detail.
1197  *
1198  *      Parameters:     iSCSI Connection, Status Class, Status Detail.
1199  *      Returns:        0 on success, -1 on error.
1200  */
1201 int iscsit_tx_login_rsp(struct iscsi_conn *conn, u8 status_class, u8 status_detail)
1202 {
1203         struct iscsi_login_rsp *hdr;
1204         struct iscsi_login *login = conn->conn_login;
1205
1206         login->login_failed = 1;
1207         iscsit_collect_login_stats(conn, status_class, status_detail);
1208
1209         memset(&login->rsp[0], 0, ISCSI_HDR_LEN);
1210
1211         hdr     = (struct iscsi_login_rsp *)&login->rsp[0];
1212         hdr->opcode             = ISCSI_OP_LOGIN_RSP;
1213         hdr->status_class       = status_class;
1214         hdr->status_detail      = status_detail;
1215         hdr->itt                = conn->login_itt;
1216
1217         return conn->conn_transport->iscsit_put_login_tx(conn, login, 0);
1218 }
1219
1220 void iscsit_print_session_params(struct iscsi_session *sess)
1221 {
1222         struct iscsi_conn *conn;
1223
1224         pr_debug("-----------------------------[Session Params for"
1225                 " SID: %u]-----------------------------\n", sess->sid);
1226         spin_lock_bh(&sess->conn_lock);
1227         list_for_each_entry(conn, &sess->sess_conn_list, conn_list)
1228                 iscsi_dump_conn_ops(conn->conn_ops);
1229         spin_unlock_bh(&sess->conn_lock);
1230
1231         iscsi_dump_sess_ops(sess->sess_ops);
1232 }
1233
1234 static int iscsit_do_rx_data(
1235         struct iscsi_conn *conn,
1236         struct iscsi_data_count *count)
1237 {
1238         int data = count->data_length, rx_loop = 0, total_rx = 0;
1239         struct msghdr msg;
1240
1241         if (!conn || !conn->sock || !conn->conn_ops)
1242                 return -1;
1243
1244         memset(&msg, 0, sizeof(struct msghdr));
1245         iov_iter_kvec(&msg.msg_iter, READ, count->iov, count->iov_count, data);
1246
1247         while (msg_data_left(&msg)) {
1248                 rx_loop = sock_recvmsg(conn->sock, &msg, MSG_WAITALL);
1249                 if (rx_loop <= 0) {
1250                         pr_debug("rx_loop: %d total_rx: %d\n",
1251                                 rx_loop, total_rx);
1252                         return rx_loop;
1253                 }
1254                 total_rx += rx_loop;
1255                 pr_debug("rx_loop: %d, total_rx: %d, data: %d\n",
1256                                 rx_loop, total_rx, data);
1257         }
1258
1259         return total_rx;
1260 }
1261
1262 int rx_data(
1263         struct iscsi_conn *conn,
1264         struct kvec *iov,
1265         int iov_count,
1266         int data)
1267 {
1268         struct iscsi_data_count c;
1269
1270         if (!conn || !conn->sock || !conn->conn_ops)
1271                 return -1;
1272
1273         memset(&c, 0, sizeof(struct iscsi_data_count));
1274         c.iov = iov;
1275         c.iov_count = iov_count;
1276         c.data_length = data;
1277         c.type = ISCSI_RX_DATA;
1278
1279         return iscsit_do_rx_data(conn, &c);
1280 }
1281
1282 int tx_data(
1283         struct iscsi_conn *conn,
1284         struct kvec *iov,
1285         int iov_count,
1286         int data)
1287 {
1288         struct msghdr msg;
1289         int total_tx = 0;
1290
1291         if (!conn || !conn->sock || !conn->conn_ops)
1292                 return -1;
1293
1294         if (data <= 0) {
1295                 pr_err("Data length is: %d\n", data);
1296                 return -1;
1297         }
1298
1299         memset(&msg, 0, sizeof(struct msghdr));
1300
1301         iov_iter_kvec(&msg.msg_iter, WRITE, iov, iov_count, data);
1302
1303         while (msg_data_left(&msg)) {
1304                 int tx_loop = sock_sendmsg(conn->sock, &msg);
1305                 if (tx_loop <= 0) {
1306                         pr_debug("tx_loop: %d total_tx %d\n",
1307                                 tx_loop, total_tx);
1308                         return tx_loop;
1309                 }
1310                 total_tx += tx_loop;
1311                 pr_debug("tx_loop: %d, total_tx: %d, data: %d\n",
1312                                         tx_loop, total_tx, data);
1313         }
1314
1315         return total_tx;
1316 }
1317
1318 void iscsit_collect_login_stats(
1319         struct iscsi_conn *conn,
1320         u8 status_class,
1321         u8 status_detail)
1322 {
1323         struct iscsi_param *intrname = NULL;
1324         struct iscsi_tiqn *tiqn;
1325         struct iscsi_login_stats *ls;
1326
1327         tiqn = iscsit_snmp_get_tiqn(conn);
1328         if (!tiqn)
1329                 return;
1330
1331         ls = &tiqn->login_stats;
1332
1333         spin_lock(&ls->lock);
1334         if (status_class == ISCSI_STATUS_CLS_SUCCESS)
1335                 ls->accepts++;
1336         else if (status_class == ISCSI_STATUS_CLS_REDIRECT) {
1337                 ls->redirects++;
1338                 ls->last_fail_type = ISCSI_LOGIN_FAIL_REDIRECT;
1339         } else if ((status_class == ISCSI_STATUS_CLS_INITIATOR_ERR)  &&
1340                  (status_detail == ISCSI_LOGIN_STATUS_AUTH_FAILED)) {
1341                 ls->authenticate_fails++;
1342                 ls->last_fail_type =  ISCSI_LOGIN_FAIL_AUTHENTICATE;
1343         } else if ((status_class == ISCSI_STATUS_CLS_INITIATOR_ERR)  &&
1344                  (status_detail == ISCSI_LOGIN_STATUS_TGT_FORBIDDEN)) {
1345                 ls->authorize_fails++;
1346                 ls->last_fail_type = ISCSI_LOGIN_FAIL_AUTHORIZE;
1347         } else if ((status_class == ISCSI_STATUS_CLS_INITIATOR_ERR) &&
1348                  (status_detail == ISCSI_LOGIN_STATUS_INIT_ERR)) {
1349                 ls->negotiate_fails++;
1350                 ls->last_fail_type = ISCSI_LOGIN_FAIL_NEGOTIATE;
1351         } else {
1352                 ls->other_fails++;
1353                 ls->last_fail_type = ISCSI_LOGIN_FAIL_OTHER;
1354         }
1355
1356         /* Save initiator name, ip address and time, if it is a failed login */
1357         if (status_class != ISCSI_STATUS_CLS_SUCCESS) {
1358                 if (conn->param_list)
1359                         intrname = iscsi_find_param_from_key(INITIATORNAME,
1360                                                              conn->param_list);
1361                 strlcpy(ls->last_intr_fail_name,
1362                        (intrname ? intrname->value : "Unknown"),
1363                        sizeof(ls->last_intr_fail_name));
1364
1365                 ls->last_intr_fail_ip_family = conn->login_family;
1366
1367                 ls->last_intr_fail_sockaddr = conn->login_sockaddr;
1368                 ls->last_fail_time = get_jiffies_64();
1369         }
1370
1371         spin_unlock(&ls->lock);
1372 }
1373
1374 struct iscsi_tiqn *iscsit_snmp_get_tiqn(struct iscsi_conn *conn)
1375 {
1376         struct iscsi_portal_group *tpg;
1377
1378         if (!conn)
1379                 return NULL;
1380
1381         tpg = conn->tpg;
1382         if (!tpg)
1383                 return NULL;
1384
1385         if (!tpg->tpg_tiqn)
1386                 return NULL;
1387
1388         return tpg->tpg_tiqn;
1389 }
1390
1391 void iscsit_fill_cxn_timeout_err_stats(struct iscsi_session *sess)
1392 {
1393         struct iscsi_portal_group *tpg = sess->tpg;
1394         struct iscsi_tiqn *tiqn = tpg->tpg_tiqn;
1395
1396         if (!tiqn)
1397                 return;
1398
1399         spin_lock_bh(&tiqn->sess_err_stats.lock);
1400         strlcpy(tiqn->sess_err_stats.last_sess_fail_rem_name,
1401                         sess->sess_ops->InitiatorName,
1402                         sizeof(tiqn->sess_err_stats.last_sess_fail_rem_name));
1403         tiqn->sess_err_stats.last_sess_failure_type =
1404                         ISCSI_SESS_ERR_CXN_TIMEOUT;
1405         tiqn->sess_err_stats.cxn_timeout_errors++;
1406         atomic_long_inc(&sess->conn_timeout_errors);
1407         spin_unlock_bh(&tiqn->sess_err_stats.lock);
1408 }