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                  * From RFC-3720 Section 10.3.1:
 865                  *
 866                  * "Either or both of R and W MAY be 1 when either the
 867                  *  Expected Data Transfer Length and/or Bidirectional Read
 868                  *  Expected Data Transfer Length are 0"
 869                  *
 870                  * For this case, go ahead and clear the unnecssary bits
 871                  * to avoid any confusion with ->data_direction.
 872                  */
 873                 hdr->flags &= ~ISCSI_FLAG_CMD_READ;
 874                 hdr->flags &= ~ISCSI_FLAG_CMD_WRITE;
 875
 876                 pr_warn("ISCSI_FLAG_CMD_READ or ISCSI_FLAG_CMD_WRITE"
 877                         " set when Expected Data Transfer Length is 0 for"
 878                         " CDB: 0x%02x, Fixing up flags\n", hdr->cdb[0]);
 879         }
 880
 881         if (!(hdr->flags & ISCSI_FLAG_CMD_READ) &&
 882             !(hdr->flags & ISCSI_FLAG_CMD_WRITE) && (hdr->data_length != 0)) {
 883                 pr_err("ISCSI_FLAG_CMD_READ and/or ISCSI_FLAG_CMD_WRITE"
 884                         " MUST be set if Expected Data Transfer Length is not 0."
 885                         " Bad iSCSI Initiator\n");
 886                 return iscsit_add_reject(ISCSI_REASON_BOOKMARK_INVALID, 1,
 887                                 buf, conn);
 888         }
 889
 890         if ((hdr->flags & ISCSI_FLAG_CMD_READ) &&
 891             (hdr->flags & ISCSI_FLAG_CMD_WRITE)) {
 892                 pr_err("Bidirectional operations not supported!\n");
 893                 return iscsit_add_reject(ISCSI_REASON_BOOKMARK_INVALID, 1,
 894                                 buf, conn);
 895         }
 896
 897         if (hdr->opcode & ISCSI_OP_IMMEDIATE) {
 898                 pr_err("Illegally set Immediate Bit in iSCSI Initiator"
 899                                 " Scsi Command PDU.\n");
 900                 return iscsit_add_reject(ISCSI_REASON_BOOKMARK_INVALID, 1,
 901                                 buf, conn);
 902         }
 903
 904         if (payload_length && !conn->sess->sess_ops->ImmediateData) {
 905                 pr_err("ImmediateData=No but DataSegmentLength=%u,"
 906                         " protocol error.\n", payload_length);
 907                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
 908                                 buf, conn);
 909         }
 910
 911         if ((hdr->data_length == payload_length) &&
 912             (!(hdr->flags & ISCSI_FLAG_CMD_FINAL))) {
 913                 pr_err("Expected Data Transfer Length and Length of"
 914                         " Immediate Data are the same, but ISCSI_FLAG_CMD_FINAL"
 915                         " bit is not set protocol error\n");
 916                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
 917                                 buf, conn);
 918         }
 919
 920         if (payload_length > hdr->data_length) {
 921                 pr_err("DataSegmentLength: %u is greater than"
 922                         " EDTL: %u, protocol error.\n", payload_length,
 923                                 hdr->data_length);
 924                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
 925                                 buf, conn);
 926         }
 927
 928         if (payload_length > conn->conn_ops->MaxRecvDataSegmentLength) {
 929                 pr_err("DataSegmentLength: %u is greater than"
 930                         " MaxRecvDataSegmentLength: %u, protocol error.\n",
 931                         payload_length, conn->conn_ops->MaxRecvDataSegmentLength);
 932                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
 933                                 buf, conn);
 934         }
 935
 936         if (payload_length > conn->sess->sess_ops->FirstBurstLength) {
 937                 pr_err("DataSegmentLength: %u is greater than"
 938                         " FirstBurstLength: %u, protocol error.\n",
 939                         payload_length, conn->sess->sess_ops->FirstBurstLength);
 940                 return iscsit_add_reject(ISCSI_REASON_BOOKMARK_INVALID, 1,
 941                                         buf, conn);
 942         }
 943
 944         data_direction = (hdr->flags & ISCSI_FLAG_CMD_WRITE) ? DMA_TO_DEVICE :
 945                          (hdr->flags & ISCSI_FLAG_CMD_READ) ? DMA_FROM_DEVICE :
 946                           DMA_NONE;
 947
 948         cmd = iscsit_allocate_se_cmd(conn, hdr->data_length, data_direction,
 949                                 (hdr->flags & ISCSI_FLAG_CMD_ATTR_MASK));
 950         if (!cmd)
 951                 return iscsit_add_reject(ISCSI_REASON_BOOKMARK_NO_RESOURCES, 1,
 952                                         buf, conn);
 953
 954         pr_debug("Got SCSI Command, ITT: 0x%08x, CmdSN: 0x%08x,"
 955                 " ExpXferLen: %u, Length: %u, CID: %hu\n", hdr->itt,
 956                 hdr->cmdsn, hdr->data_length, payload_length, conn->cid);
 957
 958         cmd->iscsi_opcode       = ISCSI_OP_SCSI_CMD;
 959         cmd->i_state            = ISTATE_NEW_CMD;
 960         cmd->immediate_cmd      = ((hdr->opcode & ISCSI_OP_IMMEDIATE) ? 1 : 0);
 961         cmd->immediate_data     = (payload_length) ? 1 : 0;
 962         cmd->unsolicited_data   = ((!(hdr->flags & ISCSI_FLAG_CMD_FINAL) &&
 963                                      (hdr->flags & ISCSI_FLAG_CMD_WRITE)) ? 1 : 0);
 964         if (cmd->unsolicited_data)
 965                 cmd->cmd_flags |= ICF_NON_IMMEDIATE_UNSOLICITED_DATA;
 966
 967         conn->sess->init_task_tag = cmd->init_task_tag = hdr->itt;
 968         if (hdr->flags & ISCSI_FLAG_CMD_READ) {
 969                 spin_lock_bh(&conn->sess->ttt_lock);
 970                 cmd->targ_xfer_tag = conn->sess->targ_xfer_tag++;
 971                 if (cmd->targ_xfer_tag == 0xFFFFFFFF)
 972                         cmd->targ_xfer_tag = conn->sess->targ_xfer_tag++;
 973                 spin_unlock_bh(&conn->sess->ttt_lock);
 974         } else if (hdr->flags & ISCSI_FLAG_CMD_WRITE)
 975                 cmd->targ_xfer_tag = 0xFFFFFFFF;
 976         cmd->cmd_sn             = hdr->cmdsn;
 977         cmd->exp_stat_sn        = hdr->exp_statsn;
 978         cmd->first_burst_len    = payload_length;
 979
 980         if (cmd->data_direction == DMA_FROM_DEVICE) {
 981                 struct iscsi_datain_req *dr;
 982
 983                 dr = iscsit_allocate_datain_req();
 984                 if (!dr)
 985                         return iscsit_add_reject_from_cmd(
 986                                         ISCSI_REASON_BOOKMARK_NO_RESOURCES,
 987                                         1, 1, buf, cmd);
 988
 989                 iscsit_attach_datain_req(cmd, dr);
 990         }
 991
 992         /*
 993          * The CDB is going to an se_device_t.
 994          */
 995         ret = iscsit_get_lun_for_cmd(cmd, hdr->cdb,
 996                                 get_unaligned_le64(&hdr->lun));
 997         if (ret < 0) {
 998                 if (cmd->se_cmd.scsi_sense_reason == TCM_NON_EXISTENT_LUN) {
 999                         pr_debug("Responding to non-acl'ed,"
1000                                 " non-existent or non-exported iSCSI LUN:"
1001                                 " 0x%016Lx\n", get_unaligned_le64(&hdr->lun));
1002                 }
1003                 send_check_condition = 1;
1004                 goto attach_cmd;
1005         }
1006         /*
1007          * The Initiator Node has access to the LUN (the addressing method
1008          * is handled inside of iscsit_get_lun_for_cmd()).  Now it's time to
1009          * allocate 1->N transport tasks (depending on sector count and
1010          * maximum request size the physical HBA(s) can handle.
1011          */
1012         transport_ret = transport_generic_allocate_tasks(&cmd->se_cmd, hdr->cdb);
1013         if (transport_ret == -ENOMEM) {
1014                 return iscsit_add_reject_from_cmd(
1015                                 ISCSI_REASON_BOOKMARK_NO_RESOURCES,
1016                                 1, 1, buf, cmd);
1017         } else if (transport_ret < 0) {
1018                 /*
1019                  * Unsupported SAM Opcode.  CHECK_CONDITION will be sent
1020                  * in iscsit_execute_cmd() during the CmdSN OOO Execution
1021                  * Mechinism.
1022                  */
1023                 send_check_condition = 1;
1024         } else {
1025                 cmd->data_length = cmd->se_cmd.data_length;
1026
1027                 if (iscsit_decide_list_to_build(cmd, payload_length) < 0)
1028                         return iscsit_add_reject_from_cmd(
1029                                 ISCSI_REASON_BOOKMARK_NO_RESOURCES,
1030                                 1, 1, buf, cmd);
1031         }
1032
1033 attach_cmd:
1034         spin_lock_bh(&conn->cmd_lock);
1035         list_add_tail(&cmd->i_list, &conn->conn_cmd_list);
1036         spin_unlock_bh(&conn->cmd_lock);
1037         /*
1038          * Check if we need to delay processing because of ALUA
1039          * Active/NonOptimized primary access state..
1040          */
1041         core_alua_check_nonop_delay(&cmd->se_cmd);
1042         /*
1043          * Allocate and setup SGL used with transport_generic_map_mem_to_cmd().
1044          * also call iscsit_allocate_iovecs()
1045          */
1046         ret = iscsit_alloc_buffs(cmd);
1047         if (ret < 0)
1048                 return iscsit_add_reject_from_cmd(
1049                                 ISCSI_REASON_BOOKMARK_NO_RESOURCES,
1050                                 1, 0, buf, cmd);
1051         /*
1052          * Check the CmdSN against ExpCmdSN/MaxCmdSN here if
1053          * the Immediate Bit is not set, and no Immediate
1054          * Data is attached.
1055          *
1056          * A PDU/CmdSN carrying Immediate Data can only
1057          * be processed after the DataCRC has passed.
1058          * If the DataCRC fails, the CmdSN MUST NOT
1059          * be acknowledged. (See below)
1060          */
1061         if (!cmd->immediate_data) {
1062                 cmdsn_ret = iscsit_sequence_cmd(conn, cmd, hdr->cmdsn);
1063                 if (cmdsn_ret == CMDSN_LOWER_THAN_EXP)
1064                         return 0;
1065                 else if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
1066                         return iscsit_add_reject_from_cmd(
1067                                 ISCSI_REASON_PROTOCOL_ERROR,
1068                                 1, 0, buf, cmd);
1069         }
1070
1071         iscsit_ack_from_expstatsn(conn, hdr->exp_statsn);
1072
1073         /*
1074          * If no Immediate Data is attached, it's OK to return now.
1075          */
1076         if (!cmd->immediate_data) {
1077                 if (send_check_condition)
1078                         return 0;
1079
1080                 if (cmd->unsolicited_data) {
1081                         iscsit_set_dataout_sequence_values(cmd);
1082
1083                         spin_lock_bh(&cmd->dataout_timeout_lock);
1084                         iscsit_start_dataout_timer(cmd, cmd->conn);
1085                         spin_unlock_bh(&cmd->dataout_timeout_lock);
1086                 }
1087
1088                 return 0;
1089         }
1090
1091         /*
1092          * Early CHECK_CONDITIONs never make it to the transport processing
1093          * thread.  They are processed in CmdSN order by
1094          * iscsit_check_received_cmdsn() below.
1095          */
1096         if (send_check_condition) {
1097                 immed_ret = IMMEDIATE_DATA_NORMAL_OPERATION;
1098                 dump_immediate_data = 1;
1099                 goto after_immediate_data;
1100         }
1101         /*
1102          * Call directly into transport_generic_new_cmd() to perform
1103          * the backend memory allocation.
1104          */
1105         ret = transport_generic_new_cmd(&cmd->se_cmd);
1106         if (ret < 0) {
1107                 immed_ret = IMMEDIATE_DATA_NORMAL_OPERATION;
1108                 dump_immediate_data = 1;
1109                 goto after_immediate_data;
1110         }
1111
1112         immed_ret = iscsit_handle_immediate_data(cmd, buf, payload_length);
1113 after_immediate_data:
1114         if (immed_ret == IMMEDIATE_DATA_NORMAL_OPERATION) {
1115                 /*
1116                  * A PDU/CmdSN carrying Immediate Data passed
1117                  * DataCRC, check against ExpCmdSN/MaxCmdSN if
1118                  * Immediate Bit is not set.
1119                  */
1120                 cmdsn_ret = iscsit_sequence_cmd(conn, cmd, hdr->cmdsn);
1121                 /*
1122                  * Special case for Unsupported SAM WRITE Opcodes
1123                  * and ImmediateData=Yes.
1124                  */
1125                 if (dump_immediate_data) {
1126                         if (iscsit_dump_data_payload(conn, payload_length, 1) < 0)
1127                                 return -1;
1128                 } else if (cmd->unsolicited_data) {
1129                         iscsit_set_dataout_sequence_values(cmd);
1130
1131                         spin_lock_bh(&cmd->dataout_timeout_lock);
1132                         iscsit_start_dataout_timer(cmd, cmd->conn);
1133                         spin_unlock_bh(&cmd->dataout_timeout_lock);
1134                 }
1135
1136                 if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
1137                         return iscsit_add_reject_from_cmd(
1138                                 ISCSI_REASON_PROTOCOL_ERROR,
1139                                 1, 0, buf, cmd);
1140
1141         } else if (immed_ret == IMMEDIATE_DATA_ERL1_CRC_FAILURE) {
1142                 /*
1143                  * Immediate Data failed DataCRC and ERL>=1,
1144                  * silently drop this PDU and let the initiator
1145                  * plug the CmdSN gap.
1146                  *
1147                  * FIXME: Send Unsolicited NOPIN with reserved
1148                  * TTT here to help the initiator figure out
1149                  * the missing CmdSN, although they should be
1150                  * intelligent enough to determine the missing
1151                  * CmdSN and issue a retry to plug the sequence.
1152                  */
1153                 cmd->i_state = ISTATE_REMOVE;
1154                 iscsit_add_cmd_to_immediate_queue(cmd, conn, cmd->i_state);
1155         } else /* immed_ret == IMMEDIATE_DATA_CANNOT_RECOVER */
1156                 return -1;
1157
1158         return 0;
1159 }
1160
1161 static u32 iscsit_do_crypto_hash_sg(
1162         struct hash_desc *hash,
1163         struct iscsi_cmd *cmd,
1164         u32 data_offset,
1165         u32 data_length,
1166         u32 padding,
1167         u8 *pad_bytes)
1168 {
1169         u32 data_crc;
1170         u32 i;
1171         struct scatterlist *sg;
1172         unsigned int page_off;
1173
1174         crypto_hash_init(hash);
1175
1176         sg = cmd->first_data_sg;
1177         page_off = cmd->first_data_sg_off;
1178
1179         i = 0;
1180         while (data_length) {
1181                 u32 cur_len = min_t(u32, data_length, (sg[i].length - page_off));
1182
1183                 crypto_hash_update(hash, &sg[i], cur_len);
1184
1185                 data_length -= cur_len;
1186                 page_off = 0;
1187                 i++;
1188         }
1189
1190         if (padding) {
1191                 struct scatterlist pad_sg;
1192
1193                 sg_init_one(&pad_sg, pad_bytes, padding);
1194                 crypto_hash_update(hash, &pad_sg, padding);
1195         }
1196         crypto_hash_final(hash, (u8 *) &data_crc);
1197
1198         return data_crc;
1199 }
1200
1201 static void iscsit_do_crypto_hash_buf(
1202         struct hash_desc *hash,
1203         unsigned char *buf,
1204         u32 payload_length,
1205         u32 padding,
1206         u8 *pad_bytes,
1207         u8 *data_crc)
1208 {
1209         struct scatterlist sg;
1210
1211         crypto_hash_init(hash);
1212
1213         sg_init_one(&sg, (u8 *)buf, payload_length);
1214         crypto_hash_update(hash, &sg, payload_length);
1215
1216         if (padding) {
1217                 sg_init_one(&sg, pad_bytes, padding);
1218                 crypto_hash_update(hash, &sg, padding);
1219         }
1220         crypto_hash_final(hash, data_crc);
1221 }
1222
1223 static int iscsit_handle_data_out(struct iscsi_conn *conn, unsigned char *buf)
1224 {
1225         int iov_ret, ooo_cmdsn = 0, ret;
1226         u8 data_crc_failed = 0;
1227         u32 checksum, iov_count = 0, padding = 0, rx_got = 0;
1228         u32 rx_size = 0, payload_length;
1229         struct iscsi_cmd *cmd = NULL;
1230         struct se_cmd *se_cmd;
1231         struct iscsi_data *hdr;
1232         struct kvec *iov;
1233         unsigned long flags;
1234
1235         hdr                     = (struct iscsi_data *) buf;
1236         payload_length          = ntoh24(hdr->dlength);
1237         hdr->itt                = be32_to_cpu(hdr->itt);
1238         hdr->ttt                = be32_to_cpu(hdr->ttt);
1239         hdr->exp_statsn         = be32_to_cpu(hdr->exp_statsn);
1240         hdr->datasn             = be32_to_cpu(hdr->datasn);
1241         hdr->offset             = be32_to_cpu(hdr->offset);
1242
1243         if (!payload_length) {
1244                 pr_err("DataOUT payload is ZERO, protocol error.\n");
1245                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
1246                                         buf, conn);
1247         }
1248
1249         /* iSCSI write */
1250         spin_lock_bh(&conn->sess->session_stats_lock);
1251         conn->sess->rx_data_octets += payload_length;
1252         if (conn->sess->se_sess->se_node_acl) {
1253                 spin_lock(&conn->sess->se_sess->se_node_acl->stats_lock);
1254                 conn->sess->se_sess->se_node_acl->write_bytes += payload_length;
1255                 spin_unlock(&conn->sess->se_sess->se_node_acl->stats_lock);
1256         }
1257         spin_unlock_bh(&conn->sess->session_stats_lock);
1258
1259         if (payload_length > conn->conn_ops->MaxRecvDataSegmentLength) {
1260                 pr_err("DataSegmentLength: %u is greater than"
1261                         " MaxRecvDataSegmentLength: %u\n", payload_length,
1262                         conn->conn_ops->MaxRecvDataSegmentLength);
1263                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
1264                                         buf, conn);
1265         }
1266
1267         cmd = iscsit_find_cmd_from_itt_or_dump(conn, hdr->itt,
1268                         payload_length);
1269         if (!cmd)
1270                 return 0;
1271
1272         pr_debug("Got DataOut ITT: 0x%08x, TTT: 0x%08x,"
1273                 " DataSN: 0x%08x, Offset: %u, Length: %u, CID: %hu\n",
1274                 hdr->itt, hdr->ttt, hdr->datasn, hdr->offset,
1275                 payload_length, conn->cid);
1276
1277         if (cmd->cmd_flags & ICF_GOT_LAST_DATAOUT) {
1278                 pr_err("Command ITT: 0x%08x received DataOUT after"
1279                         " last DataOUT received, dumping payload\n",
1280                         cmd->init_task_tag);
1281                 return iscsit_dump_data_payload(conn, payload_length, 1);
1282         }
1283
1284         if (cmd->data_direction != DMA_TO_DEVICE) {
1285                 pr_err("Command ITT: 0x%08x received DataOUT for a"
1286                         " NON-WRITE command.\n", cmd->init_task_tag);
1287                 return iscsit_add_reject_from_cmd(ISCSI_REASON_PROTOCOL_ERROR,
1288                                 1, 0, buf, cmd);
1289         }
1290         se_cmd = &cmd->se_cmd;
1291         iscsit_mod_dataout_timer(cmd);
1292
1293         if ((hdr->offset + payload_length) > cmd->data_length) {
1294                 pr_err("DataOut Offset: %u, Length %u greater than"
1295                         " iSCSI Command EDTL %u, protocol error.\n",
1296                         hdr->offset, payload_length, cmd->data_length);
1297                 return iscsit_add_reject_from_cmd(ISCSI_REASON_BOOKMARK_INVALID,
1298                                 1, 0, buf, cmd);
1299         }
1300
1301         if (cmd->unsolicited_data) {
1302                 int dump_unsolicited_data = 0;
1303
1304                 if (conn->sess->sess_ops->InitialR2T) {
1305                         pr_err("Received unexpected unsolicited data"
1306                                 " while InitialR2T=Yes, protocol error.\n");
1307                         transport_send_check_condition_and_sense(&cmd->se_cmd,
1308                                         TCM_UNEXPECTED_UNSOLICITED_DATA, 0);
1309                         return -1;
1310                 }
1311                 /*
1312                  * Special case for dealing with Unsolicited DataOUT
1313                  * and Unsupported SAM WRITE Opcodes and SE resource allocation
1314                  * failures;
1315                  */
1316
1317                 /* Something's amiss if we're not in WRITE_PENDING state... */
1318                 spin_lock_irqsave(&se_cmd->t_state_lock, flags);
1319                 WARN_ON(se_cmd->t_state != TRANSPORT_WRITE_PENDING);
1320                 spin_unlock_irqrestore(&se_cmd->t_state_lock, flags);
1321
1322                 spin_lock_irqsave(&se_cmd->t_state_lock, flags);
1323                 if (!(se_cmd->se_cmd_flags & SCF_SUPPORTED_SAM_OPCODE) ||
1324                      (se_cmd->se_cmd_flags & SCF_SCSI_CDB_EXCEPTION))
1325                         dump_unsolicited_data = 1;
1326                 spin_unlock_irqrestore(&se_cmd->t_state_lock, flags);
1327
1328                 if (dump_unsolicited_data) {
1329                         /*
1330                          * Check if a delayed TASK_ABORTED status needs to
1331                          * be sent now if the ISCSI_FLAG_CMD_FINAL has been
1332                          * received with the unsolicitied data out.
1333                          */
1334                         if (hdr->flags & ISCSI_FLAG_CMD_FINAL)
1335                                 iscsit_stop_dataout_timer(cmd);
1336
1337                         transport_check_aborted_status(se_cmd,
1338                                         (hdr->flags & ISCSI_FLAG_CMD_FINAL));
1339                         return iscsit_dump_data_payload(conn, payload_length, 1);
1340                 }
1341         } else {
1342                 /*
1343                  * For the normal solicited data path:
1344                  *
1345                  * Check for a delayed TASK_ABORTED status and dump any
1346                  * incoming data out payload if one exists.  Also, when the
1347                  * ISCSI_FLAG_CMD_FINAL is set to denote the end of the current
1348                  * data out sequence, we decrement outstanding_r2ts.  Once
1349                  * outstanding_r2ts reaches zero, go ahead and send the delayed
1350                  * TASK_ABORTED status.
1351                  */
1352                 if (atomic_read(&se_cmd->t_transport_aborted) != 0) {
1353                         if (hdr->flags & ISCSI_FLAG_CMD_FINAL)
1354                                 if (--cmd->outstanding_r2ts < 1) {
1355                                         iscsit_stop_dataout_timer(cmd);
1356                                         transport_check_aborted_status(
1357                                                         se_cmd, 1);
1358                                 }
1359
1360                         return iscsit_dump_data_payload(conn, payload_length, 1);
1361                 }
1362         }
1363         /*
1364          * Preform DataSN, DataSequenceInOrder, DataPDUInOrder, and
1365          * within-command recovery checks before receiving the payload.
1366          */
1367         ret = iscsit_check_pre_dataout(cmd, buf);
1368         if (ret == DATAOUT_WITHIN_COMMAND_RECOVERY)
1369                 return 0;
1370         else if (ret == DATAOUT_CANNOT_RECOVER)
1371                 return -1;
1372
1373         rx_size += payload_length;
1374         iov = &cmd->iov_data[0];
1375
1376         iov_ret = iscsit_map_iovec(cmd, iov, hdr->offset, payload_length);
1377         if (iov_ret < 0)
1378                 return -1;
1379
1380         iov_count += iov_ret;
1381
1382         padding = ((-payload_length) & 3);
1383         if (padding != 0) {
1384                 iov[iov_count].iov_base = cmd->pad_bytes;
1385                 iov[iov_count++].iov_len = padding;
1386                 rx_size += padding;
1387                 pr_debug("Receiving %u padding bytes.\n", padding);
1388         }
1389
1390         if (conn->conn_ops->DataDigest) {
1391                 iov[iov_count].iov_base = &checksum;
1392                 iov[iov_count++].iov_len = ISCSI_CRC_LEN;
1393                 rx_size += ISCSI_CRC_LEN;
1394         }
1395
1396         rx_got = rx_data(conn, &cmd->iov_data[0], iov_count, rx_size);
1397
1398         iscsit_unmap_iovec(cmd);
1399
1400         if (rx_got != rx_size)
1401                 return -1;
1402
1403         if (conn->conn_ops->DataDigest) {
1404                 u32 data_crc;
1405
1406                 data_crc = iscsit_do_crypto_hash_sg(&conn->conn_rx_hash, cmd,
1407                                                     hdr->offset, payload_length, padding,
1408                                                     cmd->pad_bytes);
1409
1410                 if (checksum != data_crc) {
1411                         pr_err("ITT: 0x%08x, Offset: %u, Length: %u,"
1412                                 " DataSN: 0x%08x, CRC32C DataDigest 0x%08x"
1413                                 " does not match computed 0x%08x\n",
1414                                 hdr->itt, hdr->offset, payload_length,
1415                                 hdr->datasn, checksum, data_crc);
1416                         data_crc_failed = 1;
1417                 } else {
1418                         pr_debug("Got CRC32C DataDigest 0x%08x for"
1419                                 " %u bytes of Data Out\n", checksum,
1420                                 payload_length);
1421                 }
1422         }
1423         /*
1424          * Increment post receive data and CRC values or perform
1425          * within-command recovery.
1426          */
1427         ret = iscsit_check_post_dataout(cmd, buf, data_crc_failed);
1428         if ((ret == DATAOUT_NORMAL) || (ret == DATAOUT_WITHIN_COMMAND_RECOVERY))
1429                 return 0;
1430         else if (ret == DATAOUT_SEND_R2T) {
1431                 iscsit_set_dataout_sequence_values(cmd);
1432                 iscsit_build_r2ts_for_cmd(cmd, conn, 0);
1433         } else if (ret == DATAOUT_SEND_TO_TRANSPORT) {
1434                 /*
1435                  * Handle extra special case for out of order
1436                  * Unsolicited Data Out.
1437                  */
1438                 spin_lock_bh(&cmd->istate_lock);
1439                 ooo_cmdsn = (cmd->cmd_flags & ICF_OOO_CMDSN);
1440                 cmd->cmd_flags |= ICF_GOT_LAST_DATAOUT;
1441                 cmd->i_state = ISTATE_RECEIVED_LAST_DATAOUT;
1442                 spin_unlock_bh(&cmd->istate_lock);
1443
1444                 iscsit_stop_dataout_timer(cmd);
1445                 return (!ooo_cmdsn) ? transport_generic_handle_data(
1446                                         &cmd->se_cmd) : 0;
1447         } else /* DATAOUT_CANNOT_RECOVER */
1448                 return -1;
1449
1450         return 0;
1451 }
1452
1453 static int iscsit_handle_nop_out(
1454         struct iscsi_conn *conn,
1455         unsigned char *buf)
1456 {
1457         unsigned char *ping_data = NULL;
1458         int cmdsn_ret, niov = 0, ret = 0, rx_got, rx_size;
1459         u32 checksum, data_crc, padding = 0, payload_length;
1460         u64 lun;
1461         struct iscsi_cmd *cmd = NULL;
1462         struct kvec *iov = NULL;
1463         struct iscsi_nopout *hdr;
1464
1465         hdr                     = (struct iscsi_nopout *) buf;
1466         payload_length          = ntoh24(hdr->dlength);
1467         lun                     = get_unaligned_le64(&hdr->lun);
1468         hdr->itt                = be32_to_cpu(hdr->itt);
1469         hdr->ttt                = be32_to_cpu(hdr->ttt);
1470         hdr->cmdsn              = be32_to_cpu(hdr->cmdsn);
1471         hdr->exp_statsn         = be32_to_cpu(hdr->exp_statsn);
1472
1473         if ((hdr->itt == 0xFFFFFFFF) && !(hdr->opcode & ISCSI_OP_IMMEDIATE)) {
1474                 pr_err("NOPOUT ITT is reserved, but Immediate Bit is"
1475                         " not set, protocol error.\n");
1476                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
1477                                         buf, conn);
1478         }
1479
1480         if (payload_length > conn->conn_ops->MaxRecvDataSegmentLength) {
1481                 pr_err("NOPOUT Ping Data DataSegmentLength: %u is"
1482                         " greater than MaxRecvDataSegmentLength: %u, protocol"
1483                         " error.\n", payload_length,
1484                         conn->conn_ops->MaxRecvDataSegmentLength);
1485                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
1486                                         buf, conn);
1487         }
1488
1489         pr_debug("Got NOPOUT Ping %s ITT: 0x%08x, TTT: 0x%09x,"
1490                 " CmdSN: 0x%08x, ExpStatSN: 0x%08x, Length: %u\n",
1491                 (hdr->itt == 0xFFFFFFFF) ? "Response" : "Request",
1492                 hdr->itt, hdr->ttt, hdr->cmdsn, hdr->exp_statsn,
1493                 payload_length);
1494         /*
1495          * This is not a response to a Unsolicited NopIN, which means
1496          * it can either be a NOPOUT ping request (with a valid ITT),
1497          * or a NOPOUT not requesting a NOPIN (with a reserved ITT).
1498          * Either way, make sure we allocate an struct iscsi_cmd, as both
1499          * can contain ping data.
1500          */
1501         if (hdr->ttt == 0xFFFFFFFF) {
1502                 cmd = iscsit_allocate_cmd(conn, GFP_KERNEL);
1503                 if (!cmd)
1504                         return iscsit_add_reject(
1505                                         ISCSI_REASON_BOOKMARK_NO_RESOURCES,
1506                                         1, buf, conn);
1507
1508                 cmd->iscsi_opcode       = ISCSI_OP_NOOP_OUT;
1509                 cmd->i_state            = ISTATE_SEND_NOPIN;
1510                 cmd->immediate_cmd      = ((hdr->opcode & ISCSI_OP_IMMEDIATE) ?
1511                                                 1 : 0);
1512                 conn->sess->init_task_tag = cmd->init_task_tag = hdr->itt;
1513                 cmd->targ_xfer_tag      = 0xFFFFFFFF;
1514                 cmd->cmd_sn             = hdr->cmdsn;
1515                 cmd->exp_stat_sn        = hdr->exp_statsn;
1516                 cmd->data_direction     = DMA_NONE;
1517         }
1518
1519         if (payload_length && (hdr->ttt == 0xFFFFFFFF)) {
1520                 rx_size = payload_length;
1521                 ping_data = kzalloc(payload_length + 1, GFP_KERNEL);
1522                 if (!ping_data) {
1523                         pr_err("Unable to allocate memory for"
1524                                 " NOPOUT ping data.\n");
1525                         ret = -1;
1526                         goto out;
1527                 }
1528
1529                 iov = &cmd->iov_misc[0];
1530                 iov[niov].iov_base      = ping_data;
1531                 iov[niov++].iov_len     = payload_length;
1532
1533                 padding = ((-payload_length) & 3);
1534                 if (padding != 0) {
1535                         pr_debug("Receiving %u additional bytes"
1536                                 " for padding.\n", padding);
1537                         iov[niov].iov_base      = &cmd->pad_bytes;
1538                         iov[niov++].iov_len     = padding;
1539                         rx_size += padding;
1540                 }
1541                 if (conn->conn_ops->DataDigest) {
1542                         iov[niov].iov_base      = &checksum;
1543                         iov[niov++].iov_len     = ISCSI_CRC_LEN;
1544                         rx_size += ISCSI_CRC_LEN;
1545                 }
1546
1547                 rx_got = rx_data(conn, &cmd->iov_misc[0], niov, rx_size);
1548                 if (rx_got != rx_size) {
1549                         ret = -1;
1550                         goto out;
1551                 }
1552
1553                 if (conn->conn_ops->DataDigest) {
1554                         iscsit_do_crypto_hash_buf(&conn->conn_rx_hash,
1555                                         ping_data, payload_length,
1556                                         padding, cmd->pad_bytes,
1557                                         (u8 *)&data_crc);
1558
1559                         if (checksum != data_crc) {
1560                                 pr_err("Ping data CRC32C DataDigest"
1561                                 " 0x%08x does not match computed 0x%08x\n",
1562                                         checksum, data_crc);
1563                                 if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
1564                                         pr_err("Unable to recover from"
1565                                         " NOPOUT Ping DataCRC failure while in"
1566                                                 " ERL=0.\n");
1567                                         ret = -1;
1568                                         goto out;
1569                                 } else {
1570                                         /*
1571                                          * Silently drop this PDU and let the
1572                                          * initiator plug the CmdSN gap.
1573                                          */
1574                                         pr_debug("Dropping NOPOUT"
1575                                         " Command CmdSN: 0x%08x due to"
1576                                         " DataCRC error.\n", hdr->cmdsn);
1577                                         ret = 0;
1578                                         goto out;
1579                                 }
1580                         } else {
1581                                 pr_debug("Got CRC32C DataDigest"
1582                                 " 0x%08x for %u bytes of ping data.\n",
1583                                         checksum, payload_length);
1584                         }
1585                 }
1586
1587                 ping_data[payload_length] = '\0';
1588                 /*
1589                  * Attach ping data to struct iscsi_cmd->buf_ptr.
1590                  */
1591                 cmd->buf_ptr = (void *)ping_data;
1592                 cmd->buf_ptr_size = payload_length;
1593
1594                 pr_debug("Got %u bytes of NOPOUT ping"
1595                         " data.\n", payload_length);
1596                 pr_debug("Ping Data: \"%s\"\n", ping_data);
1597         }
1598
1599         if (hdr->itt != 0xFFFFFFFF) {
1600                 if (!cmd) {
1601                         pr_err("Checking CmdSN for NOPOUT,"
1602                                 " but cmd is NULL!\n");
1603                         return -1;
1604                 }
1605                 /*
1606                  * Initiator is expecting a NopIN ping reply,
1607                  */
1608                 spin_lock_bh(&conn->cmd_lock);
1609                 list_add_tail(&cmd->i_list, &conn->conn_cmd_list);
1610                 spin_unlock_bh(&conn->cmd_lock);
1611
1612                 iscsit_ack_from_expstatsn(conn, hdr->exp_statsn);
1613
1614                 if (hdr->opcode & ISCSI_OP_IMMEDIATE) {
1615                         iscsit_add_cmd_to_response_queue(cmd, conn,
1616                                         cmd->i_state);
1617                         return 0;
1618                 }
1619
1620                 cmdsn_ret = iscsit_sequence_cmd(conn, cmd, hdr->cmdsn);
1621                 if (cmdsn_ret == CMDSN_LOWER_THAN_EXP) {
1622                         ret = 0;
1623                         goto ping_out;
1624                 }
1625                 if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
1626                         return iscsit_add_reject_from_cmd(
1627                                         ISCSI_REASON_PROTOCOL_ERROR,
1628                                         1, 0, buf, cmd);
1629
1630                 return 0;
1631         }
1632
1633         if (hdr->ttt != 0xFFFFFFFF) {
1634                 /*
1635                  * This was a response to a unsolicited NOPIN ping.
1636                  */
1637                 cmd = iscsit_find_cmd_from_ttt(conn, hdr->ttt);
1638                 if (!cmd)
1639                         return -1;
1640
1641                 iscsit_stop_nopin_response_timer(conn);
1642
1643                 cmd->i_state = ISTATE_REMOVE;
1644                 iscsit_add_cmd_to_immediate_queue(cmd, conn, cmd->i_state);
1645                 iscsit_start_nopin_timer(conn);
1646         } else {
1647                 /*
1648                  * Initiator is not expecting a NOPIN is response.
1649                  * Just ignore for now.
1650                  *
1651                  * iSCSI v19-91 10.18
1652                  * "A NOP-OUT may also be used to confirm a changed
1653                  *  ExpStatSN if another PDU will not be available
1654                  *  for a long time."
1655                  */
1656                 ret = 0;
1657                 goto out;
1658         }
1659
1660         return 0;
1661 out:
1662         if (cmd)
1663                 iscsit_release_cmd(cmd);
1664 ping_out:
1665         kfree(ping_data);
1666         return ret;
1667 }
1668
1669 static int iscsit_handle_task_mgt_cmd(
1670         struct iscsi_conn *conn,
1671         unsigned char *buf)
1672 {
1673         struct iscsi_cmd *cmd;
1674         struct se_tmr_req *se_tmr;
1675         struct iscsi_tmr_req *tmr_req;
1676         struct iscsi_tm *hdr;
1677         u32 payload_length;
1678         int out_of_order_cmdsn = 0;
1679         int ret;
1680         u8 function;
1681
1682         hdr                     = (struct iscsi_tm *) buf;
1683         payload_length          = ntoh24(hdr->dlength);
1684         hdr->itt                = be32_to_cpu(hdr->itt);
1685         hdr->rtt                = be32_to_cpu(hdr->rtt);
1686         hdr->cmdsn              = be32_to_cpu(hdr->cmdsn);
1687         hdr->exp_statsn         = be32_to_cpu(hdr->exp_statsn);
1688         hdr->refcmdsn           = be32_to_cpu(hdr->refcmdsn);
1689         hdr->exp_datasn         = be32_to_cpu(hdr->exp_datasn);
1690         hdr->flags &= ~ISCSI_FLAG_CMD_FINAL;
1691         function = hdr->flags;
1692
1693         pr_debug("Got Task Management Request ITT: 0x%08x, CmdSN:"
1694                 " 0x%08x, Function: 0x%02x, RefTaskTag: 0x%08x, RefCmdSN:"
1695                 " 0x%08x, CID: %hu\n", hdr->itt, hdr->cmdsn, function,
1696                 hdr->rtt, hdr->refcmdsn, conn->cid);
1697
1698         if ((function != ISCSI_TM_FUNC_ABORT_TASK) &&
1699             ((function != ISCSI_TM_FUNC_TASK_REASSIGN) &&
1700              (hdr->rtt != ISCSI_RESERVED_TAG))) {
1701                 pr_err("RefTaskTag should be set to 0xFFFFFFFF.\n");
1702                 hdr->rtt = ISCSI_RESERVED_TAG;
1703         }
1704
1705         if ((function == ISCSI_TM_FUNC_TASK_REASSIGN) &&
1706                         !(hdr->opcode & ISCSI_OP_IMMEDIATE)) {
1707                 pr_err("Task Management Request TASK_REASSIGN not"
1708                         " issued as immediate command, bad iSCSI Initiator"
1709                                 "implementation\n");
1710                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
1711                                         buf, conn);
1712         }
1713         if ((function != ISCSI_TM_FUNC_ABORT_TASK) &&
1714             (hdr->refcmdsn != ISCSI_RESERVED_TAG))
1715                 hdr->refcmdsn = ISCSI_RESERVED_TAG;
1716
1717         cmd = iscsit_allocate_se_cmd_for_tmr(conn, function);
1718         if (!cmd)
1719                 return iscsit_add_reject(ISCSI_REASON_BOOKMARK_NO_RESOURCES,
1720                                         1, buf, conn);
1721
1722         cmd->iscsi_opcode       = ISCSI_OP_SCSI_TMFUNC;
1723         cmd->i_state            = ISTATE_SEND_TASKMGTRSP;
1724         cmd->immediate_cmd      = ((hdr->opcode & ISCSI_OP_IMMEDIATE) ? 1 : 0);
1725         cmd->init_task_tag      = hdr->itt;
1726         cmd->targ_xfer_tag      = 0xFFFFFFFF;
1727         cmd->cmd_sn             = hdr->cmdsn;
1728         cmd->exp_stat_sn        = hdr->exp_statsn;
1729         se_tmr                  = cmd->se_cmd.se_tmr_req;
1730         tmr_req                 = cmd->tmr_req;
1731         /*
1732          * Locate the struct se_lun for all TMRs not related to ERL=2 TASK_REASSIGN
1733          */
1734         if (function != ISCSI_TM_FUNC_TASK_REASSIGN) {
1735                 ret = iscsit_get_lun_for_tmr(cmd,
1736                                 get_unaligned_le64(&hdr->lun));
1737                 if (ret < 0) {
1738                         cmd->se_cmd.se_cmd_flags |= SCF_SCSI_CDB_EXCEPTION;
1739                         se_tmr->response = ISCSI_TMF_RSP_NO_LUN;
1740                         goto attach;
1741                 }
1742         }
1743
1744         switch (function) {
1745         case ISCSI_TM_FUNC_ABORT_TASK:
1746                 se_tmr->response = iscsit_tmr_abort_task(cmd, buf);
1747                 if (se_tmr->response != ISCSI_TMF_RSP_COMPLETE) {
1748                         cmd->se_cmd.se_cmd_flags |= SCF_SCSI_CDB_EXCEPTION;
1749                         goto attach;
1750                 }
1751                 break;
1752         case ISCSI_TM_FUNC_ABORT_TASK_SET:
1753         case ISCSI_TM_FUNC_CLEAR_ACA:
1754         case ISCSI_TM_FUNC_CLEAR_TASK_SET:
1755         case ISCSI_TM_FUNC_LOGICAL_UNIT_RESET:
1756                 break;
1757         case ISCSI_TM_FUNC_TARGET_WARM_RESET:
1758                 if (iscsit_tmr_task_warm_reset(conn, tmr_req, buf) < 0) {
1759                         cmd->se_cmd.se_cmd_flags |= SCF_SCSI_CDB_EXCEPTION;
1760                         se_tmr->response = ISCSI_TMF_RSP_AUTH_FAILED;
1761                         goto attach;
1762                 }
1763                 break;
1764         case ISCSI_TM_FUNC_TARGET_COLD_RESET:
1765                 if (iscsit_tmr_task_cold_reset(conn, tmr_req, buf) < 0) {
1766                         cmd->se_cmd.se_cmd_flags |= SCF_SCSI_CDB_EXCEPTION;
1767                         se_tmr->response = ISCSI_TMF_RSP_AUTH_FAILED;
1768                         goto attach;
1769                 }
1770                 break;
1771         case ISCSI_TM_FUNC_TASK_REASSIGN:
1772                 se_tmr->response = iscsit_tmr_task_reassign(cmd, buf);
1773                 /*
1774                  * Perform sanity checks on the ExpDataSN only if the
1775                  * TASK_REASSIGN was successful.
1776                  */
1777                 if (se_tmr->response != ISCSI_TMF_RSP_COMPLETE)
1778                         break;
1779
1780                 if (iscsit_check_task_reassign_expdatasn(tmr_req, conn) < 0)
1781                         return iscsit_add_reject_from_cmd(
1782                                         ISCSI_REASON_BOOKMARK_INVALID, 1, 1,
1783                                         buf, cmd);
1784                 break;
1785         default:
1786                 pr_err("Unknown TMR function: 0x%02x, protocol"
1787                         " error.\n", function);
1788                 cmd->se_cmd.se_cmd_flags |= SCF_SCSI_CDB_EXCEPTION;
1789                 se_tmr->response = ISCSI_TMF_RSP_NOT_SUPPORTED;
1790                 goto attach;
1791         }
1792
1793         if ((function != ISCSI_TM_FUNC_TASK_REASSIGN) &&
1794             (se_tmr->response == ISCSI_TMF_RSP_COMPLETE))
1795                 se_tmr->call_transport = 1;
1796 attach:
1797         spin_lock_bh(&conn->cmd_lock);
1798         list_add_tail(&cmd->i_list, &conn->conn_cmd_list);
1799         spin_unlock_bh(&conn->cmd_lock);
1800
1801         if (!(hdr->opcode & ISCSI_OP_IMMEDIATE)) {
1802                 int cmdsn_ret = iscsit_sequence_cmd(conn, cmd, hdr->cmdsn);
1803                 if (cmdsn_ret == CMDSN_HIGHER_THAN_EXP)
1804                         out_of_order_cmdsn = 1;
1805                 else if (cmdsn_ret == CMDSN_LOWER_THAN_EXP)
1806                         return 0;
1807                 else if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
1808                         return iscsit_add_reject_from_cmd(
1809                                         ISCSI_REASON_PROTOCOL_ERROR,
1810                                         1, 0, buf, cmd);
1811         }
1812         iscsit_ack_from_expstatsn(conn, hdr->exp_statsn);
1813
1814         if (out_of_order_cmdsn || !(hdr->opcode & ISCSI_OP_IMMEDIATE))
1815                 return 0;
1816         /*
1817          * Found the referenced task, send to transport for processing.
1818          */
1819         if (se_tmr->call_transport)
1820                 return transport_generic_handle_tmr(&cmd->se_cmd);
1821
1822         /*
1823          * Could not find the referenced LUN, task, or Task Management
1824          * command not authorized or supported.  Change state and
1825          * let the tx_thread send the response.
1826          *
1827          * For connection recovery, this is also the default action for
1828          * TMR TASK_REASSIGN.
1829          */
1830         iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
1831         return 0;
1832 }
1833
1834 /* #warning FIXME: Support Text Command parameters besides SendTargets */
1835 static int iscsit_handle_text_cmd(
1836         struct iscsi_conn *conn,
1837         unsigned char *buf)
1838 {
1839         char *text_ptr, *text_in;
1840         int cmdsn_ret, niov = 0, rx_got, rx_size;
1841         u32 checksum = 0, data_crc = 0, payload_length;
1842         u32 padding = 0, pad_bytes = 0, text_length = 0;
1843         struct iscsi_cmd *cmd;
1844         struct kvec iov[3];
1845         struct iscsi_text *hdr;
1846
1847         hdr                     = (struct iscsi_text *) buf;
1848         payload_length          = ntoh24(hdr->dlength);
1849         hdr->itt                = be32_to_cpu(hdr->itt);
1850         hdr->ttt                = be32_to_cpu(hdr->ttt);
1851         hdr->cmdsn              = be32_to_cpu(hdr->cmdsn);
1852         hdr->exp_statsn         = be32_to_cpu(hdr->exp_statsn);
1853
1854         if (payload_length > conn->conn_ops->MaxRecvDataSegmentLength) {
1855                 pr_err("Unable to accept text parameter length: %u"
1856                         "greater than MaxRecvDataSegmentLength %u.\n",
1857                        payload_length, conn->conn_ops->MaxRecvDataSegmentLength);
1858                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
1859                                         buf, conn);
1860         }
1861
1862         pr_debug("Got Text Request: ITT: 0x%08x, CmdSN: 0x%08x,"
1863                 " ExpStatSN: 0x%08x, Length: %u\n", hdr->itt, hdr->cmdsn,
1864                 hdr->exp_statsn, payload_length);
1865
1866         rx_size = text_length = payload_length;
1867         if (text_length) {
1868                 text_in = kzalloc(text_length, GFP_KERNEL);
1869                 if (!text_in) {
1870                         pr_err("Unable to allocate memory for"
1871                                 " incoming text parameters\n");
1872                         return -1;
1873                 }
1874
1875                 memset(iov, 0, 3 * sizeof(struct kvec));
1876                 iov[niov].iov_base      = text_in;
1877                 iov[niov++].iov_len     = text_length;
1878
1879                 padding = ((-payload_length) & 3);
1880                 if (padding != 0) {
1881                         iov[niov].iov_base = &pad_bytes;
1882                         iov[niov++].iov_len  = padding;
1883                         rx_size += padding;
1884                         pr_debug("Receiving %u additional bytes"
1885                                         " for padding.\n", padding);
1886                 }
1887                 if (conn->conn_ops->DataDigest) {
1888                         iov[niov].iov_base      = &checksum;
1889                         iov[niov++].iov_len     = ISCSI_CRC_LEN;
1890                         rx_size += ISCSI_CRC_LEN;
1891                 }
1892
1893                 rx_got = rx_data(conn, &iov[0], niov, rx_size);
1894                 if (rx_got != rx_size) {
1895                         kfree(text_in);
1896                         return -1;
1897                 }
1898
1899                 if (conn->conn_ops->DataDigest) {
1900                         iscsit_do_crypto_hash_buf(&conn->conn_rx_hash,
1901                                         text_in, text_length,
1902                                         padding, (u8 *)&pad_bytes,
1903                                         (u8 *)&data_crc);
1904
1905                         if (checksum != data_crc) {
1906                                 pr_err("Text data CRC32C DataDigest"
1907                                         " 0x%08x does not match computed"
1908                                         " 0x%08x\n", checksum, data_crc);
1909                                 if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
1910                                         pr_err("Unable to recover from"
1911                                         " Text Data digest failure while in"
1912                                                 " ERL=0.\n");
1913                                         kfree(text_in);
1914                                         return -1;
1915                                 } else {
1916                                         /*
1917                                          * Silently drop this PDU and let the
1918                                          * initiator plug the CmdSN gap.
1919                                          */
1920                                         pr_debug("Dropping Text"
1921                                         " Command CmdSN: 0x%08x due to"
1922                                         " DataCRC error.\n", hdr->cmdsn);
1923                                         kfree(text_in);
1924                                         return 0;
1925                                 }
1926                         } else {
1927                                 pr_debug("Got CRC32C DataDigest"
1928                                         " 0x%08x for %u bytes of text data.\n",
1929                                                 checksum, text_length);
1930                         }
1931                 }
1932                 text_in[text_length - 1] = '\0';
1933                 pr_debug("Successfully read %d bytes of text"
1934                                 " data.\n", text_length);
1935
1936                 if (strncmp("SendTargets", text_in, 11) != 0) {
1937                         pr_err("Received Text Data that is not"
1938                                 " SendTargets, cannot continue.\n");
1939                         kfree(text_in);
1940                         return -1;
1941                 }
1942                 text_ptr = strchr(text_in, '=');
1943                 if (!text_ptr) {
1944                         pr_err("No \"=\" separator found in Text Data,"
1945                                 "  cannot continue.\n");
1946                         kfree(text_in);
1947                         return -1;
1948                 }
1949                 if (strncmp("=All", text_ptr, 4) != 0) {
1950                         pr_err("Unable to locate All value for"
1951                                 " SendTargets key,  cannot continue.\n");
1952                         kfree(text_in);
1953                         return -1;
1954                 }
1955 /*#warning Support SendTargets=(iSCSI Target Name/Nothing) values. */
1956                 kfree(text_in);
1957         }
1958
1959         cmd = iscsit_allocate_cmd(conn, GFP_KERNEL);
1960         if (!cmd)
1961                 return iscsit_add_reject(ISCSI_REASON_BOOKMARK_NO_RESOURCES,
1962                                         1, buf, conn);
1963
1964         cmd->iscsi_opcode       = ISCSI_OP_TEXT;
1965         cmd->i_state            = ISTATE_SEND_TEXTRSP;
1966         cmd->immediate_cmd      = ((hdr->opcode & ISCSI_OP_IMMEDIATE) ? 1 : 0);
1967         conn->sess->init_task_tag = cmd->init_task_tag  = hdr->itt;
1968         cmd->targ_xfer_tag      = 0xFFFFFFFF;
1969         cmd->cmd_sn             = hdr->cmdsn;
1970         cmd->exp_stat_sn        = hdr->exp_statsn;
1971         cmd->data_direction     = DMA_NONE;
1972
1973         spin_lock_bh(&conn->cmd_lock);
1974         list_add_tail(&cmd->i_list, &conn->conn_cmd_list);
1975         spin_unlock_bh(&conn->cmd_lock);
1976
1977         iscsit_ack_from_expstatsn(conn, hdr->exp_statsn);
1978
1979         if (!(hdr->opcode & ISCSI_OP_IMMEDIATE)) {
1980                 cmdsn_ret = iscsit_sequence_cmd(conn, cmd, hdr->cmdsn);
1981                 if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER)
1982                         return iscsit_add_reject_from_cmd(
1983                                         ISCSI_REASON_PROTOCOL_ERROR,
1984                                         1, 0, buf, cmd);
1985
1986                 return 0;
1987         }
1988
1989         return iscsit_execute_cmd(cmd, 0);
1990 }
1991
1992 int iscsit_logout_closesession(struct iscsi_cmd *cmd, struct iscsi_conn *conn)
1993 {
1994         struct iscsi_conn *conn_p;
1995         struct iscsi_session *sess = conn->sess;
1996
1997         pr_debug("Received logout request CLOSESESSION on CID: %hu"
1998                 " for SID: %u.\n", conn->cid, conn->sess->sid);
1999
2000         atomic_set(&sess->session_logout, 1);
2001         atomic_set(&conn->conn_logout_remove, 1);
2002         conn->conn_logout_reason = ISCSI_LOGOUT_REASON_CLOSE_SESSION;
2003
2004         iscsit_inc_conn_usage_count(conn);
2005         iscsit_inc_session_usage_count(sess);
2006
2007         spin_lock_bh(&sess->conn_lock);
2008         list_for_each_entry(conn_p, &sess->sess_conn_list, conn_list) {
2009                 if (conn_p->conn_state != TARG_CONN_STATE_LOGGED_IN)
2010                         continue;
2011
2012                 pr_debug("Moving to TARG_CONN_STATE_IN_LOGOUT.\n");
2013                 conn_p->conn_state = TARG_CONN_STATE_IN_LOGOUT;
2014         }
2015         spin_unlock_bh(&sess->conn_lock);
2016
2017         iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2018
2019         return 0;
2020 }
2021
2022 int iscsit_logout_closeconnection(struct iscsi_cmd *cmd, struct iscsi_conn *conn)
2023 {
2024         struct iscsi_conn *l_conn;
2025         struct iscsi_session *sess = conn->sess;
2026
2027         pr_debug("Received logout request CLOSECONNECTION for CID:"
2028                 " %hu on CID: %hu.\n", cmd->logout_cid, conn->cid);
2029
2030         /*
2031          * A Logout Request with a CLOSECONNECTION reason code for a CID
2032          * can arrive on a connection with a differing CID.
2033          */
2034         if (conn->cid == cmd->logout_cid) {
2035                 spin_lock_bh(&conn->state_lock);
2036                 pr_debug("Moving to TARG_CONN_STATE_IN_LOGOUT.\n");
2037                 conn->conn_state = TARG_CONN_STATE_IN_LOGOUT;
2038
2039                 atomic_set(&conn->conn_logout_remove, 1);
2040                 conn->conn_logout_reason = ISCSI_LOGOUT_REASON_CLOSE_CONNECTION;
2041                 iscsit_inc_conn_usage_count(conn);
2042
2043                 spin_unlock_bh(&conn->state_lock);
2044         } else {
2045                 /*
2046                  * Handle all different cid CLOSECONNECTION requests in
2047                  * iscsit_logout_post_handler_diffcid() as to give enough
2048                  * time for any non immediate command's CmdSN to be
2049                  * acknowledged on the connection in question.
2050                  *
2051                  * Here we simply make sure the CID is still around.
2052                  */
2053                 l_conn = iscsit_get_conn_from_cid(sess,
2054                                 cmd->logout_cid);
2055                 if (!l_conn) {
2056                         cmd->logout_response = ISCSI_LOGOUT_CID_NOT_FOUND;
2057                         iscsit_add_cmd_to_response_queue(cmd, conn,
2058                                         cmd->i_state);
2059                         return 0;
2060                 }
2061
2062                 iscsit_dec_conn_usage_count(l_conn);
2063         }
2064
2065         iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2066
2067         return 0;
2068 }
2069
2070 int iscsit_logout_removeconnforrecovery(struct iscsi_cmd *cmd, struct iscsi_conn *conn)
2071 {
2072         struct iscsi_session *sess = conn->sess;
2073
2074         pr_debug("Received explicit REMOVECONNFORRECOVERY logout for"
2075                 " CID: %hu on CID: %hu.\n", cmd->logout_cid, conn->cid);
2076
2077         if (sess->sess_ops->ErrorRecoveryLevel != 2) {
2078                 pr_err("Received Logout Request REMOVECONNFORRECOVERY"
2079                         " while ERL!=2.\n");
2080                 cmd->logout_response = ISCSI_LOGOUT_RECOVERY_UNSUPPORTED;
2081                 iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2082                 return 0;
2083         }
2084
2085         if (conn->cid == cmd->logout_cid) {
2086                 pr_err("Received Logout Request REMOVECONNFORRECOVERY"
2087                         " with CID: %hu on CID: %hu, implementation error.\n",
2088                                 cmd->logout_cid, conn->cid);
2089                 cmd->logout_response = ISCSI_LOGOUT_CLEANUP_FAILED;
2090                 iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2091                 return 0;
2092         }
2093
2094         iscsit_add_cmd_to_response_queue(cmd, conn, cmd->i_state);
2095
2096         return 0;
2097 }
2098
2099 static int iscsit_handle_logout_cmd(
2100         struct iscsi_conn *conn,
2101         unsigned char *buf)
2102 {
2103         int cmdsn_ret, logout_remove = 0;
2104         u8 reason_code = 0;
2105         struct iscsi_cmd *cmd;
2106         struct iscsi_logout *hdr;
2107         struct iscsi_tiqn *tiqn = iscsit_snmp_get_tiqn(conn);
2108
2109         hdr                     = (struct iscsi_logout *) buf;
2110         reason_code             = (hdr->flags & 0x7f);
2111         hdr->itt                = be32_to_cpu(hdr->itt);
2112         hdr->cid                = be16_to_cpu(hdr->cid);
2113         hdr->cmdsn              = be32_to_cpu(hdr->cmdsn);
2114         hdr->exp_statsn = be32_to_cpu(hdr->exp_statsn);
2115
2116         if (tiqn) {
2117                 spin_lock(&tiqn->logout_stats.lock);
2118                 if (reason_code == ISCSI_LOGOUT_REASON_CLOSE_SESSION)
2119                         tiqn->logout_stats.normal_logouts++;
2120                 else
2121                         tiqn->logout_stats.abnormal_logouts++;
2122                 spin_unlock(&tiqn->logout_stats.lock);
2123         }
2124
2125         pr_debug("Got Logout Request ITT: 0x%08x CmdSN: 0x%08x"
2126                 " ExpStatSN: 0x%08x Reason: 0x%02x CID: %hu on CID: %hu\n",
2127                 hdr->itt, hdr->cmdsn, hdr->exp_statsn, reason_code,
2128                 hdr->cid, conn->cid);
2129
2130         if (conn->conn_state != TARG_CONN_STATE_LOGGED_IN) {
2131                 pr_err("Received logout request on connection that"
2132                         " is not in logged in state, ignoring request.\n");
2133                 return 0;
2134         }
2135
2136         cmd = iscsit_allocate_cmd(conn, GFP_KERNEL);
2137         if (!cmd)
2138                 return iscsit_add_reject(ISCSI_REASON_BOOKMARK_NO_RESOURCES, 1,
2139                                         buf, conn);
2140
2141         cmd->iscsi_opcode       = ISCSI_OP_LOGOUT;
2142         cmd->i_state            = ISTATE_SEND_LOGOUTRSP;
2143         cmd->immediate_cmd      = ((hdr->opcode & ISCSI_OP_IMMEDIATE) ? 1 : 0);
2144         conn->sess->init_task_tag = cmd->init_task_tag  = hdr->itt;
2145         cmd->targ_xfer_tag      = 0xFFFFFFFF;
2146         cmd->cmd_sn             = hdr->cmdsn;
2147         cmd->exp_stat_sn        = hdr->exp_statsn;
2148         cmd->logout_cid         = hdr->cid;
2149         cmd->logout_reason      = reason_code;
2150         cmd->data_direction     = DMA_NONE;
2151
2152         /*
2153          * We need to sleep in these cases (by returning 1) until the Logout
2154          * Response gets sent in the tx thread.
2155          */
2156         if ((reason_code == ISCSI_LOGOUT_REASON_CLOSE_SESSION) ||
2157            ((reason_code == ISCSI_LOGOUT_REASON_CLOSE_CONNECTION) &&
2158             (hdr->cid == conn->cid)))
2159                 logout_remove = 1;
2160
2161         spin_lock_bh(&conn->cmd_lock);
2162         list_add_tail(&cmd->i_list, &conn->conn_cmd_list);
2163         spin_unlock_bh(&conn->cmd_lock);
2164
2165         if (reason_code != ISCSI_LOGOUT_REASON_RECOVERY)
2166                 iscsit_ack_from_expstatsn(conn, hdr->exp_statsn);
2167
2168         /*
2169          * Immediate commands are executed, well, immediately.
2170          * Non-Immediate Logout Commands are executed in CmdSN order.
2171          */
2172         if (hdr->opcode & ISCSI_OP_IMMEDIATE) {
2173                 int ret = iscsit_execute_cmd(cmd, 0);
2174
2175                 if (ret < 0)
2176                         return ret;
2177         } else {
2178                 cmdsn_ret = iscsit_sequence_cmd(conn, cmd, hdr->cmdsn);
2179                 if (cmdsn_ret == CMDSN_LOWER_THAN_EXP) {
2180                         logout_remove = 0;
2181                 } else if (cmdsn_ret == CMDSN_ERROR_CANNOT_RECOVER) {
2182                         return iscsit_add_reject_from_cmd(
2183                                 ISCSI_REASON_PROTOCOL_ERROR,
2184                                 1, 0, buf, cmd);
2185                 }
2186         }
2187
2188         return logout_remove;
2189 }
2190
2191 static int iscsit_handle_snack(
2192         struct iscsi_conn *conn,
2193         unsigned char *buf)
2194 {
2195         u32 unpacked_lun;
2196         u64 lun;
2197         struct iscsi_snack *hdr;
2198
2199         hdr                     = (struct iscsi_snack *) buf;
2200         hdr->flags              &= ~ISCSI_FLAG_CMD_FINAL;
2201         lun                     = get_unaligned_le64(&hdr->lun);
2202         unpacked_lun            = scsilun_to_int((struct scsi_lun *)&lun);
2203         hdr->itt                = be32_to_cpu(hdr->itt);
2204         hdr->ttt                = be32_to_cpu(hdr->ttt);
2205         hdr->exp_statsn         = be32_to_cpu(hdr->exp_statsn);
2206         hdr->begrun             = be32_to_cpu(hdr->begrun);
2207         hdr->runlength          = be32_to_cpu(hdr->runlength);
2208
2209         pr_debug("Got ISCSI_INIT_SNACK, ITT: 0x%08x, ExpStatSN:"
2210                 " 0x%08x, Type: 0x%02x, BegRun: 0x%08x, RunLength: 0x%08x,"
2211                 " CID: %hu\n", hdr->itt, hdr->exp_statsn, hdr->flags,
2212                         hdr->begrun, hdr->runlength, conn->cid);
2213
2214         if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
2215                 pr_err("Initiator sent SNACK request while in"
2216                         " ErrorRecoveryLevel=0.\n");
2217                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
2218                                         buf, conn);
2219         }
2220         /*
2221          * SNACK_DATA and SNACK_R2T are both 0,  so check which function to
2222          * call from inside iscsi_send_recovery_datain_or_r2t().
2223          */
2224         switch (hdr->flags & ISCSI_FLAG_SNACK_TYPE_MASK) {
2225         case 0:
2226                 return iscsit_handle_recovery_datain_or_r2t(conn, buf,
2227                         hdr->itt, hdr->ttt, hdr->begrun, hdr->runlength);
2228         case ISCSI_FLAG_SNACK_TYPE_STATUS:
2229                 return iscsit_handle_status_snack(conn, hdr->itt, hdr->ttt,
2230                         hdr->begrun, hdr->runlength);
2231         case ISCSI_FLAG_SNACK_TYPE_DATA_ACK:
2232                 return iscsit_handle_data_ack(conn, hdr->ttt, hdr->begrun,
2233                         hdr->runlength);
2234         case ISCSI_FLAG_SNACK_TYPE_RDATA:
2235                 /* FIXME: Support R-Data SNACK */
2236                 pr_err("R-Data SNACK Not Supported.\n");
2237                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
2238                                         buf, conn);
2239         default:
2240                 pr_err("Unknown SNACK type 0x%02x, protocol"
2241                         " error.\n", hdr->flags & 0x0f);
2242                 return iscsit_add_reject(ISCSI_REASON_PROTOCOL_ERROR, 1,
2243                                         buf, conn);
2244         }
2245
2246         return 0;
2247 }
2248
2249 static void iscsit_rx_thread_wait_for_tcp(struct iscsi_conn *conn)
2250 {
2251         if ((conn->sock->sk->sk_shutdown & SEND_SHUTDOWN) ||
2252             (conn->sock->sk->sk_shutdown & RCV_SHUTDOWN)) {
2253                 wait_for_completion_interruptible_timeout(
2254                                         &conn->rx_half_close_comp,
2255                                         ISCSI_RX_THREAD_TCP_TIMEOUT * HZ);
2256         }
2257 }
2258
2259 static int iscsit_handle_immediate_data(
2260         struct iscsi_cmd *cmd,
2261         unsigned char *buf,
2262         u32 length)
2263 {
2264         int iov_ret, rx_got = 0, rx_size = 0;
2265         u32 checksum, iov_count = 0, padding = 0;
2266         struct iscsi_conn *conn = cmd->conn;
2267         struct kvec *iov;
2268
2269         iov_ret = iscsit_map_iovec(cmd, cmd->iov_data, cmd->write_data_done, length);
2270         if (iov_ret < 0)
2271                 return IMMEDIATE_DATA_CANNOT_RECOVER;
2272
2273         rx_size = length;
2274         iov_count = iov_ret;
2275         iov = &cmd->iov_data[0];
2276
2277         padding = ((-length) & 3);
2278         if (padding != 0) {
2279                 iov[iov_count].iov_base = cmd->pad_bytes;
2280                 iov[iov_count++].iov_len = padding;
2281                 rx_size += padding;
2282         }
2283
2284         if (conn->conn_ops->DataDigest) {
2285                 iov[iov_count].iov_base         = &checksum;
2286                 iov[iov_count++].iov_len        = ISCSI_CRC_LEN;
2287                 rx_size += ISCSI_CRC_LEN;
2288         }
2289
2290         rx_got = rx_data(conn, &cmd->iov_data[0], iov_count, rx_size);
2291
2292         iscsit_unmap_iovec(cmd);
2293
2294         if (rx_got != rx_size) {
2295                 iscsit_rx_thread_wait_for_tcp(conn);
2296                 return IMMEDIATE_DATA_CANNOT_RECOVER;
2297         }
2298
2299         if (conn->conn_ops->DataDigest) {
2300                 u32 data_crc;
2301
2302                 data_crc = iscsit_do_crypto_hash_sg(&conn->conn_rx_hash, cmd,
2303                                                     cmd->write_data_done, length, padding,
2304                                                     cmd->pad_bytes);
2305
2306                 if (checksum != data_crc) {
2307                         pr_err("ImmediateData CRC32C DataDigest 0x%08x"
2308                                 " does not match computed 0x%08x\n", checksum,
2309                                 data_crc);
2310
2311                         if (!conn->sess->sess_ops->ErrorRecoveryLevel) {
2312                                 pr_err("Unable to recover from"
2313                                         " Immediate Data digest failure while"
2314                                         " in ERL=0.\n");
2315                                 iscsit_add_reject_from_cmd(
2316                                                 ISCSI_REASON_DATA_DIGEST_ERROR,
2317                                                 1, 0, buf, cmd);
2318                                 return IMMEDIATE_DATA_CANNOT_RECOVER;
2319                         } else {
2320                                 iscsit_add_reject_from_cmd(
2321                                                 ISCSI_REASON_DATA_DIGEST_ERROR,
2322                                                 0, 0, buf, cmd);
2323                                 return IMMEDIATE_DATA_ERL1_CRC_FAILURE;
2324                         }
2325                 } else {
2326                         pr_debug("Got CRC32C DataDigest 0x%08x for"
2327                                 " %u bytes of Immediate Data\n", checksum,
2328                                 length);
2329                 }
2330         }
2331
2332         cmd->write_data_done += length;
2333
2334         if (cmd->write_data_done == cmd->data_length) {
2335                 spin_lock_bh(&cmd->istate_lock);
2336                 cmd->cmd_flags |= ICF_GOT_LAST_DATAOUT;
2337                 cmd->i_state = ISTATE_RECEIVED_LAST_DATAOUT;
2338                 spin_unlock_bh(&cmd->istate_lock);
2339         }
2340
2341         return IMMEDIATE_DATA_NORMAL_OPERATION;
2342 }
2343
2344 /*
2345  *      Called with sess->conn_lock held.
2346  */
2347 /* #warning iscsi_build_conn_drop_async_message() only sends out on connections
2348         with active network interface */
2349 static void iscsit_build_conn_drop_async_message(struct iscsi_conn *conn)
2350 {
2351         struct iscsi_cmd *cmd;
2352         struct iscsi_conn *conn_p;
2353
2354         /*
2355          * Only send a Asynchronous Message on connections whos network
2356          * interface is still functional.
2357          */
2358         list_for_each_entry(conn_p, &conn->sess->sess_conn_list, conn_list) {
2359                 if (conn_p->conn_state == TARG_CONN_STATE_LOGGED_IN) {
2360                         iscsit_inc_conn_usage_count(conn_p);
2361                         break;
2362                 }
2363         }
2364
2365         if (!conn_p)
2366                 return;
2367
2368         cmd = iscsit_allocate_cmd(conn_p, GFP_ATOMIC);
2369         if (!cmd) {
2370                 iscsit_dec_conn_usage_count(conn_p);
2371                 return;
2372         }
2373
2374         cmd->logout_cid = conn->cid;
2375         cmd->iscsi_opcode = ISCSI_OP_ASYNC_EVENT;
2376         cmd->i_state = ISTATE_SEND_ASYNCMSG;
2377
2378         spin_lock_bh(&conn_p->cmd_lock);
2379         list_add_tail(&cmd->i_list, &conn_p->conn_cmd_list);
2380         spin_unlock_bh(&conn_p->cmd_lock);
2381
2382         iscsit_add_cmd_to_response_queue(cmd, conn_p, cmd->i_state);
2383         iscsit_dec_conn_usage_count(conn_p);
2384 }
2385
2386 static int iscsit_send_conn_drop_async_message(
2387         struct iscsi_cmd *cmd,
2388         struct iscsi_conn *conn)
2389 {
2390         struct iscsi_async *hdr;
2391
2392         cmd->tx_size = ISCSI_HDR_LEN;
2393         cmd->iscsi_opcode = ISCSI_OP_ASYNC_EVENT;
2394
2395         hdr                     = (struct iscsi_async *) cmd->pdu;
2396         hdr->opcode             = ISCSI_OP_ASYNC_EVENT;
2397         hdr->flags              = ISCSI_FLAG_CMD_FINAL;
2398         cmd->init_task_tag      = 0xFFFFFFFF;
2399         cmd->targ_xfer_tag      = 0xFFFFFFFF;
2400         put_unaligned_be64(0xFFFFFFFFFFFFFFFFULL, &hdr->rsvd4[0]);
2401         cmd->stat_sn            = conn->stat_sn++;
2402         hdr->statsn             = cpu_to_be32(cmd->stat_sn);
2403         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
2404         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
2405         hdr->async_event        = ISCSI_ASYNC_MSG_DROPPING_CONNECTION;
2406         hdr->param1             = cpu_to_be16(cmd->logout_cid);
2407         hdr->param2             = cpu_to_be16(conn->sess->sess_ops->DefaultTime2Wait);
2408         hdr->param3             = cpu_to_be16(conn->sess->sess_ops->DefaultTime2Retain);
2409
2410         if (conn->conn_ops->HeaderDigest) {
2411                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
2412
2413                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
2414                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
2415                                 0, NULL, (u8 *)header_digest);
2416
2417                 cmd->tx_size += ISCSI_CRC_LEN;
2418                 pr_debug("Attaching CRC32C HeaderDigest to"
2419                         " Async Message 0x%08x\n", *header_digest);
2420         }
2421
2422         cmd->iov_misc[0].iov_base       = cmd->pdu;
2423         cmd->iov_misc[0].iov_len        = cmd->tx_size;
2424         cmd->iov_misc_count             = 1;
2425
2426         pr_debug("Sending Connection Dropped Async Message StatSN:"
2427                 " 0x%08x, for CID: %hu on CID: %hu\n", cmd->stat_sn,
2428                         cmd->logout_cid, conn->cid);
2429         return 0;
2430 }
2431
2432 static int iscsit_send_data_in(
2433         struct iscsi_cmd *cmd,
2434         struct iscsi_conn *conn,
2435         int *eodr)
2436 {
2437         int iov_ret = 0, set_statsn = 0;
2438         u32 iov_count = 0, tx_size = 0;
2439         struct iscsi_datain datain;
2440         struct iscsi_datain_req *dr;
2441         struct iscsi_data_rsp *hdr;
2442         struct kvec *iov;
2443
2444         memset(&datain, 0, sizeof(struct iscsi_datain));
2445         dr = iscsit_get_datain_values(cmd, &datain);
2446         if (!dr) {
2447                 pr_err("iscsit_get_datain_values failed for ITT: 0x%08x\n",
2448                                 cmd->init_task_tag);
2449                 return -1;
2450         }
2451
2452         /*
2453          * Be paranoid and double check the logic for now.
2454          */
2455         if ((datain.offset + datain.length) > cmd->data_length) {
2456                 pr_err("Command ITT: 0x%08x, datain.offset: %u and"
2457                         " datain.length: %u exceeds cmd->data_length: %u\n",
2458                         cmd->init_task_tag, datain.offset, datain.length,
2459                                 cmd->data_length);
2460                 return -1;
2461         }
2462
2463         spin_lock_bh(&conn->sess->session_stats_lock);
2464         conn->sess->tx_data_octets += datain.length;
2465         if (conn->sess->se_sess->se_node_acl) {
2466                 spin_lock(&conn->sess->se_sess->se_node_acl->stats_lock);
2467                 conn->sess->se_sess->se_node_acl->read_bytes += datain.length;
2468                 spin_unlock(&conn->sess->se_sess->se_node_acl->stats_lock);
2469         }
2470         spin_unlock_bh(&conn->sess->session_stats_lock);
2471         /*
2472          * Special case for successfully execution w/ both DATAIN
2473          * and Sense Data.
2474          */
2475         if ((datain.flags & ISCSI_FLAG_DATA_STATUS) &&
2476             (cmd->se_cmd.se_cmd_flags & SCF_TRANSPORT_TASK_SENSE))
2477                 datain.flags &= ~ISCSI_FLAG_DATA_STATUS;
2478         else {
2479                 if ((dr->dr_complete == DATAIN_COMPLETE_NORMAL) ||
2480                     (dr->dr_complete == DATAIN_COMPLETE_CONNECTION_RECOVERY)) {
2481                         iscsit_increment_maxcmdsn(cmd, conn->sess);
2482                         cmd->stat_sn = conn->stat_sn++;
2483                         set_statsn = 1;
2484                 } else if (dr->dr_complete ==
2485                                 DATAIN_COMPLETE_WITHIN_COMMAND_RECOVERY)
2486                         set_statsn = 1;
2487         }
2488
2489         hdr     = (struct iscsi_data_rsp *) cmd->pdu;
2490         memset(hdr, 0, ISCSI_HDR_LEN);
2491         hdr->opcode             = ISCSI_OP_SCSI_DATA_IN;
2492         hdr->flags              = datain.flags;
2493         if (hdr->flags & ISCSI_FLAG_DATA_STATUS) {
2494                 if (cmd->se_cmd.se_cmd_flags & SCF_OVERFLOW_BIT) {
2495                         hdr->flags |= ISCSI_FLAG_DATA_OVERFLOW;
2496                         hdr->residual_count = cpu_to_be32(cmd->se_cmd.residual_count);
2497                 } else if (cmd->se_cmd.se_cmd_flags & SCF_UNDERFLOW_BIT) {
2498                         hdr->flags |= ISCSI_FLAG_DATA_UNDERFLOW;
2499                         hdr->residual_count = cpu_to_be32(cmd->se_cmd.residual_count);
2500                 }
2501         }
2502         hton24(hdr->dlength, datain.length);
2503         if (hdr->flags & ISCSI_FLAG_DATA_ACK)
2504                 int_to_scsilun(cmd->se_cmd.orig_fe_lun,
2505                                 (struct scsi_lun *)&hdr->lun);
2506         else
2507                 put_unaligned_le64(0xFFFFFFFFFFFFFFFFULL, &hdr->lun);
2508
2509         hdr->itt                = cpu_to_be32(cmd->init_task_tag);
2510         hdr->ttt                = (hdr->flags & ISCSI_FLAG_DATA_ACK) ?
2511                                    cpu_to_be32(cmd->targ_xfer_tag) :
2512                                    0xFFFFFFFF;
2513         hdr->statsn             = (set_statsn) ? cpu_to_be32(cmd->stat_sn) :
2514                                                 0xFFFFFFFF;
2515         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
2516         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
2517         hdr->datasn             = cpu_to_be32(datain.data_sn);
2518         hdr->offset             = cpu_to_be32(datain.offset);
2519
2520         iov = &cmd->iov_data[0];
2521         iov[iov_count].iov_base = cmd->pdu;
2522         iov[iov_count++].iov_len        = ISCSI_HDR_LEN;
2523         tx_size += ISCSI_HDR_LEN;
2524
2525         if (conn->conn_ops->HeaderDigest) {
2526                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
2527
2528                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
2529                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
2530                                 0, NULL, (u8 *)header_digest);
2531
2532                 iov[0].iov_len += ISCSI_CRC_LEN;
2533                 tx_size += ISCSI_CRC_LEN;
2534
2535                 pr_debug("Attaching CRC32 HeaderDigest"
2536                         " for DataIN PDU 0x%08x\n", *header_digest);
2537         }
2538
2539         iov_ret = iscsit_map_iovec(cmd, &cmd->iov_data[1], datain.offset, datain.length);
2540         if (iov_ret < 0)
2541                 return -1;
2542
2543         iov_count += iov_ret;
2544         tx_size += datain.length;
2545
2546         cmd->padding = ((-datain.length) & 3);
2547         if (cmd->padding) {
2548                 iov[iov_count].iov_base         = cmd->pad_bytes;
2549                 iov[iov_count++].iov_len        = cmd->padding;
2550                 tx_size += cmd->padding;
2551
2552                 pr_debug("Attaching %u padding bytes\n",
2553                                 cmd->padding);
2554         }
2555         if (conn->conn_ops->DataDigest) {
2556                 cmd->data_crc = iscsit_do_crypto_hash_sg(&conn->conn_tx_hash, cmd,
2557                          datain.offset, datain.length, cmd->padding, cmd->pad_bytes);
2558
2559                 iov[iov_count].iov_base = &cmd->data_crc;
2560                 iov[iov_count++].iov_len = ISCSI_CRC_LEN;
2561                 tx_size += ISCSI_CRC_LEN;
2562
2563                 pr_debug("Attached CRC32C DataDigest %d bytes, crc"
2564                         " 0x%08x\n", datain.length+cmd->padding, cmd->data_crc);
2565         }
2566
2567         cmd->iov_data_count = iov_count;
2568         cmd->tx_size = tx_size;
2569
2570         pr_debug("Built DataIN ITT: 0x%08x, StatSN: 0x%08x,"
2571                 " DataSN: 0x%08x, Offset: %u, Length: %u, CID: %hu\n",
2572                 cmd->init_task_tag, ntohl(hdr->statsn), ntohl(hdr->datasn),
2573                 ntohl(hdr->offset), datain.length, conn->cid);
2574
2575         if (dr->dr_complete) {
2576                 *eodr = (cmd->se_cmd.se_cmd_flags & SCF_TRANSPORT_TASK_SENSE) ?
2577                                 2 : 1;
2578                 iscsit_free_datain_req(cmd, dr);
2579         }
2580
2581         return 0;
2582 }
2583
2584 static int iscsit_send_logout_response(
2585         struct iscsi_cmd *cmd,
2586         struct iscsi_conn *conn)
2587 {
2588         int niov = 0, tx_size;
2589         struct iscsi_conn *logout_conn = NULL;
2590         struct iscsi_conn_recovery *cr = NULL;
2591         struct iscsi_session *sess = conn->sess;
2592         struct kvec *iov;
2593         struct iscsi_logout_rsp *hdr;
2594         /*
2595          * The actual shutting down of Sessions and/or Connections
2596          * for CLOSESESSION and CLOSECONNECTION Logout Requests
2597          * is done in scsi_logout_post_handler().
2598          */
2599         switch (cmd->logout_reason) {
2600         case ISCSI_LOGOUT_REASON_CLOSE_SESSION:
2601                 pr_debug("iSCSI session logout successful, setting"
2602                         " logout response to ISCSI_LOGOUT_SUCCESS.\n");
2603                 cmd->logout_response = ISCSI_LOGOUT_SUCCESS;
2604                 break;
2605         case ISCSI_LOGOUT_REASON_CLOSE_CONNECTION:
2606                 if (cmd->logout_response == ISCSI_LOGOUT_CID_NOT_FOUND)
2607                         break;
2608                 /*
2609                  * For CLOSECONNECTION logout requests carrying
2610                  * a matching logout CID -> local CID, the reference
2611                  * for the local CID will have been incremented in
2612                  * iscsi_logout_closeconnection().
2613                  *
2614                  * For CLOSECONNECTION logout requests carrying
2615                  * a different CID than the connection it arrived
2616                  * on, the connection responding to cmd->logout_cid
2617                  * is stopped in iscsit_logout_post_handler_diffcid().
2618                  */
2619
2620                 pr_debug("iSCSI CID: %hu logout on CID: %hu"
2621                         " successful.\n", cmd->logout_cid, conn->cid);
2622                 cmd->logout_response = ISCSI_LOGOUT_SUCCESS;
2623                 break;
2624         case ISCSI_LOGOUT_REASON_RECOVERY:
2625                 if ((cmd->logout_response == ISCSI_LOGOUT_RECOVERY_UNSUPPORTED) ||
2626                     (cmd->logout_response == ISCSI_LOGOUT_CLEANUP_FAILED))
2627                         break;
2628                 /*
2629                  * If the connection is still active from our point of view
2630                  * force connection recovery to occur.
2631                  */
2632                 logout_conn = iscsit_get_conn_from_cid_rcfr(sess,
2633                                 cmd->logout_cid);
2634                 if ((logout_conn)) {
2635                         iscsit_connection_reinstatement_rcfr(logout_conn);
2636                         iscsit_dec_conn_usage_count(logout_conn);
2637                 }
2638
2639                 cr = iscsit_get_inactive_connection_recovery_entry(
2640                                 conn->sess, cmd->logout_cid);
2641                 if (!cr) {
2642                         pr_err("Unable to locate CID: %hu for"
2643                         " REMOVECONNFORRECOVERY Logout Request.\n",
2644                                 cmd->logout_cid);
2645                         cmd->logout_response = ISCSI_LOGOUT_CID_NOT_FOUND;
2646                         break;
2647                 }
2648
2649                 iscsit_discard_cr_cmds_by_expstatsn(cr, cmd->exp_stat_sn);
2650
2651                 pr_debug("iSCSI REMOVECONNFORRECOVERY logout"
2652                         " for recovery for CID: %hu on CID: %hu successful.\n",
2653                                 cmd->logout_cid, conn->cid);
2654                 cmd->logout_response = ISCSI_LOGOUT_SUCCESS;
2655                 break;
2656         default:
2657                 pr_err("Unknown cmd->logout_reason: 0x%02x\n",
2658                                 cmd->logout_reason);
2659                 return -1;
2660         }
2661
2662         tx_size = ISCSI_HDR_LEN;
2663         hdr                     = (struct iscsi_logout_rsp *)cmd->pdu;
2664         memset(hdr, 0, ISCSI_HDR_LEN);
2665         hdr->opcode             = ISCSI_OP_LOGOUT_RSP;
2666         hdr->flags              |= ISCSI_FLAG_CMD_FINAL;
2667         hdr->response           = cmd->logout_response;
2668         hdr->itt                = cpu_to_be32(cmd->init_task_tag);
2669         cmd->stat_sn            = conn->stat_sn++;
2670         hdr->statsn             = cpu_to_be32(cmd->stat_sn);
2671
2672         iscsit_increment_maxcmdsn(cmd, conn->sess);
2673         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
2674         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
2675
2676         iov = &cmd->iov_misc[0];
2677         iov[niov].iov_base      = cmd->pdu;
2678         iov[niov++].iov_len     = ISCSI_HDR_LEN;
2679
2680         if (conn->conn_ops->HeaderDigest) {
2681                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
2682
2683                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
2684                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
2685                                 0, NULL, (u8 *)header_digest);
2686
2687                 iov[0].iov_len += ISCSI_CRC_LEN;
2688                 tx_size += ISCSI_CRC_LEN;
2689                 pr_debug("Attaching CRC32C HeaderDigest to"
2690                         " Logout Response 0x%08x\n", *header_digest);
2691         }
2692         cmd->iov_misc_count = niov;
2693         cmd->tx_size = tx_size;
2694
2695         pr_debug("Sending Logout Response ITT: 0x%08x StatSN:"
2696                 " 0x%08x Response: 0x%02x CID: %hu on CID: %hu\n",
2697                 cmd->init_task_tag, cmd->stat_sn, hdr->response,
2698                 cmd->logout_cid, conn->cid);
2699
2700         return 0;
2701 }
2702
2703 /*
2704  *      Unsolicited NOPIN, either requesting a response or not.
2705  */
2706 static int iscsit_send_unsolicited_nopin(
2707         struct iscsi_cmd *cmd,
2708         struct iscsi_conn *conn,
2709         int want_response)
2710 {
2711         int tx_size = ISCSI_HDR_LEN;
2712         struct iscsi_nopin *hdr;
2713
2714         hdr                     = (struct iscsi_nopin *) cmd->pdu;
2715         memset(hdr, 0, ISCSI_HDR_LEN);
2716         hdr->opcode             = ISCSI_OP_NOOP_IN;
2717         hdr->flags              |= ISCSI_FLAG_CMD_FINAL;
2718         hdr->itt                = cpu_to_be32(cmd->init_task_tag);
2719         hdr->ttt                = cpu_to_be32(cmd->targ_xfer_tag);
2720         cmd->stat_sn            = conn->stat_sn;
2721         hdr->statsn             = cpu_to_be32(cmd->stat_sn);
2722         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
2723         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
2724
2725         if (conn->conn_ops->HeaderDigest) {
2726                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
2727
2728                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
2729                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
2730                                 0, NULL, (u8 *)header_digest);
2731
2732                 tx_size += ISCSI_CRC_LEN;
2733                 pr_debug("Attaching CRC32C HeaderDigest to"
2734                         " NopIN 0x%08x\n", *header_digest);
2735         }
2736
2737         cmd->iov_misc[0].iov_base       = cmd->pdu;
2738         cmd->iov_misc[0].iov_len        = tx_size;
2739         cmd->iov_misc_count     = 1;
2740         cmd->tx_size            = tx_size;
2741
2742         pr_debug("Sending Unsolicited NOPIN TTT: 0x%08x StatSN:"
2743                 " 0x%08x CID: %hu\n", hdr->ttt, cmd->stat_sn, conn->cid);
2744
2745         return 0;
2746 }
2747
2748 static int iscsit_send_nopin_response(
2749         struct iscsi_cmd *cmd,
2750         struct iscsi_conn *conn)
2751 {
2752         int niov = 0, tx_size;
2753         u32 padding = 0;
2754         struct kvec *iov;
2755         struct iscsi_nopin *hdr;
2756
2757         tx_size = ISCSI_HDR_LEN;
2758         hdr                     = (struct iscsi_nopin *) cmd->pdu;
2759         memset(hdr, 0, ISCSI_HDR_LEN);
2760         hdr->opcode             = ISCSI_OP_NOOP_IN;
2761         hdr->flags              |= ISCSI_FLAG_CMD_FINAL;
2762         hton24(hdr->dlength, cmd->buf_ptr_size);
2763         put_unaligned_le64(0xFFFFFFFFFFFFFFFFULL, &hdr->lun);
2764         hdr->itt                = cpu_to_be32(cmd->init_task_tag);
2765         hdr->ttt                = cpu_to_be32(cmd->targ_xfer_tag);
2766         cmd->stat_sn            = conn->stat_sn++;
2767         hdr->statsn             = cpu_to_be32(cmd->stat_sn);
2768
2769         iscsit_increment_maxcmdsn(cmd, conn->sess);
2770         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
2771         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
2772
2773         iov = &cmd->iov_misc[0];
2774         iov[niov].iov_base      = cmd->pdu;
2775         iov[niov++].iov_len     = ISCSI_HDR_LEN;
2776
2777         if (conn->conn_ops->HeaderDigest) {
2778                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
2779
2780                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
2781                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
2782                                 0, NULL, (u8 *)header_digest);
2783
2784                 iov[0].iov_len += ISCSI_CRC_LEN;
2785                 tx_size += ISCSI_CRC_LEN;
2786                 pr_debug("Attaching CRC32C HeaderDigest"
2787                         " to NopIn 0x%08x\n", *header_digest);
2788         }
2789
2790         /*
2791          * NOPOUT Ping Data is attached to struct iscsi_cmd->buf_ptr.
2792          * NOPOUT DataSegmentLength is at struct iscsi_cmd->buf_ptr_size.
2793          */
2794         if (cmd->buf_ptr_size) {
2795                 iov[niov].iov_base      = cmd->buf_ptr;
2796                 iov[niov++].iov_len     = cmd->buf_ptr_size;
2797                 tx_size += cmd->buf_ptr_size;
2798
2799                 pr_debug("Echoing back %u bytes of ping"
2800                         " data.\n", cmd->buf_ptr_size);
2801
2802                 padding = ((-cmd->buf_ptr_size) & 3);
2803                 if (padding != 0) {
2804                         iov[niov].iov_base = &cmd->pad_bytes;
2805                         iov[niov++].iov_len = padding;
2806                         tx_size += padding;
2807                         pr_debug("Attaching %u additional"
2808                                 " padding bytes.\n", padding);
2809                 }
2810                 if (conn->conn_ops->DataDigest) {
2811                         iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
2812                                 cmd->buf_ptr, cmd->buf_ptr_size,
2813                                 padding, (u8 *)&cmd->pad_bytes,
2814                                 (u8 *)&cmd->data_crc);
2815
2816                         iov[niov].iov_base = &cmd->data_crc;
2817                         iov[niov++].iov_len = ISCSI_CRC_LEN;
2818                         tx_size += ISCSI_CRC_LEN;
2819                         pr_debug("Attached DataDigest for %u"
2820                                 " bytes of ping data, CRC 0x%08x\n",
2821                                 cmd->buf_ptr_size, cmd->data_crc);
2822                 }
2823         }
2824
2825         cmd->iov_misc_count = niov;
2826         cmd->tx_size = tx_size;
2827
2828         pr_debug("Sending NOPIN Response ITT: 0x%08x, TTT:"
2829                 " 0x%08x, StatSN: 0x%08x, Length %u\n", cmd->init_task_tag,
2830                 cmd->targ_xfer_tag, cmd->stat_sn, cmd->buf_ptr_size);
2831
2832         return 0;
2833 }
2834
2835 int iscsit_send_r2t(
2836         struct iscsi_cmd *cmd,
2837         struct iscsi_conn *conn)
2838 {
2839         int tx_size = 0;
2840         struct iscsi_r2t *r2t;
2841         struct iscsi_r2t_rsp *hdr;
2842
2843         r2t = iscsit_get_r2t_from_list(cmd);
2844         if (!r2t)
2845                 return -1;
2846
2847         hdr                     = (struct iscsi_r2t_rsp *) cmd->pdu;
2848         memset(hdr, 0, ISCSI_HDR_LEN);
2849         hdr->opcode             = ISCSI_OP_R2T;
2850         hdr->flags              |= ISCSI_FLAG_CMD_FINAL;
2851         int_to_scsilun(cmd->se_cmd.orig_fe_lun,
2852                         (struct scsi_lun *)&hdr->lun);
2853         hdr->itt                = cpu_to_be32(cmd->init_task_tag);
2854         spin_lock_bh(&conn->sess->ttt_lock);
2855         r2t->targ_xfer_tag      = conn->sess->targ_xfer_tag++;
2856         if (r2t->targ_xfer_tag == 0xFFFFFFFF)
2857                 r2t->targ_xfer_tag = conn->sess->targ_xfer_tag++;
2858         spin_unlock_bh(&conn->sess->ttt_lock);
2859         hdr->ttt                = cpu_to_be32(r2t->targ_xfer_tag);
2860         hdr->statsn             = cpu_to_be32(conn->stat_sn);
2861         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
2862         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
2863         hdr->r2tsn              = cpu_to_be32(r2t->r2t_sn);
2864         hdr->data_offset        = cpu_to_be32(r2t->offset);
2865         hdr->data_length        = cpu_to_be32(r2t->xfer_len);
2866
2867         cmd->iov_misc[0].iov_base       = cmd->pdu;
2868         cmd->iov_misc[0].iov_len        = ISCSI_HDR_LEN;
2869         tx_size += ISCSI_HDR_LEN;
2870
2871         if (conn->conn_ops->HeaderDigest) {
2872                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
2873
2874                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
2875                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
2876                                 0, NULL, (u8 *)header_digest);
2877
2878                 cmd->iov_misc[0].iov_len += ISCSI_CRC_LEN;
2879                 tx_size += ISCSI_CRC_LEN;
2880                 pr_debug("Attaching CRC32 HeaderDigest for R2T"
2881                         " PDU 0x%08x\n", *header_digest);
2882         }
2883
2884         pr_debug("Built %sR2T, ITT: 0x%08x, TTT: 0x%08x, StatSN:"
2885                 " 0x%08x, R2TSN: 0x%08x, Offset: %u, DDTL: %u, CID: %hu\n",
2886                 (!r2t->recovery_r2t) ? "" : "Recovery ", cmd->init_task_tag,
2887                 r2t->targ_xfer_tag, ntohl(hdr->statsn), r2t->r2t_sn,
2888                         r2t->offset, r2t->xfer_len, conn->cid);
2889
2890         cmd->iov_misc_count = 1;
2891         cmd->tx_size = tx_size;
2892
2893         spin_lock_bh(&cmd->r2t_lock);
2894         r2t->sent_r2t = 1;
2895         spin_unlock_bh(&cmd->r2t_lock);
2896
2897         return 0;
2898 }
2899
2900 /*
2901  *      type 0: Normal Operation.
2902  *      type 1: Called from Storage Transport.
2903  *      type 2: Called from iscsi_task_reassign_complete_write() for
2904  *              connection recovery.
2905  */
2906 int iscsit_build_r2ts_for_cmd(
2907         struct iscsi_cmd *cmd,
2908         struct iscsi_conn *conn,
2909         int type)
2910 {
2911         int first_r2t = 1;
2912         u32 offset = 0, xfer_len = 0;
2913
2914         spin_lock_bh(&cmd->r2t_lock);
2915         if (cmd->cmd_flags & ICF_SENT_LAST_R2T) {
2916                 spin_unlock_bh(&cmd->r2t_lock);
2917                 return 0;
2918         }
2919
2920         if (conn->sess->sess_ops->DataSequenceInOrder && (type != 2))
2921                 if (cmd->r2t_offset < cmd->write_data_done)
2922                         cmd->r2t_offset = cmd->write_data_done;
2923
2924         while (cmd->outstanding_r2ts < conn->sess->sess_ops->MaxOutstandingR2T) {
2925                 if (conn->sess->sess_ops->DataSequenceInOrder) {
2926                         offset = cmd->r2t_offset;
2927
2928                         if (first_r2t && (type == 2)) {
2929                                 xfer_len = ((offset +
2930                                              (conn->sess->sess_ops->MaxBurstLength -
2931                                              cmd->next_burst_len) >
2932                                              cmd->data_length) ?
2933                                             (cmd->data_length - offset) :
2934                                             (conn->sess->sess_ops->MaxBurstLength -
2935                                              cmd->next_burst_len));
2936                         } else {
2937                                 xfer_len = ((offset +
2938                                              conn->sess->sess_ops->MaxBurstLength) >
2939                                              cmd->data_length) ?
2940                                              (cmd->data_length - offset) :
2941                                              conn->sess->sess_ops->MaxBurstLength;
2942                         }
2943                         cmd->r2t_offset += xfer_len;
2944
2945                         if (cmd->r2t_offset == cmd->data_length)
2946                                 cmd->cmd_flags |= ICF_SENT_LAST_R2T;
2947                 } else {
2948                         struct iscsi_seq *seq;
2949
2950                         seq = iscsit_get_seq_holder_for_r2t(cmd);
2951                         if (!seq) {
2952                                 spin_unlock_bh(&cmd->r2t_lock);
2953                                 return -1;
2954                         }
2955
2956                         offset = seq->offset;
2957                         xfer_len = seq->xfer_len;
2958
2959                         if (cmd->seq_send_order == cmd->seq_count)
2960                                 cmd->cmd_flags |= ICF_SENT_LAST_R2T;
2961                 }
2962                 cmd->outstanding_r2ts++;
2963                 first_r2t = 0;
2964
2965                 if (iscsit_add_r2t_to_list(cmd, offset, xfer_len, 0, 0) < 0) {
2966                         spin_unlock_bh(&cmd->r2t_lock);
2967                         return -1;
2968                 }
2969
2970                 if (cmd->cmd_flags & ICF_SENT_LAST_R2T)
2971                         break;
2972         }
2973         spin_unlock_bh(&cmd->r2t_lock);
2974
2975         return 0;
2976 }
2977
2978 static int iscsit_send_status(
2979         struct iscsi_cmd *cmd,
2980         struct iscsi_conn *conn)
2981 {
2982         u8 iov_count = 0, recovery;
2983         u32 padding = 0, tx_size = 0;
2984         struct iscsi_scsi_rsp *hdr;
2985         struct kvec *iov;
2986
2987         recovery = (cmd->i_state != ISTATE_SEND_STATUS);
2988         if (!recovery)
2989                 cmd->stat_sn = conn->stat_sn++;
2990
2991         spin_lock_bh(&conn->sess->session_stats_lock);
2992         conn->sess->rsp_pdus++;
2993         spin_unlock_bh(&conn->sess->session_stats_lock);
2994
2995         hdr                     = (struct iscsi_scsi_rsp *) cmd->pdu;
2996         memset(hdr, 0, ISCSI_HDR_LEN);
2997         hdr->opcode             = ISCSI_OP_SCSI_CMD_RSP;
2998         hdr->flags              |= ISCSI_FLAG_CMD_FINAL;
2999         if (cmd->se_cmd.se_cmd_flags & SCF_OVERFLOW_BIT) {
3000                 hdr->flags |= ISCSI_FLAG_CMD_OVERFLOW;
3001                 hdr->residual_count = cpu_to_be32(cmd->se_cmd.residual_count);
3002         } else if (cmd->se_cmd.se_cmd_flags & SCF_UNDERFLOW_BIT) {
3003                 hdr->flags |= ISCSI_FLAG_CMD_UNDERFLOW;
3004                 hdr->residual_count = cpu_to_be32(cmd->se_cmd.residual_count);
3005         }
3006         hdr->response           = cmd->iscsi_response;
3007         hdr->cmd_status         = cmd->se_cmd.scsi_status;
3008         hdr->itt                = cpu_to_be32(cmd->init_task_tag);
3009         hdr->statsn             = cpu_to_be32(cmd->stat_sn);
3010
3011         iscsit_increment_maxcmdsn(cmd, conn->sess);
3012         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
3013         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
3014
3015         iov = &cmd->iov_misc[0];
3016         iov[iov_count].iov_base = cmd->pdu;
3017         iov[iov_count++].iov_len = ISCSI_HDR_LEN;
3018         tx_size += ISCSI_HDR_LEN;
3019
3020         /*
3021          * Attach SENSE DATA payload to iSCSI Response PDU
3022          */
3023         if (cmd->se_cmd.sense_buffer &&
3024            ((cmd->se_cmd.se_cmd_flags & SCF_TRANSPORT_TASK_SENSE) ||
3025             (cmd->se_cmd.se_cmd_flags & SCF_EMULATED_TASK_SENSE))) {
3026                 padding         = -(cmd->se_cmd.scsi_sense_length) & 3;
3027                 hton24(hdr->dlength, cmd->se_cmd.scsi_sense_length);
3028                 iov[iov_count].iov_base = cmd->se_cmd.sense_buffer;
3029                 iov[iov_count++].iov_len =
3030                                 (cmd->se_cmd.scsi_sense_length + padding);
3031                 tx_size += cmd->se_cmd.scsi_sense_length;
3032
3033                 if (padding) {
3034                         memset(cmd->se_cmd.sense_buffer +
3035                                 cmd->se_cmd.scsi_sense_length, 0, padding);
3036                         tx_size += padding;
3037                         pr_debug("Adding %u bytes of padding to"
3038                                 " SENSE.\n", padding);
3039                 }
3040
3041                 if (conn->conn_ops->DataDigest) {
3042                         iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
3043                                 cmd->se_cmd.sense_buffer,
3044                                 (cmd->se_cmd.scsi_sense_length + padding),
3045                                 0, NULL, (u8 *)&cmd->data_crc);
3046
3047                         iov[iov_count].iov_base    = &cmd->data_crc;
3048                         iov[iov_count++].iov_len     = ISCSI_CRC_LEN;
3049                         tx_size += ISCSI_CRC_LEN;
3050
3051                         pr_debug("Attaching CRC32 DataDigest for"
3052                                 " SENSE, %u bytes CRC 0x%08x\n",
3053                                 (cmd->se_cmd.scsi_sense_length + padding),
3054                                 cmd->data_crc);
3055                 }
3056
3057                 pr_debug("Attaching SENSE DATA: %u bytes to iSCSI"
3058                                 " Response PDU\n",
3059                                 cmd->se_cmd.scsi_sense_length);
3060         }
3061
3062         if (conn->conn_ops->HeaderDigest) {
3063                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
3064
3065                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
3066                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
3067                                 0, NULL, (u8 *)header_digest);
3068
3069                 iov[0].iov_len += ISCSI_CRC_LEN;
3070                 tx_size += ISCSI_CRC_LEN;
3071                 pr_debug("Attaching CRC32 HeaderDigest for Response"
3072                                 " PDU 0x%08x\n", *header_digest);
3073         }
3074
3075         cmd->iov_misc_count = iov_count;
3076         cmd->tx_size = tx_size;
3077
3078         pr_debug("Built %sSCSI Response, ITT: 0x%08x, StatSN: 0x%08x,"
3079                 " Response: 0x%02x, SAM Status: 0x%02x, CID: %hu\n",
3080                 (!recovery) ? "" : "Recovery ", cmd->init_task_tag,
3081                 cmd->stat_sn, 0x00, cmd->se_cmd.scsi_status, conn->cid);
3082
3083         return 0;
3084 }
3085
3086 static u8 iscsit_convert_tcm_tmr_rsp(struct se_tmr_req *se_tmr)
3087 {
3088         switch (se_tmr->response) {
3089         case TMR_FUNCTION_COMPLETE:
3090                 return ISCSI_TMF_RSP_COMPLETE;
3091         case TMR_TASK_DOES_NOT_EXIST:
3092                 return ISCSI_TMF_RSP_NO_TASK;
3093         case TMR_LUN_DOES_NOT_EXIST:
3094                 return ISCSI_TMF_RSP_NO_LUN;
3095         case TMR_TASK_MGMT_FUNCTION_NOT_SUPPORTED:
3096                 return ISCSI_TMF_RSP_NOT_SUPPORTED;
3097         case TMR_FUNCTION_AUTHORIZATION_FAILED:
3098                 return ISCSI_TMF_RSP_AUTH_FAILED;
3099         case TMR_FUNCTION_REJECTED:
3100         default:
3101                 return ISCSI_TMF_RSP_REJECTED;
3102         }
3103 }
3104
3105 static int iscsit_send_task_mgt_rsp(
3106         struct iscsi_cmd *cmd,
3107         struct iscsi_conn *conn)
3108 {
3109         struct se_tmr_req *se_tmr = cmd->se_cmd.se_tmr_req;
3110         struct iscsi_tm_rsp *hdr;
3111         u32 tx_size = 0;
3112
3113         hdr                     = (struct iscsi_tm_rsp *) cmd->pdu;
3114         memset(hdr, 0, ISCSI_HDR_LEN);
3115         hdr->opcode             = ISCSI_OP_SCSI_TMFUNC_RSP;
3116         hdr->flags              = ISCSI_FLAG_CMD_FINAL;
3117         hdr->response           = iscsit_convert_tcm_tmr_rsp(se_tmr);
3118         hdr->itt                = cpu_to_be32(cmd->init_task_tag);
3119         cmd->stat_sn            = conn->stat_sn++;
3120         hdr->statsn             = cpu_to_be32(cmd->stat_sn);
3121
3122         iscsit_increment_maxcmdsn(cmd, conn->sess);
3123         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
3124         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
3125
3126         cmd->iov_misc[0].iov_base       = cmd->pdu;
3127         cmd->iov_misc[0].iov_len        = ISCSI_HDR_LEN;
3128         tx_size += ISCSI_HDR_LEN;
3129
3130         if (conn->conn_ops->HeaderDigest) {
3131                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
3132
3133                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
3134                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
3135                                 0, NULL, (u8 *)header_digest);
3136
3137                 cmd->iov_misc[0].iov_len += ISCSI_CRC_LEN;
3138                 tx_size += ISCSI_CRC_LEN;
3139                 pr_debug("Attaching CRC32 HeaderDigest for Task"
3140                         " Mgmt Response PDU 0x%08x\n", *header_digest);
3141         }
3142
3143         cmd->iov_misc_count = 1;
3144         cmd->tx_size = tx_size;
3145
3146         pr_debug("Built Task Management Response ITT: 0x%08x,"
3147                 " StatSN: 0x%08x, Response: 0x%02x, CID: %hu\n",
3148                 cmd->init_task_tag, cmd->stat_sn, hdr->response, conn->cid);
3149
3150         return 0;
3151 }
3152
3153 static bool iscsit_check_inaddr_any(struct iscsi_np *np)
3154 {
3155         bool ret = false;
3156
3157         if (np->np_sockaddr.ss_family == AF_INET6) {
3158                 const struct sockaddr_in6 sin6 = {
3159                         .sin6_addr = IN6ADDR_ANY_INIT };
3160                 struct sockaddr_in6 *sock_in6 =
3161                          (struct sockaddr_in6 *)&np->np_sockaddr;
3162
3163                 if (!memcmp(sock_in6->sin6_addr.s6_addr,
3164                                 sin6.sin6_addr.s6_addr, 16))
3165                         ret = true;
3166         } else {
3167                 struct sockaddr_in * sock_in =
3168                         (struct sockaddr_in *)&np->np_sockaddr;
3169
3170                 if (sock_in->sin_addr.s_addr == INADDR_ANY)
3171                         ret = true;
3172         }
3173
3174         return ret;
3175 }
3176
3177 static int iscsit_build_sendtargets_response(struct iscsi_cmd *cmd)
3178 {
3179         char *payload = NULL;
3180         struct iscsi_conn *conn = cmd->conn;
3181         struct iscsi_portal_group *tpg;
3182         struct iscsi_tiqn *tiqn;
3183         struct iscsi_tpg_np *tpg_np;
3184         int buffer_len, end_of_buf = 0, len = 0, payload_len = 0;
3185         unsigned char buf[256];
3186
3187         buffer_len = (conn->conn_ops->MaxRecvDataSegmentLength > 32768) ?
3188                         32768 : conn->conn_ops->MaxRecvDataSegmentLength;
3189
3190         memset(buf, 0, 256);
3191
3192         payload = kzalloc(buffer_len, GFP_KERNEL);
3193         if (!payload) {
3194                 pr_err("Unable to allocate memory for sendtargets"
3195                                 " response.\n");
3196                 return -ENOMEM;
3197         }
3198
3199         spin_lock(&tiqn_lock);
3200         list_for_each_entry(tiqn, &g_tiqn_list, tiqn_list) {
3201                 len = sprintf(buf, "TargetName=%s", tiqn->tiqn);
3202                 len += 1;
3203
3204                 if ((len + payload_len) > buffer_len) {
3205                         end_of_buf = 1;
3206                         goto eob;
3207                 }
3208                 memcpy((void *)payload + payload_len, buf, len);
3209                 payload_len += len;
3210
3211                 spin_lock(&tiqn->tiqn_tpg_lock);
3212                 list_for_each_entry(tpg, &tiqn->tiqn_tpg_list, tpg_list) {
3213
3214                         spin_lock(&tpg->tpg_state_lock);
3215                         if ((tpg->tpg_state == TPG_STATE_FREE) ||
3216                             (tpg->tpg_state == TPG_STATE_INACTIVE)) {
3217                                 spin_unlock(&tpg->tpg_state_lock);
3218                                 continue;
3219                         }
3220                         spin_unlock(&tpg->tpg_state_lock);
3221
3222                         spin_lock(&tpg->tpg_np_lock);
3223                         list_for_each_entry(tpg_np, &tpg->tpg_gnp_list,
3224                                                 tpg_np_list) {
3225                                 struct iscsi_np *np = tpg_np->tpg_np;
3226                                 bool inaddr_any = iscsit_check_inaddr_any(np);
3227
3228                                 len = sprintf(buf, "TargetAddress="
3229                                         "%s%s%s:%hu,%hu",
3230                                         (np->np_sockaddr.ss_family == AF_INET6) ?
3231                                         "[" : "", (inaddr_any == false) ?
3232                                                 np->np_ip : conn->local_ip,
3233                                         (np->np_sockaddr.ss_family == AF_INET6) ?
3234                                         "]" : "", (inaddr_any == false) ?
3235                                                 np->np_port : conn->local_port,
3236                                         tpg->tpgt);
3237                                 len += 1;
3238
3239                                 if ((len + payload_len) > buffer_len) {
3240                                         spin_unlock(&tpg->tpg_np_lock);
3241                                         spin_unlock(&tiqn->tiqn_tpg_lock);
3242                                         end_of_buf = 1;
3243                                         goto eob;
3244                                 }
3245                                 memcpy((void *)payload + payload_len, buf, len);
3246                                 payload_len += len;
3247                         }
3248                         spin_unlock(&tpg->tpg_np_lock);
3249                 }
3250                 spin_unlock(&tiqn->tiqn_tpg_lock);
3251 eob:
3252                 if (end_of_buf)
3253                         break;
3254         }
3255         spin_unlock(&tiqn_lock);
3256
3257         cmd->buf_ptr = payload;
3258
3259         return payload_len;
3260 }
3261
3262 /*
3263  *      FIXME: Add support for F_BIT and C_BIT when the length is longer than
3264  *      MaxRecvDataSegmentLength.
3265  */
3266 static int iscsit_send_text_rsp(
3267         struct iscsi_cmd *cmd,
3268         struct iscsi_conn *conn)
3269 {
3270         struct iscsi_text_rsp *hdr;
3271         struct kvec *iov;
3272         u32 padding = 0, tx_size = 0;
3273         int text_length, iov_count = 0;
3274
3275         text_length = iscsit_build_sendtargets_response(cmd);
3276         if (text_length < 0)
3277                 return text_length;
3278
3279         padding = ((-text_length) & 3);
3280         if (padding != 0) {
3281                 memset(cmd->buf_ptr + text_length, 0, padding);
3282                 pr_debug("Attaching %u additional bytes for"
3283                         " padding.\n", padding);
3284         }
3285
3286         hdr                     = (struct iscsi_text_rsp *) cmd->pdu;
3287         memset(hdr, 0, ISCSI_HDR_LEN);
3288         hdr->opcode             = ISCSI_OP_TEXT_RSP;
3289         hdr->flags              |= ISCSI_FLAG_CMD_FINAL;
3290         hton24(hdr->dlength, text_length);
3291         hdr->itt                = cpu_to_be32(cmd->init_task_tag);
3292         hdr->ttt                = cpu_to_be32(cmd->targ_xfer_tag);
3293         cmd->stat_sn            = conn->stat_sn++;
3294         hdr->statsn             = cpu_to_be32(cmd->stat_sn);
3295
3296         iscsit_increment_maxcmdsn(cmd, conn->sess);
3297         hdr->exp_cmdsn          = cpu_to_be32(conn->sess->exp_cmd_sn);
3298         hdr->max_cmdsn          = cpu_to_be32(conn->sess->max_cmd_sn);
3299
3300         iov = &cmd->iov_misc[0];
3301
3302         iov[iov_count].iov_base = cmd->pdu;
3303         iov[iov_count++].iov_len = ISCSI_HDR_LEN;
3304         iov[iov_count].iov_base = cmd->buf_ptr;
3305         iov[iov_count++].iov_len = text_length + padding;
3306
3307         tx_size += (ISCSI_HDR_LEN + text_length + padding);
3308
3309         if (conn->conn_ops->HeaderDigest) {
3310                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
3311
3312                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
3313                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
3314                                 0, NULL, (u8 *)header_digest);
3315
3316                 iov[0].iov_len += ISCSI_CRC_LEN;
3317                 tx_size += ISCSI_CRC_LEN;
3318                 pr_debug("Attaching CRC32 HeaderDigest for"
3319                         " Text Response PDU 0x%08x\n", *header_digest);
3320         }
3321
3322         if (conn->conn_ops->DataDigest) {
3323                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
3324                                 cmd->buf_ptr, (text_length + padding),
3325                                 0, NULL, (u8 *)&cmd->data_crc);
3326
3327                 iov[iov_count].iov_base = &cmd->data_crc;
3328                 iov[iov_count++].iov_len = ISCSI_CRC_LEN;
3329                 tx_size += ISCSI_CRC_LEN;
3330
3331                 pr_debug("Attaching DataDigest for %u bytes of text"
3332                         " data, CRC 0x%08x\n", (text_length + padding),
3333                         cmd->data_crc);
3334         }
3335
3336         cmd->iov_misc_count = iov_count;
3337         cmd->tx_size = tx_size;
3338
3339         pr_debug("Built Text Response: ITT: 0x%08x, StatSN: 0x%08x,"
3340                 " Length: %u, CID: %hu\n", cmd->init_task_tag, cmd->stat_sn,
3341                         text_length, conn->cid);
3342         return 0;
3343 }
3344
3345 static int iscsit_send_reject(
3346         struct iscsi_cmd *cmd,
3347         struct iscsi_conn *conn)
3348 {
3349         u32 iov_count = 0, tx_size = 0;
3350         struct iscsi_reject *hdr;
3351         struct kvec *iov;
3352
3353         hdr                     = (struct iscsi_reject *) cmd->pdu;
3354         hdr->opcode             = ISCSI_OP_REJECT;
3355         hdr->flags              |= ISCSI_FLAG_CMD_FINAL;
3356         hton24(hdr->dlength, ISCSI_HDR_LEN);
3357         hdr->ffffffff           = 0xffffffff;
3358         cmd->stat_sn            = conn->stat_sn++;
3359         hdr->statsn             = cpu_to_be32(cmd->stat_sn);
3360         hdr->exp_cmdsn  = cpu_to_be32(conn->sess->exp_cmd_sn);
3361         hdr->max_cmdsn  = cpu_to_be32(conn->sess->max_cmd_sn);
3362
3363         iov = &cmd->iov_misc[0];
3364
3365         iov[iov_count].iov_base = cmd->pdu;
3366         iov[iov_count++].iov_len = ISCSI_HDR_LEN;
3367         iov[iov_count].iov_base = cmd->buf_ptr;
3368         iov[iov_count++].iov_len = ISCSI_HDR_LEN;
3369
3370         tx_size = (ISCSI_HDR_LEN + ISCSI_HDR_LEN);
3371
3372         if (conn->conn_ops->HeaderDigest) {
3373                 u32 *header_digest = (u32 *)&cmd->pdu[ISCSI_HDR_LEN];
3374
3375                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
3376                                 (unsigned char *)hdr, ISCSI_HDR_LEN,
3377                                 0, NULL, (u8 *)header_digest);
3378
3379                 iov[0].iov_len += ISCSI_CRC_LEN;
3380                 tx_size += ISCSI_CRC_LEN;
3381                 pr_debug("Attaching CRC32 HeaderDigest for"
3382                         " REJECT PDU 0x%08x\n", *header_digest);
3383         }
3384
3385         if (conn->conn_ops->DataDigest) {
3386                 iscsit_do_crypto_hash_buf(&conn->conn_tx_hash,
3387                                 (unsigned char *)cmd->buf_ptr, ISCSI_HDR_LEN,
3388                                 0, NULL, (u8 *)&cmd->data_crc);
3389
3390                 iov[iov_count].iov_base = &cmd->data_crc;
3391                 iov[iov_count++].iov_len  = ISCSI_CRC_LEN;
3392                 tx_size += ISCSI_CRC_LEN;
3393                 pr_debug("Attaching CRC32 DataDigest for REJECT"
3394                                 " PDU 0x%08x\n", cmd->data_crc);
3395         }
3396
3397         cmd->iov_misc_count = iov_count;
3398         cmd->tx_size = tx_size;
3399
3400         pr_debug("Built Reject PDU StatSN: 0x%08x, Reason: 0x%02x,"
3401                 " CID: %hu\n", ntohl(hdr->statsn), hdr->reason, conn->cid);
3402
3403         return 0;
3404 }
3405
3406 static void iscsit_tx_thread_wait_for_tcp(struct iscsi_conn *conn)
3407 {
3408         if ((conn->sock->sk->sk_shutdown & SEND_SHUTDOWN) ||
3409             (conn->sock->sk->sk_shutdown & RCV_SHUTDOWN)) {
3410                 wait_for_completion_interruptible_timeout(
3411                                         &conn->tx_half_close_comp,
3412                                         ISCSI_TX_THREAD_TCP_TIMEOUT * HZ);
3413         }
3414 }
3415
3416 #ifdef CONFIG_SMP
3417
3418 void iscsit_thread_get_cpumask(struct iscsi_conn *conn)
3419 {
3420         struct iscsi_thread_set *ts = conn->thread_set;
3421         int ord, cpu;
3422         /*
3423          * thread_id is assigned from iscsit_global->ts_bitmap from
3424          * within iscsi_thread_set.c:iscsi_allocate_thread_sets()
3425          *
3426          * Here we use thread_id to determine which CPU that this
3427          * iSCSI connection's iscsi_thread_set will be scheduled to
3428          * execute upon.
3429          */
3430         ord = ts->thread_id % cpumask_weight(cpu_online_mask);
3431 #if 0
3432         pr_debug(">>>>>>>>>>>>>>>>>>>> Generated ord: %d from"
3433                         " thread_id: %d\n", ord, ts->thread_id);
3434 #endif
3435         for_each_online_cpu(cpu) {
3436                 if (ord-- == 0) {
3437                         cpumask_set_cpu(cpu, conn->conn_cpumask);
3438                         return;
3439                 }
3440         }
3441         /*
3442          * This should never be reached..
3443          */
3444         dump_stack();
3445         cpumask_setall(conn->conn_cpumask);
3446 }
3447
3448 static inline void iscsit_thread_check_cpumask(
3449         struct iscsi_conn *conn,
3450         struct task_struct *p,
3451         int mode)
3452 {
3453         char buf[128];
3454         /*
3455          * mode == 1 signals iscsi_target_tx_thread() usage.
3456          * mode == 0 signals iscsi_target_rx_thread() usage.
3457          */
3458         if (mode == 1) {
3459                 if (!conn->conn_tx_reset_cpumask)
3460                         return;
3461                 conn->conn_tx_reset_cpumask = 0;
3462         } else {
3463                 if (!conn->conn_rx_reset_cpumask)
3464                         return;
3465                 conn->conn_rx_reset_cpumask = 0;
3466         }
3467         /*
3468          * Update the CPU mask for this single kthread so that
3469          * both TX and RX kthreads are scheduled to run on the
3470          * same CPU.
3471          */
3472         memset(buf, 0, 128);
3473         cpumask_scnprintf(buf, 128, conn->conn_cpumask);
3474 #if 0
3475         pr_debug(">>>>>>>>>>>>>> Calling set_cpus_allowed_ptr():"
3476                         " %s for %s\n", buf, p->comm);
3477 #endif
3478         set_cpus_allowed_ptr(p, conn->conn_cpumask);
3479 }
3480
3481 #else
3482
3483 void iscsit_thread_get_cpumask(struct iscsi_conn *conn)
3484 {
3485         return;
3486 }
3487
3488 #define iscsit_thread_check_cpumask(X, Y, Z) ({})
3489 #endif /* CONFIG_SMP */
3490
3491 int iscsi_target_tx_thread(void *arg)
3492 {
3493         u8 state;
3494         int eodr = 0;
3495         int ret = 0;
3496         int sent_status = 0;
3497         int use_misc = 0;
3498         int map_sg = 0;
3499         struct iscsi_cmd *cmd = NULL;
3500         struct iscsi_conn *conn;
3501         struct iscsi_queue_req *qr = NULL;
3502         struct se_cmd *se_cmd;
3503         struct iscsi_thread_set *ts = (struct iscsi_thread_set *)arg;
3504         /*
3505          * Allow ourselves to be interrupted by SIGINT so that a
3506          * connection recovery / failure event can be triggered externally.
3507          */
3508         allow_signal(SIGINT);
3509
3510 restart:
3511         conn = iscsi_tx_thread_pre_handler(ts);
3512         if (!conn)
3513                 goto out;
3514
3515         eodr = map_sg = ret = sent_status = use_misc = 0;
3516
3517         while (!kthread_should_stop()) {
3518                 /*
3519                  * Ensure that both TX and RX per connection kthreads
3520                  * are scheduled to run on the same CPU.
3521                  */
3522                 iscsit_thread_check_cpumask(conn, current, 1);
3523
3524                 wait_event_interruptible(conn->queues_wq,
3525                                          !iscsit_conn_all_queues_empty(conn) ||
3526                                          ts->status == ISCSI_THREAD_SET_RESET);
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);