drivers/target/iscsi/iscsi_target.c

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