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