156205a8152350f066b23e46c79cc0b07293ceae
[platform/adaptation/renesas_rcar/renesas_kernel.git] / drivers / infiniband / ulp / isert / ib_isert.c
1 /*******************************************************************************
2  * This file contains iSCSI extentions for RDMA (iSER) Verbs
3  *
4  * (c) Copyright 2013 Datera, Inc.
5  *
6  * Nicholas A. Bellinger <nab@linux-iscsi.org>
7  *
8  * This program is free software; you can redistribute it and/or modify
9  * it under the terms of the GNU General Public License as published by
10  * the Free Software Foundation; either version 2 of the License, or
11  * (at your option) any later version.
12  *
13  * This program is distributed in the hope that it will be useful,
14  * but WITHOUT ANY WARRANTY; without even the implied warranty of
15  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
16  * GNU General Public License for more details.
17  ****************************************************************************/
18
19 #include <linux/string.h>
20 #include <linux/module.h>
21 #include <linux/scatterlist.h>
22 #include <linux/socket.h>
23 #include <linux/in.h>
24 #include <linux/in6.h>
25 #include <linux/llist.h>
26 #include <rdma/ib_verbs.h>
27 #include <rdma/rdma_cm.h>
28 #include <target/target_core_base.h>
29 #include <target/target_core_fabric.h>
30 #include <target/iscsi/iscsi_transport.h>
31 #include <linux/semaphore.h>
32
33 #include "isert_proto.h"
34 #include "ib_isert.h"
35
36 #define ISERT_MAX_CONN          8
37 #define ISER_MAX_RX_CQ_LEN      (ISERT_QP_MAX_RECV_DTOS * ISERT_MAX_CONN)
38 #define ISER_MAX_TX_CQ_LEN      (ISERT_QP_MAX_REQ_DTOS  * ISERT_MAX_CONN)
39
40 static DEFINE_MUTEX(device_list_mutex);
41 static LIST_HEAD(device_list);
42 static struct workqueue_struct *isert_rx_wq;
43 static struct workqueue_struct *isert_comp_wq;
44
45 static void
46 isert_unmap_cmd(struct isert_cmd *isert_cmd, struct isert_conn *isert_conn);
47 static int
48 isert_map_rdma(struct iscsi_conn *conn, struct iscsi_cmd *cmd,
49                struct isert_rdma_wr *wr);
50 static void
51 isert_unreg_rdma(struct isert_cmd *isert_cmd, struct isert_conn *isert_conn);
52 static int
53 isert_reg_rdma(struct iscsi_conn *conn, struct iscsi_cmd *cmd,
54                struct isert_rdma_wr *wr);
55
56 static void
57 isert_qp_event_callback(struct ib_event *e, void *context)
58 {
59         struct isert_conn *isert_conn = (struct isert_conn *)context;
60
61         pr_err("isert_qp_event_callback event: %d\n", e->event);
62         switch (e->event) {
63         case IB_EVENT_COMM_EST:
64                 rdma_notify(isert_conn->conn_cm_id, IB_EVENT_COMM_EST);
65                 break;
66         case IB_EVENT_QP_LAST_WQE_REACHED:
67                 pr_warn("Reached TX IB_EVENT_QP_LAST_WQE_REACHED:\n");
68                 break;
69         default:
70                 break;
71         }
72 }
73
74 static int
75 isert_query_device(struct ib_device *ib_dev, struct ib_device_attr *devattr)
76 {
77         int ret;
78
79         ret = ib_query_device(ib_dev, devattr);
80         if (ret) {
81                 pr_err("ib_query_device() failed: %d\n", ret);
82                 return ret;
83         }
84         pr_debug("devattr->max_sge: %d\n", devattr->max_sge);
85         pr_debug("devattr->max_sge_rd: %d\n", devattr->max_sge_rd);
86
87         return 0;
88 }
89
90 static int
91 isert_conn_setup_qp(struct isert_conn *isert_conn, struct rdma_cm_id *cma_id)
92 {
93         struct isert_device *device = isert_conn->conn_device;
94         struct ib_qp_init_attr attr;
95         int ret, index, min_index = 0;
96
97         mutex_lock(&device_list_mutex);
98         for (index = 0; index < device->cqs_used; index++)
99                 if (device->cq_active_qps[index] <
100                     device->cq_active_qps[min_index])
101                         min_index = index;
102         device->cq_active_qps[min_index]++;
103         pr_debug("isert_conn_setup_qp: Using min_index: %d\n", min_index);
104         mutex_unlock(&device_list_mutex);
105
106         memset(&attr, 0, sizeof(struct ib_qp_init_attr));
107         attr.event_handler = isert_qp_event_callback;
108         attr.qp_context = isert_conn;
109         attr.send_cq = device->dev_tx_cq[min_index];
110         attr.recv_cq = device->dev_rx_cq[min_index];
111         attr.cap.max_send_wr = ISERT_QP_MAX_REQ_DTOS;
112         attr.cap.max_recv_wr = ISERT_QP_MAX_RECV_DTOS;
113         /*
114          * FIXME: Use devattr.max_sge - 2 for max_send_sge as
115          * work-around for RDMA_READ..
116          */
117         attr.cap.max_send_sge = device->dev_attr.max_sge - 2;
118         isert_conn->max_sge = attr.cap.max_send_sge;
119
120         attr.cap.max_recv_sge = 1;
121         attr.sq_sig_type = IB_SIGNAL_REQ_WR;
122         attr.qp_type = IB_QPT_RC;
123
124         pr_debug("isert_conn_setup_qp cma_id->device: %p\n",
125                  cma_id->device);
126         pr_debug("isert_conn_setup_qp conn_pd->device: %p\n",
127                  isert_conn->conn_pd->device);
128
129         ret = rdma_create_qp(cma_id, isert_conn->conn_pd, &attr);
130         if (ret) {
131                 pr_err("rdma_create_qp failed for cma_id %d\n", ret);
132                 return ret;
133         }
134         isert_conn->conn_qp = cma_id->qp;
135         pr_debug("rdma_create_qp() returned success >>>>>>>>>>>>>>>>>>>>>>>>>.\n");
136
137         return 0;
138 }
139
140 static void
141 isert_cq_event_callback(struct ib_event *e, void *context)
142 {
143         pr_debug("isert_cq_event_callback event: %d\n", e->event);
144 }
145
146 static int
147 isert_alloc_rx_descriptors(struct isert_conn *isert_conn)
148 {
149         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
150         struct iser_rx_desc *rx_desc;
151         struct ib_sge *rx_sg;
152         u64 dma_addr;
153         int i, j;
154
155         isert_conn->conn_rx_descs = kzalloc(ISERT_QP_MAX_RECV_DTOS *
156                                 sizeof(struct iser_rx_desc), GFP_KERNEL);
157         if (!isert_conn->conn_rx_descs)
158                 goto fail;
159
160         rx_desc = isert_conn->conn_rx_descs;
161
162         for (i = 0; i < ISERT_QP_MAX_RECV_DTOS; i++, rx_desc++)  {
163                 dma_addr = ib_dma_map_single(ib_dev, (void *)rx_desc,
164                                         ISER_RX_PAYLOAD_SIZE, DMA_FROM_DEVICE);
165                 if (ib_dma_mapping_error(ib_dev, dma_addr))
166                         goto dma_map_fail;
167
168                 rx_desc->dma_addr = dma_addr;
169
170                 rx_sg = &rx_desc->rx_sg;
171                 rx_sg->addr = rx_desc->dma_addr;
172                 rx_sg->length = ISER_RX_PAYLOAD_SIZE;
173                 rx_sg->lkey = isert_conn->conn_mr->lkey;
174         }
175
176         isert_conn->conn_rx_desc_head = 0;
177         return 0;
178
179 dma_map_fail:
180         rx_desc = isert_conn->conn_rx_descs;
181         for (j = 0; j < i; j++, rx_desc++) {
182                 ib_dma_unmap_single(ib_dev, rx_desc->dma_addr,
183                                     ISER_RX_PAYLOAD_SIZE, DMA_FROM_DEVICE);
184         }
185         kfree(isert_conn->conn_rx_descs);
186         isert_conn->conn_rx_descs = NULL;
187 fail:
188         return -ENOMEM;
189 }
190
191 static void
192 isert_free_rx_descriptors(struct isert_conn *isert_conn)
193 {
194         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
195         struct iser_rx_desc *rx_desc;
196         int i;
197
198         if (!isert_conn->conn_rx_descs)
199                 return;
200
201         rx_desc = isert_conn->conn_rx_descs;
202         for (i = 0; i < ISERT_QP_MAX_RECV_DTOS; i++, rx_desc++)  {
203                 ib_dma_unmap_single(ib_dev, rx_desc->dma_addr,
204                                     ISER_RX_PAYLOAD_SIZE, DMA_FROM_DEVICE);
205         }
206
207         kfree(isert_conn->conn_rx_descs);
208         isert_conn->conn_rx_descs = NULL;
209 }
210
211 static void isert_cq_tx_work(struct work_struct *);
212 static void isert_cq_tx_callback(struct ib_cq *, void *);
213 static void isert_cq_rx_work(struct work_struct *);
214 static void isert_cq_rx_callback(struct ib_cq *, void *);
215
216 static int
217 isert_create_device_ib_res(struct isert_device *device)
218 {
219         struct ib_device *ib_dev = device->ib_device;
220         struct isert_cq_desc *cq_desc;
221         struct ib_device_attr *dev_attr;
222         int ret = 0, i, j;
223
224         dev_attr = &device->dev_attr;
225         ret = isert_query_device(ib_dev, dev_attr);
226         if (ret)
227                 return ret;
228
229         /* asign function handlers */
230         if (dev_attr->device_cap_flags & IB_DEVICE_MEM_MGT_EXTENSIONS) {
231                 device->use_fastreg = 1;
232                 device->reg_rdma_mem = isert_reg_rdma;
233                 device->unreg_rdma_mem = isert_unreg_rdma;
234         } else {
235                 device->use_fastreg = 0;
236                 device->reg_rdma_mem = isert_map_rdma;
237                 device->unreg_rdma_mem = isert_unmap_cmd;
238         }
239
240         device->cqs_used = min_t(int, num_online_cpus(),
241                                  device->ib_device->num_comp_vectors);
242         device->cqs_used = min(ISERT_MAX_CQ, device->cqs_used);
243         pr_debug("Using %d CQs, device %s supports %d vectors support "
244                  "Fast registration %d\n",
245                  device->cqs_used, device->ib_device->name,
246                  device->ib_device->num_comp_vectors, device->use_fastreg);
247         device->cq_desc = kzalloc(sizeof(struct isert_cq_desc) *
248                                 device->cqs_used, GFP_KERNEL);
249         if (!device->cq_desc) {
250                 pr_err("Unable to allocate device->cq_desc\n");
251                 return -ENOMEM;
252         }
253         cq_desc = device->cq_desc;
254
255         for (i = 0; i < device->cqs_used; i++) {
256                 cq_desc[i].device = device;
257                 cq_desc[i].cq_index = i;
258
259                 INIT_WORK(&cq_desc[i].cq_rx_work, isert_cq_rx_work);
260                 device->dev_rx_cq[i] = ib_create_cq(device->ib_device,
261                                                 isert_cq_rx_callback,
262                                                 isert_cq_event_callback,
263                                                 (void *)&cq_desc[i],
264                                                 ISER_MAX_RX_CQ_LEN, i);
265                 if (IS_ERR(device->dev_rx_cq[i])) {
266                         ret = PTR_ERR(device->dev_rx_cq[i]);
267                         device->dev_rx_cq[i] = NULL;
268                         goto out_cq;
269                 }
270
271                 INIT_WORK(&cq_desc[i].cq_tx_work, isert_cq_tx_work);
272                 device->dev_tx_cq[i] = ib_create_cq(device->ib_device,
273                                                 isert_cq_tx_callback,
274                                                 isert_cq_event_callback,
275                                                 (void *)&cq_desc[i],
276                                                 ISER_MAX_TX_CQ_LEN, i);
277                 if (IS_ERR(device->dev_tx_cq[i])) {
278                         ret = PTR_ERR(device->dev_tx_cq[i]);
279                         device->dev_tx_cq[i] = NULL;
280                         goto out_cq;
281                 }
282
283                 ret = ib_req_notify_cq(device->dev_rx_cq[i], IB_CQ_NEXT_COMP);
284                 if (ret)
285                         goto out_cq;
286
287                 ret = ib_req_notify_cq(device->dev_tx_cq[i], IB_CQ_NEXT_COMP);
288                 if (ret)
289                         goto out_cq;
290         }
291
292         return 0;
293
294 out_cq:
295         for (j = 0; j < i; j++) {
296                 cq_desc = &device->cq_desc[j];
297
298                 if (device->dev_rx_cq[j]) {
299                         cancel_work_sync(&cq_desc->cq_rx_work);
300                         ib_destroy_cq(device->dev_rx_cq[j]);
301                 }
302                 if (device->dev_tx_cq[j]) {
303                         cancel_work_sync(&cq_desc->cq_tx_work);
304                         ib_destroy_cq(device->dev_tx_cq[j]);
305                 }
306         }
307         kfree(device->cq_desc);
308
309         return ret;
310 }
311
312 static void
313 isert_free_device_ib_res(struct isert_device *device)
314 {
315         struct isert_cq_desc *cq_desc;
316         int i;
317
318         for (i = 0; i < device->cqs_used; i++) {
319                 cq_desc = &device->cq_desc[i];
320
321                 cancel_work_sync(&cq_desc->cq_rx_work);
322                 cancel_work_sync(&cq_desc->cq_tx_work);
323                 ib_destroy_cq(device->dev_rx_cq[i]);
324                 ib_destroy_cq(device->dev_tx_cq[i]);
325                 device->dev_rx_cq[i] = NULL;
326                 device->dev_tx_cq[i] = NULL;
327         }
328
329         kfree(device->cq_desc);
330 }
331
332 static void
333 isert_device_try_release(struct isert_device *device)
334 {
335         mutex_lock(&device_list_mutex);
336         device->refcount--;
337         if (!device->refcount) {
338                 isert_free_device_ib_res(device);
339                 list_del(&device->dev_node);
340                 kfree(device);
341         }
342         mutex_unlock(&device_list_mutex);
343 }
344
345 static struct isert_device *
346 isert_device_find_by_ib_dev(struct rdma_cm_id *cma_id)
347 {
348         struct isert_device *device;
349         int ret;
350
351         mutex_lock(&device_list_mutex);
352         list_for_each_entry(device, &device_list, dev_node) {
353                 if (device->ib_device->node_guid == cma_id->device->node_guid) {
354                         device->refcount++;
355                         mutex_unlock(&device_list_mutex);
356                         return device;
357                 }
358         }
359
360         device = kzalloc(sizeof(struct isert_device), GFP_KERNEL);
361         if (!device) {
362                 mutex_unlock(&device_list_mutex);
363                 return ERR_PTR(-ENOMEM);
364         }
365
366         INIT_LIST_HEAD(&device->dev_node);
367
368         device->ib_device = cma_id->device;
369         ret = isert_create_device_ib_res(device);
370         if (ret) {
371                 kfree(device);
372                 mutex_unlock(&device_list_mutex);
373                 return ERR_PTR(ret);
374         }
375
376         device->refcount++;
377         list_add_tail(&device->dev_node, &device_list);
378         mutex_unlock(&device_list_mutex);
379
380         return device;
381 }
382
383 static void
384 isert_conn_free_fastreg_pool(struct isert_conn *isert_conn)
385 {
386         struct fast_reg_descriptor *fr_desc, *tmp;
387         int i = 0;
388
389         if (list_empty(&isert_conn->conn_fr_pool))
390                 return;
391
392         pr_debug("Freeing conn %p fastreg pool", isert_conn);
393
394         list_for_each_entry_safe(fr_desc, tmp,
395                                  &isert_conn->conn_fr_pool, list) {
396                 list_del(&fr_desc->list);
397                 ib_free_fast_reg_page_list(fr_desc->data_frpl);
398                 ib_dereg_mr(fr_desc->data_mr);
399                 kfree(fr_desc);
400                 ++i;
401         }
402
403         if (i < isert_conn->conn_fr_pool_size)
404                 pr_warn("Pool still has %d regions registered\n",
405                         isert_conn->conn_fr_pool_size - i);
406 }
407
408 static int
409 isert_create_fr_desc(struct ib_device *ib_device, struct ib_pd *pd,
410                      struct fast_reg_descriptor *fr_desc)
411 {
412         fr_desc->data_frpl = ib_alloc_fast_reg_page_list(ib_device,
413                                                          ISCSI_ISER_SG_TABLESIZE);
414         if (IS_ERR(fr_desc->data_frpl)) {
415                 pr_err("Failed to allocate data frpl err=%ld\n",
416                        PTR_ERR(fr_desc->data_frpl));
417                 return PTR_ERR(fr_desc->data_frpl);
418         }
419
420         fr_desc->data_mr = ib_alloc_fast_reg_mr(pd, ISCSI_ISER_SG_TABLESIZE);
421         if (IS_ERR(fr_desc->data_mr)) {
422                 pr_err("Failed to allocate data frmr err=%ld\n",
423                        PTR_ERR(fr_desc->data_mr));
424                 ib_free_fast_reg_page_list(fr_desc->data_frpl);
425                 return PTR_ERR(fr_desc->data_mr);
426         }
427         pr_debug("Create fr_desc %p page_list %p\n",
428                  fr_desc, fr_desc->data_frpl->page_list);
429
430         fr_desc->valid = true;
431
432         return 0;
433 }
434
435 static int
436 isert_conn_create_fastreg_pool(struct isert_conn *isert_conn)
437 {
438         struct fast_reg_descriptor *fr_desc;
439         struct isert_device *device = isert_conn->conn_device;
440         struct se_session *se_sess = isert_conn->conn->sess->se_sess;
441         struct se_node_acl *se_nacl = se_sess->se_node_acl;
442         int i, ret, tag_num;
443         /*
444          * Setup the number of FRMRs based upon the number of tags
445          * available to session in iscsi_target_locate_portal().
446          */
447         tag_num = max_t(u32, ISCSIT_MIN_TAGS, se_nacl->queue_depth);
448         tag_num = (tag_num * 2) + ISCSIT_EXTRA_TAGS;
449
450         isert_conn->conn_fr_pool_size = 0;
451         for (i = 0; i < tag_num; i++) {
452                 fr_desc = kzalloc(sizeof(*fr_desc), GFP_KERNEL);
453                 if (!fr_desc) {
454                         pr_err("Failed to allocate fast_reg descriptor\n");
455                         ret = -ENOMEM;
456                         goto err;
457                 }
458
459                 ret = isert_create_fr_desc(device->ib_device,
460                                            isert_conn->conn_pd, fr_desc);
461                 if (ret) {
462                         pr_err("Failed to create fastreg descriptor err=%d\n",
463                                ret);
464                         kfree(fr_desc);
465                         goto err;
466                 }
467
468                 list_add_tail(&fr_desc->list, &isert_conn->conn_fr_pool);
469                 isert_conn->conn_fr_pool_size++;
470         }
471
472         pr_debug("Creating conn %p fastreg pool size=%d",
473                  isert_conn, isert_conn->conn_fr_pool_size);
474
475         return 0;
476
477 err:
478         isert_conn_free_fastreg_pool(isert_conn);
479         return ret;
480 }
481
482 static int
483 isert_connect_request(struct rdma_cm_id *cma_id, struct rdma_cm_event *event)
484 {
485         struct iscsi_np *np = cma_id->context;
486         struct isert_np *isert_np = np->np_context;
487         struct isert_conn *isert_conn;
488         struct isert_device *device;
489         struct ib_device *ib_dev = cma_id->device;
490         int ret = 0;
491
492         spin_lock_bh(&np->np_thread_lock);
493         if (!np->enabled) {
494                 spin_unlock_bh(&np->np_thread_lock);
495                 pr_debug("iscsi_np is not enabled, reject connect request\n");
496                 return rdma_reject(cma_id, NULL, 0);
497         }
498         spin_unlock_bh(&np->np_thread_lock);
499
500         pr_debug("Entering isert_connect_request cma_id: %p, context: %p\n",
501                  cma_id, cma_id->context);
502
503         isert_conn = kzalloc(sizeof(struct isert_conn), GFP_KERNEL);
504         if (!isert_conn) {
505                 pr_err("Unable to allocate isert_conn\n");
506                 return -ENOMEM;
507         }
508         isert_conn->state = ISER_CONN_INIT;
509         INIT_LIST_HEAD(&isert_conn->conn_accept_node);
510         init_completion(&isert_conn->conn_login_comp);
511         init_completion(&isert_conn->conn_wait);
512         init_completion(&isert_conn->conn_wait_comp_err);
513         kref_init(&isert_conn->conn_kref);
514         kref_get(&isert_conn->conn_kref);
515         mutex_init(&isert_conn->conn_mutex);
516         spin_lock_init(&isert_conn->conn_lock);
517         INIT_LIST_HEAD(&isert_conn->conn_fr_pool);
518
519         cma_id->context = isert_conn;
520         isert_conn->conn_cm_id = cma_id;
521         isert_conn->responder_resources = event->param.conn.responder_resources;
522         isert_conn->initiator_depth = event->param.conn.initiator_depth;
523         pr_debug("Using responder_resources: %u initiator_depth: %u\n",
524                  isert_conn->responder_resources, isert_conn->initiator_depth);
525
526         isert_conn->login_buf = kzalloc(ISCSI_DEF_MAX_RECV_SEG_LEN +
527                                         ISER_RX_LOGIN_SIZE, GFP_KERNEL);
528         if (!isert_conn->login_buf) {
529                 pr_err("Unable to allocate isert_conn->login_buf\n");
530                 ret = -ENOMEM;
531                 goto out;
532         }
533
534         isert_conn->login_req_buf = isert_conn->login_buf;
535         isert_conn->login_rsp_buf = isert_conn->login_buf +
536                                     ISCSI_DEF_MAX_RECV_SEG_LEN;
537         pr_debug("Set login_buf: %p login_req_buf: %p login_rsp_buf: %p\n",
538                  isert_conn->login_buf, isert_conn->login_req_buf,
539                  isert_conn->login_rsp_buf);
540
541         isert_conn->login_req_dma = ib_dma_map_single(ib_dev,
542                                 (void *)isert_conn->login_req_buf,
543                                 ISCSI_DEF_MAX_RECV_SEG_LEN, DMA_FROM_DEVICE);
544
545         ret = ib_dma_mapping_error(ib_dev, isert_conn->login_req_dma);
546         if (ret) {
547                 pr_err("ib_dma_mapping_error failed for login_req_dma: %d\n",
548                        ret);
549                 isert_conn->login_req_dma = 0;
550                 goto out_login_buf;
551         }
552
553         isert_conn->login_rsp_dma = ib_dma_map_single(ib_dev,
554                                         (void *)isert_conn->login_rsp_buf,
555                                         ISER_RX_LOGIN_SIZE, DMA_TO_DEVICE);
556
557         ret = ib_dma_mapping_error(ib_dev, isert_conn->login_rsp_dma);
558         if (ret) {
559                 pr_err("ib_dma_mapping_error failed for login_rsp_dma: %d\n",
560                        ret);
561                 isert_conn->login_rsp_dma = 0;
562                 goto out_req_dma_map;
563         }
564
565         device = isert_device_find_by_ib_dev(cma_id);
566         if (IS_ERR(device)) {
567                 ret = PTR_ERR(device);
568                 goto out_rsp_dma_map;
569         }
570
571         isert_conn->conn_device = device;
572         isert_conn->conn_pd = ib_alloc_pd(isert_conn->conn_device->ib_device);
573         if (IS_ERR(isert_conn->conn_pd)) {
574                 ret = PTR_ERR(isert_conn->conn_pd);
575                 pr_err("ib_alloc_pd failed for conn %p: ret=%d\n",
576                        isert_conn, ret);
577                 goto out_pd;
578         }
579
580         isert_conn->conn_mr = ib_get_dma_mr(isert_conn->conn_pd,
581                                            IB_ACCESS_LOCAL_WRITE);
582         if (IS_ERR(isert_conn->conn_mr)) {
583                 ret = PTR_ERR(isert_conn->conn_mr);
584                 pr_err("ib_get_dma_mr failed for conn %p: ret=%d\n",
585                        isert_conn, ret);
586                 goto out_mr;
587         }
588
589         ret = isert_conn_setup_qp(isert_conn, cma_id);
590         if (ret)
591                 goto out_conn_dev;
592
593         mutex_lock(&isert_np->np_accept_mutex);
594         list_add_tail(&isert_conn->conn_accept_node, &isert_np->np_accept_list);
595         mutex_unlock(&isert_np->np_accept_mutex);
596
597         pr_debug("isert_connect_request() up np_sem np: %p\n", np);
598         up(&isert_np->np_sem);
599         return 0;
600
601 out_conn_dev:
602         ib_dereg_mr(isert_conn->conn_mr);
603 out_mr:
604         ib_dealloc_pd(isert_conn->conn_pd);
605 out_pd:
606         isert_device_try_release(device);
607 out_rsp_dma_map:
608         ib_dma_unmap_single(ib_dev, isert_conn->login_rsp_dma,
609                             ISER_RX_LOGIN_SIZE, DMA_TO_DEVICE);
610 out_req_dma_map:
611         ib_dma_unmap_single(ib_dev, isert_conn->login_req_dma,
612                             ISCSI_DEF_MAX_RECV_SEG_LEN, DMA_FROM_DEVICE);
613 out_login_buf:
614         kfree(isert_conn->login_buf);
615 out:
616         kfree(isert_conn);
617         return ret;
618 }
619
620 static void
621 isert_connect_release(struct isert_conn *isert_conn)
622 {
623         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
624         struct isert_device *device = isert_conn->conn_device;
625         int cq_index;
626
627         pr_debug("Entering isert_connect_release(): >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>\n");
628
629         if (device && device->use_fastreg)
630                 isert_conn_free_fastreg_pool(isert_conn);
631
632         if (isert_conn->conn_qp) {
633                 cq_index = ((struct isert_cq_desc *)
634                         isert_conn->conn_qp->recv_cq->cq_context)->cq_index;
635                 pr_debug("isert_connect_release: cq_index: %d\n", cq_index);
636                 isert_conn->conn_device->cq_active_qps[cq_index]--;
637
638                 rdma_destroy_qp(isert_conn->conn_cm_id);
639         }
640
641         isert_free_rx_descriptors(isert_conn);
642         rdma_destroy_id(isert_conn->conn_cm_id);
643
644         ib_dereg_mr(isert_conn->conn_mr);
645         ib_dealloc_pd(isert_conn->conn_pd);
646
647         if (isert_conn->login_buf) {
648                 ib_dma_unmap_single(ib_dev, isert_conn->login_rsp_dma,
649                                     ISER_RX_LOGIN_SIZE, DMA_TO_DEVICE);
650                 ib_dma_unmap_single(ib_dev, isert_conn->login_req_dma,
651                                     ISCSI_DEF_MAX_RECV_SEG_LEN,
652                                     DMA_FROM_DEVICE);
653                 kfree(isert_conn->login_buf);
654         }
655         kfree(isert_conn);
656
657         if (device)
658                 isert_device_try_release(device);
659
660         pr_debug("Leaving isert_connect_release >>>>>>>>>>>>\n");
661 }
662
663 static void
664 isert_connected_handler(struct rdma_cm_id *cma_id)
665 {
666         return;
667 }
668
669 static void
670 isert_release_conn_kref(struct kref *kref)
671 {
672         struct isert_conn *isert_conn = container_of(kref,
673                                 struct isert_conn, conn_kref);
674
675         pr_debug("Calling isert_connect_release for final kref %s/%d\n",
676                  current->comm, current->pid);
677
678         isert_connect_release(isert_conn);
679 }
680
681 static void
682 isert_put_conn(struct isert_conn *isert_conn)
683 {
684         kref_put(&isert_conn->conn_kref, isert_release_conn_kref);
685 }
686
687 static void
688 isert_disconnect_work(struct work_struct *work)
689 {
690         struct isert_conn *isert_conn = container_of(work,
691                                 struct isert_conn, conn_logout_work);
692
693         pr_debug("isert_disconnect_work(): >>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>>\n");
694         mutex_lock(&isert_conn->conn_mutex);
695         if (isert_conn->state == ISER_CONN_UP)
696                 isert_conn->state = ISER_CONN_TERMINATING;
697
698         if (isert_conn->post_recv_buf_count == 0 &&
699             atomic_read(&isert_conn->post_send_buf_count) == 0) {
700                 mutex_unlock(&isert_conn->conn_mutex);
701                 goto wake_up;
702         }
703         if (!isert_conn->conn_cm_id) {
704                 mutex_unlock(&isert_conn->conn_mutex);
705                 isert_put_conn(isert_conn);
706                 return;
707         }
708
709         if (isert_conn->disconnect) {
710                 /* Send DREQ/DREP towards our initiator */
711                 rdma_disconnect(isert_conn->conn_cm_id);
712         }
713
714         mutex_unlock(&isert_conn->conn_mutex);
715
716 wake_up:
717         complete(&isert_conn->conn_wait);
718         isert_put_conn(isert_conn);
719 }
720
721 static void
722 isert_disconnected_handler(struct rdma_cm_id *cma_id, bool disconnect)
723 {
724         struct isert_conn *isert_conn = (struct isert_conn *)cma_id->context;
725
726         isert_conn->disconnect = disconnect;
727         INIT_WORK(&isert_conn->conn_logout_work, isert_disconnect_work);
728         schedule_work(&isert_conn->conn_logout_work);
729 }
730
731 static int
732 isert_cma_handler(struct rdma_cm_id *cma_id, struct rdma_cm_event *event)
733 {
734         int ret = 0;
735         bool disconnect = false;
736
737         pr_debug("isert_cma_handler: event %d status %d conn %p id %p\n",
738                  event->event, event->status, cma_id->context, cma_id);
739
740         switch (event->event) {
741         case RDMA_CM_EVENT_CONNECT_REQUEST:
742                 ret = isert_connect_request(cma_id, event);
743                 break;
744         case RDMA_CM_EVENT_ESTABLISHED:
745                 isert_connected_handler(cma_id);
746                 break;
747         case RDMA_CM_EVENT_ADDR_CHANGE:    /* FALLTHRU */
748         case RDMA_CM_EVENT_DISCONNECTED:   /* FALLTHRU */
749         case RDMA_CM_EVENT_DEVICE_REMOVAL: /* FALLTHRU */
750                 disconnect = true;
751         case RDMA_CM_EVENT_TIMEWAIT_EXIT:  /* FALLTHRU */
752                 isert_disconnected_handler(cma_id, disconnect);
753                 break;
754         case RDMA_CM_EVENT_CONNECT_ERROR:
755         default:
756                 pr_err("Unhandled RDMA CMA event: %d\n", event->event);
757                 break;
758         }
759
760         if (ret != 0) {
761                 pr_err("isert_cma_handler failed RDMA_CM_EVENT: 0x%08x %d\n",
762                        event->event, ret);
763                 dump_stack();
764         }
765
766         return ret;
767 }
768
769 static int
770 isert_post_recv(struct isert_conn *isert_conn, u32 count)
771 {
772         struct ib_recv_wr *rx_wr, *rx_wr_failed;
773         int i, ret;
774         unsigned int rx_head = isert_conn->conn_rx_desc_head;
775         struct iser_rx_desc *rx_desc;
776
777         for (rx_wr = isert_conn->conn_rx_wr, i = 0; i < count; i++, rx_wr++) {
778                 rx_desc         = &isert_conn->conn_rx_descs[rx_head];
779                 rx_wr->wr_id    = (unsigned long)rx_desc;
780                 rx_wr->sg_list  = &rx_desc->rx_sg;
781                 rx_wr->num_sge  = 1;
782                 rx_wr->next     = rx_wr + 1;
783                 rx_head = (rx_head + 1) & (ISERT_QP_MAX_RECV_DTOS - 1);
784         }
785
786         rx_wr--;
787         rx_wr->next = NULL; /* mark end of work requests list */
788
789         isert_conn->post_recv_buf_count += count;
790         ret = ib_post_recv(isert_conn->conn_qp, isert_conn->conn_rx_wr,
791                                 &rx_wr_failed);
792         if (ret) {
793                 pr_err("ib_post_recv() failed with ret: %d\n", ret);
794                 isert_conn->post_recv_buf_count -= count;
795         } else {
796                 pr_debug("isert_post_recv(): Posted %d RX buffers\n", count);
797                 isert_conn->conn_rx_desc_head = rx_head;
798         }
799         return ret;
800 }
801
802 static int
803 isert_post_send(struct isert_conn *isert_conn, struct iser_tx_desc *tx_desc)
804 {
805         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
806         struct ib_send_wr send_wr, *send_wr_failed;
807         int ret;
808
809         ib_dma_sync_single_for_device(ib_dev, tx_desc->dma_addr,
810                                       ISER_HEADERS_LEN, DMA_TO_DEVICE);
811
812         send_wr.next    = NULL;
813         send_wr.wr_id   = (unsigned long)tx_desc;
814         send_wr.sg_list = tx_desc->tx_sg;
815         send_wr.num_sge = tx_desc->num_sge;
816         send_wr.opcode  = IB_WR_SEND;
817         send_wr.send_flags = IB_SEND_SIGNALED;
818
819         atomic_inc(&isert_conn->post_send_buf_count);
820
821         ret = ib_post_send(isert_conn->conn_qp, &send_wr, &send_wr_failed);
822         if (ret) {
823                 pr_err("ib_post_send() failed, ret: %d\n", ret);
824                 atomic_dec(&isert_conn->post_send_buf_count);
825         }
826
827         return ret;
828 }
829
830 static void
831 isert_create_send_desc(struct isert_conn *isert_conn,
832                        struct isert_cmd *isert_cmd,
833                        struct iser_tx_desc *tx_desc)
834 {
835         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
836
837         ib_dma_sync_single_for_cpu(ib_dev, tx_desc->dma_addr,
838                                    ISER_HEADERS_LEN, DMA_TO_DEVICE);
839
840         memset(&tx_desc->iser_header, 0, sizeof(struct iser_hdr));
841         tx_desc->iser_header.flags = ISER_VER;
842
843         tx_desc->num_sge = 1;
844         tx_desc->isert_cmd = isert_cmd;
845
846         if (tx_desc->tx_sg[0].lkey != isert_conn->conn_mr->lkey) {
847                 tx_desc->tx_sg[0].lkey = isert_conn->conn_mr->lkey;
848                 pr_debug("tx_desc %p lkey mismatch, fixing\n", tx_desc);
849         }
850 }
851
852 static int
853 isert_init_tx_hdrs(struct isert_conn *isert_conn,
854                    struct iser_tx_desc *tx_desc)
855 {
856         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
857         u64 dma_addr;
858
859         dma_addr = ib_dma_map_single(ib_dev, (void *)tx_desc,
860                         ISER_HEADERS_LEN, DMA_TO_DEVICE);
861         if (ib_dma_mapping_error(ib_dev, dma_addr)) {
862                 pr_err("ib_dma_mapping_error() failed\n");
863                 return -ENOMEM;
864         }
865
866         tx_desc->dma_addr = dma_addr;
867         tx_desc->tx_sg[0].addr  = tx_desc->dma_addr;
868         tx_desc->tx_sg[0].length = ISER_HEADERS_LEN;
869         tx_desc->tx_sg[0].lkey = isert_conn->conn_mr->lkey;
870
871         pr_debug("isert_init_tx_hdrs: Setup tx_sg[0].addr: 0x%llx length: %u"
872                  " lkey: 0x%08x\n", tx_desc->tx_sg[0].addr,
873                  tx_desc->tx_sg[0].length, tx_desc->tx_sg[0].lkey);
874
875         return 0;
876 }
877
878 static void
879 isert_init_send_wr(struct isert_conn *isert_conn, struct isert_cmd *isert_cmd,
880                    struct ib_send_wr *send_wr, bool coalesce)
881 {
882         struct iser_tx_desc *tx_desc = &isert_cmd->tx_desc;
883
884         isert_cmd->rdma_wr.iser_ib_op = ISER_IB_SEND;
885         send_wr->wr_id = (unsigned long)&isert_cmd->tx_desc;
886         send_wr->opcode = IB_WR_SEND;
887         send_wr->sg_list = &tx_desc->tx_sg[0];
888         send_wr->num_sge = isert_cmd->tx_desc.num_sge;
889         /*
890          * Coalesce send completion interrupts by only setting IB_SEND_SIGNALED
891          * bit for every ISERT_COMP_BATCH_COUNT number of ib_post_send() calls.
892          */
893         mutex_lock(&isert_conn->conn_mutex);
894         if (coalesce && isert_conn->state == ISER_CONN_UP &&
895             ++isert_conn->conn_comp_batch < ISERT_COMP_BATCH_COUNT) {
896                 tx_desc->llnode_active = true;
897                 llist_add(&tx_desc->comp_llnode, &isert_conn->conn_comp_llist);
898                 mutex_unlock(&isert_conn->conn_mutex);
899                 return;
900         }
901         isert_conn->conn_comp_batch = 0;
902         tx_desc->comp_llnode_batch = llist_del_all(&isert_conn->conn_comp_llist);
903         mutex_unlock(&isert_conn->conn_mutex);
904
905         send_wr->send_flags = IB_SEND_SIGNALED;
906 }
907
908 static int
909 isert_rdma_post_recvl(struct isert_conn *isert_conn)
910 {
911         struct ib_recv_wr rx_wr, *rx_wr_fail;
912         struct ib_sge sge;
913         int ret;
914
915         memset(&sge, 0, sizeof(struct ib_sge));
916         sge.addr = isert_conn->login_req_dma;
917         sge.length = ISER_RX_LOGIN_SIZE;
918         sge.lkey = isert_conn->conn_mr->lkey;
919
920         pr_debug("Setup sge: addr: %llx length: %d 0x%08x\n",
921                 sge.addr, sge.length, sge.lkey);
922
923         memset(&rx_wr, 0, sizeof(struct ib_recv_wr));
924         rx_wr.wr_id = (unsigned long)isert_conn->login_req_buf;
925         rx_wr.sg_list = &sge;
926         rx_wr.num_sge = 1;
927
928         isert_conn->post_recv_buf_count++;
929         ret = ib_post_recv(isert_conn->conn_qp, &rx_wr, &rx_wr_fail);
930         if (ret) {
931                 pr_err("ib_post_recv() failed: %d\n", ret);
932                 isert_conn->post_recv_buf_count--;
933         }
934
935         pr_debug("ib_post_recv(): returned success >>>>>>>>>>>>>>>>>>>>>>>>\n");
936         return ret;
937 }
938
939 static int
940 isert_put_login_tx(struct iscsi_conn *conn, struct iscsi_login *login,
941                    u32 length)
942 {
943         struct isert_conn *isert_conn = conn->context;
944         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
945         struct iser_tx_desc *tx_desc = &isert_conn->conn_login_tx_desc;
946         int ret;
947
948         isert_create_send_desc(isert_conn, NULL, tx_desc);
949
950         memcpy(&tx_desc->iscsi_header, &login->rsp[0],
951                sizeof(struct iscsi_hdr));
952
953         isert_init_tx_hdrs(isert_conn, tx_desc);
954
955         if (length > 0) {
956                 struct ib_sge *tx_dsg = &tx_desc->tx_sg[1];
957
958                 ib_dma_sync_single_for_cpu(ib_dev, isert_conn->login_rsp_dma,
959                                            length, DMA_TO_DEVICE);
960
961                 memcpy(isert_conn->login_rsp_buf, login->rsp_buf, length);
962
963                 ib_dma_sync_single_for_device(ib_dev, isert_conn->login_rsp_dma,
964                                               length, DMA_TO_DEVICE);
965
966                 tx_dsg->addr    = isert_conn->login_rsp_dma;
967                 tx_dsg->length  = length;
968                 tx_dsg->lkey    = isert_conn->conn_mr->lkey;
969                 tx_desc->num_sge = 2;
970         }
971         if (!login->login_failed) {
972                 if (login->login_complete) {
973                         if (isert_conn->conn_device->use_fastreg) {
974                                 ret = isert_conn_create_fastreg_pool(isert_conn);
975                                 if (ret) {
976                                         pr_err("Conn: %p failed to create"
977                                                " fastreg pool\n", isert_conn);
978                                         return ret;
979                                 }
980                         }
981
982                         ret = isert_alloc_rx_descriptors(isert_conn);
983                         if (ret)
984                                 return ret;
985
986                         ret = isert_post_recv(isert_conn, ISERT_MIN_POSTED_RX);
987                         if (ret)
988                                 return ret;
989
990                         isert_conn->state = ISER_CONN_UP;
991                         goto post_send;
992                 }
993
994                 ret = isert_rdma_post_recvl(isert_conn);
995                 if (ret)
996                         return ret;
997         }
998 post_send:
999         ret = isert_post_send(isert_conn, tx_desc);
1000         if (ret)
1001                 return ret;
1002
1003         return 0;
1004 }
1005
1006 static void
1007 isert_rx_login_req(struct iser_rx_desc *rx_desc, int rx_buflen,
1008                    struct isert_conn *isert_conn)
1009 {
1010         struct iscsi_conn *conn = isert_conn->conn;
1011         struct iscsi_login *login = conn->conn_login;
1012         int size;
1013
1014         if (!login) {
1015                 pr_err("conn->conn_login is NULL\n");
1016                 dump_stack();
1017                 return;
1018         }
1019
1020         if (login->first_request) {
1021                 struct iscsi_login_req *login_req =
1022                         (struct iscsi_login_req *)&rx_desc->iscsi_header;
1023                 /*
1024                  * Setup the initial iscsi_login values from the leading
1025                  * login request PDU.
1026                  */
1027                 login->leading_connection = (!login_req->tsih) ? 1 : 0;
1028                 login->current_stage =
1029                         (login_req->flags & ISCSI_FLAG_LOGIN_CURRENT_STAGE_MASK)
1030                          >> 2;
1031                 login->version_min      = login_req->min_version;
1032                 login->version_max      = login_req->max_version;
1033                 memcpy(login->isid, login_req->isid, 6);
1034                 login->cmd_sn           = be32_to_cpu(login_req->cmdsn);
1035                 login->init_task_tag    = login_req->itt;
1036                 login->initial_exp_statsn = be32_to_cpu(login_req->exp_statsn);
1037                 login->cid              = be16_to_cpu(login_req->cid);
1038                 login->tsih             = be16_to_cpu(login_req->tsih);
1039         }
1040
1041         memcpy(&login->req[0], (void *)&rx_desc->iscsi_header, ISCSI_HDR_LEN);
1042
1043         size = min(rx_buflen, MAX_KEY_VALUE_PAIRS);
1044         pr_debug("Using login payload size: %d, rx_buflen: %d MAX_KEY_VALUE_PAIRS: %d\n",
1045                  size, rx_buflen, MAX_KEY_VALUE_PAIRS);
1046         memcpy(login->req_buf, &rx_desc->data[0], size);
1047
1048         if (login->first_request) {
1049                 complete(&isert_conn->conn_login_comp);
1050                 return;
1051         }
1052         schedule_delayed_work(&conn->login_work, 0);
1053 }
1054
1055 static struct iscsi_cmd
1056 *isert_allocate_cmd(struct iscsi_conn *conn)
1057 {
1058         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
1059         struct isert_cmd *isert_cmd;
1060         struct iscsi_cmd *cmd;
1061
1062         cmd = iscsit_allocate_cmd(conn, TASK_INTERRUPTIBLE);
1063         if (!cmd) {
1064                 pr_err("Unable to allocate iscsi_cmd + isert_cmd\n");
1065                 return NULL;
1066         }
1067         isert_cmd = iscsit_priv_cmd(cmd);
1068         isert_cmd->conn = isert_conn;
1069         isert_cmd->iscsi_cmd = cmd;
1070
1071         return cmd;
1072 }
1073
1074 static int
1075 isert_handle_scsi_cmd(struct isert_conn *isert_conn,
1076                       struct isert_cmd *isert_cmd, struct iscsi_cmd *cmd,
1077                       struct iser_rx_desc *rx_desc, unsigned char *buf)
1078 {
1079         struct iscsi_conn *conn = isert_conn->conn;
1080         struct iscsi_scsi_req *hdr = (struct iscsi_scsi_req *)buf;
1081         struct scatterlist *sg;
1082         int imm_data, imm_data_len, unsol_data, sg_nents, rc;
1083         bool dump_payload = false;
1084
1085         rc = iscsit_setup_scsi_cmd(conn, cmd, buf);
1086         if (rc < 0)
1087                 return rc;
1088
1089         imm_data = cmd->immediate_data;
1090         imm_data_len = cmd->first_burst_len;
1091         unsol_data = cmd->unsolicited_data;
1092
1093         rc = iscsit_process_scsi_cmd(conn, cmd, hdr);
1094         if (rc < 0) {
1095                 return 0;
1096         } else if (rc > 0) {
1097                 dump_payload = true;
1098                 goto sequence_cmd;
1099         }
1100
1101         if (!imm_data)
1102                 return 0;
1103
1104         sg = &cmd->se_cmd.t_data_sg[0];
1105         sg_nents = max(1UL, DIV_ROUND_UP(imm_data_len, PAGE_SIZE));
1106
1107         pr_debug("Copying Immediate SG: %p sg_nents: %u from %p imm_data_len: %d\n",
1108                  sg, sg_nents, &rx_desc->data[0], imm_data_len);
1109
1110         sg_copy_from_buffer(sg, sg_nents, &rx_desc->data[0], imm_data_len);
1111
1112         cmd->write_data_done += imm_data_len;
1113
1114         if (cmd->write_data_done == cmd->se_cmd.data_length) {
1115                 spin_lock_bh(&cmd->istate_lock);
1116                 cmd->cmd_flags |= ICF_GOT_LAST_DATAOUT;
1117                 cmd->i_state = ISTATE_RECEIVED_LAST_DATAOUT;
1118                 spin_unlock_bh(&cmd->istate_lock);
1119         }
1120
1121 sequence_cmd:
1122         rc = iscsit_sequence_cmd(conn, cmd, buf, hdr->cmdsn);
1123
1124         if (!rc && dump_payload == false && unsol_data)
1125                 iscsit_set_unsoliticed_dataout(cmd);
1126         else if (dump_payload && imm_data)
1127                 target_put_sess_cmd(conn->sess->se_sess, &cmd->se_cmd);
1128
1129         return 0;
1130 }
1131
1132 static int
1133 isert_handle_iscsi_dataout(struct isert_conn *isert_conn,
1134                            struct iser_rx_desc *rx_desc, unsigned char *buf)
1135 {
1136         struct scatterlist *sg_start;
1137         struct iscsi_conn *conn = isert_conn->conn;
1138         struct iscsi_cmd *cmd = NULL;
1139         struct iscsi_data *hdr = (struct iscsi_data *)buf;
1140         u32 unsol_data_len = ntoh24(hdr->dlength);
1141         int rc, sg_nents, sg_off, page_off;
1142
1143         rc = iscsit_check_dataout_hdr(conn, buf, &cmd);
1144         if (rc < 0)
1145                 return rc;
1146         else if (!cmd)
1147                 return 0;
1148         /*
1149          * FIXME: Unexpected unsolicited_data out
1150          */
1151         if (!cmd->unsolicited_data) {
1152                 pr_err("Received unexpected solicited data payload\n");
1153                 dump_stack();
1154                 return -1;
1155         }
1156
1157         pr_debug("Unsolicited DataOut unsol_data_len: %u, write_data_done: %u, data_length: %u\n",
1158                  unsol_data_len, cmd->write_data_done, cmd->se_cmd.data_length);
1159
1160         sg_off = cmd->write_data_done / PAGE_SIZE;
1161         sg_start = &cmd->se_cmd.t_data_sg[sg_off];
1162         sg_nents = max(1UL, DIV_ROUND_UP(unsol_data_len, PAGE_SIZE));
1163         page_off = cmd->write_data_done % PAGE_SIZE;
1164         /*
1165          * FIXME: Non page-aligned unsolicited_data out
1166          */
1167         if (page_off) {
1168                 pr_err("Received unexpected non-page aligned data payload\n");
1169                 dump_stack();
1170                 return -1;
1171         }
1172         pr_debug("Copying DataOut: sg_start: %p, sg_off: %u sg_nents: %u from %p %u\n",
1173                  sg_start, sg_off, sg_nents, &rx_desc->data[0], unsol_data_len);
1174
1175         sg_copy_from_buffer(sg_start, sg_nents, &rx_desc->data[0],
1176                             unsol_data_len);
1177
1178         rc = iscsit_check_dataout_payload(cmd, hdr, false);
1179         if (rc < 0)
1180                 return rc;
1181
1182         return 0;
1183 }
1184
1185 static int
1186 isert_handle_nop_out(struct isert_conn *isert_conn, struct isert_cmd *isert_cmd,
1187                      struct iscsi_cmd *cmd, struct iser_rx_desc *rx_desc,
1188                      unsigned char *buf)
1189 {
1190         struct iscsi_conn *conn = isert_conn->conn;
1191         struct iscsi_nopout *hdr = (struct iscsi_nopout *)buf;
1192         int rc;
1193
1194         rc = iscsit_setup_nop_out(conn, cmd, hdr);
1195         if (rc < 0)
1196                 return rc;
1197         /*
1198          * FIXME: Add support for NOPOUT payload using unsolicited RDMA payload
1199          */
1200
1201         return iscsit_process_nop_out(conn, cmd, hdr);
1202 }
1203
1204 static int
1205 isert_handle_text_cmd(struct isert_conn *isert_conn, struct isert_cmd *isert_cmd,
1206                       struct iscsi_cmd *cmd, struct iser_rx_desc *rx_desc,
1207                       struct iscsi_text *hdr)
1208 {
1209         struct iscsi_conn *conn = isert_conn->conn;
1210         u32 payload_length = ntoh24(hdr->dlength);
1211         int rc;
1212         unsigned char *text_in;
1213
1214         rc = iscsit_setup_text_cmd(conn, cmd, hdr);
1215         if (rc < 0)
1216                 return rc;
1217
1218         text_in = kzalloc(payload_length, GFP_KERNEL);
1219         if (!text_in) {
1220                 pr_err("Unable to allocate text_in of payload_length: %u\n",
1221                        payload_length);
1222                 return -ENOMEM;
1223         }
1224         cmd->text_in_ptr = text_in;
1225
1226         memcpy(cmd->text_in_ptr, &rx_desc->data[0], payload_length);
1227
1228         return iscsit_process_text_cmd(conn, cmd, hdr);
1229 }
1230
1231 static int
1232 isert_rx_opcode(struct isert_conn *isert_conn, struct iser_rx_desc *rx_desc,
1233                 uint32_t read_stag, uint64_t read_va,
1234                 uint32_t write_stag, uint64_t write_va)
1235 {
1236         struct iscsi_hdr *hdr = &rx_desc->iscsi_header;
1237         struct iscsi_conn *conn = isert_conn->conn;
1238         struct iscsi_session *sess = conn->sess;
1239         struct iscsi_cmd *cmd;
1240         struct isert_cmd *isert_cmd;
1241         int ret = -EINVAL;
1242         u8 opcode = (hdr->opcode & ISCSI_OPCODE_MASK);
1243
1244         if (sess->sess_ops->SessionType &&
1245            (!(opcode & ISCSI_OP_TEXT) || !(opcode & ISCSI_OP_LOGOUT))) {
1246                 pr_err("Got illegal opcode: 0x%02x in SessionType=Discovery,"
1247                        " ignoring\n", opcode);
1248                 return 0;
1249         }
1250
1251         switch (opcode) {
1252         case ISCSI_OP_SCSI_CMD:
1253                 cmd = isert_allocate_cmd(conn);
1254                 if (!cmd)
1255                         break;
1256
1257                 isert_cmd = iscsit_priv_cmd(cmd);
1258                 isert_cmd->read_stag = read_stag;
1259                 isert_cmd->read_va = read_va;
1260                 isert_cmd->write_stag = write_stag;
1261                 isert_cmd->write_va = write_va;
1262
1263                 ret = isert_handle_scsi_cmd(isert_conn, isert_cmd, cmd,
1264                                         rx_desc, (unsigned char *)hdr);
1265                 break;
1266         case ISCSI_OP_NOOP_OUT:
1267                 cmd = isert_allocate_cmd(conn);
1268                 if (!cmd)
1269                         break;
1270
1271                 isert_cmd = iscsit_priv_cmd(cmd);
1272                 ret = isert_handle_nop_out(isert_conn, isert_cmd, cmd,
1273                                            rx_desc, (unsigned char *)hdr);
1274                 break;
1275         case ISCSI_OP_SCSI_DATA_OUT:
1276                 ret = isert_handle_iscsi_dataout(isert_conn, rx_desc,
1277                                                 (unsigned char *)hdr);
1278                 break;
1279         case ISCSI_OP_SCSI_TMFUNC:
1280                 cmd = isert_allocate_cmd(conn);
1281                 if (!cmd)
1282                         break;
1283
1284                 ret = iscsit_handle_task_mgt_cmd(conn, cmd,
1285                                                 (unsigned char *)hdr);
1286                 break;
1287         case ISCSI_OP_LOGOUT:
1288                 cmd = isert_allocate_cmd(conn);
1289                 if (!cmd)
1290                         break;
1291
1292                 ret = iscsit_handle_logout_cmd(conn, cmd, (unsigned char *)hdr);
1293                 if (ret > 0)
1294                         wait_for_completion_timeout(&conn->conn_logout_comp,
1295                                                     SECONDS_FOR_LOGOUT_COMP *
1296                                                     HZ);
1297                 break;
1298         case ISCSI_OP_TEXT:
1299                 cmd = isert_allocate_cmd(conn);
1300                 if (!cmd)
1301                         break;
1302
1303                 isert_cmd = iscsit_priv_cmd(cmd);
1304                 ret = isert_handle_text_cmd(isert_conn, isert_cmd, cmd,
1305                                             rx_desc, (struct iscsi_text *)hdr);
1306                 break;
1307         default:
1308                 pr_err("Got unknown iSCSI OpCode: 0x%02x\n", opcode);
1309                 dump_stack();
1310                 break;
1311         }
1312
1313         return ret;
1314 }
1315
1316 static void
1317 isert_rx_do_work(struct iser_rx_desc *rx_desc, struct isert_conn *isert_conn)
1318 {
1319         struct iser_hdr *iser_hdr = &rx_desc->iser_header;
1320         uint64_t read_va = 0, write_va = 0;
1321         uint32_t read_stag = 0, write_stag = 0;
1322         int rc;
1323
1324         switch (iser_hdr->flags & 0xF0) {
1325         case ISCSI_CTRL:
1326                 if (iser_hdr->flags & ISER_RSV) {
1327                         read_stag = be32_to_cpu(iser_hdr->read_stag);
1328                         read_va = be64_to_cpu(iser_hdr->read_va);
1329                         pr_debug("ISER_RSV: read_stag: 0x%08x read_va: 0x%16llx\n",
1330                                  read_stag, (unsigned long long)read_va);
1331                 }
1332                 if (iser_hdr->flags & ISER_WSV) {
1333                         write_stag = be32_to_cpu(iser_hdr->write_stag);
1334                         write_va = be64_to_cpu(iser_hdr->write_va);
1335                         pr_debug("ISER_WSV: write__stag: 0x%08x write_va: 0x%16llx\n",
1336                                  write_stag, (unsigned long long)write_va);
1337                 }
1338
1339                 pr_debug("ISER ISCSI_CTRL PDU\n");
1340                 break;
1341         case ISER_HELLO:
1342                 pr_err("iSER Hello message\n");
1343                 break;
1344         default:
1345                 pr_warn("Unknown iSER hdr flags: 0x%02x\n", iser_hdr->flags);
1346                 break;
1347         }
1348
1349         rc = isert_rx_opcode(isert_conn, rx_desc,
1350                              read_stag, read_va, write_stag, write_va);
1351 }
1352
1353 static void
1354 isert_rx_completion(struct iser_rx_desc *desc, struct isert_conn *isert_conn,
1355                     unsigned long xfer_len)
1356 {
1357         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
1358         struct iscsi_hdr *hdr;
1359         u64 rx_dma;
1360         int rx_buflen, outstanding;
1361
1362         if ((char *)desc == isert_conn->login_req_buf) {
1363                 rx_dma = isert_conn->login_req_dma;
1364                 rx_buflen = ISER_RX_LOGIN_SIZE;
1365                 pr_debug("ISER login_buf: Using rx_dma: 0x%llx, rx_buflen: %d\n",
1366                          rx_dma, rx_buflen);
1367         } else {
1368                 rx_dma = desc->dma_addr;
1369                 rx_buflen = ISER_RX_PAYLOAD_SIZE;
1370                 pr_debug("ISER req_buf: Using rx_dma: 0x%llx, rx_buflen: %d\n",
1371                          rx_dma, rx_buflen);
1372         }
1373
1374         ib_dma_sync_single_for_cpu(ib_dev, rx_dma, rx_buflen, DMA_FROM_DEVICE);
1375
1376         hdr = &desc->iscsi_header;
1377         pr_debug("iSCSI opcode: 0x%02x, ITT: 0x%08x, flags: 0x%02x dlen: %d\n",
1378                  hdr->opcode, hdr->itt, hdr->flags,
1379                  (int)(xfer_len - ISER_HEADERS_LEN));
1380
1381         if ((char *)desc == isert_conn->login_req_buf)
1382                 isert_rx_login_req(desc, xfer_len - ISER_HEADERS_LEN,
1383                                    isert_conn);
1384         else
1385                 isert_rx_do_work(desc, isert_conn);
1386
1387         ib_dma_sync_single_for_device(ib_dev, rx_dma, rx_buflen,
1388                                       DMA_FROM_DEVICE);
1389
1390         isert_conn->post_recv_buf_count--;
1391         pr_debug("iSERT: Decremented post_recv_buf_count: %d\n",
1392                  isert_conn->post_recv_buf_count);
1393
1394         if ((char *)desc == isert_conn->login_req_buf)
1395                 return;
1396
1397         outstanding = isert_conn->post_recv_buf_count;
1398         if (outstanding + ISERT_MIN_POSTED_RX <= ISERT_QP_MAX_RECV_DTOS) {
1399                 int err, count = min(ISERT_QP_MAX_RECV_DTOS - outstanding,
1400                                 ISERT_MIN_POSTED_RX);
1401                 err = isert_post_recv(isert_conn, count);
1402                 if (err) {
1403                         pr_err("isert_post_recv() count: %d failed, %d\n",
1404                                count, err);
1405                 }
1406         }
1407 }
1408
1409 static void
1410 isert_unmap_cmd(struct isert_cmd *isert_cmd, struct isert_conn *isert_conn)
1411 {
1412         struct isert_rdma_wr *wr = &isert_cmd->rdma_wr;
1413         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
1414
1415         pr_debug("isert_unmap_cmd: %p\n", isert_cmd);
1416         if (wr->sge) {
1417                 pr_debug("isert_unmap_cmd: %p unmap_sg op\n", isert_cmd);
1418                 ib_dma_unmap_sg(ib_dev, wr->sge, wr->num_sge,
1419                                 (wr->iser_ib_op == ISER_IB_RDMA_WRITE) ?
1420                                 DMA_TO_DEVICE : DMA_FROM_DEVICE);
1421                 wr->sge = NULL;
1422         }
1423
1424         if (wr->send_wr) {
1425                 pr_debug("isert_unmap_cmd: %p free send_wr\n", isert_cmd);
1426                 kfree(wr->send_wr);
1427                 wr->send_wr = NULL;
1428         }
1429
1430         if (wr->ib_sge) {
1431                 pr_debug("isert_unmap_cmd: %p free ib_sge\n", isert_cmd);
1432                 kfree(wr->ib_sge);
1433                 wr->ib_sge = NULL;
1434         }
1435 }
1436
1437 static void
1438 isert_unreg_rdma(struct isert_cmd *isert_cmd, struct isert_conn *isert_conn)
1439 {
1440         struct isert_rdma_wr *wr = &isert_cmd->rdma_wr;
1441         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
1442         LIST_HEAD(unmap_list);
1443
1444         pr_debug("unreg_fastreg_cmd: %p\n", isert_cmd);
1445
1446         if (wr->fr_desc) {
1447                 pr_debug("unreg_fastreg_cmd: %p free fr_desc %p\n",
1448                          isert_cmd, wr->fr_desc);
1449                 spin_lock_bh(&isert_conn->conn_lock);
1450                 list_add_tail(&wr->fr_desc->list, &isert_conn->conn_fr_pool);
1451                 spin_unlock_bh(&isert_conn->conn_lock);
1452                 wr->fr_desc = NULL;
1453         }
1454
1455         if (wr->sge) {
1456                 pr_debug("unreg_fastreg_cmd: %p unmap_sg op\n", isert_cmd);
1457                 ib_dma_unmap_sg(ib_dev, wr->sge, wr->num_sge,
1458                                 (wr->iser_ib_op == ISER_IB_RDMA_WRITE) ?
1459                                 DMA_TO_DEVICE : DMA_FROM_DEVICE);
1460                 wr->sge = NULL;
1461         }
1462
1463         wr->ib_sge = NULL;
1464         wr->send_wr = NULL;
1465 }
1466
1467 static void
1468 isert_put_cmd(struct isert_cmd *isert_cmd, bool comp_err)
1469 {
1470         struct iscsi_cmd *cmd = isert_cmd->iscsi_cmd;
1471         struct isert_conn *isert_conn = isert_cmd->conn;
1472         struct iscsi_conn *conn = isert_conn->conn;
1473         struct isert_device *device = isert_conn->conn_device;
1474
1475         pr_debug("Entering isert_put_cmd: %p\n", isert_cmd);
1476
1477         switch (cmd->iscsi_opcode) {
1478         case ISCSI_OP_SCSI_CMD:
1479                 spin_lock_bh(&conn->cmd_lock);
1480                 if (!list_empty(&cmd->i_conn_node))
1481                         list_del_init(&cmd->i_conn_node);
1482                 spin_unlock_bh(&conn->cmd_lock);
1483
1484                 if (cmd->data_direction == DMA_TO_DEVICE) {
1485                         iscsit_stop_dataout_timer(cmd);
1486                         /*
1487                          * Check for special case during comp_err where
1488                          * WRITE_PENDING has been handed off from core,
1489                          * but requires an extra target_put_sess_cmd()
1490                          * before transport_generic_free_cmd() below.
1491                          */
1492                         if (comp_err &&
1493                             cmd->se_cmd.t_state == TRANSPORT_WRITE_PENDING) {
1494                                 struct se_cmd *se_cmd = &cmd->se_cmd;
1495
1496                                 target_put_sess_cmd(se_cmd->se_sess, se_cmd);
1497                         }
1498                 }
1499
1500                 device->unreg_rdma_mem(isert_cmd, isert_conn);
1501                 transport_generic_free_cmd(&cmd->se_cmd, 0);
1502                 break;
1503         case ISCSI_OP_SCSI_TMFUNC:
1504                 spin_lock_bh(&conn->cmd_lock);
1505                 if (!list_empty(&cmd->i_conn_node))
1506                         list_del_init(&cmd->i_conn_node);
1507                 spin_unlock_bh(&conn->cmd_lock);
1508
1509                 transport_generic_free_cmd(&cmd->se_cmd, 0);
1510                 break;
1511         case ISCSI_OP_REJECT:
1512         case ISCSI_OP_NOOP_OUT:
1513         case ISCSI_OP_TEXT:
1514                 spin_lock_bh(&conn->cmd_lock);
1515                 if (!list_empty(&cmd->i_conn_node))
1516                         list_del_init(&cmd->i_conn_node);
1517                 spin_unlock_bh(&conn->cmd_lock);
1518
1519                 /*
1520                  * Handle special case for REJECT when iscsi_add_reject*() has
1521                  * overwritten the original iscsi_opcode assignment, and the
1522                  * associated cmd->se_cmd needs to be released.
1523                  */
1524                 if (cmd->se_cmd.se_tfo != NULL) {
1525                         pr_debug("Calling transport_generic_free_cmd from"
1526                                  " isert_put_cmd for 0x%02x\n",
1527                                  cmd->iscsi_opcode);
1528                         transport_generic_free_cmd(&cmd->se_cmd, 0);
1529                         break;
1530                 }
1531                 /*
1532                  * Fall-through
1533                  */
1534         default:
1535                 iscsit_release_cmd(cmd);
1536                 break;
1537         }
1538 }
1539
1540 static void
1541 isert_unmap_tx_desc(struct iser_tx_desc *tx_desc, struct ib_device *ib_dev)
1542 {
1543         if (tx_desc->dma_addr != 0) {
1544                 pr_debug("Calling ib_dma_unmap_single for tx_desc->dma_addr\n");
1545                 ib_dma_unmap_single(ib_dev, tx_desc->dma_addr,
1546                                     ISER_HEADERS_LEN, DMA_TO_DEVICE);
1547                 tx_desc->dma_addr = 0;
1548         }
1549 }
1550
1551 static void
1552 isert_completion_put(struct iser_tx_desc *tx_desc, struct isert_cmd *isert_cmd,
1553                      struct ib_device *ib_dev, bool comp_err)
1554 {
1555         if (isert_cmd->pdu_buf_dma != 0) {
1556                 pr_debug("Calling ib_dma_unmap_single for isert_cmd->pdu_buf_dma\n");
1557                 ib_dma_unmap_single(ib_dev, isert_cmd->pdu_buf_dma,
1558                                     isert_cmd->pdu_buf_len, DMA_TO_DEVICE);
1559                 isert_cmd->pdu_buf_dma = 0;
1560         }
1561
1562         isert_unmap_tx_desc(tx_desc, ib_dev);
1563         isert_put_cmd(isert_cmd, comp_err);
1564 }
1565
1566 static void
1567 isert_completion_rdma_read(struct iser_tx_desc *tx_desc,
1568                            struct isert_cmd *isert_cmd)
1569 {
1570         struct isert_rdma_wr *wr = &isert_cmd->rdma_wr;
1571         struct iscsi_cmd *cmd = isert_cmd->iscsi_cmd;
1572         struct se_cmd *se_cmd = &cmd->se_cmd;
1573         struct isert_conn *isert_conn = isert_cmd->conn;
1574         struct isert_device *device = isert_conn->conn_device;
1575
1576         iscsit_stop_dataout_timer(cmd);
1577         device->unreg_rdma_mem(isert_cmd, isert_conn);
1578         cmd->write_data_done = wr->cur_rdma_length;
1579         wr->send_wr_num = 0;
1580
1581         pr_debug("Cmd: %p RDMA_READ comp calling execute_cmd\n", isert_cmd);
1582         spin_lock_bh(&cmd->istate_lock);
1583         cmd->cmd_flags |= ICF_GOT_LAST_DATAOUT;
1584         cmd->i_state = ISTATE_RECEIVED_LAST_DATAOUT;
1585         spin_unlock_bh(&cmd->istate_lock);
1586
1587         target_execute_cmd(se_cmd);
1588 }
1589
1590 static void
1591 isert_do_control_comp(struct work_struct *work)
1592 {
1593         struct isert_cmd *isert_cmd = container_of(work,
1594                         struct isert_cmd, comp_work);
1595         struct isert_conn *isert_conn = isert_cmd->conn;
1596         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
1597         struct iscsi_cmd *cmd = isert_cmd->iscsi_cmd;
1598
1599         switch (cmd->i_state) {
1600         case ISTATE_SEND_TASKMGTRSP:
1601                 pr_debug("Calling iscsit_tmr_post_handler >>>>>>>>>>>>>>>>>\n");
1602
1603                 atomic_dec(&isert_conn->post_send_buf_count);
1604                 iscsit_tmr_post_handler(cmd, cmd->conn);
1605
1606                 cmd->i_state = ISTATE_SENT_STATUS;
1607                 isert_completion_put(&isert_cmd->tx_desc, isert_cmd, ib_dev, false);
1608                 break;
1609         case ISTATE_SEND_REJECT:
1610                 pr_debug("Got isert_do_control_comp ISTATE_SEND_REJECT: >>>\n");
1611                 atomic_dec(&isert_conn->post_send_buf_count);
1612
1613                 cmd->i_state = ISTATE_SENT_STATUS;
1614                 isert_completion_put(&isert_cmd->tx_desc, isert_cmd, ib_dev, false);
1615                 break;
1616         case ISTATE_SEND_LOGOUTRSP:
1617                 pr_debug("Calling iscsit_logout_post_handler >>>>>>>>>>>>>>\n");
1618
1619                 atomic_dec(&isert_conn->post_send_buf_count);
1620                 iscsit_logout_post_handler(cmd, cmd->conn);
1621                 break;
1622         case ISTATE_SEND_TEXTRSP:
1623                 atomic_dec(&isert_conn->post_send_buf_count);
1624                 cmd->i_state = ISTATE_SENT_STATUS;
1625                 isert_completion_put(&isert_cmd->tx_desc, isert_cmd, ib_dev, false);
1626                 break;
1627         default:
1628                 pr_err("Unknown do_control_comp i_state %d\n", cmd->i_state);
1629                 dump_stack();
1630                 break;
1631         }
1632 }
1633
1634 static void
1635 isert_response_completion(struct iser_tx_desc *tx_desc,
1636                           struct isert_cmd *isert_cmd,
1637                           struct isert_conn *isert_conn,
1638                           struct ib_device *ib_dev)
1639 {
1640         struct iscsi_cmd *cmd = isert_cmd->iscsi_cmd;
1641         struct isert_rdma_wr *wr = &isert_cmd->rdma_wr;
1642
1643         if (cmd->i_state == ISTATE_SEND_TASKMGTRSP ||
1644             cmd->i_state == ISTATE_SEND_LOGOUTRSP ||
1645             cmd->i_state == ISTATE_SEND_REJECT ||
1646             cmd->i_state == ISTATE_SEND_TEXTRSP) {
1647                 isert_unmap_tx_desc(tx_desc, ib_dev);
1648
1649                 INIT_WORK(&isert_cmd->comp_work, isert_do_control_comp);
1650                 queue_work(isert_comp_wq, &isert_cmd->comp_work);
1651                 return;
1652         }
1653         atomic_sub(wr->send_wr_num + 1, &isert_conn->post_send_buf_count);
1654
1655         cmd->i_state = ISTATE_SENT_STATUS;
1656         isert_completion_put(tx_desc, isert_cmd, ib_dev, false);
1657 }
1658
1659 static void
1660 __isert_send_completion(struct iser_tx_desc *tx_desc,
1661                         struct isert_conn *isert_conn)
1662 {
1663         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
1664         struct isert_cmd *isert_cmd = tx_desc->isert_cmd;
1665         struct isert_rdma_wr *wr;
1666
1667         if (!isert_cmd) {
1668                 atomic_dec(&isert_conn->post_send_buf_count);
1669                 isert_unmap_tx_desc(tx_desc, ib_dev);
1670                 return;
1671         }
1672         wr = &isert_cmd->rdma_wr;
1673
1674         switch (wr->iser_ib_op) {
1675         case ISER_IB_RECV:
1676                 pr_err("isert_send_completion: Got ISER_IB_RECV\n");
1677                 dump_stack();
1678                 break;
1679         case ISER_IB_SEND:
1680                 pr_debug("isert_send_completion: Got ISER_IB_SEND\n");
1681                 isert_response_completion(tx_desc, isert_cmd,
1682                                           isert_conn, ib_dev);
1683                 break;
1684         case ISER_IB_RDMA_WRITE:
1685                 pr_err("isert_send_completion: Got ISER_IB_RDMA_WRITE\n");
1686                 dump_stack();
1687                 break;
1688         case ISER_IB_RDMA_READ:
1689                 pr_debug("isert_send_completion: Got ISER_IB_RDMA_READ:\n");
1690
1691                 atomic_sub(wr->send_wr_num, &isert_conn->post_send_buf_count);
1692                 isert_completion_rdma_read(tx_desc, isert_cmd);
1693                 break;
1694         default:
1695                 pr_err("Unknown wr->iser_ib_op: 0x%02x\n", wr->iser_ib_op);
1696                 dump_stack();
1697                 break;
1698         }
1699 }
1700
1701 static void
1702 isert_send_completion(struct iser_tx_desc *tx_desc,
1703                       struct isert_conn *isert_conn)
1704 {
1705         struct llist_node *llnode = tx_desc->comp_llnode_batch;
1706         struct iser_tx_desc *t;
1707         /*
1708          * Drain coalesced completion llist starting from comp_llnode_batch
1709          * setup in isert_init_send_wr(), and then complete trailing tx_desc.
1710          */
1711         while (llnode) {
1712                 t = llist_entry(llnode, struct iser_tx_desc, comp_llnode);
1713                 llnode = llist_next(llnode);
1714                 __isert_send_completion(t, isert_conn);
1715         }
1716         __isert_send_completion(tx_desc, isert_conn);
1717 }
1718
1719 static void
1720 isert_cq_drain_comp_llist(struct isert_conn *isert_conn, struct ib_device *ib_dev)
1721 {
1722         struct llist_node *llnode;
1723         struct isert_rdma_wr *wr;
1724         struct iser_tx_desc *t;
1725
1726         mutex_lock(&isert_conn->conn_mutex);
1727         llnode = llist_del_all(&isert_conn->conn_comp_llist);
1728         isert_conn->conn_comp_batch = 0;
1729         mutex_unlock(&isert_conn->conn_mutex);
1730
1731         while (llnode) {
1732                 t = llist_entry(llnode, struct iser_tx_desc, comp_llnode);
1733                 llnode = llist_next(llnode);
1734                 wr = &t->isert_cmd->rdma_wr;
1735
1736                 atomic_sub(wr->send_wr_num + 1, &isert_conn->post_send_buf_count);
1737                 isert_completion_put(t, t->isert_cmd, ib_dev, true);
1738         }
1739 }
1740
1741 static void
1742 isert_cq_tx_comp_err(struct iser_tx_desc *tx_desc, struct isert_conn *isert_conn)
1743 {
1744         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
1745         struct isert_cmd *isert_cmd = tx_desc->isert_cmd;
1746         struct llist_node *llnode = tx_desc->comp_llnode_batch;
1747         struct isert_rdma_wr *wr;
1748         struct iser_tx_desc *t;
1749
1750         while (llnode) {
1751                 t = llist_entry(llnode, struct iser_tx_desc, comp_llnode);
1752                 llnode = llist_next(llnode);
1753                 wr = &t->isert_cmd->rdma_wr;
1754
1755                 atomic_sub(wr->send_wr_num + 1, &isert_conn->post_send_buf_count);
1756                 isert_completion_put(t, t->isert_cmd, ib_dev, true);
1757         }
1758         tx_desc->comp_llnode_batch = NULL;
1759
1760         if (!isert_cmd)
1761                 isert_unmap_tx_desc(tx_desc, ib_dev);
1762         else
1763                 isert_completion_put(tx_desc, isert_cmd, ib_dev, true);
1764 }
1765
1766 static void
1767 isert_cq_rx_comp_err(struct isert_conn *isert_conn)
1768 {
1769         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
1770         struct iscsi_conn *conn = isert_conn->conn;
1771
1772         if (isert_conn->post_recv_buf_count)
1773                 return;
1774
1775         isert_cq_drain_comp_llist(isert_conn, ib_dev);
1776
1777         if (conn->sess) {
1778                 target_sess_cmd_list_set_waiting(conn->sess->se_sess);
1779                 target_wait_for_sess_cmds(conn->sess->se_sess);
1780         }
1781
1782         while (atomic_read(&isert_conn->post_send_buf_count))
1783                 msleep(3000);
1784
1785         mutex_lock(&isert_conn->conn_mutex);
1786         isert_conn->state = ISER_CONN_DOWN;
1787         mutex_unlock(&isert_conn->conn_mutex);
1788
1789         iscsit_cause_connection_reinstatement(isert_conn->conn, 0);
1790
1791         complete(&isert_conn->conn_wait_comp_err);
1792 }
1793
1794 static void
1795 isert_cq_tx_work(struct work_struct *work)
1796 {
1797         struct isert_cq_desc *cq_desc = container_of(work,
1798                                 struct isert_cq_desc, cq_tx_work);
1799         struct isert_device *device = cq_desc->device;
1800         int cq_index = cq_desc->cq_index;
1801         struct ib_cq *tx_cq = device->dev_tx_cq[cq_index];
1802         struct isert_conn *isert_conn;
1803         struct iser_tx_desc *tx_desc;
1804         struct ib_wc wc;
1805
1806         while (ib_poll_cq(tx_cq, 1, &wc) == 1) {
1807                 tx_desc = (struct iser_tx_desc *)(unsigned long)wc.wr_id;
1808                 isert_conn = wc.qp->qp_context;
1809
1810                 if (wc.status == IB_WC_SUCCESS) {
1811                         isert_send_completion(tx_desc, isert_conn);
1812                 } else {
1813                         pr_debug("TX wc.status != IB_WC_SUCCESS >>>>>>>>>>>>>>\n");
1814                         pr_debug("TX wc.status: 0x%08x\n", wc.status);
1815                         pr_debug("TX wc.vendor_err: 0x%08x\n", wc.vendor_err);
1816
1817                         if (wc.wr_id != ISER_FASTREG_LI_WRID) {
1818                                 if (tx_desc->llnode_active)
1819                                         continue;
1820
1821                                 atomic_dec(&isert_conn->post_send_buf_count);
1822                                 isert_cq_tx_comp_err(tx_desc, isert_conn);
1823                         }
1824                 }
1825         }
1826
1827         ib_req_notify_cq(tx_cq, IB_CQ_NEXT_COMP);
1828 }
1829
1830 static void
1831 isert_cq_tx_callback(struct ib_cq *cq, void *context)
1832 {
1833         struct isert_cq_desc *cq_desc = (struct isert_cq_desc *)context;
1834
1835         queue_work(isert_comp_wq, &cq_desc->cq_tx_work);
1836 }
1837
1838 static void
1839 isert_cq_rx_work(struct work_struct *work)
1840 {
1841         struct isert_cq_desc *cq_desc = container_of(work,
1842                         struct isert_cq_desc, cq_rx_work);
1843         struct isert_device *device = cq_desc->device;
1844         int cq_index = cq_desc->cq_index;
1845         struct ib_cq *rx_cq = device->dev_rx_cq[cq_index];
1846         struct isert_conn *isert_conn;
1847         struct iser_rx_desc *rx_desc;
1848         struct ib_wc wc;
1849         unsigned long xfer_len;
1850
1851         while (ib_poll_cq(rx_cq, 1, &wc) == 1) {
1852                 rx_desc = (struct iser_rx_desc *)(unsigned long)wc.wr_id;
1853                 isert_conn = wc.qp->qp_context;
1854
1855                 if (wc.status == IB_WC_SUCCESS) {
1856                         xfer_len = (unsigned long)wc.byte_len;
1857                         isert_rx_completion(rx_desc, isert_conn, xfer_len);
1858                 } else {
1859                         pr_debug("RX wc.status != IB_WC_SUCCESS >>>>>>>>>>>>>>\n");
1860                         if (wc.status != IB_WC_WR_FLUSH_ERR) {
1861                                 pr_debug("RX wc.status: 0x%08x\n", wc.status);
1862                                 pr_debug("RX wc.vendor_err: 0x%08x\n",
1863                                          wc.vendor_err);
1864                         }
1865                         isert_conn->post_recv_buf_count--;
1866                         isert_cq_rx_comp_err(isert_conn);
1867                 }
1868         }
1869
1870         ib_req_notify_cq(rx_cq, IB_CQ_NEXT_COMP);
1871 }
1872
1873 static void
1874 isert_cq_rx_callback(struct ib_cq *cq, void *context)
1875 {
1876         struct isert_cq_desc *cq_desc = (struct isert_cq_desc *)context;
1877
1878         queue_work(isert_rx_wq, &cq_desc->cq_rx_work);
1879 }
1880
1881 static int
1882 isert_post_response(struct isert_conn *isert_conn, struct isert_cmd *isert_cmd)
1883 {
1884         struct ib_send_wr *wr_failed;
1885         int ret;
1886
1887         atomic_inc(&isert_conn->post_send_buf_count);
1888
1889         ret = ib_post_send(isert_conn->conn_qp, &isert_cmd->tx_desc.send_wr,
1890                            &wr_failed);
1891         if (ret) {
1892                 pr_err("ib_post_send failed with %d\n", ret);
1893                 atomic_dec(&isert_conn->post_send_buf_count);
1894                 return ret;
1895         }
1896         return ret;
1897 }
1898
1899 static int
1900 isert_put_response(struct iscsi_conn *conn, struct iscsi_cmd *cmd)
1901 {
1902         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
1903         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
1904         struct ib_send_wr *send_wr = &isert_cmd->tx_desc.send_wr;
1905         struct iscsi_scsi_rsp *hdr = (struct iscsi_scsi_rsp *)
1906                                 &isert_cmd->tx_desc.iscsi_header;
1907
1908         isert_create_send_desc(isert_conn, isert_cmd, &isert_cmd->tx_desc);
1909         iscsit_build_rsp_pdu(cmd, conn, true, hdr);
1910         isert_init_tx_hdrs(isert_conn, &isert_cmd->tx_desc);
1911         /*
1912          * Attach SENSE DATA payload to iSCSI Response PDU
1913          */
1914         if (cmd->se_cmd.sense_buffer &&
1915             ((cmd->se_cmd.se_cmd_flags & SCF_TRANSPORT_TASK_SENSE) ||
1916             (cmd->se_cmd.se_cmd_flags & SCF_EMULATED_TASK_SENSE))) {
1917                 struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
1918                 struct ib_sge *tx_dsg = &isert_cmd->tx_desc.tx_sg[1];
1919                 u32 padding, pdu_len;
1920
1921                 put_unaligned_be16(cmd->se_cmd.scsi_sense_length,
1922                                    cmd->sense_buffer);
1923                 cmd->se_cmd.scsi_sense_length += sizeof(__be16);
1924
1925                 padding = -(cmd->se_cmd.scsi_sense_length) & 3;
1926                 hton24(hdr->dlength, (u32)cmd->se_cmd.scsi_sense_length);
1927                 pdu_len = cmd->se_cmd.scsi_sense_length + padding;
1928
1929                 isert_cmd->pdu_buf_dma = ib_dma_map_single(ib_dev,
1930                                 (void *)cmd->sense_buffer, pdu_len,
1931                                 DMA_TO_DEVICE);
1932
1933                 isert_cmd->pdu_buf_len = pdu_len;
1934                 tx_dsg->addr    = isert_cmd->pdu_buf_dma;
1935                 tx_dsg->length  = pdu_len;
1936                 tx_dsg->lkey    = isert_conn->conn_mr->lkey;
1937                 isert_cmd->tx_desc.num_sge = 2;
1938         }
1939
1940         isert_init_send_wr(isert_conn, isert_cmd, send_wr, true);
1941
1942         pr_debug("Posting SCSI Response IB_WR_SEND >>>>>>>>>>>>>>>>>>>>>>\n");
1943
1944         return isert_post_response(isert_conn, isert_cmd);
1945 }
1946
1947 static int
1948 isert_put_nopin(struct iscsi_cmd *cmd, struct iscsi_conn *conn,
1949                 bool nopout_response)
1950 {
1951         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
1952         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
1953         struct ib_send_wr *send_wr = &isert_cmd->tx_desc.send_wr;
1954
1955         isert_create_send_desc(isert_conn, isert_cmd, &isert_cmd->tx_desc);
1956         iscsit_build_nopin_rsp(cmd, conn, (struct iscsi_nopin *)
1957                                &isert_cmd->tx_desc.iscsi_header,
1958                                nopout_response);
1959         isert_init_tx_hdrs(isert_conn, &isert_cmd->tx_desc);
1960         isert_init_send_wr(isert_conn, isert_cmd, send_wr, false);
1961
1962         pr_debug("Posting NOPIN Response IB_WR_SEND >>>>>>>>>>>>>>>>>>>>>>\n");
1963
1964         return isert_post_response(isert_conn, isert_cmd);
1965 }
1966
1967 static int
1968 isert_put_logout_rsp(struct iscsi_cmd *cmd, struct iscsi_conn *conn)
1969 {
1970         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
1971         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
1972         struct ib_send_wr *send_wr = &isert_cmd->tx_desc.send_wr;
1973
1974         isert_create_send_desc(isert_conn, isert_cmd, &isert_cmd->tx_desc);
1975         iscsit_build_logout_rsp(cmd, conn, (struct iscsi_logout_rsp *)
1976                                 &isert_cmd->tx_desc.iscsi_header);
1977         isert_init_tx_hdrs(isert_conn, &isert_cmd->tx_desc);
1978         isert_init_send_wr(isert_conn, isert_cmd, send_wr, false);
1979
1980         pr_debug("Posting Logout Response IB_WR_SEND >>>>>>>>>>>>>>>>>>>>>>\n");
1981
1982         return isert_post_response(isert_conn, isert_cmd);
1983 }
1984
1985 static int
1986 isert_put_tm_rsp(struct iscsi_cmd *cmd, struct iscsi_conn *conn)
1987 {
1988         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
1989         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
1990         struct ib_send_wr *send_wr = &isert_cmd->tx_desc.send_wr;
1991
1992         isert_create_send_desc(isert_conn, isert_cmd, &isert_cmd->tx_desc);
1993         iscsit_build_task_mgt_rsp(cmd, conn, (struct iscsi_tm_rsp *)
1994                                   &isert_cmd->tx_desc.iscsi_header);
1995         isert_init_tx_hdrs(isert_conn, &isert_cmd->tx_desc);
1996         isert_init_send_wr(isert_conn, isert_cmd, send_wr, false);
1997
1998         pr_debug("Posting Task Management Response IB_WR_SEND >>>>>>>>>>>>>>>>>>>>>>\n");
1999
2000         return isert_post_response(isert_conn, isert_cmd);
2001 }
2002
2003 static int
2004 isert_put_reject(struct iscsi_cmd *cmd, struct iscsi_conn *conn)
2005 {
2006         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
2007         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
2008         struct ib_send_wr *send_wr = &isert_cmd->tx_desc.send_wr;
2009         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
2010         struct ib_sge *tx_dsg = &isert_cmd->tx_desc.tx_sg[1];
2011         struct iscsi_reject *hdr =
2012                 (struct iscsi_reject *)&isert_cmd->tx_desc.iscsi_header;
2013
2014         isert_create_send_desc(isert_conn, isert_cmd, &isert_cmd->tx_desc);
2015         iscsit_build_reject(cmd, conn, hdr);
2016         isert_init_tx_hdrs(isert_conn, &isert_cmd->tx_desc);
2017
2018         hton24(hdr->dlength, ISCSI_HDR_LEN);
2019         isert_cmd->pdu_buf_dma = ib_dma_map_single(ib_dev,
2020                         (void *)cmd->buf_ptr, ISCSI_HDR_LEN,
2021                         DMA_TO_DEVICE);
2022         isert_cmd->pdu_buf_len = ISCSI_HDR_LEN;
2023         tx_dsg->addr    = isert_cmd->pdu_buf_dma;
2024         tx_dsg->length  = ISCSI_HDR_LEN;
2025         tx_dsg->lkey    = isert_conn->conn_mr->lkey;
2026         isert_cmd->tx_desc.num_sge = 2;
2027
2028         isert_init_send_wr(isert_conn, isert_cmd, send_wr, false);
2029
2030         pr_debug("Posting Reject IB_WR_SEND >>>>>>>>>>>>>>>>>>>>>>\n");
2031
2032         return isert_post_response(isert_conn, isert_cmd);
2033 }
2034
2035 static int
2036 isert_put_text_rsp(struct iscsi_cmd *cmd, struct iscsi_conn *conn)
2037 {
2038         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
2039         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
2040         struct ib_send_wr *send_wr = &isert_cmd->tx_desc.send_wr;
2041         struct iscsi_text_rsp *hdr =
2042                 (struct iscsi_text_rsp *)&isert_cmd->tx_desc.iscsi_header;
2043         u32 txt_rsp_len;
2044         int rc;
2045
2046         isert_create_send_desc(isert_conn, isert_cmd, &isert_cmd->tx_desc);
2047         rc = iscsit_build_text_rsp(cmd, conn, hdr, ISCSI_INFINIBAND);
2048         if (rc < 0)
2049                 return rc;
2050
2051         txt_rsp_len = rc;
2052         isert_init_tx_hdrs(isert_conn, &isert_cmd->tx_desc);
2053
2054         if (txt_rsp_len) {
2055                 struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
2056                 struct ib_sge *tx_dsg = &isert_cmd->tx_desc.tx_sg[1];
2057                 void *txt_rsp_buf = cmd->buf_ptr;
2058
2059                 isert_cmd->pdu_buf_dma = ib_dma_map_single(ib_dev,
2060                                 txt_rsp_buf, txt_rsp_len, DMA_TO_DEVICE);
2061
2062                 isert_cmd->pdu_buf_len = txt_rsp_len;
2063                 tx_dsg->addr    = isert_cmd->pdu_buf_dma;
2064                 tx_dsg->length  = txt_rsp_len;
2065                 tx_dsg->lkey    = isert_conn->conn_mr->lkey;
2066                 isert_cmd->tx_desc.num_sge = 2;
2067         }
2068         isert_init_send_wr(isert_conn, isert_cmd, send_wr, false);
2069
2070         pr_debug("Posting Text Response IB_WR_SEND >>>>>>>>>>>>>>>>>>>>>>\n");
2071
2072         return isert_post_response(isert_conn, isert_cmd);
2073 }
2074
2075 static int
2076 isert_build_rdma_wr(struct isert_conn *isert_conn, struct isert_cmd *isert_cmd,
2077                     struct ib_sge *ib_sge, struct ib_send_wr *send_wr,
2078                     u32 data_left, u32 offset)
2079 {
2080         struct iscsi_cmd *cmd = isert_cmd->iscsi_cmd;
2081         struct scatterlist *sg_start, *tmp_sg;
2082         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
2083         u32 sg_off, page_off;
2084         int i = 0, sg_nents;
2085
2086         sg_off = offset / PAGE_SIZE;
2087         sg_start = &cmd->se_cmd.t_data_sg[sg_off];
2088         sg_nents = min(cmd->se_cmd.t_data_nents - sg_off, isert_conn->max_sge);
2089         page_off = offset % PAGE_SIZE;
2090
2091         send_wr->sg_list = ib_sge;
2092         send_wr->num_sge = sg_nents;
2093         send_wr->wr_id = (unsigned long)&isert_cmd->tx_desc;
2094         /*
2095          * Perform mapping of TCM scatterlist memory ib_sge dma_addr.
2096          */
2097         for_each_sg(sg_start, tmp_sg, sg_nents, i) {
2098                 pr_debug("ISER RDMA from SGL dma_addr: 0x%16llx dma_len: %u, page_off: %u\n",
2099                          (unsigned long long)tmp_sg->dma_address,
2100                          tmp_sg->length, page_off);
2101
2102                 ib_sge->addr = ib_sg_dma_address(ib_dev, tmp_sg) + page_off;
2103                 ib_sge->length = min_t(u32, data_left,
2104                                 ib_sg_dma_len(ib_dev, tmp_sg) - page_off);
2105                 ib_sge->lkey = isert_conn->conn_mr->lkey;
2106
2107                 pr_debug("RDMA ib_sge: addr: 0x%16llx  length: %u lkey: %08x\n",
2108                          ib_sge->addr, ib_sge->length, ib_sge->lkey);
2109                 page_off = 0;
2110                 data_left -= ib_sge->length;
2111                 ib_sge++;
2112                 pr_debug("Incrementing ib_sge pointer to %p\n", ib_sge);
2113         }
2114
2115         pr_debug("Set outgoing sg_list: %p num_sg: %u from TCM SGLs\n",
2116                  send_wr->sg_list, send_wr->num_sge);
2117
2118         return sg_nents;
2119 }
2120
2121 static int
2122 isert_map_rdma(struct iscsi_conn *conn, struct iscsi_cmd *cmd,
2123                struct isert_rdma_wr *wr)
2124 {
2125         struct se_cmd *se_cmd = &cmd->se_cmd;
2126         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
2127         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
2128         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
2129         struct ib_send_wr *send_wr;
2130         struct ib_sge *ib_sge;
2131         struct scatterlist *sg_start;
2132         u32 sg_off = 0, sg_nents;
2133         u32 offset = 0, data_len, data_left, rdma_write_max, va_offset = 0;
2134         int ret = 0, count, i, ib_sge_cnt;
2135
2136         if (wr->iser_ib_op == ISER_IB_RDMA_WRITE) {
2137                 data_left = se_cmd->data_length;
2138         } else {
2139                 sg_off = cmd->write_data_done / PAGE_SIZE;
2140                 data_left = se_cmd->data_length - cmd->write_data_done;
2141                 offset = cmd->write_data_done;
2142                 isert_cmd->tx_desc.isert_cmd = isert_cmd;
2143         }
2144
2145         sg_start = &cmd->se_cmd.t_data_sg[sg_off];
2146         sg_nents = se_cmd->t_data_nents - sg_off;
2147
2148         count = ib_dma_map_sg(ib_dev, sg_start, sg_nents,
2149                               (wr->iser_ib_op == ISER_IB_RDMA_WRITE) ?
2150                               DMA_TO_DEVICE : DMA_FROM_DEVICE);
2151         if (unlikely(!count)) {
2152                 pr_err("Cmd: %p unrable to map SGs\n", isert_cmd);
2153                 return -EINVAL;
2154         }
2155         wr->sge = sg_start;
2156         wr->num_sge = sg_nents;
2157         wr->cur_rdma_length = data_left;
2158         pr_debug("Mapped cmd: %p count: %u sg: %p sg_nents: %u rdma_len %d\n",
2159                  isert_cmd, count, sg_start, sg_nents, data_left);
2160
2161         ib_sge = kzalloc(sizeof(struct ib_sge) * sg_nents, GFP_KERNEL);
2162         if (!ib_sge) {
2163                 pr_warn("Unable to allocate ib_sge\n");
2164                 ret = -ENOMEM;
2165                 goto unmap_sg;
2166         }
2167         wr->ib_sge = ib_sge;
2168
2169         wr->send_wr_num = DIV_ROUND_UP(sg_nents, isert_conn->max_sge);
2170         wr->send_wr = kzalloc(sizeof(struct ib_send_wr) * wr->send_wr_num,
2171                                 GFP_KERNEL);
2172         if (!wr->send_wr) {
2173                 pr_debug("Unable to allocate wr->send_wr\n");
2174                 ret = -ENOMEM;
2175                 goto unmap_sg;
2176         }
2177
2178         wr->isert_cmd = isert_cmd;
2179         rdma_write_max = isert_conn->max_sge * PAGE_SIZE;
2180
2181         for (i = 0; i < wr->send_wr_num; i++) {
2182                 send_wr = &isert_cmd->rdma_wr.send_wr[i];
2183                 data_len = min(data_left, rdma_write_max);
2184
2185                 send_wr->send_flags = 0;
2186                 if (wr->iser_ib_op == ISER_IB_RDMA_WRITE) {
2187                         send_wr->opcode = IB_WR_RDMA_WRITE;
2188                         send_wr->wr.rdma.remote_addr = isert_cmd->read_va + offset;
2189                         send_wr->wr.rdma.rkey = isert_cmd->read_stag;
2190                         if (i + 1 == wr->send_wr_num)
2191                                 send_wr->next = &isert_cmd->tx_desc.send_wr;
2192                         else
2193                                 send_wr->next = &wr->send_wr[i + 1];
2194                 } else {
2195                         send_wr->opcode = IB_WR_RDMA_READ;
2196                         send_wr->wr.rdma.remote_addr = isert_cmd->write_va + va_offset;
2197                         send_wr->wr.rdma.rkey = isert_cmd->write_stag;
2198                         if (i + 1 == wr->send_wr_num)
2199                                 send_wr->send_flags = IB_SEND_SIGNALED;
2200                         else
2201                                 send_wr->next = &wr->send_wr[i + 1];
2202                 }
2203
2204                 ib_sge_cnt = isert_build_rdma_wr(isert_conn, isert_cmd, ib_sge,
2205                                         send_wr, data_len, offset);
2206                 ib_sge += ib_sge_cnt;
2207
2208                 offset += data_len;
2209                 va_offset += data_len;
2210                 data_left -= data_len;
2211         }
2212
2213         return 0;
2214 unmap_sg:
2215         ib_dma_unmap_sg(ib_dev, sg_start, sg_nents,
2216                         (wr->iser_ib_op == ISER_IB_RDMA_WRITE) ?
2217                         DMA_TO_DEVICE : DMA_FROM_DEVICE);
2218         return ret;
2219 }
2220
2221 static int
2222 isert_map_fr_pagelist(struct ib_device *ib_dev,
2223                       struct scatterlist *sg_start, int sg_nents, u64 *fr_pl)
2224 {
2225         u64 start_addr, end_addr, page, chunk_start = 0;
2226         struct scatterlist *tmp_sg;
2227         int i = 0, new_chunk, last_ent, n_pages;
2228
2229         n_pages = 0;
2230         new_chunk = 1;
2231         last_ent = sg_nents - 1;
2232         for_each_sg(sg_start, tmp_sg, sg_nents, i) {
2233                 start_addr = ib_sg_dma_address(ib_dev, tmp_sg);
2234                 if (new_chunk)
2235                         chunk_start = start_addr;
2236                 end_addr = start_addr + ib_sg_dma_len(ib_dev, tmp_sg);
2237
2238                 pr_debug("SGL[%d] dma_addr: 0x%16llx len: %u\n",
2239                          i, (unsigned long long)tmp_sg->dma_address,
2240                          tmp_sg->length);
2241
2242                 if ((end_addr & ~PAGE_MASK) && i < last_ent) {
2243                         new_chunk = 0;
2244                         continue;
2245                 }
2246                 new_chunk = 1;
2247
2248                 page = chunk_start & PAGE_MASK;
2249                 do {
2250                         fr_pl[n_pages++] = page;
2251                         pr_debug("Mapped page_list[%d] page_addr: 0x%16llx\n",
2252                                  n_pages - 1, page);
2253                         page += PAGE_SIZE;
2254                 } while (page < end_addr);
2255         }
2256
2257         return n_pages;
2258 }
2259
2260 static int
2261 isert_fast_reg_mr(struct fast_reg_descriptor *fr_desc,
2262                   struct isert_conn *isert_conn, struct scatterlist *sg_start,
2263                   struct ib_sge *ib_sge, u32 sg_nents, u32 offset,
2264                   unsigned int data_len)
2265 {
2266         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
2267         struct ib_send_wr fr_wr, inv_wr;
2268         struct ib_send_wr *bad_wr, *wr = NULL;
2269         int ret, pagelist_len;
2270         u32 page_off;
2271         u8 key;
2272
2273         sg_nents = min_t(unsigned int, sg_nents, ISCSI_ISER_SG_TABLESIZE);
2274         page_off = offset % PAGE_SIZE;
2275
2276         pr_debug("Use fr_desc %p sg_nents %d offset %u\n",
2277                  fr_desc, sg_nents, offset);
2278
2279         pagelist_len = isert_map_fr_pagelist(ib_dev, sg_start, sg_nents,
2280                                              &fr_desc->data_frpl->page_list[0]);
2281
2282         if (!fr_desc->valid) {
2283                 memset(&inv_wr, 0, sizeof(inv_wr));
2284                 inv_wr.wr_id = ISER_FASTREG_LI_WRID;
2285                 inv_wr.opcode = IB_WR_LOCAL_INV;
2286                 inv_wr.ex.invalidate_rkey = fr_desc->data_mr->rkey;
2287                 wr = &inv_wr;
2288                 /* Bump the key */
2289                 key = (u8)(fr_desc->data_mr->rkey & 0x000000FF);
2290                 ib_update_fast_reg_key(fr_desc->data_mr, ++key);
2291         }
2292
2293         /* Prepare FASTREG WR */
2294         memset(&fr_wr, 0, sizeof(fr_wr));
2295         fr_wr.wr_id = ISER_FASTREG_LI_WRID;
2296         fr_wr.opcode = IB_WR_FAST_REG_MR;
2297         fr_wr.wr.fast_reg.iova_start =
2298                 fr_desc->data_frpl->page_list[0] + page_off;
2299         fr_wr.wr.fast_reg.page_list = fr_desc->data_frpl;
2300         fr_wr.wr.fast_reg.page_list_len = pagelist_len;
2301         fr_wr.wr.fast_reg.page_shift = PAGE_SHIFT;
2302         fr_wr.wr.fast_reg.length = data_len;
2303         fr_wr.wr.fast_reg.rkey = fr_desc->data_mr->rkey;
2304         fr_wr.wr.fast_reg.access_flags = IB_ACCESS_LOCAL_WRITE;
2305
2306         if (!wr)
2307                 wr = &fr_wr;
2308         else
2309                 wr->next = &fr_wr;
2310
2311         ret = ib_post_send(isert_conn->conn_qp, wr, &bad_wr);
2312         if (ret) {
2313                 pr_err("fast registration failed, ret:%d\n", ret);
2314                 return ret;
2315         }
2316         fr_desc->valid = false;
2317
2318         ib_sge->lkey = fr_desc->data_mr->lkey;
2319         ib_sge->addr = fr_desc->data_frpl->page_list[0] + page_off;
2320         ib_sge->length = data_len;
2321
2322         pr_debug("RDMA ib_sge: addr: 0x%16llx  length: %u lkey: %08x\n",
2323                  ib_sge->addr, ib_sge->length, ib_sge->lkey);
2324
2325         return ret;
2326 }
2327
2328 static int
2329 isert_reg_rdma(struct iscsi_conn *conn, struct iscsi_cmd *cmd,
2330                struct isert_rdma_wr *wr)
2331 {
2332         struct se_cmd *se_cmd = &cmd->se_cmd;
2333         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
2334         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
2335         struct ib_device *ib_dev = isert_conn->conn_cm_id->device;
2336         struct ib_send_wr *send_wr;
2337         struct ib_sge *ib_sge;
2338         struct scatterlist *sg_start;
2339         struct fast_reg_descriptor *fr_desc;
2340         u32 sg_off = 0, sg_nents;
2341         u32 offset = 0, data_len, data_left, rdma_write_max;
2342         int ret = 0, count;
2343         unsigned long flags;
2344
2345         if (wr->iser_ib_op == ISER_IB_RDMA_WRITE) {
2346                 data_left = se_cmd->data_length;
2347         } else {
2348                 offset = cmd->write_data_done;
2349                 sg_off = offset / PAGE_SIZE;
2350                 data_left = se_cmd->data_length - cmd->write_data_done;
2351                 isert_cmd->tx_desc.isert_cmd = isert_cmd;
2352         }
2353
2354         sg_start = &cmd->se_cmd.t_data_sg[sg_off];
2355         sg_nents = se_cmd->t_data_nents - sg_off;
2356
2357         count = ib_dma_map_sg(ib_dev, sg_start, sg_nents,
2358                               (wr->iser_ib_op == ISER_IB_RDMA_WRITE) ?
2359                               DMA_TO_DEVICE : DMA_FROM_DEVICE);
2360         if (unlikely(!count)) {
2361                 pr_err("Cmd: %p unrable to map SGs\n", isert_cmd);
2362                 return -EINVAL;
2363         }
2364         wr->sge = sg_start;
2365         wr->num_sge = sg_nents;
2366         pr_debug("Mapped cmd: %p count: %u sg: %p sg_nents: %u rdma_len %d\n",
2367                  isert_cmd, count, sg_start, sg_nents, data_left);
2368
2369         memset(&wr->s_ib_sge, 0, sizeof(*ib_sge));
2370         ib_sge = &wr->s_ib_sge;
2371         wr->ib_sge = ib_sge;
2372
2373         wr->send_wr_num = 1;
2374         memset(&wr->s_send_wr, 0, sizeof(*send_wr));
2375         wr->send_wr = &wr->s_send_wr;
2376
2377         wr->isert_cmd = isert_cmd;
2378         rdma_write_max = ISCSI_ISER_SG_TABLESIZE * PAGE_SIZE;
2379
2380         send_wr = &isert_cmd->rdma_wr.s_send_wr;
2381         send_wr->sg_list = ib_sge;
2382         send_wr->num_sge = 1;
2383         send_wr->wr_id = (unsigned long)&isert_cmd->tx_desc;
2384         if (wr->iser_ib_op == ISER_IB_RDMA_WRITE) {
2385                 send_wr->opcode = IB_WR_RDMA_WRITE;
2386                 send_wr->wr.rdma.remote_addr = isert_cmd->read_va;
2387                 send_wr->wr.rdma.rkey = isert_cmd->read_stag;
2388                 send_wr->send_flags = 0;
2389                 send_wr->next = &isert_cmd->tx_desc.send_wr;
2390         } else {
2391                 send_wr->opcode = IB_WR_RDMA_READ;
2392                 send_wr->wr.rdma.remote_addr = isert_cmd->write_va;
2393                 send_wr->wr.rdma.rkey = isert_cmd->write_stag;
2394                 send_wr->send_flags = IB_SEND_SIGNALED;
2395         }
2396
2397         data_len = min(data_left, rdma_write_max);
2398         wr->cur_rdma_length = data_len;
2399
2400         /* if there is a single dma entry, dma mr is sufficient */
2401         if (count == 1) {
2402                 ib_sge->addr = ib_sg_dma_address(ib_dev, &sg_start[0]);
2403                 ib_sge->length = ib_sg_dma_len(ib_dev, &sg_start[0]);
2404                 ib_sge->lkey = isert_conn->conn_mr->lkey;
2405                 wr->fr_desc = NULL;
2406         } else {
2407                 spin_lock_irqsave(&isert_conn->conn_lock, flags);
2408                 fr_desc = list_first_entry(&isert_conn->conn_fr_pool,
2409                                            struct fast_reg_descriptor, list);
2410                 list_del(&fr_desc->list);
2411                 spin_unlock_irqrestore(&isert_conn->conn_lock, flags);
2412                 wr->fr_desc = fr_desc;
2413
2414                 ret = isert_fast_reg_mr(fr_desc, isert_conn, sg_start,
2415                                         ib_sge, sg_nents, offset, data_len);
2416                 if (ret) {
2417                         list_add_tail(&fr_desc->list, &isert_conn->conn_fr_pool);
2418                         goto unmap_sg;
2419                 }
2420         }
2421
2422         return 0;
2423
2424 unmap_sg:
2425         ib_dma_unmap_sg(ib_dev, sg_start, sg_nents,
2426                         (wr->iser_ib_op == ISER_IB_RDMA_WRITE) ?
2427                         DMA_TO_DEVICE : DMA_FROM_DEVICE);
2428         return ret;
2429 }
2430
2431 static int
2432 isert_put_datain(struct iscsi_conn *conn, struct iscsi_cmd *cmd)
2433 {
2434         struct se_cmd *se_cmd = &cmd->se_cmd;
2435         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
2436         struct isert_rdma_wr *wr = &isert_cmd->rdma_wr;
2437         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
2438         struct isert_device *device = isert_conn->conn_device;
2439         struct ib_send_wr *wr_failed;
2440         int rc;
2441
2442         pr_debug("Cmd: %p RDMA_WRITE data_length: %u\n",
2443                  isert_cmd, se_cmd->data_length);
2444         wr->iser_ib_op = ISER_IB_RDMA_WRITE;
2445         rc = device->reg_rdma_mem(conn, cmd, wr);
2446         if (rc) {
2447                 pr_err("Cmd: %p failed to prepare RDMA res\n", isert_cmd);
2448                 return rc;
2449         }
2450
2451         /*
2452          * Build isert_conn->tx_desc for iSCSI response PDU and attach
2453          */
2454         isert_create_send_desc(isert_conn, isert_cmd, &isert_cmd->tx_desc);
2455         iscsit_build_rsp_pdu(cmd, conn, true, (struct iscsi_scsi_rsp *)
2456                              &isert_cmd->tx_desc.iscsi_header);
2457         isert_init_tx_hdrs(isert_conn, &isert_cmd->tx_desc);
2458         isert_init_send_wr(isert_conn, isert_cmd,
2459                            &isert_cmd->tx_desc.send_wr, true);
2460
2461         atomic_add(wr->send_wr_num + 1, &isert_conn->post_send_buf_count);
2462
2463         rc = ib_post_send(isert_conn->conn_qp, wr->send_wr, &wr_failed);
2464         if (rc) {
2465                 pr_warn("ib_post_send() failed for IB_WR_RDMA_WRITE\n");
2466                 atomic_sub(wr->send_wr_num + 1, &isert_conn->post_send_buf_count);
2467         }
2468         pr_debug("Cmd: %p posted RDMA_WRITE + Response for iSER Data READ\n",
2469                  isert_cmd);
2470
2471         return 1;
2472 }
2473
2474 static int
2475 isert_get_dataout(struct iscsi_conn *conn, struct iscsi_cmd *cmd, bool recovery)
2476 {
2477         struct se_cmd *se_cmd = &cmd->se_cmd;
2478         struct isert_cmd *isert_cmd = iscsit_priv_cmd(cmd);
2479         struct isert_rdma_wr *wr = &isert_cmd->rdma_wr;
2480         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
2481         struct isert_device *device = isert_conn->conn_device;
2482         struct ib_send_wr *wr_failed;
2483         int rc;
2484
2485         pr_debug("Cmd: %p RDMA_READ data_length: %u write_data_done: %u\n",
2486                  isert_cmd, se_cmd->data_length, cmd->write_data_done);
2487         wr->iser_ib_op = ISER_IB_RDMA_READ;
2488         rc = device->reg_rdma_mem(conn, cmd, wr);
2489         if (rc) {
2490                 pr_err("Cmd: %p failed to prepare RDMA res\n", isert_cmd);
2491                 return rc;
2492         }
2493
2494         atomic_add(wr->send_wr_num, &isert_conn->post_send_buf_count);
2495
2496         rc = ib_post_send(isert_conn->conn_qp, wr->send_wr, &wr_failed);
2497         if (rc) {
2498                 pr_warn("ib_post_send() failed for IB_WR_RDMA_READ\n");
2499                 atomic_sub(wr->send_wr_num, &isert_conn->post_send_buf_count);
2500         }
2501         pr_debug("Cmd: %p posted RDMA_READ memory for ISER Data WRITE\n",
2502                  isert_cmd);
2503
2504         return 0;
2505 }
2506
2507 static int
2508 isert_immediate_queue(struct iscsi_conn *conn, struct iscsi_cmd *cmd, int state)
2509 {
2510         int ret;
2511
2512         switch (state) {
2513         case ISTATE_SEND_NOPIN_WANT_RESPONSE:
2514                 ret = isert_put_nopin(cmd, conn, false);
2515                 break;
2516         default:
2517                 pr_err("Unknown immediate state: 0x%02x\n", state);
2518                 ret = -EINVAL;
2519                 break;
2520         }
2521
2522         return ret;
2523 }
2524
2525 static int
2526 isert_response_queue(struct iscsi_conn *conn, struct iscsi_cmd *cmd, int state)
2527 {
2528         int ret;
2529
2530         switch (state) {
2531         case ISTATE_SEND_LOGOUTRSP:
2532                 ret = isert_put_logout_rsp(cmd, conn);
2533                 if (!ret) {
2534                         pr_debug("Returning iSER Logout -EAGAIN\n");
2535                         ret = -EAGAIN;
2536                 }
2537                 break;
2538         case ISTATE_SEND_NOPIN:
2539                 ret = isert_put_nopin(cmd, conn, true);
2540                 break;
2541         case ISTATE_SEND_TASKMGTRSP:
2542                 ret = isert_put_tm_rsp(cmd, conn);
2543                 break;
2544         case ISTATE_SEND_REJECT:
2545                 ret = isert_put_reject(cmd, conn);
2546                 break;
2547         case ISTATE_SEND_TEXTRSP:
2548                 ret = isert_put_text_rsp(cmd, conn);
2549                 break;
2550         case ISTATE_SEND_STATUS:
2551                 /*
2552                  * Special case for sending non GOOD SCSI status from TX thread
2553                  * context during pre se_cmd excecution failure.
2554                  */
2555                 ret = isert_put_response(conn, cmd);
2556                 break;
2557         default:
2558                 pr_err("Unknown response state: 0x%02x\n", state);
2559                 ret = -EINVAL;
2560                 break;
2561         }
2562
2563         return ret;
2564 }
2565
2566 static int
2567 isert_setup_np(struct iscsi_np *np,
2568                struct __kernel_sockaddr_storage *ksockaddr)
2569 {
2570         struct isert_np *isert_np;
2571         struct rdma_cm_id *isert_lid;
2572         struct sockaddr *sa;
2573         int ret;
2574
2575         isert_np = kzalloc(sizeof(struct isert_np), GFP_KERNEL);
2576         if (!isert_np) {
2577                 pr_err("Unable to allocate struct isert_np\n");
2578                 return -ENOMEM;
2579         }
2580         sema_init(&isert_np->np_sem, 0);
2581         mutex_init(&isert_np->np_accept_mutex);
2582         INIT_LIST_HEAD(&isert_np->np_accept_list);
2583         init_completion(&isert_np->np_login_comp);
2584
2585         sa = (struct sockaddr *)ksockaddr;
2586         pr_debug("ksockaddr: %p, sa: %p\n", ksockaddr, sa);
2587         /*
2588          * Setup the np->np_sockaddr from the passed sockaddr setup
2589          * in iscsi_target_configfs.c code..
2590          */
2591         memcpy(&np->np_sockaddr, ksockaddr,
2592                sizeof(struct __kernel_sockaddr_storage));
2593
2594         isert_lid = rdma_create_id(isert_cma_handler, np, RDMA_PS_TCP,
2595                                 IB_QPT_RC);
2596         if (IS_ERR(isert_lid)) {
2597                 pr_err("rdma_create_id() for isert_listen_handler failed: %ld\n",
2598                        PTR_ERR(isert_lid));
2599                 ret = PTR_ERR(isert_lid);
2600                 goto out;
2601         }
2602
2603         ret = rdma_bind_addr(isert_lid, sa);
2604         if (ret) {
2605                 pr_err("rdma_bind_addr() for isert_lid failed: %d\n", ret);
2606                 goto out_lid;
2607         }
2608
2609         ret = rdma_listen(isert_lid, ISERT_RDMA_LISTEN_BACKLOG);
2610         if (ret) {
2611                 pr_err("rdma_listen() for isert_lid failed: %d\n", ret);
2612                 goto out_lid;
2613         }
2614
2615         isert_np->np_cm_id = isert_lid;
2616         np->np_context = isert_np;
2617         pr_debug("Setup isert_lid->context: %p\n", isert_lid->context);
2618
2619         return 0;
2620
2621 out_lid:
2622         rdma_destroy_id(isert_lid);
2623 out:
2624         kfree(isert_np);
2625         return ret;
2626 }
2627
2628 static int
2629 isert_rdma_accept(struct isert_conn *isert_conn)
2630 {
2631         struct rdma_cm_id *cm_id = isert_conn->conn_cm_id;
2632         struct rdma_conn_param cp;
2633         int ret;
2634
2635         memset(&cp, 0, sizeof(struct rdma_conn_param));
2636         cp.responder_resources = isert_conn->responder_resources;
2637         cp.initiator_depth = isert_conn->initiator_depth;
2638         cp.retry_count = 7;
2639         cp.rnr_retry_count = 7;
2640
2641         pr_debug("Before rdma_accept >>>>>>>>>>>>>>>>>>>>.\n");
2642
2643         ret = rdma_accept(cm_id, &cp);
2644         if (ret) {
2645                 pr_err("rdma_accept() failed with: %d\n", ret);
2646                 return ret;
2647         }
2648
2649         pr_debug("After rdma_accept >>>>>>>>>>>>>>>>>>>>>.\n");
2650
2651         return 0;
2652 }
2653
2654 static int
2655 isert_get_login_rx(struct iscsi_conn *conn, struct iscsi_login *login)
2656 {
2657         struct isert_conn *isert_conn = (struct isert_conn *)conn->context;
2658         int ret;
2659
2660         pr_debug("isert_get_login_rx before conn_login_comp conn: %p\n", conn);
2661         /*
2662          * For login requests after the first PDU, isert_rx_login_req() will
2663          * kick schedule_delayed_work(&conn->login_work) as the packet is
2664          * received, which turns this callback from iscsi_target_do_login_rx()
2665          * into a NOP.
2666          */
2667         if (!login->first_request)
2668                 return 0;
2669
2670         ret = wait_for_completion_interruptible(&isert_conn->conn_login_comp);
2671         if (ret)
2672                 return ret;
2673
2674         pr_debug("isert_get_login_rx processing login->req: %p\n", login->req);
2675         return 0;
2676 }
2677
2678 static void
2679 isert_set_conn_info(struct iscsi_np *np, struct iscsi_conn *conn,
2680                     struct isert_conn *isert_conn)
2681 {
2682         struct rdma_cm_id *cm_id = isert_conn->conn_cm_id;
2683         struct rdma_route *cm_route = &cm_id->route;
2684         struct sockaddr_in *sock_in;
2685         struct sockaddr_in6 *sock_in6;
2686
2687         conn->login_family = np->np_sockaddr.ss_family;
2688
2689         if (np->np_sockaddr.ss_family == AF_INET6) {
2690                 sock_in6 = (struct sockaddr_in6 *)&cm_route->addr.dst_addr;
2691                 snprintf(conn->login_ip, sizeof(conn->login_ip), "%pI6c",
2692                          &sock_in6->sin6_addr.in6_u);
2693                 conn->login_port = ntohs(sock_in6->sin6_port);
2694
2695                 sock_in6 = (struct sockaddr_in6 *)&cm_route->addr.src_addr;
2696                 snprintf(conn->local_ip, sizeof(conn->local_ip), "%pI6c",
2697                          &sock_in6->sin6_addr.in6_u);
2698                 conn->local_port = ntohs(sock_in6->sin6_port);
2699         } else {
2700                 sock_in = (struct sockaddr_in *)&cm_route->addr.dst_addr;
2701                 sprintf(conn->login_ip, "%pI4",
2702                         &sock_in->sin_addr.s_addr);
2703                 conn->login_port = ntohs(sock_in->sin_port);
2704
2705                 sock_in = (struct sockaddr_in *)&cm_route->addr.src_addr;
2706                 sprintf(conn->local_ip, "%pI4",
2707                         &sock_in->sin_addr.s_addr);
2708                 conn->local_port = ntohs(sock_in->sin_port);
2709         }
2710 }
2711
2712 static int
2713 isert_accept_np(struct iscsi_np *np, struct iscsi_conn *conn)
2714 {
2715         struct isert_np *isert_np = (struct isert_np *)np->np_context;
2716         struct isert_conn *isert_conn;
2717         int max_accept = 0, ret;
2718
2719 accept_wait:
2720         ret = down_interruptible(&isert_np->np_sem);
2721         if (max_accept > 5)
2722                 return -ENODEV;
2723
2724         spin_lock_bh(&np->np_thread_lock);
2725         if (np->np_thread_state >= ISCSI_NP_THREAD_RESET) {
2726                 spin_unlock_bh(&np->np_thread_lock);
2727                 pr_debug("np_thread_state %d for isert_accept_np\n",
2728                          np->np_thread_state);
2729                 /**
2730                  * No point in stalling here when np_thread
2731                  * is in state RESET/SHUTDOWN/EXIT - bail
2732                  **/
2733                 return -ENODEV;
2734         }
2735         spin_unlock_bh(&np->np_thread_lock);
2736
2737         mutex_lock(&isert_np->np_accept_mutex);
2738         if (list_empty(&isert_np->np_accept_list)) {
2739                 mutex_unlock(&isert_np->np_accept_mutex);
2740                 max_accept++;
2741                 goto accept_wait;
2742         }
2743         isert_conn = list_first_entry(&isert_np->np_accept_list,
2744                         struct isert_conn, conn_accept_node);
2745         list_del_init(&isert_conn->conn_accept_node);
2746         mutex_unlock(&isert_np->np_accept_mutex);
2747
2748         conn->context = isert_conn;
2749         isert_conn->conn = conn;
2750         max_accept = 0;
2751
2752         ret = isert_rdma_post_recvl(isert_conn);
2753         if (ret)
2754                 return ret;
2755
2756         ret = isert_rdma_accept(isert_conn);
2757         if (ret)
2758                 return ret;
2759
2760         isert_set_conn_info(np, conn, isert_conn);
2761
2762         pr_debug("Processing isert_accept_np: isert_conn: %p\n", isert_conn);
2763         return 0;
2764 }
2765
2766 static void
2767 isert_free_np(struct iscsi_np *np)
2768 {
2769         struct isert_np *isert_np = (struct isert_np *)np->np_context;
2770
2771         rdma_destroy_id(isert_np->np_cm_id);
2772
2773         np->np_context = NULL;
2774         kfree(isert_np);
2775 }
2776
2777 static void isert_wait_conn(struct iscsi_conn *conn)
2778 {
2779         struct isert_conn *isert_conn = conn->context;
2780
2781         pr_debug("isert_wait_conn: Starting \n");
2782
2783         mutex_lock(&isert_conn->conn_mutex);
2784         if (isert_conn->conn_cm_id) {
2785                 pr_debug("Calling rdma_disconnect from isert_wait_conn\n");
2786                 rdma_disconnect(isert_conn->conn_cm_id);
2787         }
2788         /*
2789          * Only wait for conn_wait_comp_err if the isert_conn made it
2790          * into full feature phase..
2791          */
2792         if (isert_conn->state == ISER_CONN_INIT) {
2793                 mutex_unlock(&isert_conn->conn_mutex);
2794                 return;
2795         }
2796         if (isert_conn->state == ISER_CONN_UP)
2797                 isert_conn->state = ISER_CONN_TERMINATING;
2798         mutex_unlock(&isert_conn->conn_mutex);
2799
2800         wait_for_completion(&isert_conn->conn_wait_comp_err);
2801
2802         wait_for_completion(&isert_conn->conn_wait);
2803 }
2804
2805 static void isert_free_conn(struct iscsi_conn *conn)
2806 {
2807         struct isert_conn *isert_conn = conn->context;
2808
2809         isert_put_conn(isert_conn);
2810 }
2811
2812 static struct iscsit_transport iser_target_transport = {
2813         .name                   = "IB/iSER",
2814         .transport_type         = ISCSI_INFINIBAND,
2815         .priv_size              = sizeof(struct isert_cmd),
2816         .owner                  = THIS_MODULE,
2817         .iscsit_setup_np        = isert_setup_np,
2818         .iscsit_accept_np       = isert_accept_np,
2819         .iscsit_free_np         = isert_free_np,
2820         .iscsit_wait_conn       = isert_wait_conn,
2821         .iscsit_free_conn       = isert_free_conn,
2822         .iscsit_get_login_rx    = isert_get_login_rx,
2823         .iscsit_put_login_tx    = isert_put_login_tx,
2824         .iscsit_immediate_queue = isert_immediate_queue,
2825         .iscsit_response_queue  = isert_response_queue,
2826         .iscsit_get_dataout     = isert_get_dataout,
2827         .iscsit_queue_data_in   = isert_put_datain,
2828         .iscsit_queue_status    = isert_put_response,
2829 };
2830
2831 static int __init isert_init(void)
2832 {
2833         int ret;
2834
2835         isert_rx_wq = alloc_workqueue("isert_rx_wq", 0, 0);
2836         if (!isert_rx_wq) {
2837                 pr_err("Unable to allocate isert_rx_wq\n");
2838                 return -ENOMEM;
2839         }
2840
2841         isert_comp_wq = alloc_workqueue("isert_comp_wq", 0, 0);
2842         if (!isert_comp_wq) {
2843                 pr_err("Unable to allocate isert_comp_wq\n");
2844                 ret = -ENOMEM;
2845                 goto destroy_rx_wq;
2846         }
2847
2848         iscsit_register_transport(&iser_target_transport);
2849         pr_debug("iSER_TARGET[0] - Loaded iser_target_transport\n");
2850         return 0;
2851
2852 destroy_rx_wq:
2853         destroy_workqueue(isert_rx_wq);
2854         return ret;
2855 }
2856
2857 static void __exit isert_exit(void)
2858 {
2859         flush_scheduled_work();
2860         destroy_workqueue(isert_comp_wq);
2861         destroy_workqueue(isert_rx_wq);
2862         iscsit_unregister_transport(&iser_target_transport);
2863         pr_debug("iSER_TARGET[0] - Released iser_target_transport\n");
2864 }
2865
2866 MODULE_DESCRIPTION("iSER-Target for mainline target infrastructure");
2867 MODULE_VERSION("0.1");
2868 MODULE_AUTHOR("nab@Linux-iSCSI.org");
2869 MODULE_LICENSE("GPL");
2870
2871 module_init(isert_init);
2872 module_exit(isert_exit);