drivers/target/iscsi/iscsi_target.c

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