xref: /openbmc/linux/drivers/scsi/iscsi_tcp.c (revision 4800cd83)
1 /*
2  * iSCSI Initiator over TCP/IP Data-Path
3  *
4  * Copyright (C) 2004 Dmitry Yusupov
5  * Copyright (C) 2004 Alex Aizman
6  * Copyright (C) 2005 - 2006 Mike Christie
7  * Copyright (C) 2006 Red Hat, Inc.  All rights reserved.
8  * maintained by open-iscsi@googlegroups.com
9  *
10  * This program is free software; you can redistribute it and/or modify
11  * it under the terms of the GNU General Public License as published
12  * by the Free Software Foundation; either version 2 of the License, or
13  * (at your option) any later version.
14  *
15  * This program is distributed in the hope that it will be useful, but
16  * WITHOUT ANY WARRANTY; without even the implied warranty of
17  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
18  * General Public License for more details.
19  *
20  * See the file COPYING included with this distribution for more details.
21  *
22  * Credits:
23  *	Christoph Hellwig
24  *	FUJITA Tomonori
25  *	Arne Redlich
26  *	Zhenyu Wang
27  */
28 
29 #include <linux/types.h>
30 #include <linux/inet.h>
31 #include <linux/slab.h>
32 #include <linux/file.h>
33 #include <linux/blkdev.h>
34 #include <linux/crypto.h>
35 #include <linux/delay.h>
36 #include <linux/kfifo.h>
37 #include <linux/scatterlist.h>
38 #include <net/tcp.h>
39 #include <scsi/scsi_cmnd.h>
40 #include <scsi/scsi_device.h>
41 #include <scsi/scsi_host.h>
42 #include <scsi/scsi.h>
43 #include <scsi/scsi_transport_iscsi.h>
44 
45 #include "iscsi_tcp.h"
46 
47 MODULE_AUTHOR("Mike Christie <michaelc@cs.wisc.edu>, "
48 	      "Dmitry Yusupov <dmitry_yus@yahoo.com>, "
49 	      "Alex Aizman <itn780@yahoo.com>");
50 MODULE_DESCRIPTION("iSCSI/TCP data-path");
51 MODULE_LICENSE("GPL");
52 
53 static struct scsi_transport_template *iscsi_sw_tcp_scsi_transport;
54 static struct scsi_host_template iscsi_sw_tcp_sht;
55 static struct iscsi_transport iscsi_sw_tcp_transport;
56 
57 static unsigned int iscsi_max_lun = 512;
58 module_param_named(max_lun, iscsi_max_lun, uint, S_IRUGO);
59 
60 static int iscsi_sw_tcp_dbg;
61 module_param_named(debug_iscsi_tcp, iscsi_sw_tcp_dbg, int,
62 		   S_IRUGO | S_IWUSR);
63 MODULE_PARM_DESC(debug_iscsi_tcp, "Turn on debugging for iscsi_tcp module "
64 		 "Set to 1 to turn on, and zero to turn off. Default is off.");
65 
66 #define ISCSI_SW_TCP_DBG(_conn, dbg_fmt, arg...)		\
67 	do {							\
68 		if (iscsi_sw_tcp_dbg)				\
69 			iscsi_conn_printk(KERN_INFO, _conn,	\
70 					     "%s " dbg_fmt,	\
71 					     __func__, ##arg);	\
72 	} while (0);
73 
74 
75 /**
76  * iscsi_sw_tcp_recv - TCP receive in sendfile fashion
77  * @rd_desc: read descriptor
78  * @skb: socket buffer
79  * @offset: offset in skb
80  * @len: skb->len - offset
81  */
82 static int iscsi_sw_tcp_recv(read_descriptor_t *rd_desc, struct sk_buff *skb,
83 			     unsigned int offset, size_t len)
84 {
85 	struct iscsi_conn *conn = rd_desc->arg.data;
86 	unsigned int consumed, total_consumed = 0;
87 	int status;
88 
89 	ISCSI_SW_TCP_DBG(conn, "in %d bytes\n", skb->len - offset);
90 
91 	do {
92 		status = 0;
93 		consumed = iscsi_tcp_recv_skb(conn, skb, offset, 0, &status);
94 		offset += consumed;
95 		total_consumed += consumed;
96 	} while (consumed != 0 && status != ISCSI_TCP_SKB_DONE);
97 
98 	ISCSI_SW_TCP_DBG(conn, "read %d bytes status %d\n",
99 			 skb->len - offset, status);
100 	return total_consumed;
101 }
102 
103 /**
104  * iscsi_sw_sk_state_check - check socket state
105  * @sk: socket
106  *
107  * If the socket is in CLOSE or CLOSE_WAIT we should
108  * not close the connection if there is still some
109  * data pending.
110  */
111 static inline int iscsi_sw_sk_state_check(struct sock *sk)
112 {
113 	struct iscsi_conn *conn = (struct iscsi_conn*)sk->sk_user_data;
114 
115 	if ((sk->sk_state == TCP_CLOSE_WAIT || sk->sk_state == TCP_CLOSE) &&
116 	    !atomic_read(&sk->sk_rmem_alloc)) {
117 		ISCSI_SW_TCP_DBG(conn, "TCP_CLOSE|TCP_CLOSE_WAIT\n");
118 		iscsi_conn_failure(conn, ISCSI_ERR_TCP_CONN_CLOSE);
119 		return -ECONNRESET;
120 	}
121 	return 0;
122 }
123 
124 static void iscsi_sw_tcp_data_ready(struct sock *sk, int flag)
125 {
126 	struct iscsi_conn *conn = sk->sk_user_data;
127 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
128 	read_descriptor_t rd_desc;
129 
130 	read_lock(&sk->sk_callback_lock);
131 
132 	/*
133 	 * Use rd_desc to pass 'conn' to iscsi_tcp_recv.
134 	 * We set count to 1 because we want the network layer to
135 	 * hand us all the skbs that are available. iscsi_tcp_recv
136 	 * handled pdus that cross buffers or pdus that still need data.
137 	 */
138 	rd_desc.arg.data = conn;
139 	rd_desc.count = 1;
140 	tcp_read_sock(sk, &rd_desc, iscsi_sw_tcp_recv);
141 
142 	iscsi_sw_sk_state_check(sk);
143 
144 	read_unlock(&sk->sk_callback_lock);
145 
146 	/* If we had to (atomically) map a highmem page,
147 	 * unmap it now. */
148 	iscsi_tcp_segment_unmap(&tcp_conn->in.segment);
149 }
150 
151 static void iscsi_sw_tcp_state_change(struct sock *sk)
152 {
153 	struct iscsi_tcp_conn *tcp_conn;
154 	struct iscsi_sw_tcp_conn *tcp_sw_conn;
155 	struct iscsi_conn *conn;
156 	struct iscsi_session *session;
157 	void (*old_state_change)(struct sock *);
158 
159 	read_lock(&sk->sk_callback_lock);
160 
161 	conn = (struct iscsi_conn*)sk->sk_user_data;
162 	session = conn->session;
163 
164 	iscsi_sw_sk_state_check(sk);
165 
166 	tcp_conn = conn->dd_data;
167 	tcp_sw_conn = tcp_conn->dd_data;
168 	old_state_change = tcp_sw_conn->old_state_change;
169 
170 	read_unlock(&sk->sk_callback_lock);
171 
172 	old_state_change(sk);
173 }
174 
175 /**
176  * iscsi_write_space - Called when more output buffer space is available
177  * @sk: socket space is available for
178  **/
179 static void iscsi_sw_tcp_write_space(struct sock *sk)
180 {
181 	struct iscsi_conn *conn = (struct iscsi_conn*)sk->sk_user_data;
182 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
183 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
184 
185 	tcp_sw_conn->old_write_space(sk);
186 	ISCSI_SW_TCP_DBG(conn, "iscsi_write_space\n");
187 	iscsi_conn_queue_work(conn);
188 }
189 
190 static void iscsi_sw_tcp_conn_set_callbacks(struct iscsi_conn *conn)
191 {
192 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
193 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
194 	struct sock *sk = tcp_sw_conn->sock->sk;
195 
196 	/* assign new callbacks */
197 	write_lock_bh(&sk->sk_callback_lock);
198 	sk->sk_user_data = conn;
199 	tcp_sw_conn->old_data_ready = sk->sk_data_ready;
200 	tcp_sw_conn->old_state_change = sk->sk_state_change;
201 	tcp_sw_conn->old_write_space = sk->sk_write_space;
202 	sk->sk_data_ready = iscsi_sw_tcp_data_ready;
203 	sk->sk_state_change = iscsi_sw_tcp_state_change;
204 	sk->sk_write_space = iscsi_sw_tcp_write_space;
205 	write_unlock_bh(&sk->sk_callback_lock);
206 }
207 
208 static void
209 iscsi_sw_tcp_conn_restore_callbacks(struct iscsi_conn *conn)
210 {
211 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
212 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
213 	struct sock *sk = tcp_sw_conn->sock->sk;
214 
215 	/* restore socket callbacks, see also: iscsi_conn_set_callbacks() */
216 	write_lock_bh(&sk->sk_callback_lock);
217 	sk->sk_user_data    = NULL;
218 	sk->sk_data_ready   = tcp_sw_conn->old_data_ready;
219 	sk->sk_state_change = tcp_sw_conn->old_state_change;
220 	sk->sk_write_space  = tcp_sw_conn->old_write_space;
221 	sk->sk_no_check	 = 0;
222 	write_unlock_bh(&sk->sk_callback_lock);
223 }
224 
225 /**
226  * iscsi_sw_tcp_xmit_segment - transmit segment
227  * @tcp_conn: the iSCSI TCP connection
228  * @segment: the buffer to transmnit
229  *
230  * This function transmits as much of the buffer as
231  * the network layer will accept, and returns the number of
232  * bytes transmitted.
233  *
234  * If CRC hashing is enabled, the function will compute the
235  * hash as it goes. When the entire segment has been transmitted,
236  * it will retrieve the hash value and send it as well.
237  */
238 static int iscsi_sw_tcp_xmit_segment(struct iscsi_tcp_conn *tcp_conn,
239 				     struct iscsi_segment *segment)
240 {
241 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
242 	struct socket *sk = tcp_sw_conn->sock;
243 	unsigned int copied = 0;
244 	int r = 0;
245 
246 	while (!iscsi_tcp_segment_done(tcp_conn, segment, 0, r)) {
247 		struct scatterlist *sg;
248 		unsigned int offset, copy;
249 		int flags = 0;
250 
251 		r = 0;
252 		offset = segment->copied;
253 		copy = segment->size - offset;
254 
255 		if (segment->total_copied + segment->size < segment->total_size)
256 			flags |= MSG_MORE;
257 
258 		/* Use sendpage if we can; else fall back to sendmsg */
259 		if (!segment->data) {
260 			sg = segment->sg;
261 			offset += segment->sg_offset + sg->offset;
262 			r = tcp_sw_conn->sendpage(sk, sg_page(sg), offset,
263 						  copy, flags);
264 		} else {
265 			struct msghdr msg = { .msg_flags = flags };
266 			struct kvec iov = {
267 				.iov_base = segment->data + offset,
268 				.iov_len = copy
269 			};
270 
271 			r = kernel_sendmsg(sk, &msg, &iov, 1, copy);
272 		}
273 
274 		if (r < 0) {
275 			iscsi_tcp_segment_unmap(segment);
276 			return r;
277 		}
278 		copied += r;
279 	}
280 	return copied;
281 }
282 
283 /**
284  * iscsi_sw_tcp_xmit - TCP transmit
285  **/
286 static int iscsi_sw_tcp_xmit(struct iscsi_conn *conn)
287 {
288 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
289 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
290 	struct iscsi_segment *segment = &tcp_sw_conn->out.segment;
291 	unsigned int consumed = 0;
292 	int rc = 0;
293 
294 	while (1) {
295 		rc = iscsi_sw_tcp_xmit_segment(tcp_conn, segment);
296 		/*
297 		 * We may not have been able to send data because the conn
298 		 * is getting stopped. libiscsi will know so propogate err
299 		 * for it to do the right thing.
300 		 */
301 		if (rc == -EAGAIN)
302 			return rc;
303 		else if (rc < 0) {
304 			rc = ISCSI_ERR_XMIT_FAILED;
305 			goto error;
306 		} else if (rc == 0)
307 			break;
308 
309 		consumed += rc;
310 
311 		if (segment->total_copied >= segment->total_size) {
312 			if (segment->done != NULL) {
313 				rc = segment->done(tcp_conn, segment);
314 				if (rc != 0)
315 					goto error;
316 			}
317 		}
318 	}
319 
320 	ISCSI_SW_TCP_DBG(conn, "xmit %d bytes\n", consumed);
321 
322 	conn->txdata_octets += consumed;
323 	return consumed;
324 
325 error:
326 	/* Transmit error. We could initiate error recovery
327 	 * here. */
328 	ISCSI_SW_TCP_DBG(conn, "Error sending PDU, errno=%d\n", rc);
329 	iscsi_conn_failure(conn, rc);
330 	return -EIO;
331 }
332 
333 /**
334  * iscsi_tcp_xmit_qlen - return the number of bytes queued for xmit
335  */
336 static inline int iscsi_sw_tcp_xmit_qlen(struct iscsi_conn *conn)
337 {
338 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
339 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
340 	struct iscsi_segment *segment = &tcp_sw_conn->out.segment;
341 
342 	return segment->total_copied - segment->total_size;
343 }
344 
345 static int iscsi_sw_tcp_pdu_xmit(struct iscsi_task *task)
346 {
347 	struct iscsi_conn *conn = task->conn;
348 	int rc;
349 
350 	while (iscsi_sw_tcp_xmit_qlen(conn)) {
351 		rc = iscsi_sw_tcp_xmit(conn);
352 		if (rc == 0)
353 			return -EAGAIN;
354 		if (rc < 0)
355 			return rc;
356 	}
357 
358 	return 0;
359 }
360 
361 /*
362  * This is called when we're done sending the header.
363  * Simply copy the data_segment to the send segment, and return.
364  */
365 static int iscsi_sw_tcp_send_hdr_done(struct iscsi_tcp_conn *tcp_conn,
366 				      struct iscsi_segment *segment)
367 {
368 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
369 
370 	tcp_sw_conn->out.segment = tcp_sw_conn->out.data_segment;
371 	ISCSI_SW_TCP_DBG(tcp_conn->iscsi_conn,
372 			 "Header done. Next segment size %u total_size %u\n",
373 			 tcp_sw_conn->out.segment.size,
374 			 tcp_sw_conn->out.segment.total_size);
375 	return 0;
376 }
377 
378 static void iscsi_sw_tcp_send_hdr_prep(struct iscsi_conn *conn, void *hdr,
379 				       size_t hdrlen)
380 {
381 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
382 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
383 
384 	ISCSI_SW_TCP_DBG(conn, "%s\n", conn->hdrdgst_en ?
385 			 "digest enabled" : "digest disabled");
386 
387 	/* Clear the data segment - needs to be filled in by the
388 	 * caller using iscsi_tcp_send_data_prep() */
389 	memset(&tcp_sw_conn->out.data_segment, 0,
390 	       sizeof(struct iscsi_segment));
391 
392 	/* If header digest is enabled, compute the CRC and
393 	 * place the digest into the same buffer. We make
394 	 * sure that both iscsi_tcp_task and mtask have
395 	 * sufficient room.
396 	 */
397 	if (conn->hdrdgst_en) {
398 		iscsi_tcp_dgst_header(&tcp_sw_conn->tx_hash, hdr, hdrlen,
399 				      hdr + hdrlen);
400 		hdrlen += ISCSI_DIGEST_SIZE;
401 	}
402 
403 	/* Remember header pointer for later, when we need
404 	 * to decide whether there's a payload to go along
405 	 * with the header. */
406 	tcp_sw_conn->out.hdr = hdr;
407 
408 	iscsi_segment_init_linear(&tcp_sw_conn->out.segment, hdr, hdrlen,
409 				  iscsi_sw_tcp_send_hdr_done, NULL);
410 }
411 
412 /*
413  * Prepare the send buffer for the payload data.
414  * Padding and checksumming will all be taken care
415  * of by the iscsi_segment routines.
416  */
417 static int
418 iscsi_sw_tcp_send_data_prep(struct iscsi_conn *conn, struct scatterlist *sg,
419 			    unsigned int count, unsigned int offset,
420 			    unsigned int len)
421 {
422 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
423 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
424 	struct hash_desc *tx_hash = NULL;
425 	unsigned int hdr_spec_len;
426 
427 	ISCSI_SW_TCP_DBG(conn, "offset=%d, datalen=%d %s\n", offset, len,
428 			 conn->datadgst_en ?
429 			 "digest enabled" : "digest disabled");
430 
431 	/* Make sure the datalen matches what the caller
432 	   said he would send. */
433 	hdr_spec_len = ntoh24(tcp_sw_conn->out.hdr->dlength);
434 	WARN_ON(iscsi_padded(len) != iscsi_padded(hdr_spec_len));
435 
436 	if (conn->datadgst_en)
437 		tx_hash = &tcp_sw_conn->tx_hash;
438 
439 	return iscsi_segment_seek_sg(&tcp_sw_conn->out.data_segment,
440 				     sg, count, offset, len,
441 				     NULL, tx_hash);
442 }
443 
444 static void
445 iscsi_sw_tcp_send_linear_data_prep(struct iscsi_conn *conn, void *data,
446 				   size_t len)
447 {
448 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
449 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
450 	struct hash_desc *tx_hash = NULL;
451 	unsigned int hdr_spec_len;
452 
453 	ISCSI_SW_TCP_DBG(conn, "datalen=%zd %s\n", len, conn->datadgst_en ?
454 			 "digest enabled" : "digest disabled");
455 
456 	/* Make sure the datalen matches what the caller
457 	   said he would send. */
458 	hdr_spec_len = ntoh24(tcp_sw_conn->out.hdr->dlength);
459 	WARN_ON(iscsi_padded(len) != iscsi_padded(hdr_spec_len));
460 
461 	if (conn->datadgst_en)
462 		tx_hash = &tcp_sw_conn->tx_hash;
463 
464 	iscsi_segment_init_linear(&tcp_sw_conn->out.data_segment,
465 				data, len, NULL, tx_hash);
466 }
467 
468 static int iscsi_sw_tcp_pdu_init(struct iscsi_task *task,
469 				 unsigned int offset, unsigned int count)
470 {
471 	struct iscsi_conn *conn = task->conn;
472 	int err = 0;
473 
474 	iscsi_sw_tcp_send_hdr_prep(conn, task->hdr, task->hdr_len);
475 
476 	if (!count)
477 		return 0;
478 
479 	if (!task->sc)
480 		iscsi_sw_tcp_send_linear_data_prep(conn, task->data, count);
481 	else {
482 		struct scsi_data_buffer *sdb = scsi_out(task->sc);
483 
484 		err = iscsi_sw_tcp_send_data_prep(conn, sdb->table.sgl,
485 						  sdb->table.nents, offset,
486 						  count);
487 	}
488 
489 	if (err) {
490 		/* got invalid offset/len */
491 		return -EIO;
492 	}
493 	return 0;
494 }
495 
496 static int iscsi_sw_tcp_pdu_alloc(struct iscsi_task *task, uint8_t opcode)
497 {
498 	struct iscsi_tcp_task *tcp_task = task->dd_data;
499 
500 	task->hdr = task->dd_data + sizeof(*tcp_task);
501 	task->hdr_max = sizeof(struct iscsi_sw_tcp_hdrbuf) - ISCSI_DIGEST_SIZE;
502 	return 0;
503 }
504 
505 static struct iscsi_cls_conn *
506 iscsi_sw_tcp_conn_create(struct iscsi_cls_session *cls_session,
507 			 uint32_t conn_idx)
508 {
509 	struct iscsi_conn *conn;
510 	struct iscsi_cls_conn *cls_conn;
511 	struct iscsi_tcp_conn *tcp_conn;
512 	struct iscsi_sw_tcp_conn *tcp_sw_conn;
513 
514 	cls_conn = iscsi_tcp_conn_setup(cls_session, sizeof(*tcp_sw_conn),
515 					conn_idx);
516 	if (!cls_conn)
517 		return NULL;
518 	conn = cls_conn->dd_data;
519 	tcp_conn = conn->dd_data;
520 	tcp_sw_conn = tcp_conn->dd_data;
521 
522 	tcp_sw_conn->tx_hash.tfm = crypto_alloc_hash("crc32c", 0,
523 						     CRYPTO_ALG_ASYNC);
524 	tcp_sw_conn->tx_hash.flags = 0;
525 	if (IS_ERR(tcp_sw_conn->tx_hash.tfm))
526 		goto free_conn;
527 
528 	tcp_sw_conn->rx_hash.tfm = crypto_alloc_hash("crc32c", 0,
529 						     CRYPTO_ALG_ASYNC);
530 	tcp_sw_conn->rx_hash.flags = 0;
531 	if (IS_ERR(tcp_sw_conn->rx_hash.tfm))
532 		goto free_tx_tfm;
533 	tcp_conn->rx_hash = &tcp_sw_conn->rx_hash;
534 
535 	return cls_conn;
536 
537 free_tx_tfm:
538 	crypto_free_hash(tcp_sw_conn->tx_hash.tfm);
539 free_conn:
540 	iscsi_conn_printk(KERN_ERR, conn,
541 			  "Could not create connection due to crc32c "
542 			  "loading error. Make sure the crc32c "
543 			  "module is built as a module or into the "
544 			  "kernel\n");
545 	iscsi_tcp_conn_teardown(cls_conn);
546 	return NULL;
547 }
548 
549 static void iscsi_sw_tcp_release_conn(struct iscsi_conn *conn)
550 {
551 	struct iscsi_session *session = conn->session;
552 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
553 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
554 	struct socket *sock = tcp_sw_conn->sock;
555 
556 	if (!sock)
557 		return;
558 
559 	sock_hold(sock->sk);
560 	iscsi_sw_tcp_conn_restore_callbacks(conn);
561 	sock_put(sock->sk);
562 
563 	spin_lock_bh(&session->lock);
564 	tcp_sw_conn->sock = NULL;
565 	spin_unlock_bh(&session->lock);
566 	sockfd_put(sock);
567 }
568 
569 static void iscsi_sw_tcp_conn_destroy(struct iscsi_cls_conn *cls_conn)
570 {
571 	struct iscsi_conn *conn = cls_conn->dd_data;
572 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
573 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
574 
575 	iscsi_sw_tcp_release_conn(conn);
576 
577 	if (tcp_sw_conn->tx_hash.tfm)
578 		crypto_free_hash(tcp_sw_conn->tx_hash.tfm);
579 	if (tcp_sw_conn->rx_hash.tfm)
580 		crypto_free_hash(tcp_sw_conn->rx_hash.tfm);
581 
582 	iscsi_tcp_conn_teardown(cls_conn);
583 }
584 
585 static void iscsi_sw_tcp_conn_stop(struct iscsi_cls_conn *cls_conn, int flag)
586 {
587 	struct iscsi_conn *conn = cls_conn->dd_data;
588 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
589 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
590 	struct socket *sock = tcp_sw_conn->sock;
591 
592 	/* userspace may have goofed up and not bound us */
593 	if (!sock)
594 		return;
595 	/*
596 	 * Make sure our recv side is stopped.
597 	 * Older tools called conn stop before ep_disconnect
598 	 * so IO could still be coming in.
599 	 */
600 	write_lock_bh(&tcp_sw_conn->sock->sk->sk_callback_lock);
601 	set_bit(ISCSI_SUSPEND_BIT, &conn->suspend_rx);
602 	write_unlock_bh(&tcp_sw_conn->sock->sk->sk_callback_lock);
603 
604 	sock->sk->sk_err = EIO;
605 	wake_up_interruptible(sk_sleep(sock->sk));
606 
607 	iscsi_conn_stop(cls_conn, flag);
608 	iscsi_sw_tcp_release_conn(conn);
609 }
610 
611 static int iscsi_sw_tcp_get_addr(struct iscsi_conn *conn, struct socket *sock,
612 				 char *buf, int *port,
613 				 int (*getname)(struct socket *,
614 						struct sockaddr *,
615 						int *addrlen))
616 {
617 	struct sockaddr_storage *addr;
618 	struct sockaddr_in6 *sin6;
619 	struct sockaddr_in *sin;
620 	int rc = 0, len;
621 
622 	addr = kmalloc(sizeof(*addr), GFP_KERNEL);
623 	if (!addr)
624 		return -ENOMEM;
625 
626 	if (getname(sock, (struct sockaddr *) addr, &len)) {
627 		rc = -ENODEV;
628 		goto free_addr;
629 	}
630 
631 	switch (addr->ss_family) {
632 	case AF_INET:
633 		sin = (struct sockaddr_in *)addr;
634 		spin_lock_bh(&conn->session->lock);
635 		sprintf(buf, "%pI4", &sin->sin_addr.s_addr);
636 		*port = be16_to_cpu(sin->sin_port);
637 		spin_unlock_bh(&conn->session->lock);
638 		break;
639 	case AF_INET6:
640 		sin6 = (struct sockaddr_in6 *)addr;
641 		spin_lock_bh(&conn->session->lock);
642 		sprintf(buf, "%pI6", &sin6->sin6_addr);
643 		*port = be16_to_cpu(sin6->sin6_port);
644 		spin_unlock_bh(&conn->session->lock);
645 		break;
646 	}
647 free_addr:
648 	kfree(addr);
649 	return rc;
650 }
651 
652 static int
653 iscsi_sw_tcp_conn_bind(struct iscsi_cls_session *cls_session,
654 		       struct iscsi_cls_conn *cls_conn, uint64_t transport_eph,
655 		       int is_leading)
656 {
657 	struct Scsi_Host *shost = iscsi_session_to_shost(cls_session);
658 	struct iscsi_host *ihost = shost_priv(shost);
659 	struct iscsi_conn *conn = cls_conn->dd_data;
660 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
661 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
662 	struct sock *sk;
663 	struct socket *sock;
664 	int err;
665 
666 	/* lookup for existing socket */
667 	sock = sockfd_lookup((int)transport_eph, &err);
668 	if (!sock) {
669 		iscsi_conn_printk(KERN_ERR, conn,
670 				  "sockfd_lookup failed %d\n", err);
671 		return -EEXIST;
672 	}
673 	/*
674 	 * copy these values now because if we drop the session
675 	 * userspace may still want to query the values since we will
676 	 * be using them for the reconnect
677 	 */
678 	err = iscsi_sw_tcp_get_addr(conn, sock, conn->portal_address,
679 				    &conn->portal_port, kernel_getpeername);
680 	if (err)
681 		goto free_socket;
682 
683 	err = iscsi_sw_tcp_get_addr(conn, sock, ihost->local_address,
684 				    &ihost->local_port, kernel_getsockname);
685 	if (err)
686 		goto free_socket;
687 
688 	err = iscsi_conn_bind(cls_session, cls_conn, is_leading);
689 	if (err)
690 		goto free_socket;
691 
692 	/* bind iSCSI connection and socket */
693 	tcp_sw_conn->sock = sock;
694 
695 	/* setup Socket parameters */
696 	sk = sock->sk;
697 	sk->sk_reuse = 1;
698 	sk->sk_sndtimeo = 15 * HZ; /* FIXME: make it configurable */
699 	sk->sk_allocation = GFP_ATOMIC;
700 
701 	iscsi_sw_tcp_conn_set_callbacks(conn);
702 	tcp_sw_conn->sendpage = tcp_sw_conn->sock->ops->sendpage;
703 	/*
704 	 * set receive state machine into initial state
705 	 */
706 	iscsi_tcp_hdr_recv_prep(tcp_conn);
707 	return 0;
708 
709 free_socket:
710 	sockfd_put(sock);
711 	return err;
712 }
713 
714 static int iscsi_sw_tcp_conn_set_param(struct iscsi_cls_conn *cls_conn,
715 				       enum iscsi_param param, char *buf,
716 				       int buflen)
717 {
718 	struct iscsi_conn *conn = cls_conn->dd_data;
719 	struct iscsi_session *session = conn->session;
720 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
721 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
722 	int value;
723 
724 	switch(param) {
725 	case ISCSI_PARAM_HDRDGST_EN:
726 		iscsi_set_param(cls_conn, param, buf, buflen);
727 		break;
728 	case ISCSI_PARAM_DATADGST_EN:
729 		iscsi_set_param(cls_conn, param, buf, buflen);
730 		tcp_sw_conn->sendpage = conn->datadgst_en ?
731 			sock_no_sendpage : tcp_sw_conn->sock->ops->sendpage;
732 		break;
733 	case ISCSI_PARAM_MAX_R2T:
734 		sscanf(buf, "%d", &value);
735 		if (value <= 0 || !is_power_of_2(value))
736 			return -EINVAL;
737 		if (session->max_r2t == value)
738 			break;
739 		iscsi_tcp_r2tpool_free(session);
740 		iscsi_set_param(cls_conn, param, buf, buflen);
741 		if (iscsi_tcp_r2tpool_alloc(session))
742 			return -ENOMEM;
743 		break;
744 	default:
745 		return iscsi_set_param(cls_conn, param, buf, buflen);
746 	}
747 
748 	return 0;
749 }
750 
751 static int iscsi_sw_tcp_conn_get_param(struct iscsi_cls_conn *cls_conn,
752 				       enum iscsi_param param, char *buf)
753 {
754 	struct iscsi_conn *conn = cls_conn->dd_data;
755 	int len;
756 
757 	switch(param) {
758 	case ISCSI_PARAM_CONN_PORT:
759 		spin_lock_bh(&conn->session->lock);
760 		len = sprintf(buf, "%hu\n", conn->portal_port);
761 		spin_unlock_bh(&conn->session->lock);
762 		break;
763 	case ISCSI_PARAM_CONN_ADDRESS:
764 		spin_lock_bh(&conn->session->lock);
765 		len = sprintf(buf, "%s\n", conn->portal_address);
766 		spin_unlock_bh(&conn->session->lock);
767 		break;
768 	default:
769 		return iscsi_conn_get_param(cls_conn, param, buf);
770 	}
771 
772 	return len;
773 }
774 
775 static void
776 iscsi_sw_tcp_conn_get_stats(struct iscsi_cls_conn *cls_conn,
777 			    struct iscsi_stats *stats)
778 {
779 	struct iscsi_conn *conn = cls_conn->dd_data;
780 	struct iscsi_tcp_conn *tcp_conn = conn->dd_data;
781 	struct iscsi_sw_tcp_conn *tcp_sw_conn = tcp_conn->dd_data;
782 
783 	stats->custom_length = 3;
784 	strcpy(stats->custom[0].desc, "tx_sendpage_failures");
785 	stats->custom[0].value = tcp_sw_conn->sendpage_failures_cnt;
786 	strcpy(stats->custom[1].desc, "rx_discontiguous_hdr");
787 	stats->custom[1].value = tcp_sw_conn->discontiguous_hdr_cnt;
788 	strcpy(stats->custom[2].desc, "eh_abort_cnt");
789 	stats->custom[2].value = conn->eh_abort_cnt;
790 
791 	iscsi_tcp_conn_get_stats(cls_conn, stats);
792 }
793 
794 static struct iscsi_cls_session *
795 iscsi_sw_tcp_session_create(struct iscsi_endpoint *ep, uint16_t cmds_max,
796 			    uint16_t qdepth, uint32_t initial_cmdsn)
797 {
798 	struct iscsi_cls_session *cls_session;
799 	struct iscsi_session *session;
800 	struct Scsi_Host *shost;
801 
802 	if (ep) {
803 		printk(KERN_ERR "iscsi_tcp: invalid ep %p.\n", ep);
804 		return NULL;
805 	}
806 
807 	shost = iscsi_host_alloc(&iscsi_sw_tcp_sht, 0, 1);
808 	if (!shost)
809 		return NULL;
810 	shost->transportt = iscsi_sw_tcp_scsi_transport;
811 	shost->cmd_per_lun = qdepth;
812 	shost->max_lun = iscsi_max_lun;
813 	shost->max_id = 0;
814 	shost->max_channel = 0;
815 	shost->max_cmd_len = SCSI_MAX_VARLEN_CDB_SIZE;
816 
817 	if (iscsi_host_add(shost, NULL))
818 		goto free_host;
819 
820 	cls_session = iscsi_session_setup(&iscsi_sw_tcp_transport, shost,
821 					  cmds_max, 0,
822 					  sizeof(struct iscsi_tcp_task) +
823 					  sizeof(struct iscsi_sw_tcp_hdrbuf),
824 					  initial_cmdsn, 0);
825 	if (!cls_session)
826 		goto remove_host;
827 	session = cls_session->dd_data;
828 
829 	shost->can_queue = session->scsi_cmds_max;
830 	if (iscsi_tcp_r2tpool_alloc(session))
831 		goto remove_session;
832 	return cls_session;
833 
834 remove_session:
835 	iscsi_session_teardown(cls_session);
836 remove_host:
837 	iscsi_host_remove(shost);
838 free_host:
839 	iscsi_host_free(shost);
840 	return NULL;
841 }
842 
843 static void iscsi_sw_tcp_session_destroy(struct iscsi_cls_session *cls_session)
844 {
845 	struct Scsi_Host *shost = iscsi_session_to_shost(cls_session);
846 
847 	iscsi_tcp_r2tpool_free(cls_session->dd_data);
848 	iscsi_session_teardown(cls_session);
849 
850 	iscsi_host_remove(shost);
851 	iscsi_host_free(shost);
852 }
853 
854 static int iscsi_sw_tcp_slave_alloc(struct scsi_device *sdev)
855 {
856 	set_bit(QUEUE_FLAG_BIDI, &sdev->request_queue->queue_flags);
857 	return 0;
858 }
859 
860 static int iscsi_sw_tcp_slave_configure(struct scsi_device *sdev)
861 {
862 	blk_queue_bounce_limit(sdev->request_queue, BLK_BOUNCE_ANY);
863 	blk_queue_dma_alignment(sdev->request_queue, 0);
864 	return 0;
865 }
866 
867 static struct scsi_host_template iscsi_sw_tcp_sht = {
868 	.module			= THIS_MODULE,
869 	.name			= "iSCSI Initiator over TCP/IP",
870 	.queuecommand           = iscsi_queuecommand,
871 	.change_queue_depth	= iscsi_change_queue_depth,
872 	.can_queue		= ISCSI_DEF_XMIT_CMDS_MAX - 1,
873 	.sg_tablesize		= 4096,
874 	.max_sectors		= 0xFFFF,
875 	.cmd_per_lun		= ISCSI_DEF_CMD_PER_LUN,
876 	.eh_abort_handler       = iscsi_eh_abort,
877 	.eh_device_reset_handler= iscsi_eh_device_reset,
878 	.eh_target_reset_handler = iscsi_eh_recover_target,
879 	.use_clustering         = DISABLE_CLUSTERING,
880 	.slave_alloc            = iscsi_sw_tcp_slave_alloc,
881 	.slave_configure        = iscsi_sw_tcp_slave_configure,
882 	.target_alloc		= iscsi_target_alloc,
883 	.proc_name		= "iscsi_tcp",
884 	.this_id		= -1,
885 };
886 
887 static struct iscsi_transport iscsi_sw_tcp_transport = {
888 	.owner			= THIS_MODULE,
889 	.name			= "tcp",
890 	.caps			= CAP_RECOVERY_L0 | CAP_MULTI_R2T | CAP_HDRDGST
891 				  | CAP_DATADGST,
892 	.param_mask		= ISCSI_MAX_RECV_DLENGTH |
893 				  ISCSI_MAX_XMIT_DLENGTH |
894 				  ISCSI_HDRDGST_EN |
895 				  ISCSI_DATADGST_EN |
896 				  ISCSI_INITIAL_R2T_EN |
897 				  ISCSI_MAX_R2T |
898 				  ISCSI_IMM_DATA_EN |
899 				  ISCSI_FIRST_BURST |
900 				  ISCSI_MAX_BURST |
901 				  ISCSI_PDU_INORDER_EN |
902 				  ISCSI_DATASEQ_INORDER_EN |
903 				  ISCSI_ERL |
904 				  ISCSI_CONN_PORT |
905 				  ISCSI_CONN_ADDRESS |
906 				  ISCSI_EXP_STATSN |
907 				  ISCSI_PERSISTENT_PORT |
908 				  ISCSI_PERSISTENT_ADDRESS |
909 				  ISCSI_TARGET_NAME | ISCSI_TPGT |
910 				  ISCSI_USERNAME | ISCSI_PASSWORD |
911 				  ISCSI_USERNAME_IN | ISCSI_PASSWORD_IN |
912 				  ISCSI_FAST_ABORT | ISCSI_ABORT_TMO |
913 				  ISCSI_LU_RESET_TMO | ISCSI_TGT_RESET_TMO |
914 				  ISCSI_PING_TMO | ISCSI_RECV_TMO |
915 				  ISCSI_IFACE_NAME | ISCSI_INITIATOR_NAME,
916 	.host_param_mask	= ISCSI_HOST_HWADDRESS | ISCSI_HOST_IPADDRESS |
917 				  ISCSI_HOST_INITIATOR_NAME |
918 				  ISCSI_HOST_NETDEV_NAME,
919 	/* session management */
920 	.create_session		= iscsi_sw_tcp_session_create,
921 	.destroy_session	= iscsi_sw_tcp_session_destroy,
922 	/* connection management */
923 	.create_conn		= iscsi_sw_tcp_conn_create,
924 	.bind_conn		= iscsi_sw_tcp_conn_bind,
925 	.destroy_conn		= iscsi_sw_tcp_conn_destroy,
926 	.set_param		= iscsi_sw_tcp_conn_set_param,
927 	.get_conn_param		= iscsi_sw_tcp_conn_get_param,
928 	.get_session_param	= iscsi_session_get_param,
929 	.start_conn		= iscsi_conn_start,
930 	.stop_conn		= iscsi_sw_tcp_conn_stop,
931 	/* iscsi host params */
932 	.get_host_param		= iscsi_host_get_param,
933 	.set_host_param		= iscsi_host_set_param,
934 	/* IO */
935 	.send_pdu		= iscsi_conn_send_pdu,
936 	.get_stats		= iscsi_sw_tcp_conn_get_stats,
937 	/* iscsi task/cmd helpers */
938 	.init_task		= iscsi_tcp_task_init,
939 	.xmit_task		= iscsi_tcp_task_xmit,
940 	.cleanup_task		= iscsi_tcp_cleanup_task,
941 	/* low level pdu helpers */
942 	.xmit_pdu		= iscsi_sw_tcp_pdu_xmit,
943 	.init_pdu		= iscsi_sw_tcp_pdu_init,
944 	.alloc_pdu		= iscsi_sw_tcp_pdu_alloc,
945 	/* recovery */
946 	.session_recovery_timedout = iscsi_session_recovery_timedout,
947 };
948 
949 static int __init iscsi_sw_tcp_init(void)
950 {
951 	if (iscsi_max_lun < 1) {
952 		printk(KERN_ERR "iscsi_tcp: Invalid max_lun value of %u\n",
953 		       iscsi_max_lun);
954 		return -EINVAL;
955 	}
956 
957 	iscsi_sw_tcp_scsi_transport = iscsi_register_transport(
958 						&iscsi_sw_tcp_transport);
959 	if (!iscsi_sw_tcp_scsi_transport)
960 		return -ENODEV;
961 
962 	return 0;
963 }
964 
965 static void __exit iscsi_sw_tcp_exit(void)
966 {
967 	iscsi_unregister_transport(&iscsi_sw_tcp_transport);
968 }
969 
970 module_init(iscsi_sw_tcp_init);
971 module_exit(iscsi_sw_tcp_exit);
972