1// SPDX-License-Identifier: GPL-2.0-or-later
2/*******************************************************************************
3 * This file contains main functions related to the iSCSI Target Core Driver.
4 *
5 * (c) Copyright 2007-2013 Datera, Inc.
6 *
7 * Author: Nicholas A. Bellinger <nab@linux-iscsi.org>
8 *
9 ******************************************************************************/
10
11#include <crypto/hash.h>
12#include <linux/string.h>
13#include <linux/kthread.h>
14#include <linux/completion.h>
15#include <linux/module.h>
16#include <linux/vmalloc.h>
17#include <linux/idr.h>
18#include <linux/delay.h>
19#include <linux/sched/signal.h>
20#include <asm/unaligned.h>
21#include <linux/inet.h>
22#include <net/ipv6.h>
23#include <scsi/scsi_proto.h>
24#include <scsi/iscsi_proto.h>
25#include <scsi/scsi_tcq.h>
26#include <target/target_core_base.h>
27#include <target/target_core_fabric.h>
28
29#include <target/target_core_backend.h>
30#include <target/iscsi/iscsi_target_core.h>
31#include "iscsi_target_parameters.h"
32#include "iscsi_target_seq_pdu_list.h"
33#include "iscsi_target_datain_values.h"
34#include "iscsi_target_erl0.h"
35#include "iscsi_target_erl1.h"
36#include "iscsi_target_erl2.h"
37#include "iscsi_target_login.h"
38#include "iscsi_target_tmr.h"
39#include "iscsi_target_tpg.h"
40#include "iscsi_target_util.h"
41#include "iscsi_target.h"
42#include "iscsi_target_device.h"
43#include <target/iscsi/iscsi_target_stat.h>
44
45#include <target/iscsi/iscsi_transport.h>
46
47static LIST_HEAD(g_tiqn_list);
48static LIST_HEAD(g_np_list);
49static DEFINE_SPINLOCK(tiqn_lock);
50static DEFINE_MUTEX(np_lock);
51
52static struct idr tiqn_idr;
53DEFINE_IDA(sess_ida);
54struct mutex auth_id_lock;
55
56struct iscsit_global *iscsit_global;
57
58struct kmem_cache *lio_qr_cache;
59struct kmem_cache *lio_dr_cache;
60struct kmem_cache *lio_ooo_cache;
61struct kmem_cache *lio_r2t_cache;
62
63static int iscsit_handle_immediate_data(struct iscsit_cmd *,
64			struct iscsi_scsi_req *, u32);
65
66struct iscsi_tiqn *iscsit_get_tiqn_for_login(unsigned char *buf)
67{
68	struct iscsi_tiqn *tiqn = NULL;
69
70	spin_lock(&tiqn_lock);
71	list_for_each_entry(tiqn, &g_tiqn_list, tiqn_list) {
72		if (!strcmp(tiqn->tiqn, buf)) {
73
74			spin_lock(&tiqn->tiqn_state_lock);
75			if (tiqn->tiqn_state == TIQN_STATE_ACTIVE) {
76				tiqn->tiqn_access_count++;
77				spin_unlock(&tiqn->tiqn_state_lock);
78				spin_unlock(&tiqn_lock);
79				return tiqn;
80			}
81			spin_unlock(&tiqn->tiqn_state_lock);
82		}
83	}
84	spin_unlock(&tiqn_lock);
85
86	return NULL;
87}
88
89static int iscsit_set_tiqn_shutdown(struct iscsi_tiqn *tiqn)
90{
91	spin_lock(&tiqn->tiqn_state_lock);
92	if (tiqn->tiqn_state == TIQN_STATE_ACTIVE) {
93		tiqn->tiqn_state = TIQN_STATE_SHUTDOWN;
94		spin_unlock(&tiqn->tiqn_state_lock);
95		return 0;
96	}
97	spin_unlock(&tiqn->tiqn_state_lock);
98
99	return -1;
100}
101
102void iscsit_put_tiqn_for_login(struct iscsi_tiqn *tiqn)
103{
104	spin_lock(&tiqn->tiqn_state_lock);
105	tiqn->tiqn_access_count--;
106	spin_unlock(&tiqn->tiqn_state_lock);
107}
108
109/*
110 * Note that IQN formatting is expected to be done in userspace, and
111 * no explict IQN format checks are done here.
112 */
113struct iscsi_tiqn *iscsit_add_tiqn(unsigned char *buf)
114{
115	struct iscsi_tiqn *tiqn = NULL;
116	int ret;
117
118	if (strlen(buf) >= ISCSI_IQN_LEN) {
119		pr_err("Target IQN exceeds %d bytes\n",
120				ISCSI_IQN_LEN);
121		return ERR_PTR(-EINVAL);
122	}
123
124	tiqn = kzalloc(sizeof(*tiqn), GFP_KERNEL);
125	if (!tiqn)
126		return ERR_PTR(-ENOMEM);
127
128	sprintf(tiqn->tiqn, "%s", buf);
129	INIT_LIST_HEAD(&tiqn->tiqn_list);
130	INIT_LIST_HEAD(&tiqn->tiqn_tpg_list);
131	spin_lock_init(&tiqn->tiqn_state_lock);
132	spin_lock_init(&tiqn->tiqn_tpg_lock);
133	spin_lock_init(&tiqn->sess_err_stats.lock);
134	spin_lock_init(&tiqn->login_stats.lock);
135	spin_lock_init(&tiqn->logout_stats.lock);
136
137	tiqn->tiqn_state = TIQN_STATE_ACTIVE;
138
139	idr_preload(GFP_KERNEL);
140	spin_lock(&tiqn_lock);
141
142	ret = idr_alloc(&tiqn_idr, NULL, 0, 0, GFP_NOWAIT);
143	if (ret < 0) {
144		pr_err("idr_alloc() failed for tiqn->tiqn_index\n");
145		spin_unlock(&tiqn_lock);
146		idr_preload_end();
147		kfree(tiqn);
148		return ERR_PTR(ret);
149	}
150	tiqn->tiqn_index = ret;
151	list_add_tail(&tiqn->tiqn_list, &g_tiqn_list);
152
153	spin_unlock(&tiqn_lock);
154	idr_preload_end();
155
156	pr_debug("CORE[0] - Added iSCSI Target IQN: %s\n", tiqn->tiqn);
157
158	return tiqn;
159
160}
161
162static void iscsit_wait_for_tiqn(struct iscsi_tiqn *tiqn)
163{
164	/*
165	 * Wait for accesses to said struct iscsi_tiqn to end.
166	 */
167	spin_lock(&tiqn->tiqn_state_lock);
168	while (tiqn->tiqn_access_count != 0) {
169		spin_unlock(&tiqn->tiqn_state_lock);
170		msleep(10);
171		spin_lock(&tiqn->tiqn_state_lock);
172	}
173	spin_unlock(&tiqn->tiqn_state_lock);
174}
175
176void iscsit_del_tiqn(struct iscsi_tiqn *tiqn)
177{
178	/*
179	 * iscsit_set_tiqn_shutdown sets tiqn->tiqn_state = TIQN_STATE_SHUTDOWN
180	 * while holding tiqn->tiqn_state_lock.  This means that all subsequent
181	 * attempts to access this struct iscsi_tiqn will fail from both transport
182	 * fabric and control code paths.
183	 */
184	if (iscsit_set_tiqn_shutdown(tiqn) < 0) {
185		pr_err("iscsit_set_tiqn_shutdown() failed\n");
186		return;
187	}
188
189	iscsit_wait_for_tiqn(tiqn);
190
191	spin_lock(&tiqn_lock);
192	list_del(&tiqn->tiqn_list);
193	idr_remove(&tiqn_idr, tiqn->tiqn_index);
194	spin_unlock(&tiqn_lock);
195
196	pr_debug("CORE[0] - Deleted iSCSI Target IQN: %s\n",
197			tiqn->tiqn);
198	kfree(tiqn);
199}
200
201int iscsit_access_np(struct iscsi_np *np, struct iscsi_portal_group *tpg)
202{
203	int ret;
204	/*
205	 * Determine if the network portal is accepting storage traffic.
206	 */
207	spin_lock_bh(&np->np_thread_lock);
208	if (np->np_thread_state != ISCSI_NP_THREAD_ACTIVE) {
209		spin_unlock_bh(&np->np_thread_lock);
210		return -1;
211	}
212	spin_unlock_bh(&np->np_thread_lock);
213	/*
214	 * Determine if the portal group is accepting storage traffic.
215	 */
216	spin_lock_bh(&tpg->tpg_state_lock);
217	if (tpg->tpg_state != TPG_STATE_ACTIVE) {
218		spin_unlock_bh(&tpg->tpg_state_lock);
219		return -1;
220	}
221	spin_unlock_bh(&tpg->tpg_state_lock);
222
223	/*
224	 * Here we serialize access across the TIQN+TPG Tuple.
225	 */
226	ret = down_interruptible(&tpg->np_login_sem);
227	if (ret != 0)
228		return -1;
229
230	spin_lock_bh(&tpg->tpg_state_lock);
231	if (tpg->tpg_state != TPG_STATE_ACTIVE) {
232		spin_unlock_bh(&tpg->tpg_state_lock);
233		up(&tpg->np_login_sem);
234		return -1;
235	}
236	spin_unlock_bh(&tpg->tpg_state_lock);
237
238	return 0;
239}
240
241void iscsit_login_kref_put(struct kref *kref)
242{
243	struct iscsi_tpg_np *tpg_np = container_of(kref,
244				struct iscsi_tpg_np, tpg_np_kref);
245
246	complete(&tpg_np->tpg_np_comp);
247}
248
249int iscsit_deaccess_np(struct iscsi_np *np, struct iscsi_portal_group *tpg,
250		       struct iscsi_tpg_np *tpg_np)
251{
252	struct iscsi_tiqn *tiqn = tpg->tpg_tiqn;
253
254	up(&tpg->np_login_sem);
255
256	if (tpg_np)
257		kref_put(&tpg_np->tpg_np_kref, iscsit_login_kref_put);
258
259	if (tiqn)
260		iscsit_put_tiqn_for_login(tiqn);
261
262	return 0;
263}
264
265bool iscsit_check_np_match(
266	struct sockaddr_storage *sockaddr,
267	struct iscsi_np *np,
268	int network_transport)
269{
270	struct sockaddr_in *sock_in, *sock_in_e;
271	struct sockaddr_in6 *sock_in6, *sock_in6_e;
272	bool ip_match = false;
273	u16 port, port_e;
274
275	if (sockaddr->ss_family == AF_INET6) {
276		sock_in6 = (struct sockaddr_in6 *)sockaddr;
277		sock_in6_e = (struct sockaddr_in6 *)&np->np_sockaddr;
278
279		if (!memcmp(&sock_in6->sin6_addr.in6_u,
280			    &sock_in6_e->sin6_addr.in6_u,
281			    sizeof(struct in6_addr)))
282			ip_match = true;
283
284		port = ntohs(sock_in6->sin6_port);
285		port_e = ntohs(sock_in6_e->sin6_port);
286	} else {
287		sock_in = (struct sockaddr_in *)sockaddr;
288		sock_in_e = (struct sockaddr_in *)&np->np_sockaddr;
289
290		if (sock_in->sin_addr.s_addr == sock_in_e->sin_addr.s_addr)
291			ip_match = true;
292
293		port = ntohs(sock_in->sin_port);
294		port_e = ntohs(sock_in_e->sin_port);
295	}
296
297	if (ip_match && (port_e == port) &&
298	    (np->np_network_transport == network_transport))
299		return true;
300
301	return false;
302}
303
304static struct iscsi_np *iscsit_get_np(
305	struct sockaddr_storage *sockaddr,
306	int network_transport)
307{
308	struct iscsi_np *np;
309	bool match;
310
311	lockdep_assert_held(&np_lock);
312
313	list_for_each_entry(np, &g_np_list, np_list) {
314		spin_lock_bh(&np->np_thread_lock);
315		if (np->np_thread_state != ISCSI_NP_THREAD_ACTIVE) {
316			spin_unlock_bh(&np->np_thread_lock);
317			continue;
318		}
319
320		match = iscsit_check_np_match(sockaddr, np, network_transport);
321		if (match) {
322			/*
323			 * Increment the np_exports reference count now to
324			 * prevent iscsit_del_np() below from being called
325			 * while iscsi_tpg_add_network_portal() is called.
326			 */
327			np->np_exports++;
328			spin_unlock_bh(&np->np_thread_lock);
329			return np;
330		}
331		spin_unlock_bh(&np->np_thread_lock);
332	}
333
334	return NULL;
335}
336
337struct iscsi_np *iscsit_add_np(
338	struct sockaddr_storage *sockaddr,
339	int network_transport)
340{
341	struct iscsi_np *np;
342	int ret;
343
344	mutex_lock(&np_lock);
345
346	/*
347	 * Locate the existing struct iscsi_np if already active..
348	 */
349	np = iscsit_get_np(sockaddr, network_transport);
350	if (np) {
351		mutex_unlock(&np_lock);
352		return np;
353	}
354
355	np = kzalloc(sizeof(*np), GFP_KERNEL);
356	if (!np) {
357		mutex_unlock(&np_lock);
358		return ERR_PTR(-ENOMEM);
359	}
360
361	np->np_flags |= NPF_IP_NETWORK;
362	np->np_network_transport = network_transport;
363	spin_lock_init(&np->np_thread_lock);
364	init_completion(&np->np_restart_comp);
365	INIT_LIST_HEAD(&np->np_list);
366
367	ret = iscsi_target_setup_login_socket(np, sockaddr);
368	if (ret != 0) {
369		kfree(np);
370		mutex_unlock(&np_lock);
371		return ERR_PTR(ret);
372	}
373
374	np->np_thread = kthread_run(iscsi_target_login_thread, np, "iscsi_np");
375	if (IS_ERR(np->np_thread)) {
376		pr_err("Unable to create kthread: iscsi_np\n");
377		ret = PTR_ERR(np->np_thread);
378		kfree(np);
379		mutex_unlock(&np_lock);
380		return ERR_PTR(ret);
381	}
382	/*
383	 * Increment the np_exports reference count now to prevent
384	 * iscsit_del_np() below from being run while a new call to
385	 * iscsi_tpg_add_network_portal() for a matching iscsi_np is
386	 * active.  We don't need to hold np->np_thread_lock at this
387	 * point because iscsi_np has not been added to g_np_list yet.
388	 */
389	np->np_exports = 1;
390	np->np_thread_state = ISCSI_NP_THREAD_ACTIVE;
391
392	list_add_tail(&np->np_list, &g_np_list);
393	mutex_unlock(&np_lock);
394
395	pr_debug("CORE[0] - Added Network Portal: %pISpc on %s\n",
396		&np->np_sockaddr, np->np_transport->name);
397
398	return np;
399}
400
401int iscsit_reset_np_thread(
402	struct iscsi_np *np,
403	struct iscsi_tpg_np *tpg_np,
404	struct iscsi_portal_group *tpg,
405	bool shutdown)
406{
407	spin_lock_bh(&np->np_thread_lock);
408	if (np->np_thread_state == ISCSI_NP_THREAD_INACTIVE) {
409		spin_unlock_bh(&np->np_thread_lock);
410		return 0;
411	}
412	np->np_thread_state = ISCSI_NP_THREAD_RESET;
413	atomic_inc(&np->np_reset_count);
414
415	if (np->np_thread) {
416		spin_unlock_bh(&np->np_thread_lock);
417		send_sig(SIGINT, np->np_thread, 1);
418		wait_for_completion(&np->np_restart_comp);
419		spin_lock_bh(&np->np_thread_lock);
420	}
421	spin_unlock_bh(&np->np_thread_lock);
422
423	if (tpg_np && shutdown) {
424		kref_put(&tpg_np->tpg_np_kref, iscsit_login_kref_put);
425
426		wait_for_completion(&tpg_np->tpg_np_comp);
427	}
428
429	return 0;
430}
431
432static void iscsit_free_np(struct iscsi_np *np)
433{
434	if (np->np_socket)
435		sock_release(np->np_socket);
436}
437
438int iscsit_del_np(struct iscsi_np *np)
439{
440	spin_lock_bh(&np->np_thread_lock);
441	np->np_exports--;
442	if (np->np_exports) {
443		np->enabled = true;
444		spin_unlock_bh(&np->np_thread_lock);
445		return 0;
446	}
447	np->np_thread_state = ISCSI_NP_THREAD_SHUTDOWN;
448	spin_unlock_bh(&np->np_thread_lock);
449
450	if (np->np_thread) {
451		/*
452		 * We need to send the signal to wakeup Linux/Net
453		 * which may be sleeping in sock_accept()..
454		 */
455		send_sig(SIGINT, np->np_thread, 1);
456		kthread_stop(np->np_thread);
457		np->np_thread = NULL;
458	}
459
460	np->np_transport->iscsit_free_np(np);
461
462	mutex_lock(&np_lock);
463	list_del(&np->np_list);
464	mutex_unlock(&np_lock);
465
466	pr_debug("CORE[0] - Removed Network Portal: %pISpc on %s\n",
467		&np->np_sockaddr, np->np_transport->name);
468
469	iscsit_put_transport(np->np_transport);
470	kfree(np);
471	return 0;
472}
473
474static void iscsit_get_rx_pdu(struct iscsit_conn *);
475
476int iscsit_queue_rsp(struct iscsit_conn *conn, struct iscsit_cmd *cmd)
477{
478	return iscsit_add_cmd_to_response_queue(cmd, cmd->conn, cmd->i_state);
479}
480EXPORT_SYMBOL(iscsit_queue_rsp);
481
482void iscsit_aborted_task(struct iscsit_conn *conn, struct iscsit_cmd *cmd)
483{
484	spin_lock_bh(&conn->cmd_lock);
485	if (!list_empty(&cmd->i_conn_node))
486		list_del_init(&cmd->i_conn_node);
487	spin_unlock_bh(&conn->cmd_lock);
488
489	__iscsit_free_cmd(cmd, true);
490}
491EXPORT_SYMBOL(iscsit_aborted_task);
492
493static void iscsit_do_crypto_hash_buf(struct ahash_request *, const void *,
494				      u32, u32, const void *, void *);
495static void iscsit_tx_thread_wait_for_tcp(struct iscsit_conn *);
496
497static int
498iscsit_xmit_nondatain_pdu(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
499			  const void *data_buf, u32 data_buf_len)
500{
501	struct iscsi_hdr *hdr = (struct iscsi_hdr *)cmd->pdu;
502	struct kvec *iov;
503	u32 niov = 0, tx_size = ISCSI_HDR_LEN;
504	int ret;
505
506	iov = &cmd->iov_misc[0];
507	iov[niov].iov_base	= cmd->pdu;
508	iov[niov++].iov_len	= ISCSI_HDR_LEN;
509
510	if (conn->conn_ops->HeaderDigest) {
511		u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
512
513		iscsit_do_crypto_hash_buf(conn->conn_tx_hash, hdr,
514					  ISCSI_HDR_LEN, 0, NULL,
515					  header_digest);
516
517		iov[0].iov_len += ISCSI_CRC_LEN;
518		tx_size += ISCSI_CRC_LEN;
519		pr_debug("Attaching CRC32C HeaderDigest"
520			 " to opcode 0x%x 0x%08x\n",
521			 hdr->opcode, *header_digest);
522	}
523
524	if (data_buf_len) {
525		u32 padding = ((-data_buf_len) & 3);
526
527		iov[niov].iov_base	= (void *)data_buf;
528		iov[niov++].iov_len	= data_buf_len;
529		tx_size += data_buf_len;
530
531		if (padding != 0) {
532			iov[niov].iov_base = &cmd->pad_bytes;
533			iov[niov++].iov_len = padding;
534			tx_size += padding;
535			pr_debug("Attaching %u additional"
536				 " padding bytes.\n", padding);
537		}
538
539		if (conn->conn_ops->DataDigest) {
540			iscsit_do_crypto_hash_buf(conn->conn_tx_hash,
541						  data_buf, data_buf_len,
542						  padding, &cmd->pad_bytes,
543						  &cmd->data_crc);
544
545			iov[niov].iov_base = &cmd->data_crc;
546			iov[niov++].iov_len = ISCSI_CRC_LEN;
547			tx_size += ISCSI_CRC_LEN;
548			pr_debug("Attached DataDigest for %u"
549				 " bytes opcode 0x%x, CRC 0x%08x\n",
550				 data_buf_len, hdr->opcode, cmd->data_crc);
551		}
552	}
553
554	cmd->iov_misc_count = niov;
555	cmd->tx_size = tx_size;
556
557	ret = iscsit_send_tx_data(cmd, conn, 1);
558	if (ret < 0) {
559		iscsit_tx_thread_wait_for_tcp(conn);
560		return ret;
561	}
562
563	return 0;
564}
565
566static int iscsit_map_iovec(struct iscsit_cmd *cmd, struct kvec *iov, int nvec,
567			    u32 data_offset, u32 data_length);
568static void iscsit_unmap_iovec(struct iscsit_cmd *);
569static u32 iscsit_do_crypto_hash_sg(struct ahash_request *, struct iscsit_cmd *,
570				    u32, u32, u32, u8 *);
571static int
572iscsit_xmit_datain_pdu(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
573		       const struct iscsi_datain *datain)
574{
575	struct kvec *iov;
576	u32 iov_count = 0, tx_size = 0;
577	int ret, iov_ret;
578
579	iov = &cmd->iov_data[0];
580	iov[iov_count].iov_base	= cmd->pdu;
581	iov[iov_count++].iov_len = ISCSI_HDR_LEN;
582	tx_size += ISCSI_HDR_LEN;
583
584	if (conn->conn_ops->HeaderDigest) {
585		u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
586
587		iscsit_do_crypto_hash_buf(conn->conn_tx_hash, cmd->pdu,
588					  ISCSI_HDR_LEN, 0, NULL,
589					  header_digest);
590
591		iov[0].iov_len += ISCSI_CRC_LEN;
592		tx_size += ISCSI_CRC_LEN;
593
594		pr_debug("Attaching CRC32 HeaderDigest for DataIN PDU 0x%08x\n",
595			 *header_digest);
596	}
597
598	iov_ret = iscsit_map_iovec(cmd, &cmd->iov_data[iov_count],
599				   cmd->orig_iov_data_count - (iov_count + 2),
600				   datain->offset, datain->length);
601	if (iov_ret < 0)
602		return -1;
603
604	iov_count += iov_ret;
605	tx_size += datain->length;
606
607	cmd->padding = ((-datain->length) & 3);
608	if (cmd->padding) {
609		iov[iov_count].iov_base		= cmd->pad_bytes;
610		iov[iov_count++].iov_len	= cmd->padding;
611		tx_size += cmd->padding;
612
613		pr_debug("Attaching %u padding bytes\n", cmd->padding);
614	}
615
616	if (conn->conn_ops->DataDigest) {
617		cmd->data_crc = iscsit_do_crypto_hash_sg(conn->conn_tx_hash,
618							 cmd, datain->offset,
619							 datain->length,
620							 cmd->padding,
621							 cmd->pad_bytes);
622
623		iov[iov_count].iov_base	= &cmd->data_crc;
624		iov[iov_count++].iov_len = ISCSI_CRC_LEN;
625		tx_size += ISCSI_CRC_LEN;
626
627		pr_debug("Attached CRC32C DataDigest %d bytes, crc 0x%08x\n",
628			 datain->length + cmd->padding, cmd->data_crc);
629	}
630
631	cmd->iov_data_count = iov_count;
632	cmd->tx_size = tx_size;
633
634	ret = iscsit_fe_sendpage_sg(cmd, conn);
635
636	iscsit_unmap_iovec(cmd);
637
638	if (ret < 0) {
639		iscsit_tx_thread_wait_for_tcp(conn);
640		return ret;
641	}
642
643	return 0;
644}
645
646static int iscsit_xmit_pdu(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
647			   struct iscsi_datain_req *dr, const void *buf,
648			   u32 buf_len)
649{
650	if (dr)
651		return iscsit_xmit_datain_pdu(conn, cmd, buf);
652	else
653		return iscsit_xmit_nondatain_pdu(conn, cmd, buf, buf_len);
654}
655
656static enum target_prot_op iscsit_get_sup_prot_ops(struct iscsit_conn *conn)
657{
658	return TARGET_PROT_NORMAL;
659}
660
661static struct iscsit_transport iscsi_target_transport = {
662	.name			= "iSCSI/TCP",
663	.transport_type		= ISCSI_TCP,
664	.rdma_shutdown		= false,
665	.owner			= NULL,
666	.iscsit_setup_np	= iscsit_setup_np,
667	.iscsit_accept_np	= iscsit_accept_np,
668	.iscsit_free_np		= iscsit_free_np,
669	.iscsit_get_login_rx	= iscsit_get_login_rx,
670	.iscsit_put_login_tx	= iscsit_put_login_tx,
671	.iscsit_get_dataout	= iscsit_build_r2ts_for_cmd,
672	.iscsit_immediate_queue	= iscsit_immediate_queue,
673	.iscsit_response_queue	= iscsit_response_queue,
674	.iscsit_queue_data_in	= iscsit_queue_rsp,
675	.iscsit_queue_status	= iscsit_queue_rsp,
676	.iscsit_aborted_task	= iscsit_aborted_task,
677	.iscsit_xmit_pdu	= iscsit_xmit_pdu,
678	.iscsit_get_rx_pdu	= iscsit_get_rx_pdu,
679	.iscsit_get_sup_prot_ops = iscsit_get_sup_prot_ops,
680};
681
682static int __init iscsi_target_init_module(void)
683{
684	int ret = 0, size;
685
686	pr_debug("iSCSI-Target "ISCSIT_VERSION"\n");
687	iscsit_global = kzalloc(sizeof(*iscsit_global), GFP_KERNEL);
688	if (!iscsit_global)
689		return -1;
690
691	spin_lock_init(&iscsit_global->ts_bitmap_lock);
692	mutex_init(&auth_id_lock);
693	idr_init(&tiqn_idr);
694
695	ret = target_register_template(&iscsi_ops);
696	if (ret)
697		goto out;
698
699	size = BITS_TO_LONGS(ISCSIT_BITMAP_BITS) * sizeof(long);
700	iscsit_global->ts_bitmap = vzalloc(size);
701	if (!iscsit_global->ts_bitmap)
702		goto configfs_out;
703
704	if (!zalloc_cpumask_var(&iscsit_global->allowed_cpumask, GFP_KERNEL)) {
705		pr_err("Unable to allocate iscsit_global->allowed_cpumask\n");
706		goto bitmap_out;
707	}
708	cpumask_setall(iscsit_global->allowed_cpumask);
709
710	lio_qr_cache = kmem_cache_create("lio_qr_cache",
711			sizeof(struct iscsi_queue_req),
712			__alignof__(struct iscsi_queue_req), 0, NULL);
713	if (!lio_qr_cache) {
714		pr_err("Unable to kmem_cache_create() for"
715				" lio_qr_cache\n");
716		goto cpumask_out;
717	}
718
719	lio_dr_cache = kmem_cache_create("lio_dr_cache",
720			sizeof(struct iscsi_datain_req),
721			__alignof__(struct iscsi_datain_req), 0, NULL);
722	if (!lio_dr_cache) {
723		pr_err("Unable to kmem_cache_create() for"
724				" lio_dr_cache\n");
725		goto qr_out;
726	}
727
728	lio_ooo_cache = kmem_cache_create("lio_ooo_cache",
729			sizeof(struct iscsi_ooo_cmdsn),
730			__alignof__(struct iscsi_ooo_cmdsn), 0, NULL);
731	if (!lio_ooo_cache) {
732		pr_err("Unable to kmem_cache_create() for"
733				" lio_ooo_cache\n");
734		goto dr_out;
735	}
736
737	lio_r2t_cache = kmem_cache_create("lio_r2t_cache",
738			sizeof(struct iscsi_r2t), __alignof__(struct iscsi_r2t),
739			0, NULL);
740	if (!lio_r2t_cache) {
741		pr_err("Unable to kmem_cache_create() for"
742				" lio_r2t_cache\n");
743		goto ooo_out;
744	}
745
746	iscsit_register_transport(&iscsi_target_transport);
747
748	if (iscsit_load_discovery_tpg() < 0)
749		goto r2t_out;
750
751	return ret;
752r2t_out:
753	iscsit_unregister_transport(&iscsi_target_transport);
754	kmem_cache_destroy(lio_r2t_cache);
755ooo_out:
756	kmem_cache_destroy(lio_ooo_cache);
757dr_out:
758	kmem_cache_destroy(lio_dr_cache);
759qr_out:
760	kmem_cache_destroy(lio_qr_cache);
761cpumask_out:
762	free_cpumask_var(iscsit_global->allowed_cpumask);
763bitmap_out:
764	vfree(iscsit_global->ts_bitmap);
765configfs_out:
766	/* XXX: this probably wants it to be it's own unwind step.. */
767	if (iscsit_global->discovery_tpg)
768		iscsit_tpg_disable_portal_group(iscsit_global->discovery_tpg, 1);
769	target_unregister_template(&iscsi_ops);
770out:
771	kfree(iscsit_global);
772	return -ENOMEM;
773}
774
775static void __exit iscsi_target_cleanup_module(void)
776{
777	iscsit_release_discovery_tpg();
778	iscsit_unregister_transport(&iscsi_target_transport);
779	kmem_cache_destroy(lio_qr_cache);
780	kmem_cache_destroy(lio_dr_cache);
781	kmem_cache_destroy(lio_ooo_cache);
782	kmem_cache_destroy(lio_r2t_cache);
783
784	/*
785	 * Shutdown discovery sessions and disable discovery TPG
786	 */
787	if (iscsit_global->discovery_tpg)
788		iscsit_tpg_disable_portal_group(iscsit_global->discovery_tpg, 1);
789
790	target_unregister_template(&iscsi_ops);
791
792	free_cpumask_var(iscsit_global->allowed_cpumask);
793	vfree(iscsit_global->ts_bitmap);
794	kfree(iscsit_global);
795}
796
797int iscsit_add_reject(
798	struct iscsit_conn *conn,
799	u8 reason,
800	unsigned char *buf)
801{
802	struct iscsit_cmd *cmd;
803
804	cmd = iscsit_allocate_cmd(conn, TASK_INTERRUPTIBLE);
805	if (!cmd)
806		return -1;
807
808	cmd->iscsi_opcode = ISCSI_OP_REJECT;
809	cmd->reject_reason = reason;
810
811	cmd->buf_ptr = kmemdup(buf, ISCSI_HDR_LEN, GFP_KERNEL);
812	if (!cmd->buf_ptr) {
813		pr_err("Unable to allocate memory for cmd->buf_ptr\n");
814		iscsit_free_cmd(cmd, false);
815		return -1;
816	}
817
818	spin_lock_bh(&conn->cmd_lock);
819	list_add_tail(&cmd->i_conn_node, &conn->conn_cmd_list);
820	spin_unlock_bh(&conn->cmd_lock);
821
822	cmd->i_state = ISTATE_SEND_REJECT;
823	iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
824
825	return -1;
826}
827EXPORT_SYMBOL(iscsit_add_reject);
828
829static int iscsit_add_reject_from_cmd(
830	struct iscsit_cmd *cmd,
831	u8 reason,
832	bool add_to_conn,
833	unsigned char *buf)
834{
835	struct iscsit_conn *conn;
836	const bool do_put = cmd->se_cmd.se_tfo != NULL;
837
838	if (!cmd->conn) {
839		pr_err("cmd->conn is NULL for ITT: 0x%08x\n",
840				cmd->init_task_tag);
841		return -1;
842	}
843	conn = cmd->conn;
844
845	cmd->iscsi_opcode = ISCSI_OP_REJECT;
846	cmd->reject_reason = reason;
847
848	cmd->buf_ptr = kmemdup(buf, ISCSI_HDR_LEN, GFP_KERNEL);
849	if (!cmd->buf_ptr) {
850		pr_err("Unable to allocate memory for cmd->buf_ptr\n");
851		iscsit_free_cmd(cmd, false);
852		return -1;
853	}
854
855	if (add_to_conn) {
856		spin_lock_bh(&conn->cmd_lock);
857		list_add_tail(&cmd->i_conn_node, &conn->conn_cmd_list);
858		spin_unlock_bh(&conn->cmd_lock);
859	}
860
861	cmd->i_state = ISTATE_SEND_REJECT;
862	iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
863	/*
864	 * Perform the kref_put now if se_cmd has already been setup by
865	 * scsit_setup_scsi_cmd()
866	 */
867	if (do_put) {
868		pr_debug("iscsi reject: calling target_put_sess_cmd >>>>>>\n");
869		target_put_sess_cmd(&cmd->se_cmd);
870	}
871	return -1;
872}
873
874static int iscsit_add_reject_cmd(struct iscsit_cmd *cmd, u8 reason,
875				 unsigned char *buf)
876{
877	return iscsit_add_reject_from_cmd(cmd, reason, true, buf);
878}
879
880int iscsit_reject_cmd(struct iscsit_cmd *cmd, u8 reason, unsigned char *buf)
881{
882	return iscsit_add_reject_from_cmd(cmd, reason, false, buf);
883}
884EXPORT_SYMBOL(iscsit_reject_cmd);
885
886/*
887 * Map some portion of the allocated scatterlist to an iovec, suitable for
888 * kernel sockets to copy data in/out.
889 */
890static int iscsit_map_iovec(struct iscsit_cmd *cmd, struct kvec *iov, int nvec,
891			    u32 data_offset, u32 data_length)
892{
893	u32 i = 0, orig_data_length = data_length;
894	struct scatterlist *sg;
895	unsigned int page_off;
896
897	/*
898	 * We know each entry in t_data_sg contains a page.
899	 */
900	u32 ent = data_offset / PAGE_SIZE;
901
902	if (!data_length)
903		return 0;
904
905	if (ent >= cmd->se_cmd.t_data_nents) {
906		pr_err("Initial page entry out-of-bounds\n");
907		goto overflow;
908	}
909
910	sg = &cmd->se_cmd.t_data_sg[ent];
911	page_off = (data_offset % PAGE_SIZE);
912
913	cmd->first_data_sg = sg;
914	cmd->first_data_sg_off = page_off;
915
916	while (data_length) {
917		u32 cur_len;
918
919		if (WARN_ON_ONCE(!sg || i >= nvec))
920			goto overflow;
921
922		cur_len = min_t(u32, data_length, sg->length - page_off);
923
924		iov[i].iov_base = kmap(sg_page(sg)) + sg->offset + page_off;
925		iov[i].iov_len = cur_len;
926
927		data_length -= cur_len;
928		page_off = 0;
929		sg = sg_next(sg);
930		i++;
931	}
932
933	cmd->kmapped_nents = i;
934
935	return i;
936
937overflow:
938	pr_err("offset %d + length %d overflow; %d/%d; sg-list:\n",
939	       data_offset, orig_data_length, i, nvec);
940	for_each_sg(cmd->se_cmd.t_data_sg, sg,
941		    cmd->se_cmd.t_data_nents, i) {
942		pr_err("[%d] off %d len %d\n",
943		       i, sg->offset, sg->length);
944	}
945	return -1;
946}
947
948static void iscsit_unmap_iovec(struct iscsit_cmd *cmd)
949{
950	u32 i;
951	struct scatterlist *sg;
952
953	sg = cmd->first_data_sg;
954
955	for (i = 0; i < cmd->kmapped_nents; i++)
956		kunmap(sg_page(&sg[i]));
957}
958
959static void iscsit_ack_from_expstatsn(struct iscsit_conn *conn, u32 exp_statsn)
960{
961	LIST_HEAD(ack_list);
962	struct iscsit_cmd *cmd, *cmd_p;
963
964	conn->exp_statsn = exp_statsn;
965
966	if (conn->sess->sess_ops->RDMAExtensions)
967		return;
968
969	spin_lock_bh(&conn->cmd_lock);
970	list_for_each_entry_safe(cmd, cmd_p, &conn->conn_cmd_list, i_conn_node) {
971		spin_lock(&cmd->istate_lock);
972		if ((cmd->i_state == ISTATE_SENT_STATUS) &&
973		    iscsi_sna_lt(cmd->stat_sn, exp_statsn)) {
974			cmd->i_state = ISTATE_REMOVE;
975			spin_unlock(&cmd->istate_lock);
976			list_move_tail(&cmd->i_conn_node, &ack_list);
977			continue;
978		}
979		spin_unlock(&cmd->istate_lock);
980	}
981	spin_unlock_bh(&conn->cmd_lock);
982
983	list_for_each_entry_safe(cmd, cmd_p, &ack_list, i_conn_node) {
984		list_del_init(&cmd->i_conn_node);
985		iscsit_free_cmd(cmd, false);
986	}
987}
988
989static int iscsit_allocate_iovecs(struct iscsit_cmd *cmd)
990{
991	u32 iov_count = max(1UL, DIV_ROUND_UP(cmd->se_cmd.data_length, PAGE_SIZE));
992
993	iov_count += ISCSI_IOV_DATA_BUFFER;
994	cmd->iov_data = kcalloc(iov_count, sizeof(*cmd->iov_data), GFP_KERNEL);
995	if (!cmd->iov_data)
996		return -ENOMEM;
997
998	cmd->orig_iov_data_count = iov_count;
999	return 0;
1000}
1001
1002int iscsit_setup_scsi_cmd(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
1003			  unsigned char *buf)
1004{
1005	int data_direction, payload_length;
1006	struct iscsi_ecdb_ahdr *ecdb_ahdr;
1007	struct iscsi_scsi_req *hdr;
1008	int iscsi_task_attr;
1009	unsigned char *cdb;
1010	int sam_task_attr;
1011
1012	atomic_long_inc(&conn->sess->cmd_pdus);
1013
1014	hdr			= (struct iscsi_scsi_req *) buf;
1015	payload_length		= ntoh24(hdr->dlength);
1016
1017	/* FIXME; Add checks for AdditionalHeaderSegment */
1018
1019	if (!(hdr->flags & ISCSI_FLAG_CMD_WRITE) &&
1020	    !(hdr->flags & ISCSI_FLAG_CMD_FINAL)) {
1021		pr_err("ISCSI_FLAG_CMD_WRITE & ISCSI_FLAG_CMD_FINAL"
1022				" not set. Bad iSCSI Initiator.\n");
1023		return iscsit_add_reject_cmd(cmd,
1024					     ISCSI_REASON_BOOKMARK_INVALID, buf);
1025	}
1026
1027	if (((hdr->flags & ISCSI_FLAG_CMD_READ) ||
1028	     (hdr->flags & ISCSI_FLAG_CMD_WRITE)) && !hdr->data_length) {
1029		/*
1030		 * From RFC-3720 Section 10.3.1:
1031		 *
1032		 * "Either or both of R and W MAY be 1 when either the
1033		 *  Expected Data Transfer Length and/or Bidirectional Read
1034		 *  Expected Data Transfer Length are 0"
1035		 *
1036		 * For this case, go ahead and clear the unnecssary bits
1037		 * to avoid any confusion with ->data_direction.
1038		 */
1039		hdr->flags &= ~ISCSI_FLAG_CMD_READ;
1040		hdr->flags &= ~ISCSI_FLAG_CMD_WRITE;
1041
1042		pr_warn("ISCSI_FLAG_CMD_READ or ISCSI_FLAG_CMD_WRITE"
1043			" set when Expected Data Transfer Length is 0 for"
1044			" CDB: 0x%02x, Fixing up flags\n", hdr->cdb[0]);
1045	}
1046
1047	if (!(hdr->flags & ISCSI_FLAG_CMD_READ) &&
1048	    !(hdr->flags & ISCSI_FLAG_CMD_WRITE) && (hdr->data_length != 0)) {
1049		pr_err("ISCSI_FLAG_CMD_READ and/or ISCSI_FLAG_CMD_WRITE"
1050			" MUST be set if Expected Data Transfer Length is not 0."
1051			" Bad iSCSI Initiator\n");
1052		return iscsit_add_reject_cmd(cmd,
1053					     ISCSI_REASON_BOOKMARK_INVALID, buf);
1054	}
1055
1056	if ((hdr->flags & ISCSI_FLAG_CMD_READ) &&
1057	    (hdr->flags & ISCSI_FLAG_CMD_WRITE)) {
1058		pr_err("Bidirectional operations not supported!\n");
1059		return iscsit_add_reject_cmd(cmd,
1060					     ISCSI_REASON_BOOKMARK_INVALID, buf);
1061	}
1062
1063	if (hdr->opcode & ISCSI_OP_IMMEDIATE) {
1064		pr_err("Illegally set Immediate Bit in iSCSI Initiator"
1065				" Scsi Command PDU.\n");
1066		return iscsit_add_reject_cmd(cmd,
1067					     ISCSI_REASON_BOOKMARK_INVALID, buf);
1068	}
1069
1070	if (payload_length && !conn->sess->sess_ops->ImmediateData) {
1071		pr_err("ImmediateData=No but DataSegmentLength=%u,"
1072			" protocol error.\n", payload_length);
1073		return iscsit_add_reject_cmd(cmd,
1074					     ISCSI_REASON_PROTOCOL_ERROR, buf);
1075	}
1076
1077	if ((be32_to_cpu(hdr->data_length) == payload_length) &&
1078	    (!(hdr->flags & ISCSI_FLAG_CMD_FINAL))) {
1079		pr_err("Expected Data Transfer Length and Length of"
1080			" Immediate Data are the same, but ISCSI_FLAG_CMD_FINAL"
1081			" bit is not set protocol error\n");
1082		return iscsit_add_reject_cmd(cmd,
1083					     ISCSI_REASON_PROTOCOL_ERROR, buf);
1084	}
1085
1086	if (payload_length > be32_to_cpu(hdr->data_length)) {
1087		pr_err("DataSegmentLength: %u is greater than"
1088			" EDTL: %u, protocol error.\n", payload_length,
1089				hdr->data_length);
1090		return iscsit_add_reject_cmd(cmd,
1091					     ISCSI_REASON_PROTOCOL_ERROR, buf);
1092	}
1093
1094	if (payload_length > conn->conn_ops->MaxXmitDataSegmentLength) {
1095		pr_err("DataSegmentLength: %u is greater than"
1096			" MaxXmitDataSegmentLength: %u, protocol error.\n",
1097			payload_length, conn->conn_ops->MaxXmitDataSegmentLength);
1098		return iscsit_add_reject_cmd(cmd,
1099					     ISCSI_REASON_PROTOCOL_ERROR, buf);
1100	}
1101
1102	if (payload_length > conn->sess->sess_ops->FirstBurstLength) {
1103		pr_err("DataSegmentLength: %u is greater than"
1104			" FirstBurstLength: %u, protocol error.\n",
1105			payload_length, conn->sess->sess_ops->FirstBurstLength);
1106		return iscsit_add_reject_cmd(cmd,
1107					     ISCSI_REASON_BOOKMARK_INVALID, buf);
1108	}
1109
1110	cdb = hdr->cdb;
1111
1112	if (hdr->hlength) {
1113		ecdb_ahdr = (struct iscsi_ecdb_ahdr *) (hdr + 1);
1114		if (ecdb_ahdr->ahstype != ISCSI_AHSTYPE_CDB) {
1115			pr_err("Additional Header Segment type %d not supported!\n",
1116			       ecdb_ahdr->ahstype);
1117			return iscsit_add_reject_cmd(cmd,
1118				ISCSI_REASON_CMD_NOT_SUPPORTED, buf);
1119		}
1120
1121		cdb = kmalloc(be16_to_cpu(ecdb_ahdr->ahslength) + 15,
1122			      GFP_KERNEL);
1123		if (cdb == NULL)
1124			return iscsit_add_reject_cmd(cmd,
1125				ISCSI_REASON_BOOKMARK_NO_RESOURCES, buf);
1126		memcpy(cdb, hdr->cdb, ISCSI_CDB_SIZE);
1127		memcpy(cdb + ISCSI_CDB_SIZE, ecdb_ahdr->ecdb,
1128		       be16_to_cpu(ecdb_ahdr->ahslength) - 1);
1129	}
1130
1131	data_direction = (hdr->flags & ISCSI_FLAG_CMD_WRITE) ? DMA_TO_DEVICE :
1132			 (hdr->flags & ISCSI_FLAG_CMD_READ) ? DMA_FROM_DEVICE :
1133			  DMA_NONE;
1134
1135	cmd->data_direction = data_direction;
1136	iscsi_task_attr = hdr->flags & ISCSI_FLAG_CMD_ATTR_MASK;
1137	/*
1138	 * Figure out the SAM Task Attribute for the incoming SCSI CDB
1139	 */
1140	if ((iscsi_task_attr == ISCSI_ATTR_UNTAGGED) ||
1141	    (iscsi_task_attr == ISCSI_ATTR_SIMPLE))
1142		sam_task_attr = TCM_SIMPLE_TAG;
1143	else if (iscsi_task_attr == ISCSI_ATTR_ORDERED)
1144		sam_task_attr = TCM_ORDERED_TAG;
1145	else if (iscsi_task_attr == ISCSI_ATTR_HEAD_OF_QUEUE)
1146		sam_task_attr = TCM_HEAD_TAG;
1147	else if (iscsi_task_attr == ISCSI_ATTR_ACA)
1148		sam_task_attr = TCM_ACA_TAG;
1149	else {
1150		pr_debug("Unknown iSCSI Task Attribute: 0x%02x, using"
1151			" TCM_SIMPLE_TAG\n", iscsi_task_attr);
1152		sam_task_attr = TCM_SIMPLE_TAG;
1153	}
1154
1155	cmd->iscsi_opcode	= ISCSI_OP_SCSI_CMD;
1156	cmd->i_state		= ISTATE_NEW_CMD;
1157	cmd->immediate_cmd	= ((hdr->opcode & ISCSI_OP_IMMEDIATE) ? 1 : 0);
1158	cmd->immediate_data	= (payload_length) ? 1 : 0;
1159	cmd->unsolicited_data	= ((!(hdr->flags & ISCSI_FLAG_CMD_FINAL) &&
1160				     (hdr->flags & ISCSI_FLAG_CMD_WRITE)) ? 1 : 0);
1161	if (cmd->unsolicited_data)
1162		cmd->cmd_flags |= ICF_NON_IMMEDIATE_UNSOLICITED_DATA;
1163
1164	conn->sess->init_task_tag = cmd->init_task_tag = hdr->itt;
1165	if (hdr->flags & ISCSI_FLAG_CMD_READ)
1166		cmd->targ_xfer_tag = session_get_next_ttt(conn->sess);
1167	else
1168		cmd->targ_xfer_tag = 0xFFFFFFFF;
1169	cmd->cmd_sn		= be32_to_cpu(hdr->cmdsn);
1170	cmd->exp_stat_sn	= be32_to_cpu(hdr->exp_statsn);
1171	cmd->first_burst_len	= payload_length;
1172
1173	if (!conn->sess->sess_ops->RDMAExtensions &&
1174	     cmd->data_direction == DMA_FROM_DEVICE) {
1175		struct iscsi_datain_req *dr;
1176
1177		dr = iscsit_allocate_datain_req();
1178		if (!dr) {
1179			if (cdb != hdr->cdb)
1180				kfree(cdb);
1181			return iscsit_add_reject_cmd(cmd,
1182					ISCSI_REASON_BOOKMARK_NO_RESOURCES, buf);
1183		}
1184
1185		iscsit_attach_datain_req(cmd, dr);
1186	}
1187
1188	/*
1189	 * Initialize struct se_cmd descriptor from target_core_mod infrastructure
1190	 */
1191	__target_init_cmd(&cmd->se_cmd, &iscsi_ops,
1192			  conn->sess->se_sess, be32_to_cpu(hdr->data_length),
1193			  cmd->data_direction, sam_task_attr,
1194			  cmd->sense_buffer + 2, scsilun_to_int(&hdr->lun),
1195			  conn->cmd_cnt);
1196
1197	pr_debug("Got SCSI Command, ITT: 0x%08x, CmdSN: 0x%08x,"
1198		" ExpXferLen: %u, Length: %u, CID: %hu\n", hdr->itt,
1199		hdr->cmdsn, be32_to_cpu(hdr->data_length), payload_length,
1200		conn->cid);
1201
1202	target_get_sess_cmd(&cmd->se_cmd, true);
1203
1204	cmd->se_cmd.tag = (__force u32)cmd->init_task_tag;
1205	cmd->sense_reason = target_cmd_init_cdb(&cmd->se_cmd, cdb,
1206						GFP_KERNEL);
1207
1208	if (cdb != hdr->cdb)
1209		kfree(cdb);
1210
1211	if (cmd->sense_reason) {
1212		if (cmd->sense_reason == TCM_OUT_OF_RESOURCES) {
1213			return iscsit_add_reject_cmd(cmd,
1214				ISCSI_REASON_BOOKMARK_NO_RESOURCES, buf);
1215		}
1216
1217		goto attach_cmd;
1218	}
1219
1220	cmd->sense_reason = transport_lookup_cmd_lun(&cmd->se_cmd);
1221	if (cmd->sense_reason)
1222		goto attach_cmd;
1223
1224	cmd->sense_reason = target_cmd_parse_cdb(&cmd->se_cmd);
1225	if (cmd->sense_reason)
1226		goto attach_cmd;
1227
1228	if (iscsit_build_pdu_and_seq_lists(cmd, payload_length) < 0) {
1229		return iscsit_add_reject_cmd(cmd,
1230				ISCSI_REASON_BOOKMARK_NO_RESOURCES, buf);
1231	}
1232
1233attach_cmd:
1234	spin_lock_bh(&conn->cmd_lock);
1235	list_add_tail(&cmd->i_conn_node, &conn->conn_cmd_list);
1236	spin_unlock_bh(&conn->cmd_lock);
1237	return 0;
1238}
1239EXPORT_SYMBOL(iscsit_setup_scsi_cmd);
1240
1241void iscsit_set_unsolicited_dataout(struct iscsit_cmd *cmd)
1242{
1243	iscsit_set_dataout_sequence_values(cmd);
1244
1245	spin_lock_bh(&cmd->dataout_timeout_lock);
1246	iscsit_start_dataout_timer(cmd, cmd->conn);
1247	spin_unlock_bh(&cmd->dataout_timeout_lock);
1248}
1249EXPORT_SYMBOL(iscsit_set_unsolicited_dataout);
1250
1251int iscsit_process_scsi_cmd(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
1252			    struct iscsi_scsi_req *hdr)
1253{
1254	int cmdsn_ret = 0;
1255	/*
1256	 * Check the CmdSN against ExpCmdSN/MaxCmdSN here if
1257	 * the Immediate Bit is not set, and no Immediate
1258	 * Data is attached.
1259	 *
1260	 * A PDU/CmdSN carrying Immediate Data can only
1261	 * be processed after the DataCRC has passed.
1262	 * If the DataCRC fails, the CmdSN MUST NOT
1263	 * be acknowledged. (See below)
1264	 */
1265	if (!cmd->immediate_data) {
1266		cmdsn_ret = iscsit_sequence_cmd(conn, cmd,
1267					(unsigned char *)hdr, hdr->cmdsn);
1268		if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
1269			return -1;
1270		else if (cmdsn_ret == CMDSN_LOWER_THAN_EXP) {
1271			target_put_sess_cmd(&cmd->se_cmd);
1272			return 0;
1273		}
1274	}
1275
1276	iscsit_ack_from_expstatsn(conn, be32_to_cpu(hdr->exp_statsn));
1277
1278	/*
1279	 * If no Immediate Data is attached, it's OK to return now.
1280	 */
1281	if (!cmd->immediate_data) {
1282		if (!cmd->sense_reason && cmd->unsolicited_data)
1283			iscsit_set_unsolicited_dataout(cmd);
1284		if (!cmd->sense_reason)
1285			return 0;
1286
1287		target_put_sess_cmd(&cmd->se_cmd);
1288		return 0;
1289	}
1290
1291	/*
1292	 * Early CHECK_CONDITIONs with ImmediateData never make it to command
1293	 * execution.  These exceptions are processed in CmdSN order using
1294	 * iscsit_check_received_cmdsn() in iscsit_get_immediate_data() below.
1295	 */
1296	if (cmd->sense_reason)
1297		return 1;
1298	/*
1299	 * Call directly into transport_generic_new_cmd() to perform
1300	 * the backend memory allocation.
1301	 */
1302	cmd->sense_reason = transport_generic_new_cmd(&cmd->se_cmd);
1303	if (cmd->sense_reason)
1304		return 1;
1305
1306	return 0;
1307}
1308EXPORT_SYMBOL(iscsit_process_scsi_cmd);
1309
1310static int
1311iscsit_get_immediate_data(struct iscsit_cmd *cmd, struct iscsi_scsi_req *hdr,
1312			  bool dump_payload)
1313{
1314	int cmdsn_ret = 0, immed_ret = IMMEDIATE_DATA_NORMAL_OPERATION;
1315	int rc;
1316
1317	/*
1318	 * Special case for Unsupported SAM WRITE Opcodes and ImmediateData=Yes.
1319	 */
1320	if (dump_payload) {
1321		u32 length = min(cmd->se_cmd.data_length - cmd->write_data_done,
1322				 cmd->first_burst_len);
1323
1324		pr_debug("Dumping min(%d - %d, %d) = %d bytes of immediate data\n",
1325			 cmd->se_cmd.data_length, cmd->write_data_done,
1326			 cmd->first_burst_len, length);
1327		rc = iscsit_dump_data_payload(cmd->conn, length, 1);
1328		pr_debug("Finished dumping immediate data\n");
1329		if (rc < 0)
1330			immed_ret = IMMEDIATE_DATA_CANNOT_RECOVER;
1331	} else {
1332		immed_ret = iscsit_handle_immediate_data(cmd, hdr,
1333							 cmd->first_burst_len);
1334	}
1335
1336	if (immed_ret == IMMEDIATE_DATA_NORMAL_OPERATION) {
1337		/*
1338		 * A PDU/CmdSN carrying Immediate Data passed
1339		 * DataCRC, check against ExpCmdSN/MaxCmdSN if
1340		 * Immediate Bit is not set.
1341		 */
1342		cmdsn_ret = iscsit_sequence_cmd(cmd->conn, cmd,
1343					(unsigned char *)hdr, hdr->cmdsn);
1344		if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
1345			return -1;
1346
1347		if (cmd->sense_reason || cmdsn_ret == CMDSN_LOWER_THAN_EXP) {
1348			target_put_sess_cmd(&cmd->se_cmd);
1349
1350			return 0;
1351		} else if (cmd->unsolicited_data)
1352			iscsit_set_unsolicited_dataout(cmd);
1353
1354	} else if (immed_ret == IMMEDIATE_DATA_ERL1_CRC_FAILURE) {
1355		/*
1356		 * Immediate Data failed DataCRC and ERL>=1,
1357		 * silently drop this PDU and let the initiator
1358		 * plug the CmdSN gap.
1359		 *
1360		 * FIXME: Send Unsolicited NOPIN with reserved
1361		 * TTT here to help the initiator figure out
1362		 * the missing CmdSN, although they should be
1363		 * intelligent enough to determine the missing
1364		 * CmdSN and issue a retry to plug the sequence.
1365		 */
1366		cmd->i_state = ISTATE_REMOVE;
1367		iscsit_add_cmd_to_immediate_queue(cmd, cmd->conn, cmd->i_state);
1368	} else /* immed_ret == IMMEDIATE_DATA_CANNOT_RECOVER */
1369		return -1;
1370
1371	return 0;
1372}
1373
1374static int
1375iscsit_handle_scsi_cmd(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
1376			   unsigned char *buf)
1377{
1378	struct iscsi_scsi_req *hdr = (struct iscsi_scsi_req *)buf;
1379	int rc, immed_data;
1380	bool dump_payload = false;
1381
1382	rc = iscsit_setup_scsi_cmd(conn, cmd, buf);
1383	if (rc < 0)
1384		return 0;
1385	/*
1386	 * Allocation iovecs needed for struct socket operations for
1387	 * traditional iSCSI block I/O.
1388	 */
1389	if (iscsit_allocate_iovecs(cmd) < 0) {
1390		return iscsit_reject_cmd(cmd,
1391				ISCSI_REASON_BOOKMARK_NO_RESOURCES, buf);
1392	}
1393	immed_data = cmd->immediate_data;
1394
1395	rc = iscsit_process_scsi_cmd(conn, cmd, hdr);
1396	if (rc < 0)
1397		return rc;
1398	else if (rc > 0)
1399		dump_payload = true;
1400
1401	if (!immed_data)
1402		return 0;
1403
1404	return iscsit_get_immediate_data(cmd, hdr, dump_payload);
1405}
1406
1407static u32 iscsit_do_crypto_hash_sg(
1408	struct ahash_request *hash,
1409	struct iscsit_cmd *cmd,
1410	u32 data_offset,
1411	u32 data_length,
1412	u32 padding,
1413	u8 *pad_bytes)
1414{
1415	u32 data_crc;
1416	struct scatterlist *sg;
1417	unsigned int page_off;
1418
1419	crypto_ahash_init(hash);
1420
1421	sg = cmd->first_data_sg;
1422	page_off = cmd->first_data_sg_off;
1423
1424	if (data_length && page_off) {
1425		struct scatterlist first_sg;
1426		u32 len = min_t(u32, data_length, sg->length - page_off);
1427
1428		sg_init_table(&first_sg, 1);
1429		sg_set_page(&first_sg, sg_page(sg), len, sg->offset + page_off);
1430
1431		ahash_request_set_crypt(hash, &first_sg, NULL, len);
1432		crypto_ahash_update(hash);
1433
1434		data_length -= len;
1435		sg = sg_next(sg);
1436	}
1437
1438	while (data_length) {
1439		u32 cur_len = min_t(u32, data_length, sg->length);
1440
1441		ahash_request_set_crypt(hash, sg, NULL, cur_len);
1442		crypto_ahash_update(hash);
1443
1444		data_length -= cur_len;
1445		/* iscsit_map_iovec has already checked for invalid sg pointers */
1446		sg = sg_next(sg);
1447	}
1448
1449	if (padding) {
1450		struct scatterlist pad_sg;
1451
1452		sg_init_one(&pad_sg, pad_bytes, padding);
1453		ahash_request_set_crypt(hash, &pad_sg, (u8 *)&data_crc,
1454					padding);
1455		crypto_ahash_finup(hash);
1456	} else {
1457		ahash_request_set_crypt(hash, NULL, (u8 *)&data_crc, 0);
1458		crypto_ahash_final(hash);
1459	}
1460
1461	return data_crc;
1462}
1463
1464static void iscsit_do_crypto_hash_buf(struct ahash_request *hash,
1465	const void *buf, u32 payload_length, u32 padding,
1466	const void *pad_bytes, void *data_crc)
1467{
1468	struct scatterlist sg[2];
1469
1470	sg_init_table(sg, ARRAY_SIZE(sg));
1471	sg_set_buf(sg, buf, payload_length);
1472	if (padding)
1473		sg_set_buf(sg + 1, pad_bytes, padding);
1474
1475	ahash_request_set_crypt(hash, sg, data_crc, payload_length + padding);
1476
1477	crypto_ahash_digest(hash);
1478}
1479
1480int
1481__iscsit_check_dataout_hdr(struct iscsit_conn *conn, void *buf,
1482			   struct iscsit_cmd *cmd, u32 payload_length,
1483			   bool *success)
1484{
1485	struct iscsi_data *hdr = buf;
1486	struct se_cmd *se_cmd;
1487	int rc;
1488
1489	/* iSCSI write */
1490	atomic_long_add(payload_length, &conn->sess->rx_data_octets);
1491
1492	pr_debug("Got DataOut ITT: 0x%08x, TTT: 0x%08x,"
1493		" DataSN: 0x%08x, Offset: %u, Length: %u, CID: %hu\n",
1494		hdr->itt, hdr->ttt, hdr->datasn, ntohl(hdr->offset),
1495		payload_length, conn->cid);
1496
1497	if (cmd->cmd_flags & ICF_GOT_LAST_DATAOUT) {
1498		pr_err("Command ITT: 0x%08x received DataOUT after"
1499			" last DataOUT received, dumping payload\n",
1500			cmd->init_task_tag);
1501		return iscsit_dump_data_payload(conn, payload_length, 1);
1502	}
1503
1504	if (cmd->data_direction != DMA_TO_DEVICE) {
1505		pr_err("Command ITT: 0x%08x received DataOUT for a"
1506			" NON-WRITE command.\n", cmd->init_task_tag);
1507		return iscsit_dump_data_payload(conn, payload_length, 1);
1508	}
1509	se_cmd = &cmd->se_cmd;
1510	iscsit_mod_dataout_timer(cmd);
1511
1512	if ((be32_to_cpu(hdr->offset) + payload_length) > cmd->se_cmd.data_length) {
1513		pr_err("DataOut Offset: %u, Length %u greater than iSCSI Command EDTL %u, protocol error.\n",
1514		       be32_to_cpu(hdr->offset), payload_length,
1515		       cmd->se_cmd.data_length);
1516		return iscsit_reject_cmd(cmd, ISCSI_REASON_BOOKMARK_INVALID, buf);
1517	}
1518
1519	if (cmd->unsolicited_data) {
1520		int dump_unsolicited_data = 0;
1521
1522		if (conn->sess->sess_ops->InitialR2T) {
1523			pr_err("Received unexpected unsolicited data"
1524				" while InitialR2T=Yes, protocol error.\n");
1525			transport_send_check_condition_and_sense(&cmd->se_cmd,
1526					TCM_UNEXPECTED_UNSOLICITED_DATA, 0);
1527			return -1;
1528		}
1529		/*
1530		 * Special case for dealing with Unsolicited DataOUT
1531		 * and Unsupported SAM WRITE Opcodes and SE resource allocation
1532		 * failures;
1533		 */
1534
1535		/* Something's amiss if we're not in WRITE_PENDING state... */
1536		WARN_ON(se_cmd->t_state != TRANSPORT_WRITE_PENDING);
1537		if (!(se_cmd->se_cmd_flags & SCF_SUPPORTED_SAM_OPCODE))
1538			dump_unsolicited_data = 1;
1539
1540		if (dump_unsolicited_data) {
1541			/*
1542			 * Check if a delayed TASK_ABORTED status needs to
1543			 * be sent now if the ISCSI_FLAG_CMD_FINAL has been
1544			 * received with the unsolicited data out.
1545			 */
1546			if (hdr->flags & ISCSI_FLAG_CMD_FINAL)
1547				iscsit_stop_dataout_timer(cmd);
1548
1549			return iscsit_dump_data_payload(conn, payload_length, 1);
1550		}
1551	} else {
1552		/*
1553		 * For the normal solicited data path:
1554		 *
1555		 * Check for a delayed TASK_ABORTED status and dump any
1556		 * incoming data out payload if one exists.  Also, when the
1557		 * ISCSI_FLAG_CMD_FINAL is set to denote the end of the current
1558		 * data out sequence, we decrement outstanding_r2ts.  Once
1559		 * outstanding_r2ts reaches zero, go ahead and send the delayed
1560		 * TASK_ABORTED status.
1561		 */
1562		if (se_cmd->transport_state & CMD_T_ABORTED) {
1563			if (hdr->flags & ISCSI_FLAG_CMD_FINAL &&
1564			    --cmd->outstanding_r2ts < 1)
1565				iscsit_stop_dataout_timer(cmd);
1566
1567			return iscsit_dump_data_payload(conn, payload_length, 1);
1568		}
1569	}
1570	/*
1571	 * Perform DataSN, DataSequenceInOrder, DataPDUInOrder, and
1572	 * within-command recovery checks before receiving the payload.
1573	 */
1574	rc = iscsit_check_pre_dataout(cmd, buf);
1575	if (rc == DATAOUT_WITHIN_COMMAND_RECOVERY)
1576		return 0;
1577	else if (rc == DATAOUT_CANNOT_RECOVER)
1578		return -1;
1579	*success = true;
1580	return 0;
1581}
1582EXPORT_SYMBOL(__iscsit_check_dataout_hdr);
1583
1584int
1585iscsit_check_dataout_hdr(struct iscsit_conn *conn, void *buf,
1586			 struct iscsit_cmd **out_cmd)
1587{
1588	struct iscsi_data *hdr = buf;
1589	struct iscsit_cmd *cmd;
1590	u32 payload_length = ntoh24(hdr->dlength);
1591	int rc;
1592	bool success = false;
1593
1594	if (!payload_length) {
1595		pr_warn_ratelimited("DataOUT payload is ZERO, ignoring.\n");
1596		return 0;
1597	}
1598
1599	if (payload_length > conn->conn_ops->MaxXmitDataSegmentLength) {
1600		pr_err_ratelimited("DataSegmentLength: %u is greater than"
1601			" MaxXmitDataSegmentLength: %u\n", payload_length,
1602			conn->conn_ops->MaxXmitDataSegmentLength);
1603		return iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR, buf);
1604	}
1605
1606	cmd = iscsit_find_cmd_from_itt_or_dump(conn, hdr->itt, payload_length);
1607	if (!cmd)
1608		return 0;
1609
1610	rc = __iscsit_check_dataout_hdr(conn, buf, cmd, payload_length, &success);
1611
1612	if (success)
1613		*out_cmd = cmd;
1614
1615	return rc;
1616}
1617EXPORT_SYMBOL(iscsit_check_dataout_hdr);
1618
1619static int
1620iscsit_get_dataout(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
1621		   struct iscsi_data *hdr)
1622{
1623	struct kvec *iov;
1624	u32 checksum, iov_count = 0, padding = 0, rx_got = 0, rx_size = 0;
1625	u32 payload_length;
1626	int iov_ret, data_crc_failed = 0;
1627
1628	payload_length = min_t(u32, cmd->se_cmd.data_length,
1629			       ntoh24(hdr->dlength));
1630	rx_size += payload_length;
1631	iov = &cmd->iov_data[0];
1632
1633	iov_ret = iscsit_map_iovec(cmd, iov, cmd->orig_iov_data_count - 2,
1634				   be32_to_cpu(hdr->offset), payload_length);
1635	if (iov_ret < 0)
1636		return -1;
1637
1638	iov_count += iov_ret;
1639
1640	padding = ((-payload_length) & 3);
1641	if (padding != 0) {
1642		iov[iov_count].iov_base	= cmd->pad_bytes;
1643		iov[iov_count++].iov_len = padding;
1644		rx_size += padding;
1645		pr_debug("Receiving %u padding bytes.\n", padding);
1646	}
1647
1648	if (conn->conn_ops->DataDigest) {
1649		iov[iov_count].iov_base = &checksum;
1650		iov[iov_count++].iov_len = ISCSI_CRC_LEN;
1651		rx_size += ISCSI_CRC_LEN;
1652	}
1653
1654	WARN_ON_ONCE(iov_count > cmd->orig_iov_data_count);
1655	rx_got = rx_data(conn, &cmd->iov_data[0], iov_count, rx_size);
1656
1657	iscsit_unmap_iovec(cmd);
1658
1659	if (rx_got != rx_size)
1660		return -1;
1661
1662	if (conn->conn_ops->DataDigest) {
1663		u32 data_crc;
1664
1665		data_crc = iscsit_do_crypto_hash_sg(conn->conn_rx_hash, cmd,
1666						    be32_to_cpu(hdr->offset),
1667						    payload_length, padding,
1668						    cmd->pad_bytes);
1669
1670		if (checksum != data_crc) {
1671			pr_err("ITT: 0x%08x, Offset: %u, Length: %u,"
1672				" DataSN: 0x%08x, CRC32C DataDigest 0x%08x"
1673				" does not match computed 0x%08x\n",
1674				hdr->itt, hdr->offset, payload_length,
1675				hdr->datasn, checksum, data_crc);
1676			data_crc_failed = 1;
1677		} else {
1678			pr_debug("Got CRC32C DataDigest 0x%08x for"
1679				" %u bytes of Data Out\n", checksum,
1680				payload_length);
1681		}
1682	}
1683
1684	return data_crc_failed;
1685}
1686
1687int
1688iscsit_check_dataout_payload(struct iscsit_cmd *cmd, struct iscsi_data *hdr,
1689			     bool data_crc_failed)
1690{
1691	struct iscsit_conn *conn = cmd->conn;
1692	int rc, ooo_cmdsn;
1693	/*
1694	 * Increment post receive data and CRC values or perform
1695	 * within-command recovery.
1696	 */
1697	rc = iscsit_check_post_dataout(cmd, (unsigned char *)hdr, data_crc_failed);
1698	if ((rc == DATAOUT_NORMAL) || (rc == DATAOUT_WITHIN_COMMAND_RECOVERY))
1699		return 0;
1700	else if (rc == DATAOUT_SEND_R2T) {
1701		iscsit_set_dataout_sequence_values(cmd);
1702		conn->conn_transport->iscsit_get_dataout(conn, cmd, false);
1703	} else if (rc == DATAOUT_SEND_TO_TRANSPORT) {
1704		/*
1705		 * Handle extra special case for out of order
1706		 * Unsolicited Data Out.
1707		 */
1708		spin_lock_bh(&cmd->istate_lock);
1709		ooo_cmdsn = (cmd->cmd_flags & ICF_OOO_CMDSN);
1710		cmd->cmd_flags |= ICF_GOT_LAST_DATAOUT;
1711		cmd->i_state = ISTATE_RECEIVED_LAST_DATAOUT;
1712		spin_unlock_bh(&cmd->istate_lock);
1713
1714		iscsit_stop_dataout_timer(cmd);
1715		if (ooo_cmdsn)
1716			return 0;
1717		target_execute_cmd(&cmd->se_cmd);
1718		return 0;
1719	} else /* DATAOUT_CANNOT_RECOVER */
1720		return -1;
1721
1722	return 0;
1723}
1724EXPORT_SYMBOL(iscsit_check_dataout_payload);
1725
1726static int iscsit_handle_data_out(struct iscsit_conn *conn, unsigned char *buf)
1727{
1728	struct iscsit_cmd *cmd = NULL;
1729	struct iscsi_data *hdr = (struct iscsi_data *)buf;
1730	int rc;
1731	bool data_crc_failed = false;
1732
1733	rc = iscsit_check_dataout_hdr(conn, buf, &cmd);
1734	if (rc < 0)
1735		return 0;
1736	else if (!cmd)
1737		return 0;
1738
1739	rc = iscsit_get_dataout(conn, cmd, hdr);
1740	if (rc < 0)
1741		return rc;
1742	else if (rc > 0)
1743		data_crc_failed = true;
1744
1745	return iscsit_check_dataout_payload(cmd, hdr, data_crc_failed);
1746}
1747
1748int iscsit_setup_nop_out(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
1749			 struct iscsi_nopout *hdr)
1750{
1751	u32 payload_length = ntoh24(hdr->dlength);
1752
1753	if (!(hdr->flags & ISCSI_FLAG_CMD_FINAL)) {
1754		pr_err("NopOUT Flag's, Left Most Bit not set, protocol error.\n");
1755		if (!cmd)
1756			return iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR,
1757						 (unsigned char *)hdr);
1758
1759		return iscsit_reject_cmd(cmd, ISCSI_REASON_PROTOCOL_ERROR,
1760					 (unsigned char *)hdr);
1761	}
1762
1763	if (hdr->itt == RESERVED_ITT && !(hdr->opcode & ISCSI_OP_IMMEDIATE)) {
1764		pr_err("NOPOUT ITT is reserved, but Immediate Bit is"
1765			" not set, protocol error.\n");
1766		if (!cmd)
1767			return iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR,
1768						 (unsigned char *)hdr);
1769
1770		return iscsit_reject_cmd(cmd, ISCSI_REASON_PROTOCOL_ERROR,
1771					 (unsigned char *)hdr);
1772	}
1773
1774	if (payload_length > conn->conn_ops->MaxXmitDataSegmentLength) {
1775		pr_err("NOPOUT Ping Data DataSegmentLength: %u is"
1776			" greater than MaxXmitDataSegmentLength: %u, protocol"
1777			" error.\n", payload_length,
1778			conn->conn_ops->MaxXmitDataSegmentLength);
1779		if (!cmd)
1780			return iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR,
1781						 (unsigned char *)hdr);
1782
1783		return iscsit_reject_cmd(cmd, ISCSI_REASON_PROTOCOL_ERROR,
1784					 (unsigned char *)hdr);
1785	}
1786
1787	pr_debug("Got NOPOUT Ping %s ITT: 0x%08x, TTT: 0x%08x,"
1788		" CmdSN: 0x%08x, ExpStatSN: 0x%08x, Length: %u\n",
1789		hdr->itt == RESERVED_ITT ? "Response" : "Request",
1790		hdr->itt, hdr->ttt, hdr->cmdsn, hdr->exp_statsn,
1791		payload_length);
1792	/*
1793	 * This is not a response to a Unsolicited NopIN, which means
1794	 * it can either be a NOPOUT ping request (with a valid ITT),
1795	 * or a NOPOUT not requesting a NOPIN (with a reserved ITT).
1796	 * Either way, make sure we allocate an struct iscsit_cmd, as both
1797	 * can contain ping data.
1798	 */
1799	if (hdr->ttt == cpu_to_be32(0xFFFFFFFF)) {
1800		cmd->iscsi_opcode	= ISCSI_OP_NOOP_OUT;
1801		cmd->i_state		= ISTATE_SEND_NOPIN;
1802		cmd->immediate_cmd	= ((hdr->opcode & ISCSI_OP_IMMEDIATE) ?
1803						1 : 0);
1804		conn->sess->init_task_tag = cmd->init_task_tag = hdr->itt;
1805		cmd->targ_xfer_tag	= 0xFFFFFFFF;
1806		cmd->cmd_sn		= be32_to_cpu(hdr->cmdsn);
1807		cmd->exp_stat_sn	= be32_to_cpu(hdr->exp_statsn);
1808		cmd->data_direction	= DMA_NONE;
1809	}
1810
1811	return 0;
1812}
1813EXPORT_SYMBOL(iscsit_setup_nop_out);
1814
1815int iscsit_process_nop_out(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
1816			   struct iscsi_nopout *hdr)
1817{
1818	struct iscsit_cmd *cmd_p = NULL;
1819	int cmdsn_ret = 0;
1820	/*
1821	 * Initiator is expecting a NopIN ping reply..
1822	 */
1823	if (hdr->itt != RESERVED_ITT) {
1824		if (!cmd)
1825			return iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR,
1826						(unsigned char *)hdr);
1827
1828		spin_lock_bh(&conn->cmd_lock);
1829		list_add_tail(&cmd->i_conn_node, &conn->conn_cmd_list);
1830		spin_unlock_bh(&conn->cmd_lock);
1831
1832		iscsit_ack_from_expstatsn(conn, be32_to_cpu(hdr->exp_statsn));
1833
1834		if (hdr->opcode & ISCSI_OP_IMMEDIATE) {
1835			iscsit_add_cmd_to_response_queue(cmd, conn,
1836							 cmd->i_state);
1837			return 0;
1838		}
1839
1840		cmdsn_ret = iscsit_sequence_cmd(conn, cmd,
1841				(unsigned char *)hdr, hdr->cmdsn);
1842                if (cmdsn_ret == CMDSN_LOWER_THAN_EXP)
1843			return 0;
1844		if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
1845			return -1;
1846
1847		return 0;
1848	}
1849	/*
1850	 * This was a response to a unsolicited NOPIN ping.
1851	 */
1852	if (hdr->ttt != cpu_to_be32(0xFFFFFFFF)) {
1853		cmd_p = iscsit_find_cmd_from_ttt(conn, be32_to_cpu(hdr->ttt));
1854		if (!cmd_p)
1855			return -EINVAL;
1856
1857		iscsit_stop_nopin_response_timer(conn);
1858
1859		cmd_p->i_state = ISTATE_REMOVE;
1860		iscsit_add_cmd_to_immediate_queue(cmd_p, conn, cmd_p->i_state);
1861
1862		iscsit_start_nopin_timer(conn);
1863		return 0;
1864	}
1865	/*
1866	 * Otherwise, initiator is not expecting a NOPIN is response.
1867	 * Just ignore for now.
1868	 */
1869
1870	if (cmd)
1871		iscsit_free_cmd(cmd, false);
1872
1873        return 0;
1874}
1875EXPORT_SYMBOL(iscsit_process_nop_out);
1876
1877static int iscsit_handle_nop_out(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
1878				 unsigned char *buf)
1879{
1880	unsigned char *ping_data = NULL;
1881	struct iscsi_nopout *hdr = (struct iscsi_nopout *)buf;
1882	struct kvec *iov = NULL;
1883	u32 payload_length = ntoh24(hdr->dlength);
1884	int ret;
1885
1886	ret = iscsit_setup_nop_out(conn, cmd, hdr);
1887	if (ret < 0)
1888		return 0;
1889	/*
1890	 * Handle NOP-OUT payload for traditional iSCSI sockets
1891	 */
1892	if (payload_length && hdr->ttt == cpu_to_be32(0xFFFFFFFF)) {
1893		u32 checksum, data_crc, padding = 0;
1894		int niov = 0, rx_got, rx_size = payload_length;
1895
1896		ping_data = kzalloc(payload_length + 1, GFP_KERNEL);
1897		if (!ping_data) {
1898			ret = -1;
1899			goto out;
1900		}
1901
1902		iov = &cmd->iov_misc[0];
1903		iov[niov].iov_base	= ping_data;
1904		iov[niov++].iov_len	= payload_length;
1905
1906		padding = ((-payload_length) & 3);
1907		if (padding != 0) {
1908			pr_debug("Receiving %u additional bytes"
1909				" for padding.\n", padding);
1910			iov[niov].iov_base	= &cmd->pad_bytes;
1911			iov[niov++].iov_len	= padding;
1912			rx_size += padding;
1913		}
1914		if (conn->conn_ops->DataDigest) {
1915			iov[niov].iov_base	= &checksum;
1916			iov[niov++].iov_len	= ISCSI_CRC_LEN;
1917			rx_size += ISCSI_CRC_LEN;
1918		}
1919
1920		WARN_ON_ONCE(niov > ARRAY_SIZE(cmd->iov_misc));
1921		rx_got = rx_data(conn, &cmd->iov_misc[0], niov, rx_size);
1922		if (rx_got != rx_size) {
1923			ret = -1;
1924			goto out;
1925		}
1926
1927		if (conn->conn_ops->DataDigest) {
1928			iscsit_do_crypto_hash_buf(conn->conn_rx_hash, ping_data,
1929						  payload_length, padding,
1930						  cmd->pad_bytes, &data_crc);
1931
1932			if (checksum != data_crc) {
1933				pr_err("Ping data CRC32C DataDigest"
1934				" 0x%08x does not match computed 0x%08x\n",
1935					checksum, data_crc);
1936				if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
1937					pr_err("Unable to recover from"
1938					" NOPOUT Ping DataCRC failure while in"
1939						" ERL=0.\n");
1940					ret = -1;
1941					goto out;
1942				} else {
1943					/*
1944					 * Silently drop this PDU and let the
1945					 * initiator plug the CmdSN gap.
1946					 */
1947					pr_debug("Dropping NOPOUT"
1948					" Command CmdSN: 0x%08x due to"
1949					" DataCRC error.\n", hdr->cmdsn);
1950					ret = 0;
1951					goto out;
1952				}
1953			} else {
1954				pr_debug("Got CRC32C DataDigest"
1955				" 0x%08x for %u bytes of ping data.\n",
1956					checksum, payload_length);
1957			}
1958		}
1959
1960		ping_data[payload_length] = '\0';
1961		/*
1962		 * Attach ping data to struct iscsit_cmd->buf_ptr.
1963		 */
1964		cmd->buf_ptr = ping_data;
1965		cmd->buf_ptr_size = payload_length;
1966
1967		pr_debug("Got %u bytes of NOPOUT ping"
1968			" data.\n", payload_length);
1969		pr_debug("Ping Data: \"%s\"\n", ping_data);
1970	}
1971
1972	return iscsit_process_nop_out(conn, cmd, hdr);
1973out:
1974	if (cmd)
1975		iscsit_free_cmd(cmd, false);
1976
1977	kfree(ping_data);
1978	return ret;
1979}
1980
1981static enum tcm_tmreq_table iscsit_convert_tmf(u8 iscsi_tmf)
1982{
1983	switch (iscsi_tmf) {
1984	case ISCSI_TM_FUNC_ABORT_TASK:
1985		return TMR_ABORT_TASK;
1986	case ISCSI_TM_FUNC_ABORT_TASK_SET:
1987		return TMR_ABORT_TASK_SET;
1988	case ISCSI_TM_FUNC_CLEAR_ACA:
1989		return TMR_CLEAR_ACA;
1990	case ISCSI_TM_FUNC_CLEAR_TASK_SET:
1991		return TMR_CLEAR_TASK_SET;
1992	case ISCSI_TM_FUNC_LOGICAL_UNIT_RESET:
1993		return TMR_LUN_RESET;
1994	case ISCSI_TM_FUNC_TARGET_WARM_RESET:
1995		return TMR_TARGET_WARM_RESET;
1996	case ISCSI_TM_FUNC_TARGET_COLD_RESET:
1997		return TMR_TARGET_COLD_RESET;
1998	default:
1999		return TMR_UNKNOWN;
2000	}
2001}
2002
2003int
2004iscsit_handle_task_mgt_cmd(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
2005			   unsigned char *buf)
2006{
2007	struct se_tmr_req *se_tmr;
2008	struct iscsi_tmr_req *tmr_req;
2009	struct iscsi_tm *hdr;
2010	int out_of_order_cmdsn = 0, ret;
2011	u8 function, tcm_function = TMR_UNKNOWN;
2012
2013	hdr			= (struct iscsi_tm *) buf;
2014	hdr->flags &= ~ISCSI_FLAG_CMD_FINAL;
2015	function = hdr->flags;
2016
2017	pr_debug("Got Task Management Request ITT: 0x%08x, CmdSN:"
2018		" 0x%08x, Function: 0x%02x, RefTaskTag: 0x%08x, RefCmdSN:"
2019		" 0x%08x, CID: %hu\n", hdr->itt, hdr->cmdsn, function,
2020		hdr->rtt, hdr->refcmdsn, conn->cid);
2021
2022	if ((function != ISCSI_TM_FUNC_ABORT_TASK) &&
2023	    ((function != ISCSI_TM_FUNC_TASK_REASSIGN) &&
2024	     hdr->rtt != RESERVED_ITT)) {
2025		pr_err("RefTaskTag should be set to 0xFFFFFFFF.\n");
2026		hdr->rtt = RESERVED_ITT;
2027	}
2028
2029	if ((function == ISCSI_TM_FUNC_TASK_REASSIGN) &&
2030			!(hdr->opcode & ISCSI_OP_IMMEDIATE)) {
2031		pr_err("Task Management Request TASK_REASSIGN not"
2032			" issued as immediate command, bad iSCSI Initiator"
2033				"implementation\n");
2034		return iscsit_add_reject_cmd(cmd,
2035					     ISCSI_REASON_PROTOCOL_ERROR, buf);
2036	}
2037	if ((function != ISCSI_TM_FUNC_ABORT_TASK) &&
2038	    be32_to_cpu(hdr->refcmdsn) != ISCSI_RESERVED_TAG)
2039		hdr->refcmdsn = cpu_to_be32(ISCSI_RESERVED_TAG);
2040
2041	cmd->data_direction = DMA_NONE;
2042	cmd->tmr_req = kzalloc(sizeof(*cmd->tmr_req), GFP_KERNEL);
2043	if (!cmd->tmr_req) {
2044		return iscsit_add_reject_cmd(cmd,
2045					     ISCSI_REASON_BOOKMARK_NO_RESOURCES,
2046					     buf);
2047	}
2048
2049	__target_init_cmd(&cmd->se_cmd, &iscsi_ops,
2050			  conn->sess->se_sess, 0, DMA_NONE,
2051			  TCM_SIMPLE_TAG, cmd->sense_buffer + 2,
2052			  scsilun_to_int(&hdr->lun),
2053			  conn->cmd_cnt);
2054
2055	target_get_sess_cmd(&cmd->se_cmd, true);
2056
2057	/*
2058	 * TASK_REASSIGN for ERL=2 / connection stays inside of
2059	 * LIO-Target $FABRIC_MOD
2060	 */
2061	if (function != ISCSI_TM_FUNC_TASK_REASSIGN) {
2062		tcm_function = iscsit_convert_tmf(function);
2063		if (tcm_function == TMR_UNKNOWN) {
2064			pr_err("Unknown iSCSI TMR Function:"
2065			       " 0x%02x\n", function);
2066			return iscsit_add_reject_cmd(cmd,
2067				ISCSI_REASON_BOOKMARK_NO_RESOURCES, buf);
2068		}
2069	}
2070	ret = core_tmr_alloc_req(&cmd->se_cmd, cmd->tmr_req, tcm_function,
2071				 GFP_KERNEL);
2072	if (ret < 0)
2073		return iscsit_add_reject_cmd(cmd,
2074				ISCSI_REASON_BOOKMARK_NO_RESOURCES, buf);
2075
2076	cmd->tmr_req->se_tmr_req = cmd->se_cmd.se_tmr_req;
2077
2078	cmd->iscsi_opcode	= ISCSI_OP_SCSI_TMFUNC;
2079	cmd->i_state		= ISTATE_SEND_TASKMGTRSP;
2080	cmd->immediate_cmd	= ((hdr->opcode & ISCSI_OP_IMMEDIATE) ? 1 : 0);
2081	cmd->init_task_tag	= hdr->itt;
2082	cmd->targ_xfer_tag	= 0xFFFFFFFF;
2083	cmd->cmd_sn		= be32_to_cpu(hdr->cmdsn);
2084	cmd->exp_stat_sn	= be32_to_cpu(hdr->exp_statsn);
2085	se_tmr			= cmd->se_cmd.se_tmr_req;
2086	tmr_req			= cmd->tmr_req;
2087	/*
2088	 * Locate the struct se_lun for all TMRs not related to ERL=2 TASK_REASSIGN
2089	 */
2090	if (function != ISCSI_TM_FUNC_TASK_REASSIGN) {
2091		ret = transport_lookup_tmr_lun(&cmd->se_cmd);
2092		if (ret < 0) {
2093			se_tmr->response = ISCSI_TMF_RSP_NO_LUN;
2094			goto attach;
2095		}
2096	}
2097
2098	switch (function) {
2099	case ISCSI_TM_FUNC_ABORT_TASK:
2100		se_tmr->response = iscsit_tmr_abort_task(cmd, buf);
2101		if (se_tmr->response)
2102			goto attach;
2103		break;
2104	case ISCSI_TM_FUNC_ABORT_TASK_SET:
2105	case ISCSI_TM_FUNC_CLEAR_ACA:
2106	case ISCSI_TM_FUNC_CLEAR_TASK_SET:
2107	case ISCSI_TM_FUNC_LOGICAL_UNIT_RESET:
2108		break;
2109	case ISCSI_TM_FUNC_TARGET_WARM_RESET:
2110		if (iscsit_tmr_task_warm_reset(conn, tmr_req, buf) < 0) {
2111			se_tmr->response = ISCSI_TMF_RSP_AUTH_FAILED;
2112			goto attach;
2113		}
2114		break;
2115	case ISCSI_TM_FUNC_TARGET_COLD_RESET:
2116		if (iscsit_tmr_task_cold_reset(conn, tmr_req, buf) < 0) {
2117			se_tmr->response = ISCSI_TMF_RSP_AUTH_FAILED;
2118			goto attach;
2119		}
2120		break;
2121	case ISCSI_TM_FUNC_TASK_REASSIGN:
2122		se_tmr->response = iscsit_tmr_task_reassign(cmd, buf);
2123		/*
2124		 * Perform sanity checks on the ExpDataSN only if the
2125		 * TASK_REASSIGN was successful.
2126		 */
2127		if (se_tmr->response)
2128			break;
2129
2130		if (iscsit_check_task_reassign_expdatasn(tmr_req, conn) < 0)
2131			return iscsit_add_reject_cmd(cmd,
2132					ISCSI_REASON_BOOKMARK_INVALID, buf);
2133		break;
2134	default:
2135		pr_err("Unknown TMR function: 0x%02x, protocol"
2136			" error.\n", function);
2137		se_tmr->response = ISCSI_TMF_RSP_NOT_SUPPORTED;
2138		goto attach;
2139	}
2140
2141	if ((function != ISCSI_TM_FUNC_TASK_REASSIGN) &&
2142	    (se_tmr->response == ISCSI_TMF_RSP_COMPLETE))
2143		se_tmr->call_transport = 1;
2144attach:
2145	spin_lock_bh(&conn->cmd_lock);
2146	list_add_tail(&cmd->i_conn_node, &conn->conn_cmd_list);
2147	spin_unlock_bh(&conn->cmd_lock);
2148
2149	if (!(hdr->opcode & ISCSI_OP_IMMEDIATE)) {
2150		int cmdsn_ret = iscsit_sequence_cmd(conn, cmd, buf, hdr->cmdsn);
2151		if (cmdsn_ret == CMDSN_HIGHER_THAN_EXP) {
2152			out_of_order_cmdsn = 1;
2153		} else if (cmdsn_ret == CMDSN_LOWER_THAN_EXP) {
2154			target_put_sess_cmd(&cmd->se_cmd);
2155			return 0;
2156		} else if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER) {
2157			return -1;
2158		}
2159	}
2160	iscsit_ack_from_expstatsn(conn, be32_to_cpu(hdr->exp_statsn));
2161
2162	if (out_of_order_cmdsn || !(hdr->opcode & ISCSI_OP_IMMEDIATE))
2163		return 0;
2164	/*
2165	 * Found the referenced task, send to transport for processing.
2166	 */
2167	if (se_tmr->call_transport)
2168		return transport_generic_handle_tmr(&cmd->se_cmd);
2169
2170	/*
2171	 * Could not find the referenced LUN, task, or Task Management
2172	 * command not authorized or supported.  Change state and
2173	 * let the tx_thread send the response.
2174	 *
2175	 * For connection recovery, this is also the default action for
2176	 * TMR TASK_REASSIGN.
2177	 */
2178	iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2179	target_put_sess_cmd(&cmd->se_cmd);
2180	return 0;
2181}
2182EXPORT_SYMBOL(iscsit_handle_task_mgt_cmd);
2183
2184/* #warning FIXME: Support Text Command parameters besides SendTargets */
2185int
2186iscsit_setup_text_cmd(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
2187		      struct iscsi_text *hdr)
2188{
2189	u32 payload_length = ntoh24(hdr->dlength);
2190
2191	if (payload_length > conn->conn_ops->MaxXmitDataSegmentLength) {
2192		pr_err("Unable to accept text parameter length: %u"
2193			"greater than MaxXmitDataSegmentLength %u.\n",
2194		       payload_length, conn->conn_ops->MaxXmitDataSegmentLength);
2195		return iscsit_reject_cmd(cmd, ISCSI_REASON_PROTOCOL_ERROR,
2196					 (unsigned char *)hdr);
2197	}
2198
2199	if (!(hdr->flags & ISCSI_FLAG_CMD_FINAL) ||
2200	     (hdr->flags & ISCSI_FLAG_TEXT_CONTINUE)) {
2201		pr_err("Multi sequence text commands currently not supported\n");
2202		return iscsit_reject_cmd(cmd, ISCSI_REASON_CMD_NOT_SUPPORTED,
2203					(unsigned char *)hdr);
2204	}
2205
2206	pr_debug("Got Text Request: ITT: 0x%08x, CmdSN: 0x%08x,"
2207		" ExpStatSN: 0x%08x, Length: %u\n", hdr->itt, hdr->cmdsn,
2208		hdr->exp_statsn, payload_length);
2209
2210	cmd->iscsi_opcode	= ISCSI_OP_TEXT;
2211	cmd->i_state		= ISTATE_SEND_TEXTRSP;
2212	cmd->immediate_cmd	= ((hdr->opcode & ISCSI_OP_IMMEDIATE) ? 1 : 0);
2213	conn->sess->init_task_tag = cmd->init_task_tag  = hdr->itt;
2214	cmd->targ_xfer_tag	= 0xFFFFFFFF;
2215	cmd->cmd_sn		= be32_to_cpu(hdr->cmdsn);
2216	cmd->exp_stat_sn	= be32_to_cpu(hdr->exp_statsn);
2217	cmd->data_direction	= DMA_NONE;
2218	kfree(cmd->text_in_ptr);
2219	cmd->text_in_ptr	= NULL;
2220
2221	return 0;
2222}
2223EXPORT_SYMBOL(iscsit_setup_text_cmd);
2224
2225int
2226iscsit_process_text_cmd(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
2227			struct iscsi_text *hdr)
2228{
2229	unsigned char *text_in = cmd->text_in_ptr, *text_ptr;
2230	int cmdsn_ret;
2231
2232	if (!text_in) {
2233		cmd->targ_xfer_tag = be32_to_cpu(hdr->ttt);
2234		if (cmd->targ_xfer_tag == 0xFFFFFFFF) {
2235			pr_err("Unable to locate text_in buffer for sendtargets"
2236			       " discovery\n");
2237			goto reject;
2238		}
2239		goto empty_sendtargets;
2240	}
2241	if (strncmp("SendTargets=", text_in, 12) != 0) {
2242		pr_err("Received Text Data that is not"
2243			" SendTargets, cannot continue.\n");
2244		goto reject;
2245	}
2246	/* '=' confirmed in strncmp */
2247	text_ptr = strchr(text_in, '=');
2248	BUG_ON(!text_ptr);
2249	if (!strncmp("=All", text_ptr, 5)) {
2250		cmd->cmd_flags |= ICF_SENDTARGETS_ALL;
2251	} else if (!strncmp("=iqn.", text_ptr, 5) ||
2252		   !strncmp("=eui.", text_ptr, 5)) {
2253		cmd->cmd_flags |= ICF_SENDTARGETS_SINGLE;
2254	} else {
2255		pr_err("Unable to locate valid SendTargets%s value\n",
2256		       text_ptr);
2257		goto reject;
2258	}
2259
2260	spin_lock_bh(&conn->cmd_lock);
2261	list_add_tail(&cmd->i_conn_node, &conn->conn_cmd_list);
2262	spin_unlock_bh(&conn->cmd_lock);
2263
2264empty_sendtargets:
2265	iscsit_ack_from_expstatsn(conn, be32_to_cpu(hdr->exp_statsn));
2266
2267	if (!(hdr->opcode & ISCSI_OP_IMMEDIATE)) {
2268		cmdsn_ret = iscsit_sequence_cmd(conn, cmd,
2269				(unsigned char *)hdr, hdr->cmdsn);
2270		if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
2271			return -1;
2272
2273		return 0;
2274	}
2275
2276	return iscsit_execute_cmd(cmd, 0);
2277
2278reject:
2279	return iscsit_reject_cmd(cmd, ISCSI_REASON_PROTOCOL_ERROR,
2280				 (unsigned char *)hdr);
2281}
2282EXPORT_SYMBOL(iscsit_process_text_cmd);
2283
2284static int
2285iscsit_handle_text_cmd(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
2286		       unsigned char *buf)
2287{
2288	struct iscsi_text *hdr = (struct iscsi_text *)buf;
2289	char *text_in = NULL;
2290	u32 payload_length = ntoh24(hdr->dlength);
2291	int rx_size, rc;
2292
2293	rc = iscsit_setup_text_cmd(conn, cmd, hdr);
2294	if (rc < 0)
2295		return 0;
2296
2297	rx_size = payload_length;
2298	if (payload_length) {
2299		u32 checksum = 0, data_crc = 0;
2300		u32 padding = 0;
2301		int niov = 0, rx_got;
2302		struct kvec iov[2];
2303
2304		rx_size = ALIGN(payload_length, 4);
2305		text_in = kzalloc(rx_size, GFP_KERNEL);
2306		if (!text_in)
2307			goto reject;
2308
2309		cmd->text_in_ptr = text_in;
2310
2311		memset(iov, 0, sizeof(iov));
2312		iov[niov].iov_base	= text_in;
2313		iov[niov++].iov_len	= rx_size;
2314
2315		padding = rx_size - payload_length;
2316		if (padding)
2317			pr_debug("Receiving %u additional bytes"
2318					" for padding.\n", padding);
2319		if (conn->conn_ops->DataDigest) {
2320			iov[niov].iov_base	= &checksum;
2321			iov[niov++].iov_len	= ISCSI_CRC_LEN;
2322			rx_size += ISCSI_CRC_LEN;
2323		}
2324
2325		WARN_ON_ONCE(niov > ARRAY_SIZE(iov));
2326		rx_got = rx_data(conn, &iov[0], niov, rx_size);
2327		if (rx_got != rx_size)
2328			goto reject;
2329
2330		if (conn->conn_ops->DataDigest) {
2331			iscsit_do_crypto_hash_buf(conn->conn_rx_hash,
2332						  text_in, rx_size, 0, NULL,
2333						  &data_crc);
2334
2335			if (checksum != data_crc) {
2336				pr_err("Text data CRC32C DataDigest"
2337					" 0x%08x does not match computed"
2338					" 0x%08x\n", checksum, data_crc);
2339				if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
2340					pr_err("Unable to recover from"
2341					" Text Data digest failure while in"
2342						" ERL=0.\n");
2343					goto reject;
2344				} else {
2345					/*
2346					 * Silently drop this PDU and let the
2347					 * initiator plug the CmdSN gap.
2348					 */
2349					pr_debug("Dropping Text"
2350					" Command CmdSN: 0x%08x due to"
2351					" DataCRC error.\n", hdr->cmdsn);
2352					kfree(text_in);
2353					return 0;
2354				}
2355			} else {
2356				pr_debug("Got CRC32C DataDigest"
2357					" 0x%08x for %u bytes of text data.\n",
2358						checksum, payload_length);
2359			}
2360		}
2361		text_in[payload_length - 1] = '\0';
2362		pr_debug("Successfully read %d bytes of text"
2363				" data.\n", payload_length);
2364	}
2365
2366	return iscsit_process_text_cmd(conn, cmd, hdr);
2367
2368reject:
2369	kfree(cmd->text_in_ptr);
2370	cmd->text_in_ptr = NULL;
2371	return iscsit_reject_cmd(cmd, ISCSI_REASON_PROTOCOL_ERROR, buf);
2372}
2373
2374int iscsit_logout_closesession(struct iscsit_cmd *cmd, struct iscsit_conn *conn)
2375{
2376	struct iscsit_conn *conn_p;
2377	struct iscsit_session *sess = conn->sess;
2378
2379	pr_debug("Received logout request CLOSESESSION on CID: %hu"
2380		" for SID: %u.\n", conn->cid, conn->sess->sid);
2381
2382	atomic_set(&sess->session_logout, 1);
2383	atomic_set(&conn->conn_logout_remove, 1);
2384	conn->conn_logout_reason = ISCSI_LOGOUT_REASON_CLOSE_SESSION;
2385
2386	iscsit_inc_conn_usage_count(conn);
2387	iscsit_inc_session_usage_count(sess);
2388
2389	spin_lock_bh(&sess->conn_lock);
2390	list_for_each_entry(conn_p, &sess->sess_conn_list, conn_list) {
2391		if (conn_p->conn_state != TARG_CONN_STATE_LOGGED_IN)
2392			continue;
2393
2394		pr_debug("Moving to TARG_CONN_STATE_IN_LOGOUT.\n");
2395		conn_p->conn_state = TARG_CONN_STATE_IN_LOGOUT;
2396	}
2397	spin_unlock_bh(&sess->conn_lock);
2398
2399	iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2400
2401	return 0;
2402}
2403
2404int iscsit_logout_closeconnection(struct iscsit_cmd *cmd, struct iscsit_conn *conn)
2405{
2406	struct iscsit_conn *l_conn;
2407	struct iscsit_session *sess = conn->sess;
2408
2409	pr_debug("Received logout request CLOSECONNECTION for CID:"
2410		" %hu on CID: %hu.\n", cmd->logout_cid, conn->cid);
2411
2412	/*
2413	 * A Logout Request with a CLOSECONNECTION reason code for a CID
2414	 * can arrive on a connection with a differing CID.
2415	 */
2416	if (conn->cid == cmd->logout_cid) {
2417		spin_lock_bh(&conn->state_lock);
2418		pr_debug("Moving to TARG_CONN_STATE_IN_LOGOUT.\n");
2419		conn->conn_state = TARG_CONN_STATE_IN_LOGOUT;
2420
2421		atomic_set(&conn->conn_logout_remove, 1);
2422		conn->conn_logout_reason = ISCSI_LOGOUT_REASON_CLOSE_CONNECTION;
2423		iscsit_inc_conn_usage_count(conn);
2424
2425		spin_unlock_bh(&conn->state_lock);
2426	} else {
2427		/*
2428		 * Handle all different cid CLOSECONNECTION requests in
2429		 * iscsit_logout_post_handler_diffcid() as to give enough
2430		 * time for any non immediate command's CmdSN to be
2431		 * acknowledged on the connection in question.
2432		 *
2433		 * Here we simply make sure the CID is still around.
2434		 */
2435		l_conn = iscsit_get_conn_from_cid(sess,
2436				cmd->logout_cid);
2437		if (!l_conn) {
2438			cmd->logout_response = ISCSI_LOGOUT_CID_NOT_FOUND;
2439			iscsit_add_cmd_to_response_queue(cmd, conn,
2440					cmd->i_state);
2441			return 0;
2442		}
2443
2444		iscsit_dec_conn_usage_count(l_conn);
2445	}
2446
2447	iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2448
2449	return 0;
2450}
2451
2452int iscsit_logout_removeconnforrecovery(struct iscsit_cmd *cmd, struct iscsit_conn *conn)
2453{
2454	struct iscsit_session *sess = conn->sess;
2455
2456	pr_debug("Received explicit REMOVECONNFORRECOVERY logout for"
2457		" CID: %hu on CID: %hu.\n", cmd->logout_cid, conn->cid);
2458
2459	if (sess->sess_ops->ErrorRecoveryLevel != 2) {
2460		pr_err("Received Logout Request REMOVECONNFORRECOVERY"
2461			" while ERL!=2.\n");
2462		cmd->logout_response = ISCSI_LOGOUT_RECOVERY_UNSUPPORTED;
2463		iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2464		return 0;
2465	}
2466
2467	if (conn->cid == cmd->logout_cid) {
2468		pr_err("Received Logout Request REMOVECONNFORRECOVERY"
2469			" with CID: %hu on CID: %hu, implementation error.\n",
2470				cmd->logout_cid, conn->cid);
2471		cmd->logout_response = ISCSI_LOGOUT_CLEANUP_FAILED;
2472		iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2473		return 0;
2474	}
2475
2476	iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2477
2478	return 0;
2479}
2480
2481int
2482iscsit_handle_logout_cmd(struct iscsit_conn *conn, struct iscsit_cmd *cmd,
2483			unsigned char *buf)
2484{
2485	int cmdsn_ret, logout_remove = 0;
2486	u8 reason_code = 0;
2487	struct iscsi_logout *hdr;
2488	struct iscsi_tiqn *tiqn = iscsit_snmp_get_tiqn(conn);
2489
2490	hdr			= (struct iscsi_logout *) buf;
2491	reason_code		= (hdr->flags & 0x7f);
2492
2493	if (tiqn) {
2494		spin_lock(&tiqn->logout_stats.lock);
2495		if (reason_code == ISCSI_LOGOUT_REASON_CLOSE_SESSION)
2496			tiqn->logout_stats.normal_logouts++;
2497		else
2498			tiqn->logout_stats.abnormal_logouts++;
2499		spin_unlock(&tiqn->logout_stats.lock);
2500	}
2501
2502	pr_debug("Got Logout Request ITT: 0x%08x CmdSN: 0x%08x"
2503		" ExpStatSN: 0x%08x Reason: 0x%02x CID: %hu on CID: %hu\n",
2504		hdr->itt, hdr->cmdsn, hdr->exp_statsn, reason_code,
2505		hdr->cid, conn->cid);
2506
2507	if (conn->conn_state != TARG_CONN_STATE_LOGGED_IN) {
2508		pr_err("Received logout request on connection that"
2509			" is not in logged in state, ignoring request.\n");
2510		iscsit_free_cmd(cmd, false);
2511		return 0;
2512	}
2513
2514	cmd->iscsi_opcode       = ISCSI_OP_LOGOUT;
2515	cmd->i_state            = ISTATE_SEND_LOGOUTRSP;
2516	cmd->immediate_cmd      = ((hdr->opcode & ISCSI_OP_IMMEDIATE) ? 1 : 0);
2517	conn->sess->init_task_tag = cmd->init_task_tag  = hdr->itt;
2518	cmd->targ_xfer_tag      = 0xFFFFFFFF;
2519	cmd->cmd_sn             = be32_to_cpu(hdr->cmdsn);
2520	cmd->exp_stat_sn        = be32_to_cpu(hdr->exp_statsn);
2521	cmd->logout_cid         = be16_to_cpu(hdr->cid);
2522	cmd->logout_reason      = reason_code;
2523	cmd->data_direction     = DMA_NONE;
2524
2525	/*
2526	 * We need to sleep in these cases (by returning 1) until the Logout
2527	 * Response gets sent in the tx thread.
2528	 */
2529	if ((reason_code == ISCSI_LOGOUT_REASON_CLOSE_SESSION) ||
2530	   ((reason_code == ISCSI_LOGOUT_REASON_CLOSE_CONNECTION) &&
2531	    be16_to_cpu(hdr->cid) == conn->cid))
2532		logout_remove = 1;
2533
2534	spin_lock_bh(&conn->cmd_lock);
2535	list_add_tail(&cmd->i_conn_node, &conn->conn_cmd_list);
2536	spin_unlock_bh(&conn->cmd_lock);
2537
2538	if (reason_code != ISCSI_LOGOUT_REASON_RECOVERY)
2539		iscsit_ack_from_expstatsn(conn, be32_to_cpu(hdr->exp_statsn));
2540
2541	/*
2542	 * Immediate commands are executed, well, immediately.
2543	 * Non-Immediate Logout Commands are executed in CmdSN order.
2544	 */
2545	if (cmd->immediate_cmd) {
2546		int ret = iscsit_execute_cmd(cmd, 0);
2547
2548		if (ret < 0)
2549			return ret;
2550	} else {
2551		cmdsn_ret = iscsit_sequence_cmd(conn, cmd, buf, hdr->cmdsn);
2552		if (cmdsn_ret == CMDSN_LOWER_THAN_EXP)
2553			logout_remove = 0;
2554		else if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
2555			return -1;
2556	}
2557
2558	return logout_remove;
2559}
2560EXPORT_SYMBOL(iscsit_handle_logout_cmd);
2561
2562int iscsit_handle_snack(
2563	struct iscsit_conn *conn,
2564	unsigned char *buf)
2565{
2566	struct iscsi_snack *hdr;
2567
2568	hdr			= (struct iscsi_snack *) buf;
2569	hdr->flags		&= ~ISCSI_FLAG_CMD_FINAL;
2570
2571	pr_debug("Got ISCSI_INIT_SNACK, ITT: 0x%08x, ExpStatSN:"
2572		" 0x%08x, Type: 0x%02x, BegRun: 0x%08x, RunLength: 0x%08x,"
2573		" CID: %hu\n", hdr->itt, hdr->exp_statsn, hdr->flags,
2574			hdr->begrun, hdr->runlength, conn->cid);
2575
2576	if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
2577		pr_err("Initiator sent SNACK request while in"
2578			" ErrorRecoveryLevel=0.\n");
2579		return iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR,
2580					 buf);
2581	}
2582	/*
2583	 * SNACK_DATA and SNACK_R2T are both 0,  so check which function to
2584	 * call from inside iscsi_send_recovery_datain_or_r2t().
2585	 */
2586	switch (hdr->flags & ISCSI_FLAG_SNACK_TYPE_MASK) {
2587	case 0:
2588		return iscsit_handle_recovery_datain_or_r2t(conn, buf,
2589			hdr->itt,
2590			be32_to_cpu(hdr->ttt),
2591			be32_to_cpu(hdr->begrun),
2592			be32_to_cpu(hdr->runlength));
2593	case ISCSI_FLAG_SNACK_TYPE_STATUS:
2594		return iscsit_handle_status_snack(conn, hdr->itt,
2595			be32_to_cpu(hdr->ttt),
2596			be32_to_cpu(hdr->begrun), be32_to_cpu(hdr->runlength));
2597	case ISCSI_FLAG_SNACK_TYPE_DATA_ACK:
2598		return iscsit_handle_data_ack(conn, be32_to_cpu(hdr->ttt),
2599			be32_to_cpu(hdr->begrun),
2600			be32_to_cpu(hdr->runlength));
2601	case ISCSI_FLAG_SNACK_TYPE_RDATA:
2602		/* FIXME: Support R-Data SNACK */
2603		pr_err("R-Data SNACK Not Supported.\n");
2604		return iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR,
2605					 buf);
2606	default:
2607		pr_err("Unknown SNACK type 0x%02x, protocol"
2608			" error.\n", hdr->flags & 0x0f);
2609		return iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR,
2610					 buf);
2611	}
2612
2613	return 0;
2614}
2615EXPORT_SYMBOL(iscsit_handle_snack);
2616
2617static void iscsit_rx_thread_wait_for_tcp(struct iscsit_conn *conn)
2618{
2619	if ((conn->sock->sk->sk_shutdown & SEND_SHUTDOWN) ||
2620	    (conn->sock->sk->sk_shutdown & RCV_SHUTDOWN)) {
2621		wait_for_completion_interruptible_timeout(
2622					&conn->rx_half_close_comp,
2623					ISCSI_RX_THREAD_TCP_TIMEOUT * HZ);
2624	}
2625}
2626
2627static int iscsit_handle_immediate_data(
2628	struct iscsit_cmd *cmd,
2629	struct iscsi_scsi_req *hdr,
2630	u32 length)
2631{
2632	int iov_ret, rx_got = 0, rx_size = 0;
2633	u32 checksum, iov_count = 0, padding = 0;
2634	struct iscsit_conn *conn = cmd->conn;
2635	struct kvec *iov;
2636	void *overflow_buf = NULL;
2637
2638	BUG_ON(cmd->write_data_done > cmd->se_cmd.data_length);
2639	rx_size = min(cmd->se_cmd.data_length - cmd->write_data_done, length);
2640	iov_ret = iscsit_map_iovec(cmd, cmd->iov_data,
2641				   cmd->orig_iov_data_count - 2,
2642				   cmd->write_data_done, rx_size);
2643	if (iov_ret < 0)
2644		return IMMEDIATE_DATA_CANNOT_RECOVER;
2645
2646	iov_count = iov_ret;
2647	iov = &cmd->iov_data[0];
2648	if (rx_size < length) {
2649		/*
2650		 * Special case: length of immediate data exceeds the data
2651		 * buffer size derived from the CDB.
2652		 */
2653		overflow_buf = kmalloc(length - rx_size, GFP_KERNEL);
2654		if (!overflow_buf) {
2655			iscsit_unmap_iovec(cmd);
2656			return IMMEDIATE_DATA_CANNOT_RECOVER;
2657		}
2658		cmd->overflow_buf = overflow_buf;
2659		iov[iov_count].iov_base = overflow_buf;
2660		iov[iov_count].iov_len = length - rx_size;
2661		iov_count++;
2662		rx_size = length;
2663	}
2664
2665	padding = ((-length) & 3);
2666	if (padding != 0) {
2667		iov[iov_count].iov_base	= cmd->pad_bytes;
2668		iov[iov_count++].iov_len = padding;
2669		rx_size += padding;
2670	}
2671
2672	if (conn->conn_ops->DataDigest) {
2673		iov[iov_count].iov_base		= &checksum;
2674		iov[iov_count++].iov_len	= ISCSI_CRC_LEN;
2675		rx_size += ISCSI_CRC_LEN;
2676	}
2677
2678	WARN_ON_ONCE(iov_count > cmd->orig_iov_data_count);
2679	rx_got = rx_data(conn, &cmd->iov_data[0], iov_count, rx_size);
2680
2681	iscsit_unmap_iovec(cmd);
2682
2683	if (rx_got != rx_size) {
2684		iscsit_rx_thread_wait_for_tcp(conn);
2685		return IMMEDIATE_DATA_CANNOT_RECOVER;
2686	}
2687
2688	if (conn->conn_ops->DataDigest) {
2689		u32 data_crc;
2690
2691		data_crc = iscsit_do_crypto_hash_sg(conn->conn_rx_hash, cmd,
2692						    cmd->write_data_done, length, padding,
2693						    cmd->pad_bytes);
2694
2695		if (checksum != data_crc) {
2696			pr_err("ImmediateData CRC32C DataDigest 0x%08x"
2697				" does not match computed 0x%08x\n", checksum,
2698				data_crc);
2699
2700			if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
2701				pr_err("Unable to recover from"
2702					" Immediate Data digest failure while"
2703					" in ERL=0.\n");
2704				iscsit_reject_cmd(cmd,
2705						ISCSI_REASON_DATA_DIGEST_ERROR,
2706						(unsigned char *)hdr);
2707				return IMMEDIATE_DATA_CANNOT_RECOVER;
2708			} else {
2709				iscsit_reject_cmd(cmd,
2710						ISCSI_REASON_DATA_DIGEST_ERROR,
2711						(unsigned char *)hdr);
2712				return IMMEDIATE_DATA_ERL1_CRC_FAILURE;
2713			}
2714		} else {
2715			pr_debug("Got CRC32C DataDigest 0x%08x for"
2716				" %u bytes of Immediate Data\n", checksum,
2717				length);
2718		}
2719	}
2720
2721	cmd->write_data_done += length;
2722
2723	if (cmd->write_data_done == cmd->se_cmd.data_length) {
2724		spin_lock_bh(&cmd->istate_lock);
2725		cmd->cmd_flags |= ICF_GOT_LAST_DATAOUT;
2726		cmd->i_state = ISTATE_RECEIVED_LAST_DATAOUT;
2727		spin_unlock_bh(&cmd->istate_lock);
2728	}
2729
2730	return IMMEDIATE_DATA_NORMAL_OPERATION;
2731}
2732
2733/* #warning iscsi_build_conn_drop_async_message() only sends out on connections
2734	with active network interface */
2735static void iscsit_build_conn_drop_async_message(struct iscsit_conn *conn)
2736{
2737	struct iscsit_cmd *cmd;
2738	struct iscsit_conn *conn_p;
2739	bool found = false;
2740
2741	lockdep_assert_held(&conn->sess->conn_lock);
2742
2743	/*
2744	 * Only send a Asynchronous Message on connections whos network
2745	 * interface is still functional.
2746	 */
2747	list_for_each_entry(conn_p, &conn->sess->sess_conn_list, conn_list) {
2748		if (conn_p->conn_state == TARG_CONN_STATE_LOGGED_IN) {
2749			iscsit_inc_conn_usage_count(conn_p);
2750			found = true;
2751			break;
2752		}
2753	}
2754
2755	if (!found)
2756		return;
2757
2758	cmd = iscsit_allocate_cmd(conn_p, TASK_RUNNING);
2759	if (!cmd) {
2760		iscsit_dec_conn_usage_count(conn_p);
2761		return;
2762	}
2763
2764	cmd->logout_cid = conn->cid;
2765	cmd->iscsi_opcode = ISCSI_OP_ASYNC_EVENT;
2766	cmd->i_state = ISTATE_SEND_ASYNCMSG;
2767
2768	spin_lock_bh(&conn_p->cmd_lock);
2769	list_add_tail(&cmd->i_conn_node, &conn_p->conn_cmd_list);
2770	spin_unlock_bh(&conn_p->cmd_lock);
2771
2772	iscsit_add_cmd_to_response_queue(cmd, conn_p, cmd->i_state);
2773	iscsit_dec_conn_usage_count(conn_p);
2774}
2775
2776static int iscsit_send_conn_drop_async_message(
2777	struct iscsit_cmd *cmd,
2778	struct iscsit_conn *conn)
2779{
2780	struct iscsi_async *hdr;
2781
2782	cmd->iscsi_opcode = ISCSI_OP_ASYNC_EVENT;
2783
2784	hdr			= (struct iscsi_async *) cmd->pdu;
2785	hdr->opcode		= ISCSI_OP_ASYNC_EVENT;
2786	hdr->flags		= ISCSI_FLAG_CMD_FINAL;
2787	cmd->init_task_tag	= RESERVED_ITT;
2788	cmd->targ_xfer_tag	= 0xFFFFFFFF;
2789	put_unaligned_be64(0xFFFFFFFFFFFFFFFFULL, &hdr->rsvd4[0]);
2790	cmd->stat_sn		= conn->stat_sn++;
2791	hdr->statsn		= cpu_to_be32(cmd->stat_sn);
2792	hdr->exp_cmdsn		= cpu_to_be32(conn->sess->exp_cmd_sn);
2793	hdr->max_cmdsn		= cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
2794	hdr->async_event	= ISCSI_ASYNC_MSG_DROPPING_CONNECTION;
2795	hdr->param1		= cpu_to_be16(cmd->logout_cid);
2796	hdr->param2		= cpu_to_be16(conn->sess->sess_ops->DefaultTime2Wait);
2797	hdr->param3		= cpu_to_be16(conn->sess->sess_ops->DefaultTime2Retain);
2798
2799	pr_debug("Sending Connection Dropped Async Message StatSN:"
2800		" 0x%08x, for CID: %hu on CID: %hu\n", cmd->stat_sn,
2801			cmd->logout_cid, conn->cid);
2802
2803	return conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL, NULL, 0);
2804}
2805
2806static void iscsit_tx_thread_wait_for_tcp(struct iscsit_conn *conn)
2807{
2808	if ((conn->sock->sk->sk_shutdown & SEND_SHUTDOWN) ||
2809	    (conn->sock->sk->sk_shutdown & RCV_SHUTDOWN)) {
2810		wait_for_completion_interruptible_timeout(
2811					&conn->tx_half_close_comp,
2812					ISCSI_TX_THREAD_TCP_TIMEOUT * HZ);
2813	}
2814}
2815
2816void
2817iscsit_build_datain_pdu(struct iscsit_cmd *cmd, struct iscsit_conn *conn,
2818			struct iscsi_datain *datain, struct iscsi_data_rsp *hdr,
2819			bool set_statsn)
2820{
2821	hdr->opcode		= ISCSI_OP_SCSI_DATA_IN;
2822	hdr->flags		= datain->flags;
2823	if (hdr->flags & ISCSI_FLAG_DATA_STATUS) {
2824		if (cmd->se_cmd.se_cmd_flags & SCF_OVERFLOW_BIT) {
2825			hdr->flags |= ISCSI_FLAG_DATA_OVERFLOW;
2826			hdr->residual_count = cpu_to_be32(cmd->se_cmd.residual_count);
2827		} else if (cmd->se_cmd.se_cmd_flags & SCF_UNDERFLOW_BIT) {
2828			hdr->flags |= ISCSI_FLAG_DATA_UNDERFLOW;
2829			hdr->residual_count = cpu_to_be32(cmd->se_cmd.residual_count);
2830		}
2831	}
2832	hton24(hdr->dlength, datain->length);
2833	if (hdr->flags & ISCSI_FLAG_DATA_ACK)
2834		int_to_scsilun(cmd->se_cmd.orig_fe_lun,
2835				(struct scsi_lun *)&hdr->lun);
2836	else
2837		put_unaligned_le64(0xFFFFFFFFFFFFFFFFULL, &hdr->lun);
2838
2839	hdr->itt		= cmd->init_task_tag;
2840
2841	if (hdr->flags & ISCSI_FLAG_DATA_ACK)
2842		hdr->ttt		= cpu_to_be32(cmd->targ_xfer_tag);
2843	else
2844		hdr->ttt		= cpu_to_be32(0xFFFFFFFF);
2845	if (set_statsn)
2846		hdr->statsn		= cpu_to_be32(cmd->stat_sn);
2847	else
2848		hdr->statsn		= cpu_to_be32(0xFFFFFFFF);
2849
2850	hdr->exp_cmdsn		= cpu_to_be32(conn->sess->exp_cmd_sn);
2851	hdr->max_cmdsn		= cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
2852	hdr->datasn		= cpu_to_be32(datain->data_sn);
2853	hdr->offset		= cpu_to_be32(datain->offset);
2854
2855	pr_debug("Built DataIN ITT: 0x%08x, StatSN: 0x%08x,"
2856		" DataSN: 0x%08x, Offset: %u, Length: %u, CID: %hu\n",
2857		cmd->init_task_tag, ntohl(hdr->statsn), ntohl(hdr->datasn),
2858		ntohl(hdr->offset), datain->length, conn->cid);
2859}
2860EXPORT_SYMBOL(iscsit_build_datain_pdu);
2861
2862static int iscsit_send_datain(struct iscsit_cmd *cmd, struct iscsit_conn *conn)
2863{
2864	struct iscsi_data_rsp *hdr = (struct iscsi_data_rsp *)&cmd->pdu[0];
2865	struct iscsi_datain datain;
2866	struct iscsi_datain_req *dr;
2867	int eodr = 0, ret;
2868	bool set_statsn = false;
2869
2870	memset(&datain, 0, sizeof(struct iscsi_datain));
2871	dr = iscsit_get_datain_values(cmd, &datain);
2872	if (!dr) {
2873		pr_err("iscsit_get_datain_values failed for ITT: 0x%08x\n",
2874				cmd->init_task_tag);
2875		return -1;
2876	}
2877	/*
2878	 * Be paranoid and double check the logic for now.
2879	 */
2880	if ((datain.offset + datain.length) > cmd->se_cmd.data_length) {
2881		pr_err("Command ITT: 0x%08x, datain.offset: %u and"
2882			" datain.length: %u exceeds cmd->data_length: %u\n",
2883			cmd->init_task_tag, datain.offset, datain.length,
2884			cmd->se_cmd.data_length);
2885		return -1;
2886	}
2887
2888	atomic_long_add(datain.length, &conn->sess->tx_data_octets);
2889	/*
2890	 * Special case for successfully execution w/ both DATAIN
2891	 * and Sense Data.
2892	 */
2893	if ((datain.flags & ISCSI_FLAG_DATA_STATUS) &&
2894	    (cmd->se_cmd.se_cmd_flags & SCF_TRANSPORT_TASK_SENSE))
2895		datain.flags &= ~ISCSI_FLAG_DATA_STATUS;
2896	else {
2897		if ((dr->dr_complete == DATAIN_COMPLETE_NORMAL) ||
2898		    (dr->dr_complete == DATAIN_COMPLETE_CONNECTION_RECOVERY)) {
2899			iscsit_increment_maxcmdsn(cmd, conn->sess);
2900			cmd->stat_sn = conn->stat_sn++;
2901			set_statsn = true;
2902		} else if (dr->dr_complete ==
2903			   DATAIN_COMPLETE_WITHIN_COMMAND_RECOVERY)
2904			set_statsn = true;
2905	}
2906
2907	iscsit_build_datain_pdu(cmd, conn, &datain, hdr, set_statsn);
2908
2909	ret = conn->conn_transport->iscsit_xmit_pdu(conn, cmd, dr, &datain, 0);
2910	if (ret < 0)
2911		return ret;
2912
2913	if (dr->dr_complete) {
2914		eodr = (cmd->se_cmd.se_cmd_flags & SCF_TRANSPORT_TASK_SENSE) ?
2915				2 : 1;
2916		iscsit_free_datain_req(cmd, dr);
2917	}
2918
2919	return eodr;
2920}
2921
2922int
2923iscsit_build_logout_rsp(struct iscsit_cmd *cmd, struct iscsit_conn *conn,
2924			struct iscsi_logout_rsp *hdr)
2925{
2926	struct iscsit_conn *logout_conn = NULL;
2927	struct iscsi_conn_recovery *cr = NULL;
2928	struct iscsit_session *sess = conn->sess;
2929	/*
2930	 * The actual shutting down of Sessions and/or Connections
2931	 * for CLOSESESSION and CLOSECONNECTION Logout Requests
2932	 * is done in scsi_logout_post_handler().
2933	 */
2934	switch (cmd->logout_reason) {
2935	case ISCSI_LOGOUT_REASON_CLOSE_SESSION:
2936		pr_debug("iSCSI session logout successful, setting"
2937			" logout response to ISCSI_LOGOUT_SUCCESS.\n");
2938		cmd->logout_response = ISCSI_LOGOUT_SUCCESS;
2939		break;
2940	case ISCSI_LOGOUT_REASON_CLOSE_CONNECTION:
2941		if (cmd->logout_response == ISCSI_LOGOUT_CID_NOT_FOUND)
2942			break;
2943		/*
2944		 * For CLOSECONNECTION logout requests carrying
2945		 * a matching logout CID -> local CID, the reference
2946		 * for the local CID will have been incremented in
2947		 * iscsi_logout_closeconnection().
2948		 *
2949		 * For CLOSECONNECTION logout requests carrying
2950		 * a different CID than the connection it arrived
2951		 * on, the connection responding to cmd->logout_cid
2952		 * is stopped in iscsit_logout_post_handler_diffcid().
2953		 */
2954
2955		pr_debug("iSCSI CID: %hu logout on CID: %hu"
2956			" successful.\n", cmd->logout_cid, conn->cid);
2957		cmd->logout_response = ISCSI_LOGOUT_SUCCESS;
2958		break;
2959	case ISCSI_LOGOUT_REASON_RECOVERY:
2960		if ((cmd->logout_response == ISCSI_LOGOUT_RECOVERY_UNSUPPORTED) ||
2961		    (cmd->logout_response == ISCSI_LOGOUT_CLEANUP_FAILED))
2962			break;
2963		/*
2964		 * If the connection is still active from our point of view
2965		 * force connection recovery to occur.
2966		 */
2967		logout_conn = iscsit_get_conn_from_cid_rcfr(sess,
2968				cmd->logout_cid);
2969		if (logout_conn) {
2970			iscsit_connection_reinstatement_rcfr(logout_conn);
2971			iscsit_dec_conn_usage_count(logout_conn);
2972		}
2973
2974		cr = iscsit_get_inactive_connection_recovery_entry(
2975				conn->sess, cmd->logout_cid);
2976		if (!cr) {
2977			pr_err("Unable to locate CID: %hu for"
2978			" REMOVECONNFORRECOVERY Logout Request.\n",
2979				cmd->logout_cid);
2980			cmd->logout_response = ISCSI_LOGOUT_CID_NOT_FOUND;
2981			break;
2982		}
2983
2984		iscsit_discard_cr_cmds_by_expstatsn(cr, cmd->exp_stat_sn);
2985
2986		pr_debug("iSCSI REMOVECONNFORRECOVERY logout"
2987			" for recovery for CID: %hu on CID: %hu successful.\n",
2988				cmd->logout_cid, conn->cid);
2989		cmd->logout_response = ISCSI_LOGOUT_SUCCESS;
2990		break;
2991	default:
2992		pr_err("Unknown cmd->logout_reason: 0x%02x\n",
2993				cmd->logout_reason);
2994		return -1;
2995	}
2996
2997	hdr->opcode		= ISCSI_OP_LOGOUT_RSP;
2998	hdr->flags		|= ISCSI_FLAG_CMD_FINAL;
2999	hdr->response		= cmd->logout_response;
3000	hdr->itt		= cmd->init_task_tag;
3001	cmd->stat_sn		= conn->stat_sn++;
3002	hdr->statsn		= cpu_to_be32(cmd->stat_sn);
3003
3004	iscsit_increment_maxcmdsn(cmd, conn->sess);
3005	hdr->exp_cmdsn		= cpu_to_be32(conn->sess->exp_cmd_sn);
3006	hdr->max_cmdsn		= cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
3007
3008	pr_debug("Built Logout Response ITT: 0x%08x StatSN:"
3009		" 0x%08x Response: 0x%02x CID: %hu on CID: %hu\n",
3010		cmd->init_task_tag, cmd->stat_sn, hdr->response,
3011		cmd->logout_cid, conn->cid);
3012
3013	return 0;
3014}
3015EXPORT_SYMBOL(iscsit_build_logout_rsp);
3016
3017static int
3018iscsit_send_logout(struct iscsit_cmd *cmd, struct iscsit_conn *conn)
3019{
3020	int rc;
3021
3022	rc = iscsit_build_logout_rsp(cmd, conn,
3023			(struct iscsi_logout_rsp *)&cmd->pdu[0]);
3024	if (rc < 0)
3025		return rc;
3026
3027	return conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL, NULL, 0);
3028}
3029
3030void
3031iscsit_build_nopin_rsp(struct iscsit_cmd *cmd, struct iscsit_conn *conn,
3032		       struct iscsi_nopin *hdr, bool nopout_response)
3033{
3034	hdr->opcode		= ISCSI_OP_NOOP_IN;
3035	hdr->flags		|= ISCSI_FLAG_CMD_FINAL;
3036        hton24(hdr->dlength, cmd->buf_ptr_size);
3037	if (nopout_response)
3038		put_unaligned_le64(0xFFFFFFFFFFFFFFFFULL, &hdr->lun);
3039	hdr->itt		= cmd->init_task_tag;
3040	hdr->ttt		= cpu_to_be32(cmd->targ_xfer_tag);
3041	cmd->stat_sn		= (nopout_response) ? conn->stat_sn++ :
3042				  conn->stat_sn;
3043	hdr->statsn		= cpu_to_be32(cmd->stat_sn);
3044
3045	if (nopout_response)
3046		iscsit_increment_maxcmdsn(cmd, conn->sess);
3047
3048	hdr->exp_cmdsn		= cpu_to_be32(conn->sess->exp_cmd_sn);
3049	hdr->max_cmdsn		= cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
3050
3051	pr_debug("Built NOPIN %s Response ITT: 0x%08x, TTT: 0x%08x,"
3052		" StatSN: 0x%08x, Length %u\n", (nopout_response) ?
3053		"Solicited" : "Unsolicited", cmd->init_task_tag,
3054		cmd->targ_xfer_tag, cmd->stat_sn, cmd->buf_ptr_size);
3055}
3056EXPORT_SYMBOL(iscsit_build_nopin_rsp);
3057
3058/*
3059 *	Unsolicited NOPIN, either requesting a response or not.
3060 */
3061static int iscsit_send_unsolicited_nopin(
3062	struct iscsit_cmd *cmd,
3063	struct iscsit_conn *conn,
3064	int want_response)
3065{
3066	struct iscsi_nopin *hdr = (struct iscsi_nopin *)&cmd->pdu[0];
3067	int ret;
3068
3069	iscsit_build_nopin_rsp(cmd, conn, hdr, false);
3070
3071	pr_debug("Sending Unsolicited NOPIN TTT: 0x%08x StatSN:"
3072		" 0x%08x CID: %hu\n", hdr->ttt, cmd->stat_sn, conn->cid);
3073
3074	ret = conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL, NULL, 0);
3075	if (ret < 0)
3076		return ret;
3077
3078	spin_lock_bh(&cmd->istate_lock);
3079	cmd->i_state = want_response ?
3080		ISTATE_SENT_NOPIN_WANT_RESPONSE : ISTATE_SENT_STATUS;
3081	spin_unlock_bh(&cmd->istate_lock);
3082
3083	return 0;
3084}
3085
3086static int
3087iscsit_send_nopin(struct iscsit_cmd *cmd, struct iscsit_conn *conn)
3088{
3089	struct iscsi_nopin *hdr = (struct iscsi_nopin *)&cmd->pdu[0];
3090
3091	iscsit_build_nopin_rsp(cmd, conn, hdr, true);
3092
3093	/*
3094	 * NOPOUT Ping Data is attached to struct iscsit_cmd->buf_ptr.
3095	 * NOPOUT DataSegmentLength is at struct iscsit_cmd->buf_ptr_size.
3096	 */
3097	pr_debug("Echoing back %u bytes of ping data.\n", cmd->buf_ptr_size);
3098
3099	return conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL,
3100						     cmd->buf_ptr,
3101						     cmd->buf_ptr_size);
3102}
3103
3104static int iscsit_send_r2t(
3105	struct iscsit_cmd *cmd,
3106	struct iscsit_conn *conn)
3107{
3108	struct iscsi_r2t *r2t;
3109	struct iscsi_r2t_rsp *hdr;
3110	int ret;
3111
3112	r2t = iscsit_get_r2t_from_list(cmd);
3113	if (!r2t)
3114		return -1;
3115
3116	hdr			= (struct iscsi_r2t_rsp *) cmd->pdu;
3117	memset(hdr, 0, ISCSI_HDR_LEN);
3118	hdr->opcode		= ISCSI_OP_R2T;
3119	hdr->flags		|= ISCSI_FLAG_CMD_FINAL;
3120	int_to_scsilun(cmd->se_cmd.orig_fe_lun,
3121			(struct scsi_lun *)&hdr->lun);
3122	hdr->itt		= cmd->init_task_tag;
3123	if (conn->conn_transport->iscsit_get_r2t_ttt)
3124		conn->conn_transport->iscsit_get_r2t_ttt(conn, cmd, r2t);
3125	else
3126		r2t->targ_xfer_tag = session_get_next_ttt(conn->sess);
3127	hdr->ttt		= cpu_to_be32(r2t->targ_xfer_tag);
3128	hdr->statsn		= cpu_to_be32(conn->stat_sn);
3129	hdr->exp_cmdsn		= cpu_to_be32(conn->sess->exp_cmd_sn);
3130	hdr->max_cmdsn		= cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
3131	hdr->r2tsn		= cpu_to_be32(r2t->r2t_sn);
3132	hdr->data_offset	= cpu_to_be32(r2t->offset);
3133	hdr->data_length	= cpu_to_be32(r2t->xfer_len);
3134
3135	pr_debug("Built %sR2T, ITT: 0x%08x, TTT: 0x%08x, StatSN:"
3136		" 0x%08x, R2TSN: 0x%08x, Offset: %u, DDTL: %u, CID: %hu\n",
3137		(!r2t->recovery_r2t) ? "" : "Recovery ", cmd->init_task_tag,
3138		r2t->targ_xfer_tag, ntohl(hdr->statsn), r2t->r2t_sn,
3139			r2t->offset, r2t->xfer_len, conn->cid);
3140
3141	spin_lock_bh(&cmd->r2t_lock);
3142	r2t->sent_r2t = 1;
3143	spin_unlock_bh(&cmd->r2t_lock);
3144
3145	ret = conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL, NULL, 0);
3146	if (ret < 0) {
3147		return ret;
3148	}
3149
3150	spin_lock_bh(&cmd->dataout_timeout_lock);
3151	iscsit_start_dataout_timer(cmd, conn);
3152	spin_unlock_bh(&cmd->dataout_timeout_lock);
3153
3154	return 0;
3155}
3156
3157/*
3158 *	@recovery: If called from iscsi_task_reassign_complete_write() for
3159 *		connection recovery.
3160 */
3161int iscsit_build_r2ts_for_cmd(
3162	struct iscsit_conn *conn,
3163	struct iscsit_cmd *cmd,
3164	bool recovery)
3165{
3166	int first_r2t = 1;
3167	u32 offset = 0, xfer_len = 0;
3168
3169	spin_lock_bh(&cmd->r2t_lock);
3170	if (cmd->cmd_flags & ICF_SENT_LAST_R2T) {
3171		spin_unlock_bh(&cmd->r2t_lock);
3172		return 0;
3173	}
3174
3175	if (conn->sess->sess_ops->DataSequenceInOrder &&
3176	    !recovery)
3177		cmd->r2t_offset = max(cmd->r2t_offset, cmd->write_data_done);
3178
3179	while (cmd->outstanding_r2ts < conn->sess->sess_ops->MaxOutstandingR2T) {
3180		if (conn->sess->sess_ops->DataSequenceInOrder) {
3181			offset = cmd->r2t_offset;
3182
3183			if (first_r2t && recovery) {
3184				int new_data_end = offset +
3185					conn->sess->sess_ops->MaxBurstLength -
3186					cmd->next_burst_len;
3187
3188				if (new_data_end > cmd->se_cmd.data_length)
3189					xfer_len = cmd->se_cmd.data_length - offset;
3190				else
3191					xfer_len =
3192						conn->sess->sess_ops->MaxBurstLength -
3193						cmd->next_burst_len;
3194			} else {
3195				int new_data_end = offset +
3196					conn->sess->sess_ops->MaxBurstLength;
3197
3198				if (new_data_end > cmd->se_cmd.data_length)
3199					xfer_len = cmd->se_cmd.data_length - offset;
3200				else
3201					xfer_len = conn->sess->sess_ops->MaxBurstLength;
3202			}
3203
3204			if ((s32)xfer_len < 0) {
3205				cmd->cmd_flags |= ICF_SENT_LAST_R2T;
3206				break;
3207			}
3208
3209			cmd->r2t_offset += xfer_len;
3210
3211			if (cmd->r2t_offset == cmd->se_cmd.data_length)
3212				cmd->cmd_flags |= ICF_SENT_LAST_R2T;
3213		} else {
3214			struct iscsi_seq *seq;
3215
3216			seq = iscsit_get_seq_holder_for_r2t(cmd);
3217			if (!seq) {
3218				spin_unlock_bh(&cmd->r2t_lock);
3219				return -1;
3220			}
3221
3222			offset = seq->offset;
3223			xfer_len = seq->xfer_len;
3224
3225			if (cmd->seq_send_order == cmd->seq_count)
3226				cmd->cmd_flags |= ICF_SENT_LAST_R2T;
3227		}
3228		cmd->outstanding_r2ts++;
3229		first_r2t = 0;
3230
3231		if (iscsit_add_r2t_to_list(cmd, offset, xfer_len, 0, 0) < 0) {
3232			spin_unlock_bh(&cmd->r2t_lock);
3233			return -1;
3234		}
3235
3236		if (cmd->cmd_flags & ICF_SENT_LAST_R2T)
3237			break;
3238	}
3239	spin_unlock_bh(&cmd->r2t_lock);
3240
3241	return 0;
3242}
3243EXPORT_SYMBOL(iscsit_build_r2ts_for_cmd);
3244
3245void iscsit_build_rsp_pdu(struct iscsit_cmd *cmd, struct iscsit_conn *conn,
3246			bool inc_stat_sn, struct iscsi_scsi_rsp *hdr)
3247{
3248	if (inc_stat_sn)
3249		cmd->stat_sn = conn->stat_sn++;
3250
3251	atomic_long_inc(&conn->sess->rsp_pdus);
3252
3253	memset(hdr, 0, ISCSI_HDR_LEN);
3254	hdr->opcode		= ISCSI_OP_SCSI_CMD_RSP;
3255	hdr->flags		|= ISCSI_FLAG_CMD_FINAL;
3256	if (cmd->se_cmd.se_cmd_flags & SCF_OVERFLOW_BIT) {
3257		hdr->flags |= ISCSI_FLAG_CMD_OVERFLOW;
3258		hdr->residual_count = cpu_to_be32(cmd->se_cmd.residual_count);
3259	} else if (cmd->se_cmd.se_cmd_flags & SCF_UNDERFLOW_BIT) {
3260		hdr->flags |= ISCSI_FLAG_CMD_UNDERFLOW;
3261		hdr->residual_count = cpu_to_be32(cmd->se_cmd.residual_count);
3262	}
3263	hdr->response		= cmd->iscsi_response;
3264	hdr->cmd_status		= cmd->se_cmd.scsi_status;
3265	hdr->itt		= cmd->init_task_tag;
3266	hdr->statsn		= cpu_to_be32(cmd->stat_sn);
3267
3268	iscsit_increment_maxcmdsn(cmd, conn->sess);
3269	hdr->exp_cmdsn		= cpu_to_be32(conn->sess->exp_cmd_sn);
3270	hdr->max_cmdsn		= cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
3271
3272	pr_debug("Built SCSI Response, ITT: 0x%08x, StatSN: 0x%08x,"
3273		" Response: 0x%02x, SAM Status: 0x%02x, CID: %hu\n",
3274		cmd->init_task_tag, cmd->stat_sn, cmd->se_cmd.scsi_status,
3275		cmd->se_cmd.scsi_status, conn->cid);
3276}
3277EXPORT_SYMBOL(iscsit_build_rsp_pdu);
3278
3279static int iscsit_send_response(struct iscsit_cmd *cmd, struct iscsit_conn *conn)
3280{
3281	struct iscsi_scsi_rsp *hdr = (struct iscsi_scsi_rsp *)&cmd->pdu[0];
3282	bool inc_stat_sn = (cmd->i_state == ISTATE_SEND_STATUS);
3283	void *data_buf = NULL;
3284	u32 padding = 0, data_buf_len = 0;
3285
3286	iscsit_build_rsp_pdu(cmd, conn, inc_stat_sn, hdr);
3287
3288	/*
3289	 * Attach SENSE DATA payload to iSCSI Response PDU
3290	 */
3291	if (cmd->se_cmd.sense_buffer &&
3292	   ((cmd->se_cmd.se_cmd_flags & SCF_TRANSPORT_TASK_SENSE) ||
3293	    (cmd->se_cmd.se_cmd_flags & SCF_EMULATED_TASK_SENSE))) {
3294		put_unaligned_be16(cmd->se_cmd.scsi_sense_length, cmd->sense_buffer);
3295		cmd->se_cmd.scsi_sense_length += sizeof (__be16);
3296
3297		padding		= -(cmd->se_cmd.scsi_sense_length) & 3;
3298		hton24(hdr->dlength, (u32)cmd->se_cmd.scsi_sense_length);
3299		data_buf = cmd->sense_buffer;
3300		data_buf_len = cmd->se_cmd.scsi_sense_length + padding;
3301
3302		if (padding) {
3303			memset(cmd->sense_buffer +
3304				cmd->se_cmd.scsi_sense_length, 0, padding);
3305			pr_debug("Adding %u bytes of padding to"
3306				" SENSE.\n", padding);
3307		}
3308
3309		pr_debug("Attaching SENSE DATA: %u bytes to iSCSI"
3310				" Response PDU\n",
3311				cmd->se_cmd.scsi_sense_length);
3312	}
3313
3314	return conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL, data_buf,
3315						     data_buf_len);
3316}
3317
3318static u8 iscsit_convert_tcm_tmr_rsp(struct se_tmr_req *se_tmr)
3319{
3320	switch (se_tmr->response) {
3321	case TMR_FUNCTION_COMPLETE:
3322		return ISCSI_TMF_RSP_COMPLETE;
3323	case TMR_TASK_DOES_NOT_EXIST:
3324		return ISCSI_TMF_RSP_NO_TASK;
3325	case TMR_LUN_DOES_NOT_EXIST:
3326		return ISCSI_TMF_RSP_NO_LUN;
3327	case TMR_TASK_MGMT_FUNCTION_NOT_SUPPORTED:
3328		return ISCSI_TMF_RSP_NOT_SUPPORTED;
3329	case TMR_FUNCTION_REJECTED:
3330	default:
3331		return ISCSI_TMF_RSP_REJECTED;
3332	}
3333}
3334
3335void
3336iscsit_build_task_mgt_rsp(struct iscsit_cmd *cmd, struct iscsit_conn *conn,
3337			  struct iscsi_tm_rsp *hdr)
3338{
3339	struct se_tmr_req *se_tmr = cmd->se_cmd.se_tmr_req;
3340
3341	hdr->opcode		= ISCSI_OP_SCSI_TMFUNC_RSP;
3342	hdr->flags		= ISCSI_FLAG_CMD_FINAL;
3343	hdr->response		= iscsit_convert_tcm_tmr_rsp(se_tmr);
3344	hdr->itt		= cmd->init_task_tag;
3345	cmd->stat_sn		= conn->stat_sn++;
3346	hdr->statsn		= cpu_to_be32(cmd->stat_sn);
3347
3348	iscsit_increment_maxcmdsn(cmd, conn->sess);
3349	hdr->exp_cmdsn		= cpu_to_be32(conn->sess->exp_cmd_sn);
3350	hdr->max_cmdsn		= cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
3351
3352	pr_debug("Built Task Management Response ITT: 0x%08x,"
3353		" StatSN: 0x%08x, Response: 0x%02x, CID: %hu\n",
3354		cmd->init_task_tag, cmd->stat_sn, hdr->response, conn->cid);
3355}
3356EXPORT_SYMBOL(iscsit_build_task_mgt_rsp);
3357
3358static int
3359iscsit_send_task_mgt_rsp(struct iscsit_cmd *cmd, struct iscsit_conn *conn)
3360{
3361	struct iscsi_tm_rsp *hdr = (struct iscsi_tm_rsp *)&cmd->pdu[0];
3362
3363	iscsit_build_task_mgt_rsp(cmd, conn, hdr);
3364
3365	return conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL, NULL, 0);
3366}
3367
3368#define SENDTARGETS_BUF_LIMIT 32768U
3369
3370static int
3371iscsit_build_sendtargets_response(struct iscsit_cmd *cmd,
3372				  enum iscsit_transport_type network_transport,
3373				  int skip_bytes, bool *completed)
3374{
3375	char *payload = NULL;
3376	struct iscsit_conn *conn = cmd->conn;
3377	struct iscsi_portal_group *tpg;
3378	struct iscsi_tiqn *tiqn;
3379	struct iscsi_tpg_np *tpg_np;
3380	int buffer_len, end_of_buf = 0, len = 0, payload_len = 0;
3381	int target_name_printed;
3382	unsigned char buf[ISCSI_IQN_LEN+12]; /* iqn + "TargetName=" + \0 */
3383	unsigned char *text_in = cmd->text_in_ptr, *text_ptr = NULL;
3384	bool active;
3385
3386	buffer_len = min(conn->conn_ops->MaxRecvDataSegmentLength,
3387			 SENDTARGETS_BUF_LIMIT);
3388
3389	payload = kzalloc(buffer_len, GFP_KERNEL);
3390	if (!payload)
3391		return -ENOMEM;
3392
3393	/*
3394	 * Locate pointer to iqn./eui. string for ICF_SENDTARGETS_SINGLE
3395	 * explicit case..
3396	 */
3397	if (cmd->cmd_flags & ICF_SENDTARGETS_SINGLE) {
3398		text_ptr = strchr(text_in, '=');
3399		if (!text_ptr) {
3400			pr_err("Unable to locate '=' string in text_in:"
3401			       " %s\n", text_in);
3402			kfree(payload);
3403			return -EINVAL;
3404		}
3405		/*
3406		 * Skip over '=' character..
3407		 */
3408		text_ptr += 1;
3409	}
3410
3411	spin_lock(&tiqn_lock);
3412	list_for_each_entry(tiqn, &g_tiqn_list, tiqn_list) {
3413		if ((cmd->cmd_flags & ICF_SENDTARGETS_SINGLE) &&
3414		     strcmp(tiqn->tiqn, text_ptr)) {
3415			continue;
3416		}
3417
3418		target_name_printed = 0;
3419
3420		spin_lock(&tiqn->tiqn_tpg_lock);
3421		list_for_each_entry(tpg, &tiqn->tiqn_tpg_list, tpg_list) {
3422
3423			/* If demo_mode_discovery=0 and generate_node_acls=0
3424			 * (demo mode dislabed) do not return
3425			 * TargetName+TargetAddress unless a NodeACL exists.
3426			 */
3427
3428			if ((tpg->tpg_attrib.generate_node_acls == 0) &&
3429			    (tpg->tpg_attrib.demo_mode_discovery == 0) &&
3430			    (!target_tpg_has_node_acl(&tpg->tpg_se_tpg,
3431				cmd->conn->sess->sess_ops->InitiatorName))) {
3432				continue;
3433			}
3434
3435			spin_lock(&tpg->tpg_state_lock);
3436			active = (tpg->tpg_state == TPG_STATE_ACTIVE);
3437			spin_unlock(&tpg->tpg_state_lock);
3438
3439			if (!active && tpg->tpg_attrib.tpg_enabled_sendtargets)
3440				continue;
3441
3442			spin_lock(&tpg->tpg_np_lock);
3443			list_for_each_entry(tpg_np, &tpg->tpg_gnp_list,
3444						tpg_np_list) {
3445				struct iscsi_np *np = tpg_np->tpg_np;
3446				struct sockaddr_storage *sockaddr;
3447
3448				if (np->np_network_transport != network_transport)
3449					continue;
3450
3451				if (!target_name_printed) {
3452					len = sprintf(buf, "TargetName=%s",
3453						      tiqn->tiqn);
3454					len += 1;
3455
3456					if ((len + payload_len) > buffer_len) {
3457						spin_unlock(&tpg->tpg_np_lock);
3458						spin_unlock(&tiqn->tiqn_tpg_lock);
3459						end_of_buf = 1;
3460						goto eob;
3461					}
3462
3463					if (skip_bytes && len <= skip_bytes) {
3464						skip_bytes -= len;
3465					} else {
3466						memcpy(payload + payload_len, buf, len);
3467						payload_len += len;
3468						target_name_printed = 1;
3469						if (len > skip_bytes)
3470							skip_bytes = 0;
3471					}
3472				}
3473
3474				if (inet_addr_is_any((struct sockaddr *)&np->np_sockaddr))
3475					sockaddr = &conn->local_sockaddr;
3476				else
3477					sockaddr = &np->np_sockaddr;
3478
3479				len = sprintf(buf, "TargetAddress="
3480					      "%pISpc,%hu",
3481					      sockaddr,
3482					      tpg->tpgt);
3483				len += 1;
3484
3485				if ((len + payload_len) > buffer_len) {
3486					spin_unlock(&tpg->tpg_np_lock);
3487					spin_unlock(&tiqn->tiqn_tpg_lock);
3488					end_of_buf = 1;
3489					goto eob;
3490				}
3491
3492				if (skip_bytes && len <= skip_bytes) {
3493					skip_bytes -= len;
3494				} else {
3495					memcpy(payload + payload_len, buf, len);
3496					payload_len += len;
3497					if (len > skip_bytes)
3498						skip_bytes = 0;
3499				}
3500			}
3501			spin_unlock(&tpg->tpg_np_lock);
3502		}
3503		spin_unlock(&tiqn->tiqn_tpg_lock);
3504eob:
3505		if (end_of_buf) {
3506			*completed = false;
3507			break;
3508		}
3509
3510		if (cmd->cmd_flags & ICF_SENDTARGETS_SINGLE)
3511			break;
3512	}
3513	spin_unlock(&tiqn_lock);
3514
3515	cmd->buf_ptr = payload;
3516
3517	return payload_len;
3518}
3519
3520int
3521iscsit_build_text_rsp(struct iscsit_cmd *cmd, struct iscsit_conn *conn,
3522		      struct iscsi_text_rsp *hdr,
3523		      enum iscsit_transport_type network_transport)
3524{
3525	int text_length, padding;
3526	bool completed = true;
3527
3528	text_length = iscsit_build_sendtargets_response(cmd, network_transport,
3529							cmd->read_data_done,
3530							&completed);
3531	if (text_length < 0)
3532		return text_length;
3533
3534	if (completed) {
3535		hdr->flags = ISCSI_FLAG_CMD_FINAL;
3536	} else {
3537		hdr->flags = ISCSI_FLAG_TEXT_CONTINUE;
3538		cmd->read_data_done += text_length;
3539		if (cmd->targ_xfer_tag == 0xFFFFFFFF)
3540			cmd->targ_xfer_tag = session_get_next_ttt(conn->sess);
3541	}
3542	hdr->opcode = ISCSI_OP_TEXT_RSP;
3543	padding = ((-text_length) & 3);
3544	hton24(hdr->dlength, text_length);
3545	hdr->itt = cmd->init_task_tag;
3546	hdr->ttt = cpu_to_be32(cmd->targ_xfer_tag);
3547	cmd->stat_sn = conn->stat_sn++;
3548	hdr->statsn = cpu_to_be32(cmd->stat_sn);
3549
3550	iscsit_increment_maxcmdsn(cmd, conn->sess);
3551	/*
3552	 * Reset maxcmdsn_inc in multi-part text payload exchanges to
3553	 * correctly increment MaxCmdSN for each response answering a
3554	 * non immediate text request with a valid CmdSN.
3555	 */
3556	cmd->maxcmdsn_inc = 0;
3557	hdr->exp_cmdsn = cpu_to_be32(conn->sess->exp_cmd_sn);
3558	hdr->max_cmdsn = cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
3559
3560	pr_debug("Built Text Response: ITT: 0x%08x, TTT: 0x%08x, StatSN: 0x%08x,"
3561		" Length: %u, CID: %hu F: %d C: %d\n", cmd->init_task_tag,
3562		cmd->targ_xfer_tag, cmd->stat_sn, text_length, conn->cid,
3563		!!(hdr->flags & ISCSI_FLAG_CMD_FINAL),
3564		!!(hdr->flags & ISCSI_FLAG_TEXT_CONTINUE));
3565
3566	return text_length + padding;
3567}
3568EXPORT_SYMBOL(iscsit_build_text_rsp);
3569
3570static int iscsit_send_text_rsp(
3571	struct iscsit_cmd *cmd,
3572	struct iscsit_conn *conn)
3573{
3574	struct iscsi_text_rsp *hdr = (struct iscsi_text_rsp *)cmd->pdu;
3575	int text_length;
3576
3577	text_length = iscsit_build_text_rsp(cmd, conn, hdr,
3578				conn->conn_transport->transport_type);
3579	if (text_length < 0)
3580		return text_length;
3581
3582	return conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL,
3583						     cmd->buf_ptr,
3584						     text_length);
3585}
3586
3587void
3588iscsit_build_reject(struct iscsit_cmd *cmd, struct iscsit_conn *conn,
3589		    struct iscsi_reject *hdr)
3590{
3591	hdr->opcode		= ISCSI_OP_REJECT;
3592	hdr->reason		= cmd->reject_reason;
3593	hdr->flags		|= ISCSI_FLAG_CMD_FINAL;
3594	hton24(hdr->dlength, ISCSI_HDR_LEN);
3595	hdr->ffffffff		= cpu_to_be32(0xffffffff);
3596	cmd->stat_sn		= conn->stat_sn++;
3597	hdr->statsn		= cpu_to_be32(cmd->stat_sn);
3598	hdr->exp_cmdsn		= cpu_to_be32(conn->sess->exp_cmd_sn);
3599	hdr->max_cmdsn		= cpu_to_be32((u32) atomic_read(&conn->sess->max_cmd_sn));
3600
3601}
3602EXPORT_SYMBOL(iscsit_build_reject);
3603
3604static int iscsit_send_reject(
3605	struct iscsit_cmd *cmd,
3606	struct iscsit_conn *conn)
3607{
3608	struct iscsi_reject *hdr = (struct iscsi_reject *)&cmd->pdu[0];
3609
3610	iscsit_build_reject(cmd, conn, hdr);
3611
3612	pr_debug("Built Reject PDU StatSN: 0x%08x, Reason: 0x%02x,"
3613		" CID: %hu\n", ntohl(hdr->statsn), hdr->reason, conn->cid);
3614
3615	return conn->conn_transport->iscsit_xmit_pdu(conn, cmd, NULL,
3616						     cmd->buf_ptr,
3617						     ISCSI_HDR_LEN);
3618}
3619
3620void iscsit_thread_get_cpumask(struct iscsit_conn *conn)
3621{
3622	int ord, cpu;
3623	cpumask_var_t conn_allowed_cpumask;
3624
3625	/*
3626	 * bitmap_id is assigned from iscsit_global->ts_bitmap from
3627	 * within iscsit_start_kthreads()
3628	 *
3629	 * Here we use bitmap_id to determine which CPU that this
3630	 * iSCSI connection's RX/TX threads will be scheduled to
3631	 * execute upon.
3632	 */
3633	if (!zalloc_cpumask_var(&conn_allowed_cpumask, GFP_KERNEL)) {
3634		ord = conn->bitmap_id % cpumask_weight(cpu_online_mask);
3635		for_each_online_cpu(cpu) {
3636			if (ord-- == 0) {
3637				cpumask_set_cpu(cpu, conn->conn_cpumask);
3638				return;
3639			}
3640		}
3641	} else {
3642		cpumask_and(conn_allowed_cpumask, iscsit_global->allowed_cpumask,
3643			cpu_online_mask);
3644
3645		cpumask_clear(conn->conn_cpumask);
3646		ord = conn->bitmap_id % cpumask_weight(conn_allowed_cpumask);
3647		for_each_cpu(cpu, conn_allowed_cpumask) {
3648			if (ord-- == 0) {
3649				cpumask_set_cpu(cpu, conn->conn_cpumask);
3650				free_cpumask_var(conn_allowed_cpumask);
3651				return;
3652			}
3653		}
3654		free_cpumask_var(conn_allowed_cpumask);
3655	}
3656	/*
3657	 * This should never be reached..
3658	 */
3659	dump_stack();
3660	cpumask_setall(conn->conn_cpumask);
3661}
3662
3663static void iscsit_thread_reschedule(struct iscsit_conn *conn)
3664{
3665	/*
3666	 * If iscsit_global->allowed_cpumask modified, reschedule iSCSI
3667	 * connection's RX/TX threads update conn->allowed_cpumask.
3668	 */
3669	if (!cpumask_equal(iscsit_global->allowed_cpumask,
3670			   conn->allowed_cpumask)) {
3671		iscsit_thread_get_cpumask(conn);
3672		conn->conn_tx_reset_cpumask = 1;
3673		conn->conn_rx_reset_cpumask = 1;
3674		cpumask_copy(conn->allowed_cpumask,
3675			     iscsit_global->allowed_cpumask);
3676	}
3677}
3678
3679void iscsit_thread_check_cpumask(
3680	struct iscsit_conn *conn,
3681	struct task_struct *p,
3682	int mode)
3683{
3684	/*
3685	 * The TX and RX threads maybe call iscsit_thread_check_cpumask()
3686	 * at the same time. The RX thread might be faster and return from
3687	 * iscsit_thread_reschedule() with conn_rx_reset_cpumask set to 0.
3688	 * Then the TX thread sets it back to 1.
3689	 * The next time the RX thread loops, it sees conn_rx_reset_cpumask
3690	 * set to 1 and calls set_cpus_allowed_ptr() again and set it to 0.
3691	 */
3692	iscsit_thread_reschedule(conn);
3693
3694	/*
3695	 * mode == 1 signals iscsi_target_tx_thread() usage.
3696	 * mode == 0 signals iscsi_target_rx_thread() usage.
3697	 */
3698	if (mode == 1) {
3699		if (!conn->conn_tx_reset_cpumask)
3700			return;
3701	} else {
3702		if (!conn->conn_rx_reset_cpumask)
3703			return;
3704	}
3705
3706	/*
3707	 * Update the CPU mask for this single kthread so that
3708	 * both TX and RX kthreads are scheduled to run on the
3709	 * same CPU.
3710	 */
3711	set_cpus_allowed_ptr(p, conn->conn_cpumask);
3712	if (mode == 1)
3713		conn->conn_tx_reset_cpumask = 0;
3714	else
3715		conn->conn_rx_reset_cpumask = 0;
3716}
3717EXPORT_SYMBOL(iscsit_thread_check_cpumask);
3718
3719int
3720iscsit_immediate_queue(struct iscsit_conn *conn, struct iscsit_cmd *cmd, int state)
3721{
3722	int ret;
3723
3724	switch (state) {
3725	case ISTATE_SEND_R2T:
3726		ret = iscsit_send_r2t(cmd, conn);
3727		if (ret < 0)
3728			goto err;
3729		break;
3730	case ISTATE_REMOVE:
3731		spin_lock_bh(&conn->cmd_lock);
3732		list_del_init(&cmd->i_conn_node);
3733		spin_unlock_bh(&conn->cmd_lock);
3734
3735		iscsit_free_cmd(cmd, false);
3736		break;
3737	case ISTATE_SEND_NOPIN_WANT_RESPONSE:
3738		iscsit_mod_nopin_response_timer(conn);
3739		ret = iscsit_send_unsolicited_nopin(cmd, conn, 1);
3740		if (ret < 0)
3741			goto err;
3742		break;
3743	case ISTATE_SEND_NOPIN_NO_RESPONSE:
3744		ret = iscsit_send_unsolicited_nopin(cmd, conn, 0);
3745		if (ret < 0)
3746			goto err;
3747		break;
3748	default:
3749		pr_err("Unknown Opcode: 0x%02x ITT:"
3750		       " 0x%08x, i_state: %d on CID: %hu\n",
3751		       cmd->iscsi_opcode, cmd->init_task_tag, state,
3752		       conn->cid);
3753		goto err;
3754	}
3755
3756	return 0;
3757
3758err:
3759	return -1;
3760}
3761EXPORT_SYMBOL(iscsit_immediate_queue);
3762
3763static int
3764iscsit_handle_immediate_queue(struct iscsit_conn *conn)
3765{
3766	struct iscsit_transport *t = conn->conn_transport;
3767	struct iscsi_queue_req *qr;
3768	struct iscsit_cmd *cmd;
3769	u8 state;
3770	int ret;
3771
3772	while ((qr = iscsit_get_cmd_from_immediate_queue(conn))) {
3773		atomic_set(&conn->check_immediate_queue, 0);
3774		cmd = qr->cmd;
3775		state = qr->state;
3776		kmem_cache_free(lio_qr_cache, qr);
3777
3778		ret = t->iscsit_immediate_queue(conn, cmd, state);
3779		if (ret < 0)
3780			return ret;
3781	}
3782
3783	return 0;
3784}
3785
3786int
3787iscsit_response_queue(struct iscsit_conn *conn, struct iscsit_cmd *cmd, int state)
3788{
3789	int ret;
3790
3791check_rsp_state:
3792	switch (state) {
3793	case ISTATE_SEND_DATAIN:
3794		ret = iscsit_send_datain(cmd, conn);
3795		if (ret < 0)
3796			goto err;
3797		else if (!ret)
3798			/* more drs */
3799			goto check_rsp_state;
3800		else if (ret == 1) {
3801			/* all done */
3802			spin_lock_bh(&cmd->istate_lock);
3803			cmd->i_state = ISTATE_SENT_STATUS;
3804			spin_unlock_bh(&cmd->istate_lock);
3805
3806			if (atomic_read(&conn->check_immediate_queue))
3807				return 1;
3808
3809			return 0;
3810		} else if (ret == 2) {
3811			/* Still must send status,
3812			   SCF_TRANSPORT_TASK_SENSE was set */
3813			spin_lock_bh(&cmd->istate_lock);
3814			cmd->i_state = ISTATE_SEND_STATUS;
3815			spin_unlock_bh(&cmd->istate_lock);
3816			state = ISTATE_SEND_STATUS;
3817			goto check_rsp_state;
3818		}
3819
3820		break;
3821	case ISTATE_SEND_STATUS:
3822	case ISTATE_SEND_STATUS_RECOVERY:
3823		ret = iscsit_send_response(cmd, conn);
3824		break;
3825	case ISTATE_SEND_LOGOUTRSP:
3826		ret = iscsit_send_logout(cmd, conn);
3827		break;
3828	case ISTATE_SEND_ASYNCMSG:
3829		ret = iscsit_send_conn_drop_async_message(
3830			cmd, conn);
3831		break;
3832	case ISTATE_SEND_NOPIN:
3833		ret = iscsit_send_nopin(cmd, conn);
3834		break;
3835	case ISTATE_SEND_REJECT:
3836		ret = iscsit_send_reject(cmd, conn);
3837		break;
3838	case ISTATE_SEND_TASKMGTRSP:
3839		ret = iscsit_send_task_mgt_rsp(cmd, conn);
3840		if (ret != 0)
3841			break;
3842		ret = iscsit_tmr_post_handler(cmd, conn);
3843		if (ret != 0)
3844			iscsit_fall_back_to_erl0(conn->sess);
3845		break;
3846	case ISTATE_SEND_TEXTRSP:
3847		ret = iscsit_send_text_rsp(cmd, conn);
3848		break;
3849	default:
3850		pr_err("Unknown Opcode: 0x%02x ITT:"
3851		       " 0x%08x, i_state: %d on CID: %hu\n",
3852		       cmd->iscsi_opcode, cmd->init_task_tag,
3853		       state, conn->cid);
3854		goto err;
3855	}
3856	if (ret < 0)
3857		goto err;
3858
3859	switch (state) {
3860	case ISTATE_SEND_LOGOUTRSP:
3861		if (!iscsit_logout_post_handler(cmd, conn))
3862			return -ECONNRESET;
3863		fallthrough;
3864	case ISTATE_SEND_STATUS:
3865	case ISTATE_SEND_ASYNCMSG:
3866	case ISTATE_SEND_NOPIN:
3867	case ISTATE_SEND_STATUS_RECOVERY:
3868	case ISTATE_SEND_TEXTRSP:
3869	case ISTATE_SEND_TASKMGTRSP:
3870	case ISTATE_SEND_REJECT:
3871		spin_lock_bh(&cmd->istate_lock);
3872		cmd->i_state = ISTATE_SENT_STATUS;
3873		spin_unlock_bh(&cmd->istate_lock);
3874		break;
3875	default:
3876		pr_err("Unknown Opcode: 0x%02x ITT:"
3877		       " 0x%08x, i_state: %d on CID: %hu\n",
3878		       cmd->iscsi_opcode, cmd->init_task_tag,
3879		       cmd->i_state, conn->cid);
3880		goto err;
3881	}
3882
3883	if (atomic_read(&conn->check_immediate_queue))
3884		return 1;
3885
3886	return 0;
3887
3888err:
3889	return -1;
3890}
3891EXPORT_SYMBOL(iscsit_response_queue);
3892
3893static int iscsit_handle_response_queue(struct iscsit_conn *conn)
3894{
3895	struct iscsit_transport *t = conn->conn_transport;
3896	struct iscsi_queue_req *qr;
3897	struct iscsit_cmd *cmd;
3898	u8 state;
3899	int ret;
3900
3901	while ((qr = iscsit_get_cmd_from_response_queue(conn))) {
3902		cmd = qr->cmd;
3903		state = qr->state;
3904		kmem_cache_free(lio_qr_cache, qr);
3905
3906		ret = t->iscsit_response_queue(conn, cmd, state);
3907		if (ret == 1 || ret < 0)
3908			return ret;
3909	}
3910
3911	return 0;
3912}
3913
3914int iscsi_target_tx_thread(void *arg)
3915{
3916	int ret = 0;
3917	struct iscsit_conn *conn = arg;
3918	bool conn_freed = false;
3919
3920	/*
3921	 * Allow ourselves to be interrupted by SIGINT so that a
3922	 * connection recovery / failure event can be triggered externally.
3923	 */
3924	allow_signal(SIGINT);
3925
3926	while (!kthread_should_stop()) {
3927		/*
3928		 * Ensure that both TX and RX per connection kthreads
3929		 * are scheduled to run on the same CPU.
3930		 */
3931		iscsit_thread_check_cpumask(conn, current, 1);
3932
3933		wait_event_interruptible(conn->queues_wq,
3934					 !iscsit_conn_all_queues_empty(conn));
3935
3936		if (signal_pending(current))
3937			goto transport_err;
3938
3939get_immediate:
3940		ret = iscsit_handle_immediate_queue(conn);
3941		if (ret < 0)
3942			goto transport_err;
3943
3944		ret = iscsit_handle_response_queue(conn);
3945		if (ret == 1) {
3946			goto get_immediate;
3947		} else if (ret == -ECONNRESET) {
3948			conn_freed = true;
3949			goto out;
3950		} else if (ret < 0) {
3951			goto transport_err;
3952		}
3953	}
3954
3955transport_err:
3956	/*
3957	 * Avoid the normal connection failure code-path if this connection
3958	 * is still within LOGIN mode, and iscsi_np process context is
3959	 * responsible for cleaning up the early connection failure.
3960	 */
3961	if (conn->conn_state != TARG_CONN_STATE_IN_LOGIN)
3962		iscsit_take_action_for_connection_exit(conn, &conn_freed);
3963out:
3964	if (!conn_freed) {
3965		while (!kthread_should_stop()) {
3966			msleep(100);
3967		}
3968	}
3969	return 0;
3970}
3971
3972static int iscsi_target_rx_opcode(struct iscsit_conn *conn, unsigned char *buf)
3973{
3974	struct iscsi_hdr *hdr = (struct iscsi_hdr *)buf;
3975	struct iscsit_cmd *cmd;
3976	int ret = 0;
3977
3978	switch (hdr->opcode & ISCSI_OPCODE_MASK) {
3979	case ISCSI_OP_SCSI_CMD:
3980		cmd = iscsit_allocate_cmd(conn, TASK_INTERRUPTIBLE);
3981		if (!cmd)
3982			goto reject;
3983
3984		ret = iscsit_handle_scsi_cmd(conn, cmd, buf);
3985		break;
3986	case ISCSI_OP_SCSI_DATA_OUT:
3987		ret = iscsit_handle_data_out(conn, buf);
3988		break;
3989	case ISCSI_OP_NOOP_OUT:
3990		cmd = NULL;
3991		if (hdr->ttt == cpu_to_be32(0xFFFFFFFF)) {
3992			cmd = iscsit_allocate_cmd(conn, TASK_INTERRUPTIBLE);
3993			if (!cmd)
3994				goto reject;
3995		}
3996		ret = iscsit_handle_nop_out(conn, cmd, buf);
3997		break;
3998	case ISCSI_OP_SCSI_TMFUNC:
3999		cmd = iscsit_allocate_cmd(conn, TASK_INTERRUPTIBLE);
4000		if (!cmd)
4001			goto reject;
4002
4003		ret = iscsit_handle_task_mgt_cmd(conn, cmd, buf);
4004		break;
4005	case ISCSI_OP_TEXT:
4006		if (hdr->ttt != cpu_to_be32(0xFFFFFFFF)) {
4007			cmd = iscsit_find_cmd_from_itt(conn, hdr->itt);
4008			if (!cmd)
4009				goto reject;
4010		} else {
4011			cmd = iscsit_allocate_cmd(conn, TASK_INTERRUPTIBLE);
4012			if (!cmd)
4013				goto reject;
4014		}
4015
4016		ret = iscsit_handle_text_cmd(conn, cmd, buf);
4017		break;
4018	case ISCSI_OP_LOGOUT:
4019		cmd = iscsit_allocate_cmd(conn, TASK_INTERRUPTIBLE);
4020		if (!cmd)
4021			goto reject;
4022
4023		ret = iscsit_handle_logout_cmd(conn, cmd, buf);
4024		if (ret > 0)
4025			wait_for_completion_timeout(&conn->conn_logout_comp,
4026					SECONDS_FOR_LOGOUT_COMP * HZ);
4027		break;
4028	case ISCSI_OP_SNACK:
4029		ret = iscsit_handle_snack(conn, buf);
4030		break;
4031	default:
4032		pr_err("Got unknown iSCSI OpCode: 0x%02x\n", hdr->opcode);
4033		if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
4034			pr_err("Cannot recover from unknown"
4035			" opcode while ERL=0, closing iSCSI connection.\n");
4036			return -1;
4037		}
4038		pr_err("Unable to recover from unknown opcode while OFMarker=No,"
4039		       " closing iSCSI connection.\n");
4040		ret = -1;
4041		break;
4042	}
4043
4044	return ret;
4045reject:
4046	return iscsit_add_reject(conn, ISCSI_REASON_BOOKMARK_NO_RESOURCES, buf);
4047}
4048
4049static bool iscsi_target_check_conn_state(struct iscsit_conn *conn)
4050{
4051	bool ret;
4052
4053	spin_lock_bh(&conn->state_lock);
4054	ret = (conn->conn_state != TARG_CONN_STATE_LOGGED_IN);
4055	spin_unlock_bh(&conn->state_lock);
4056
4057	return ret;
4058}
4059
4060static void iscsit_get_rx_pdu(struct iscsit_conn *conn)
4061{
4062	int ret;
4063	u8 *buffer, *tmp_buf, opcode;
4064	u32 checksum = 0, digest = 0;
4065	struct iscsi_hdr *hdr;
4066	struct kvec iov;
4067
4068	buffer = kcalloc(ISCSI_HDR_LEN, sizeof(*buffer), GFP_KERNEL);
4069	if (!buffer)
4070		return;
4071
4072	while (!kthread_should_stop()) {
4073		/*
4074		 * Ensure that both TX and RX per connection kthreads
4075		 * are scheduled to run on the same CPU.
4076		 */
4077		iscsit_thread_check_cpumask(conn, current, 0);
4078
4079		memset(&iov, 0, sizeof(struct kvec));
4080
4081		iov.iov_base	= buffer;
4082		iov.iov_len	= ISCSI_HDR_LEN;
4083
4084		ret = rx_data(conn, &iov, 1, ISCSI_HDR_LEN);
4085		if (ret != ISCSI_HDR_LEN) {
4086			iscsit_rx_thread_wait_for_tcp(conn);
4087			break;
4088		}
4089
4090		hdr = (struct iscsi_hdr *) buffer;
4091		if (hdr->hlength) {
4092			iov.iov_len = hdr->hlength * 4;
4093			tmp_buf = krealloc(buffer,
4094					  ISCSI_HDR_LEN + iov.iov_len,
4095					  GFP_KERNEL);
4096			if (!tmp_buf)
4097				break;
4098
4099			buffer = tmp_buf;
4100			iov.iov_base = &buffer[ISCSI_HDR_LEN];
4101
4102			ret = rx_data(conn, &iov, 1, iov.iov_len);
4103			if (ret != iov.iov_len) {
4104				iscsit_rx_thread_wait_for_tcp(conn);
4105				break;
4106			}
4107		}
4108
4109		if (conn->conn_ops->HeaderDigest) {
4110			iov.iov_base	= &digest;
4111			iov.iov_len	= ISCSI_CRC_LEN;
4112
4113			ret = rx_data(conn, &iov, 1, ISCSI_CRC_LEN);
4114			if (ret != ISCSI_CRC_LEN) {
4115				iscsit_rx_thread_wait_for_tcp(conn);
4116				break;
4117			}
4118
4119			iscsit_do_crypto_hash_buf(conn->conn_rx_hash, buffer,
4120						  ISCSI_HDR_LEN, 0, NULL,
4121						  &checksum);
4122
4123			if (digest != checksum) {
4124				pr_err("HeaderDigest CRC32C failed,"
4125					" received 0x%08x, computed 0x%08x\n",
4126					digest, checksum);
4127				/*
4128				 * Set the PDU to 0xff so it will intentionally
4129				 * hit default in the switch below.
4130				 */
4131				memset(buffer, 0xff, ISCSI_HDR_LEN);
4132				atomic_long_inc(&conn->sess->conn_digest_errors);
4133			} else {
4134				pr_debug("Got HeaderDigest CRC32C"
4135						" 0x%08x\n", checksum);
4136			}
4137		}
4138
4139		if (conn->conn_state == TARG_CONN_STATE_IN_LOGOUT)
4140			break;
4141
4142		opcode = buffer[0] & ISCSI_OPCODE_MASK;
4143
4144		if (conn->sess->sess_ops->SessionType &&
4145		   ((!(opcode & ISCSI_OP_TEXT)) ||
4146		    (!(opcode & ISCSI_OP_LOGOUT)))) {
4147			pr_err("Received illegal iSCSI Opcode: 0x%02x"
4148			" while in Discovery Session, rejecting.\n", opcode);
4149			iscsit_add_reject(conn, ISCSI_REASON_PROTOCOL_ERROR,
4150					  buffer);
4151			break;
4152		}
4153
4154		ret = iscsi_target_rx_opcode(conn, buffer);
4155		if (ret < 0)
4156			break;
4157	}
4158
4159	kfree(buffer);
4160}
4161
4162int iscsi_target_rx_thread(void *arg)
4163{
4164	int rc;
4165	struct iscsit_conn *conn = arg;
4166	bool conn_freed = false;
4167
4168	/*
4169	 * Allow ourselves to be interrupted by SIGINT so that a
4170	 * connection recovery / failure event can be triggered externally.
4171	 */
4172	allow_signal(SIGINT);
4173	/*
4174	 * Wait for iscsi_post_login_handler() to complete before allowing
4175	 * incoming iscsi/tcp socket I/O, and/or failing the connection.
4176	 */
4177	rc = wait_for_completion_interruptible(&conn->rx_login_comp);
4178	if (rc < 0 || iscsi_target_check_conn_state(conn))
4179		goto out;
4180
4181	if (!conn->conn_transport->iscsit_get_rx_pdu)
4182		return 0;
4183
4184	conn->conn_transport->iscsit_get_rx_pdu(conn);
4185
4186	if (!signal_pending(current))
4187		atomic_set(&conn->transport_failed, 1);
4188	iscsit_take_action_for_connection_exit(conn, &conn_freed);
4189
4190out:
4191	if (!conn_freed) {
4192		while (!kthread_should_stop()) {
4193			msleep(100);
4194		}
4195	}
4196
4197	return 0;
4198}
4199
4200static void iscsit_release_commands_from_conn(struct iscsit_conn *conn)
4201{
4202	LIST_HEAD(tmp_list);
4203	struct iscsit_cmd *cmd = NULL, *cmd_tmp = NULL;
4204	struct iscsit_session *sess = conn->sess;
4205	/*
4206	 * We expect this function to only ever be called from either RX or TX
4207	 * thread context via iscsit_close_connection() once the other context
4208	 * has been reset -> returned sleeping pre-handler state.
4209	 */
4210	spin_lock_bh(&conn->cmd_lock);
4211	list_splice_init(&conn->conn_cmd_list, &tmp_list);
4212
4213	list_for_each_entry_safe(cmd, cmd_tmp, &tmp_list, i_conn_node) {
4214		struct se_cmd *se_cmd = &cmd->se_cmd;
4215
4216		if (!se_cmd->se_tfo)
4217			continue;
4218
4219		spin_lock_irq(&se_cmd->t_state_lock);
4220		if (se_cmd->transport_state & CMD_T_ABORTED) {
4221			if (!(se_cmd->transport_state & CMD_T_TAS))
4222				/*
4223				 * LIO's abort path owns the cleanup for this,
4224				 * so put it back on the list and let
4225				 * aborted_task handle it.
4226				 */
4227				list_move_tail(&cmd->i_conn_node,
4228					       &conn->conn_cmd_list);
4229		} else {
4230			se_cmd->transport_state |= CMD_T_FABRIC_STOP;
4231		}
4232
4233		if (cmd->se_cmd.t_state == TRANSPORT_WRITE_PENDING) {
4234			/*
4235			 * We never submitted the cmd to LIO core, so we have
4236			 * to tell LIO to perform the completion process.
4237			 */
4238			spin_unlock_irq(&se_cmd->t_state_lock);
4239			target_complete_cmd(&cmd->se_cmd, SAM_STAT_TASK_ABORTED);
4240			continue;
4241		}
4242		spin_unlock_irq(&se_cmd->t_state_lock);
4243	}
4244	spin_unlock_bh(&conn->cmd_lock);
4245
4246	list_for_each_entry_safe(cmd, cmd_tmp, &tmp_list, i_conn_node) {
4247		list_del_init(&cmd->i_conn_node);
4248
4249		iscsit_increment_maxcmdsn(cmd, sess);
4250		iscsit_free_cmd(cmd, true);
4251
4252	}
4253
4254	/*
4255	 * Wait on commands that were cleaned up via the aborted_task path.
4256	 * LLDs that implement iscsit_wait_conn will already have waited for
4257	 * commands.
4258	 */
4259	if (!conn->conn_transport->iscsit_wait_conn) {
4260		target_stop_cmd_counter(conn->cmd_cnt);
4261		target_wait_for_cmds(conn->cmd_cnt);
4262	}
4263}
4264
4265static void iscsit_stop_timers_for_cmds(
4266	struct iscsit_conn *conn)
4267{
4268	struct iscsit_cmd *cmd;
4269
4270	spin_lock_bh(&conn->cmd_lock);
4271	list_for_each_entry(cmd, &conn->conn_cmd_list, i_conn_node) {
4272		if (cmd->data_direction == DMA_TO_DEVICE)
4273			iscsit_stop_dataout_timer(cmd);
4274	}
4275	spin_unlock_bh(&conn->cmd_lock);
4276}
4277
4278int iscsit_close_connection(
4279	struct iscsit_conn *conn)
4280{
4281	int conn_logout = (conn->conn_state == TARG_CONN_STATE_IN_LOGOUT);
4282	struct iscsit_session	*sess = conn->sess;
4283
4284	pr_debug("Closing iSCSI connection CID %hu on SID:"
4285		" %u\n", conn->cid, sess->sid);
4286	/*
4287	 * Always up conn_logout_comp for the traditional TCP and HW_OFFLOAD
4288	 * case just in case the RX Thread in iscsi_target_rx_opcode() is
4289	 * sleeping and the logout response never got sent because the
4290	 * connection failed.
4291	 *
4292	 * However for iser-target, isert_wait4logout() is using conn_logout_comp
4293	 * to signal logout response TX interrupt completion.  Go ahead and skip
4294	 * this for iser since isert_rx_opcode() does not wait on logout failure,
4295	 * and to avoid iscsit_conn pointer dereference in iser-target code.
4296	 */
4297	if (!conn->conn_transport->rdma_shutdown)
4298		complete(&conn->conn_logout_comp);
4299
4300	if (!strcmp(current->comm, ISCSI_RX_THREAD_NAME)) {
4301		if (conn->tx_thread &&
4302		    cmpxchg(&conn->tx_thread_active, true, false)) {
4303			send_sig(SIGINT, conn->tx_thread, 1);
4304			kthread_stop(conn->tx_thread);
4305		}
4306	} else if (!strcmp(current->comm, ISCSI_TX_THREAD_NAME)) {
4307		if (conn->rx_thread &&
4308		    cmpxchg(&conn->rx_thread_active, true, false)) {
4309			send_sig(SIGINT, conn->rx_thread, 1);
4310			kthread_stop(conn->rx_thread);
4311		}
4312	}
4313
4314	spin_lock(&iscsit_global->ts_bitmap_lock);
4315	bitmap_release_region(iscsit_global->ts_bitmap, conn->bitmap_id,
4316			      get_order(1));
4317	spin_unlock(&iscsit_global->ts_bitmap_lock);
4318
4319	iscsit_stop_timers_for_cmds(conn);
4320	iscsit_stop_nopin_response_timer(conn);
4321	iscsit_stop_nopin_timer(conn);
4322
4323	if (conn->conn_transport->iscsit_wait_conn)
4324		conn->conn_transport->iscsit_wait_conn(conn);
4325
4326	/*
4327	 * During Connection recovery drop unacknowledged out of order
4328	 * commands for this connection, and prepare the other commands
4329	 * for reallegiance.
4330	 *
4331	 * During normal operation clear the out of order commands (but
4332	 * do not free the struct iscsi_ooo_cmdsn's) and release all
4333	 * struct iscsit_cmds.
4334	 */
4335	if (atomic_read(&conn->connection_recovery)) {
4336		iscsit_discard_unacknowledged_ooo_cmdsns_for_conn(conn);
4337		iscsit_prepare_cmds_for_reallegiance(conn);
4338	} else {
4339		iscsit_clear_ooo_cmdsns_for_conn(conn);
4340		iscsit_release_commands_from_conn(conn);
4341	}
4342	iscsit_free_queue_reqs_for_conn(conn);
4343
4344	/*
4345	 * Handle decrementing session or connection usage count if
4346	 * a logout response was not able to be sent because the
4347	 * connection failed.  Fall back to Session Recovery here.
4348	 */
4349	if (atomic_read(&conn->conn_logout_remove)) {
4350		if (conn->conn_logout_reason == ISCSI_LOGOUT_REASON_CLOSE_SESSION) {
4351			iscsit_dec_conn_usage_count(conn);
4352			iscsit_dec_session_usage_count(sess);
4353		}
4354		if (conn->conn_logout_reason == ISCSI_LOGOUT_REASON_CLOSE_CONNECTION)
4355			iscsit_dec_conn_usage_count(conn);
4356
4357		atomic_set(&conn->conn_logout_remove, 0);
4358		atomic_set(&sess->session_reinstatement, 0);
4359		atomic_set(&sess->session_fall_back_to_erl0, 1);
4360	}
4361
4362	spin_lock_bh(&sess->conn_lock);
4363	list_del(&conn->conn_list);
4364
4365	/*
4366	 * Attempt to let the Initiator know this connection failed by
4367	 * sending an Connection Dropped Async Message on another
4368	 * active connection.
4369	 */
4370	if (atomic_read(&conn->connection_recovery))
4371		iscsit_build_conn_drop_async_message(conn);
4372
4373	spin_unlock_bh(&sess->conn_lock);
4374
4375	/*
4376	 * If connection reinstatement is being performed on this connection,
4377	 * up the connection reinstatement semaphore that is being blocked on
4378	 * in iscsit_cause_connection_reinstatement().
4379	 */
4380	spin_lock_bh(&conn->state_lock);
4381	if (atomic_read(&conn->sleep_on_conn_wait_comp)) {
4382		spin_unlock_bh(&conn->state_lock);
4383		complete(&conn->conn_wait_comp);
4384		wait_for_completion(&conn->conn_post_wait_comp);
4385		spin_lock_bh(&conn->state_lock);
4386	}
4387
4388	/*
4389	 * If connection reinstatement is being performed on this connection
4390	 * by receiving a REMOVECONNFORRECOVERY logout request, up the
4391	 * connection wait rcfr semaphore that is being blocked on
4392	 * an iscsit_connection_reinstatement_rcfr().
4393	 */
4394	if (atomic_read(&conn->connection_wait_rcfr)) {
4395		spin_unlock_bh(&conn->state_lock);
4396		complete(&conn->conn_wait_rcfr_comp);
4397		wait_for_completion(&conn->conn_post_wait_comp);
4398		spin_lock_bh(&conn->state_lock);
4399	}
4400	atomic_set(&conn->connection_reinstatement, 1);
4401	spin_unlock_bh(&conn->state_lock);
4402
4403	/*
4404	 * If any other processes are accessing this connection pointer we
4405	 * must wait until they have completed.
4406	 */
4407	iscsit_check_conn_usage_count(conn);
4408
4409	ahash_request_free(conn->conn_tx_hash);
4410	if (conn->conn_rx_hash) {
4411		struct crypto_ahash *tfm;
4412
4413		tfm = crypto_ahash_reqtfm(conn->conn_rx_hash);
4414		ahash_request_free(conn->conn_rx_hash);
4415		crypto_free_ahash(tfm);
4416	}
4417
4418	if (conn->sock)
4419		sock_release(conn->sock);
4420
4421	if (conn->conn_transport->iscsit_free_conn)
4422		conn->conn_transport->iscsit_free_conn(conn);
4423
4424	pr_debug("Moving to TARG_CONN_STATE_FREE.\n");
4425	conn->conn_state = TARG_CONN_STATE_FREE;
4426	iscsit_free_conn(conn);
4427
4428	spin_lock_bh(&sess->conn_lock);
4429	atomic_dec(&sess->nconn);
4430	pr_debug("Decremented iSCSI connection count to %d from node:"
4431		" %s\n", atomic_read(&sess->nconn),
4432		sess->sess_ops->InitiatorName);
4433	/*
4434	 * Make sure that if one connection fails in an non ERL=2 iSCSI
4435	 * Session that they all fail.
4436	 */
4437	if ((sess->sess_ops->ErrorRecoveryLevel != 2) && !conn_logout &&
4438	     !atomic_read(&sess->session_logout))
4439		atomic_set(&sess->session_fall_back_to_erl0, 1);
4440
4441	/*
4442	 * If this was not the last connection in the session, and we are
4443	 * performing session reinstatement or falling back to ERL=0, call
4444	 * iscsit_stop_session() without sleeping to shutdown the other
4445	 * active connections.
4446	 */
4447	if (atomic_read(&sess->nconn)) {
4448		if (!atomic_read(&sess->session_reinstatement) &&
4449		    !atomic_read(&sess->session_fall_back_to_erl0)) {
4450			spin_unlock_bh(&sess->conn_lock);
4451			return 0;
4452		}
4453		if (!atomic_read(&sess->session_stop_active)) {
4454			atomic_set(&sess->session_stop_active, 1);
4455			spin_unlock_bh(&sess->conn_lock);
4456			iscsit_stop_session(sess, 0, 0);
4457			return 0;
4458		}
4459		spin_unlock_bh(&sess->conn_lock);
4460		return 0;
4461	}
4462
4463	/*
4464	 * If this was the last connection in the session and one of the
4465	 * following is occurring:
4466	 *
4467	 * Session Reinstatement is not being performed, and are falling back
4468	 * to ERL=0 call iscsit_close_session().
4469	 *
4470	 * Session Logout was requested.  iscsit_close_session() will be called
4471	 * elsewhere.
4472	 *
4473	 * Session Continuation is not being performed, start the Time2Retain
4474	 * handler and check if sleep_on_sess_wait_sem is active.
4475	 */
4476	if (!atomic_read(&sess->session_reinstatement) &&
4477	     atomic_read(&sess->session_fall_back_to_erl0)) {
4478		spin_unlock_bh(&sess->conn_lock);
4479		complete_all(&sess->session_wait_comp);
4480		iscsit_close_session(sess, true);
4481
4482		return 0;
4483	} else if (atomic_read(&sess->session_logout)) {
4484		pr_debug("Moving to TARG_SESS_STATE_FREE.\n");
4485		sess->session_state = TARG_SESS_STATE_FREE;
4486
4487		if (atomic_read(&sess->session_close)) {
4488			spin_unlock_bh(&sess->conn_lock);
4489			complete_all(&sess->session_wait_comp);
4490			iscsit_close_session(sess, true);
4491		} else {
4492			spin_unlock_bh(&sess->conn_lock);
4493		}
4494
4495		return 0;
4496	} else {
4497		pr_debug("Moving to TARG_SESS_STATE_FAILED.\n");
4498		sess->session_state = TARG_SESS_STATE_FAILED;
4499
4500		if (!atomic_read(&sess->session_continuation))
4501			iscsit_start_time2retain_handler(sess);
4502
4503		if (atomic_read(&sess->session_close)) {
4504			spin_unlock_bh(&sess->conn_lock);
4505			complete_all(&sess->session_wait_comp);
4506			iscsit_close_session(sess, true);
4507		} else {
4508			spin_unlock_bh(&sess->conn_lock);
4509		}
4510
4511		return 0;
4512	}
4513}
4514
4515/*
4516 * If the iSCSI Session for the iSCSI Initiator Node exists,
4517 * forcefully shutdown the iSCSI NEXUS.
4518 */
4519int iscsit_close_session(struct iscsit_session *sess, bool can_sleep)
4520{
4521	struct iscsi_portal_group *tpg = sess->tpg;
4522	struct se_portal_group *se_tpg = &tpg->tpg_se_tpg;
4523
4524	if (atomic_read(&sess->nconn)) {
4525		pr_err("%d connection(s) still exist for iSCSI session"
4526			" to %s\n", atomic_read(&sess->nconn),
4527			sess->sess_ops->InitiatorName);
4528		BUG();
4529	}
4530
4531	spin_lock_bh(&se_tpg->session_lock);
4532	atomic_set(&sess->session_logout, 1);
4533	atomic_set(&sess->session_reinstatement, 1);
4534	iscsit_stop_time2retain_timer(sess);
4535	spin_unlock_bh(&se_tpg->session_lock);
4536
4537	if (sess->sess_ops->ErrorRecoveryLevel == 2)
4538		iscsit_free_connection_recovery_entries(sess);
4539
4540	/*
4541	 * transport_deregister_session_configfs() will clear the
4542	 * struct se_node_acl->nacl_sess pointer now as a iscsi_np process context
4543	 * can be setting it again with __transport_register_session() in
4544	 * iscsi_post_login_handler() again after the iscsit_stop_session()
4545	 * completes in iscsi_np context.
4546	 */
4547	transport_deregister_session_configfs(sess->se_sess);
4548
4549	/*
4550	 * If any other processes are accessing this session pointer we must
4551	 * wait until they have completed.  If we are in an interrupt (the
4552	 * time2retain handler) and contain and active session usage count we
4553	 * restart the timer and exit.
4554	 */
4555	if (iscsit_check_session_usage_count(sess, can_sleep)) {
4556		atomic_set(&sess->session_logout, 0);
4557		iscsit_start_time2retain_handler(sess);
4558		return 0;
4559	}
4560
4561	transport_deregister_session(sess->se_sess);
4562
4563	iscsit_free_all_ooo_cmdsns(sess);
4564
4565	spin_lock_bh(&se_tpg->session_lock);
4566	pr_debug("Moving to TARG_SESS_STATE_FREE.\n");
4567	sess->session_state = TARG_SESS_STATE_FREE;
4568	pr_debug("Released iSCSI session from node: %s\n",
4569			sess->sess_ops->InitiatorName);
4570	tpg->nsessions--;
4571	if (tpg->tpg_tiqn)
4572		tpg->tpg_tiqn->tiqn_nsessions--;
4573
4574	pr_debug("Decremented number of active iSCSI Sessions on"
4575		" iSCSI TPG: %hu to %u\n", tpg->tpgt, tpg->nsessions);
4576
4577	ida_free(&sess_ida, sess->session_index);
4578	kfree(sess->sess_ops);
4579	sess->sess_ops = NULL;
4580	spin_unlock_bh(&se_tpg->session_lock);
4581
4582	kfree(sess);
4583	return 0;
4584}
4585
4586static void iscsit_logout_post_handler_closesession(
4587	struct iscsit_conn *conn)
4588{
4589	struct iscsit_session *sess = conn->sess;
4590	int sleep = 1;
4591	/*
4592	 * Traditional iscsi/tcp will invoke this logic from TX thread
4593	 * context during session logout, so clear tx_thread_active and
4594	 * sleep if iscsit_close_connection() has not already occured.
4595	 *
4596	 * Since iser-target invokes this logic from it's own workqueue,
4597	 * always sleep waiting for RX/TX thread shutdown to complete
4598	 * within iscsit_close_connection().
4599	 */
4600	if (!conn->conn_transport->rdma_shutdown) {
4601		sleep = cmpxchg(&conn->tx_thread_active, true, false);
4602		if (!sleep)
4603			return;
4604	}
4605
4606	atomic_set(&conn->conn_logout_remove, 0);
4607	complete(&conn->conn_logout_comp);
4608
4609	iscsit_dec_conn_usage_count(conn);
4610	atomic_set(&sess->session_close, 1);
4611	iscsit_stop_session(sess, sleep, sleep);
4612	iscsit_dec_session_usage_count(sess);
4613}
4614
4615static void iscsit_logout_post_handler_samecid(
4616	struct iscsit_conn *conn)
4617{
4618	int sleep = 1;
4619
4620	if (!conn->conn_transport->rdma_shutdown) {
4621		sleep = cmpxchg(&conn->tx_thread_active, true, false);
4622		if (!sleep)
4623			return;
4624	}
4625
4626	atomic_set(&conn->conn_logout_remove, 0);
4627	complete(&conn->conn_logout_comp);
4628
4629	iscsit_cause_connection_reinstatement(conn, sleep);
4630	iscsit_dec_conn_usage_count(conn);
4631}
4632
4633static void iscsit_logout_post_handler_diffcid(
4634	struct iscsit_conn *conn,
4635	u16 cid)
4636{
4637	struct iscsit_conn *l_conn;
4638	struct iscsit_session *sess = conn->sess;
4639	bool conn_found = false;
4640
4641	if (!sess)
4642		return;
4643
4644	spin_lock_bh(&sess->conn_lock);
4645	list_for_each_entry(l_conn, &sess->sess_conn_list, conn_list) {
4646		if (l_conn->cid == cid) {
4647			iscsit_inc_conn_usage_count(l_conn);
4648			conn_found = true;
4649			break;
4650		}
4651	}
4652	spin_unlock_bh(&sess->conn_lock);
4653
4654	if (!conn_found)
4655		return;
4656
4657	if (l_conn->sock)
4658		l_conn->sock->ops->shutdown(l_conn->sock, RCV_SHUTDOWN);
4659
4660	spin_lock_bh(&l_conn->state_lock);
4661	pr_debug("Moving to TARG_CONN_STATE_IN_LOGOUT.\n");
4662	l_conn->conn_state = TARG_CONN_STATE_IN_LOGOUT;
4663	spin_unlock_bh(&l_conn->state_lock);
4664
4665	iscsit_cause_connection_reinstatement(l_conn, 1);
4666	iscsit_dec_conn_usage_count(l_conn);
4667}
4668
4669/*
4670 *	Return of 0 causes the TX thread to restart.
4671 */
4672int iscsit_logout_post_handler(
4673	struct iscsit_cmd *cmd,
4674	struct iscsit_conn *conn)
4675{
4676	int ret = 0;
4677
4678	switch (cmd->logout_reason) {
4679	case ISCSI_LOGOUT_REASON_CLOSE_SESSION:
4680		switch (cmd->logout_response) {
4681		case ISCSI_LOGOUT_SUCCESS:
4682		case ISCSI_LOGOUT_CLEANUP_FAILED:
4683		default:
4684			iscsit_logout_post_handler_closesession(conn);
4685			break;
4686		}
4687		break;
4688	case ISCSI_LOGOUT_REASON_CLOSE_CONNECTION:
4689		if (conn->cid == cmd->logout_cid) {
4690			switch (cmd->logout_response) {
4691			case ISCSI_LOGOUT_SUCCESS:
4692			case ISCSI_LOGOUT_CLEANUP_FAILED:
4693			default:
4694				iscsit_logout_post_handler_samecid(conn);
4695				break;
4696			}
4697		} else {
4698			switch (cmd->logout_response) {
4699			case ISCSI_LOGOUT_SUCCESS:
4700				iscsit_logout_post_handler_diffcid(conn,
4701					cmd->logout_cid);
4702				break;
4703			case ISCSI_LOGOUT_CID_NOT_FOUND:
4704			case ISCSI_LOGOUT_CLEANUP_FAILED:
4705			default:
4706				break;
4707			}
4708			ret = 1;
4709		}
4710		break;
4711	case ISCSI_LOGOUT_REASON_RECOVERY:
4712		switch (cmd->logout_response) {
4713		case ISCSI_LOGOUT_SUCCESS:
4714		case ISCSI_LOGOUT_CID_NOT_FOUND:
4715		case ISCSI_LOGOUT_RECOVERY_UNSUPPORTED:
4716		case ISCSI_LOGOUT_CLEANUP_FAILED:
4717		default:
4718			break;
4719		}
4720		ret = 1;
4721		break;
4722	default:
4723		break;
4724
4725	}
4726	return ret;
4727}
4728EXPORT_SYMBOL(iscsit_logout_post_handler);
4729
4730void iscsit_fail_session(struct iscsit_session *sess)
4731{
4732	struct iscsit_conn *conn;
4733
4734	spin_lock_bh(&sess->conn_lock);
4735	list_for_each_entry(conn, &sess->sess_conn_list, conn_list) {
4736		pr_debug("Moving to TARG_CONN_STATE_CLEANUP_WAIT.\n");
4737		conn->conn_state = TARG_CONN_STATE_CLEANUP_WAIT;
4738	}
4739	spin_unlock_bh(&sess->conn_lock);
4740
4741	pr_debug("Moving to TARG_SESS_STATE_FAILED.\n");
4742	sess->session_state = TARG_SESS_STATE_FAILED;
4743}
4744
4745void iscsit_stop_session(
4746	struct iscsit_session *sess,
4747	int session_sleep,
4748	int connection_sleep)
4749{
4750	u16 conn_count = atomic_read(&sess->nconn);
4751	struct iscsit_conn *conn, *conn_tmp = NULL;
4752	int is_last;
4753
4754	spin_lock_bh(&sess->conn_lock);
4755
4756	if (connection_sleep) {
4757		list_for_each_entry_safe(conn, conn_tmp, &sess->sess_conn_list,
4758				conn_list) {
4759			if (conn_count == 0)
4760				break;
4761
4762			if (list_is_last(&conn->conn_list, &sess->sess_conn_list)) {
4763				is_last = 1;
4764			} else {
4765				iscsit_inc_conn_usage_count(conn_tmp);
4766				is_last = 0;
4767			}
4768			iscsit_inc_conn_usage_count(conn);
4769
4770			spin_unlock_bh(&sess->conn_lock);
4771			iscsit_cause_connection_reinstatement(conn, 1);
4772			spin_lock_bh(&sess->conn_lock);
4773
4774			iscsit_dec_conn_usage_count(conn);
4775			if (is_last == 0)
4776				iscsit_dec_conn_usage_count(conn_tmp);
4777			conn_count--;
4778		}
4779	} else {
4780		list_for_each_entry(conn, &sess->sess_conn_list, conn_list)
4781			iscsit_cause_connection_reinstatement(conn, 0);
4782	}
4783
4784	if (session_sleep && atomic_read(&sess->nconn)) {
4785		spin_unlock_bh(&sess->conn_lock);
4786		wait_for_completion(&sess->session_wait_comp);
4787	} else
4788		spin_unlock_bh(&sess->conn_lock);
4789}
4790
4791int iscsit_release_sessions_for_tpg(struct iscsi_portal_group *tpg, int force)
4792{
4793	struct iscsit_session *sess;
4794	struct se_portal_group *se_tpg = &tpg->tpg_se_tpg;
4795	struct se_session *se_sess, *se_sess_tmp;
4796	LIST_HEAD(free_list);
4797	int session_count = 0;
4798
4799	spin_lock_bh(&se_tpg->session_lock);
4800	if (tpg->nsessions && !force) {
4801		spin_unlock_bh(&se_tpg->session_lock);
4802		return -1;
4803	}
4804
4805	list_for_each_entry_safe(se_sess, se_sess_tmp, &se_tpg->tpg_sess_list,
4806			sess_list) {
4807		sess = (struct iscsit_session *)se_sess->fabric_sess_ptr;
4808
4809		spin_lock(&sess->conn_lock);
4810		if (atomic_read(&sess->session_fall_back_to_erl0) ||
4811		    atomic_read(&sess->session_logout) ||
4812		    atomic_read(&sess->session_close) ||
4813		    (sess->time2retain_timer_flags & ISCSI_TF_EXPIRED)) {
4814			spin_unlock(&sess->conn_lock);
4815			continue;
4816		}
4817		iscsit_inc_session_usage_count(sess);
4818		atomic_set(&sess->session_reinstatement, 1);
4819		atomic_set(&sess->session_fall_back_to_erl0, 1);
4820		atomic_set(&sess->session_close, 1);
4821		spin_unlock(&sess->conn_lock);
4822
4823		list_move_tail(&se_sess->sess_list, &free_list);
4824	}
4825	spin_unlock_bh(&se_tpg->session_lock);
4826
4827	list_for_each_entry_safe(se_sess, se_sess_tmp, &free_list, sess_list) {
4828		sess = (struct iscsit_session *)se_sess->fabric_sess_ptr;
4829
4830		list_del_init(&se_sess->sess_list);
4831		iscsit_stop_session(sess, 1, 1);
4832		iscsit_dec_session_usage_count(sess);
4833		session_count++;
4834	}
4835
4836	pr_debug("Released %d iSCSI Session(s) from Target Portal"
4837			" Group: %hu\n", session_count, tpg->tpgt);
4838	return 0;
4839}
4840
4841MODULE_DESCRIPTION("iSCSI-Target Driver for mainline target infrastructure");
4842MODULE_VERSION("4.1.x");
4843MODULE_AUTHOR("nab@Linux-iSCSI.org");
4844MODULE_LICENSE("GPL");
4845
4846module_init(iscsi_target_init_module);
4847module_exit(iscsi_target_cleanup_module);
4848