lnet/klnds/gnilnd/gnilnd_cb.c

   1 /*
   2  * Copyright (C) 2004 Cluster File Systems, Inc.
   3  *
   4  * Copyright (C) 2009-2012 Cray, Inc.
   5  *
   6  *   Derived from work by Eric Barton <eric@bartonsoftware.com>
   7  *   Author: James Shimek <jshimek@cray.com>
   8  *   Author: Nic Henke <nic@cray.com>
   9  *
  10  *   This file is part of Lustre, http://www.lustre.org.
  11  *
  12  *   Lustre is free software; you can redistribute it and/or
  13  *   modify it under the terms of version 2 of the GNU General Public
  14  *   License as published by the Free Software Foundation.
  15  *
  16  *   Lustre is distributed in the hope that it will be useful,
  17  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
  18  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  19  *   GNU General Public License for more details.
  20  *
  21  *   You should have received a copy of the GNU General Public License
  22  *   along with Lustre; if not, write to the Free Software
  23  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  24  *
  25  */
  26
  27 #include <asm/page.h>
  28 #include <linux/nmi.h>
  29 #include "gnilnd.h"
  30
  31 /* this is useful when needed to debug wire corruption. */
  32 static void
  33 kgnilnd_dump_blob(int level, char *prefix, void *buf, int len) {
  34         __u64 *ptr;
  35
  36         ptr = (__u64 *) buf;
  37
  38         while (len > 0) {
  39                 if (len >= 32) {
  40                         CDEBUG(level,
  41                                "%s 0x%p: 0x%16.16llx 0x%16.16llx 0x%16.16llx 0x%16.16llx\n",
  42                                prefix, ptr, *(ptr), *(ptr + 1), *(ptr + 2), *(ptr + 3));
  43                         ptr += 4;
  44                         len -= 32;
  45                 } else if (len >= 16) {
  46                         CDEBUG(level,
  47                                "%s 0x%p: 0x%16.16llx 0x%16.16llx\n",
  48                                prefix, ptr, *(ptr), *(ptr + 1));
  49                         ptr += 2;
  50                         len -= 16;
  51                 } else {
  52                         CDEBUG(level, "%s 0x%p: 0x%16.16llx\n",
  53                                prefix, ptr, *(ptr));
  54                         ptr++;
  55                         len -= 8;
  56                 }
  57         }
  58 }
  59
  60 static void
  61 kgnilnd_dump_msg(int mask, kgn_msg_t *msg)
  62 {
  63         CDEBUG(mask, "0x%8.8x 0x%4.4x 0x%4.4x 0x%16.16llx"
  64                 " 0x%16.16llx 0x%8.8x 0x%4.4x 0x%4.4x 0x%8.8x\n",
  65                 msg->gnm_magic, msg->gnm_version,
  66                 msg->gnm_type, msg->gnm_srcnid,
  67                 msg->gnm_connstamp, msg->gnm_seq,
  68                 msg->gnm_cksum, msg->gnm_payload_cksum,
  69                 msg->gnm_payload_len);
  70 }
  71
  72 void
  73 kgnilnd_schedule_device(kgn_device_t *dev)
  74 {
  75         short         already_live = 0;
  76
  77         /* we'll only want to wake if the scheduler thread
  78          * has come around and set ready to zero */
  79         already_live = cmpxchg(&dev->gnd_ready, GNILND_DEV_IDLE, GNILND_DEV_IRQ);
  80
  81         if (!already_live) {
  82                 wake_up_all(&dev->gnd_waitq);
  83         }
  84         return;
  85 }
  86
  87 void kgnilnd_schedule_device_timer(unsigned long arg)
  88 {
  89         kgn_device_t *dev = (kgn_device_t *) arg;
  90
  91         kgnilnd_schedule_device(dev);
  92 }
  93
  94 void
  95 kgnilnd_device_callback(__u32 devid, __u64 arg)
  96 {
  97         kgn_device_t *dev;
  98         int           index = (int) arg;
  99
 100         if (index >= kgnilnd_data.kgn_ndevs) {
 101                 /* use _EMERG instead of an LBUG to prevent LBUG'ing in
 102                  * interrupt context. */
 103                 LCONSOLE_EMERG("callback for unknown device %d->%d\n",
 104                                 devid, index);
 105                 return;
 106         }
 107
 108         dev = &kgnilnd_data.kgn_devices[index];
 109         /* just basic sanity */
 110         if (dev->gnd_id == devid) {
 111                 kgnilnd_schedule_device(dev);
 112         } else {
 113                 LCONSOLE_EMERG("callback for bad device %d devid %d\n",
 114                                 dev->gnd_id, devid);
 115         }
 116 }
 117
 118 /* sched_intent values:
 119  * < 0 : do not reschedule under any circumstances
 120  * == 0: reschedule if someone marked him WANTS_SCHED
 121  * > 0 : force a reschedule */
 122 /* Return code 0 means it did not schedule the conn, 1
 123  * means it successfully scheduled the conn.
 124  */
 125
 126 int
 127 kgnilnd_schedule_process_conn(kgn_conn_t *conn, int sched_intent)
 128 {
 129         int     conn_sched;
 130
 131         /* move back to IDLE but save previous state.
 132          * if we see WANTS_SCHED, we'll call kgnilnd_schedule_conn and
 133          * let the xchg there handle any racing callers to get it
 134          * onto gnd_ready_conns */
 135
 136         conn_sched = xchg(&conn->gnc_scheduled, GNILND_CONN_IDLE);
 137         LASSERTF(conn_sched == GNILND_CONN_WANTS_SCHED ||
 138                  conn_sched == GNILND_CONN_PROCESS,
 139                  "conn %p after process in bad state: %d\n",
 140                  conn, conn_sched);
 141
 142         if (sched_intent >= 0) {
 143                 if ((sched_intent > 0 || (conn_sched == GNILND_CONN_WANTS_SCHED))) {
 144                         return kgnilnd_schedule_conn_refheld(conn, 1);
 145                 }
 146         }
 147         return 0;
 148 }
 149
 150 /* Return of 0 for conn not scheduled, 1 returned if conn was scheduled or marked
 151  * as scheduled */
 152
 153 int
 154 _kgnilnd_schedule_conn(kgn_conn_t *conn, const char *caller, int line, int refheld)
 155 {
 156         kgn_device_t        *dev = conn->gnc_device;
 157         int                  sched;
 158         int                  rc;
 159
 160         sched = xchg(&conn->gnc_scheduled, GNILND_CONN_WANTS_SCHED);
 161         /* we only care about the last person who marked want_sched since they
 162          * are most likely the culprit
 163          */
 164         memcpy(conn->gnc_sched_caller, caller, sizeof(conn->gnc_sched_caller));
 165         conn->gnc_sched_line = line;
 166         /* if we are IDLE, add to list - only one guy sees IDLE and "wins"
 167          * the chance to put it onto gnd_ready_conns.
 168          * otherwise, leave marked as WANTS_SCHED and the thread that "owns"
 169          *  the conn in process_conns will take care of moving it back to
 170          *  SCHED when it is done processing */
 171
 172         if (sched == GNILND_CONN_IDLE) {
 173                 /* if the conn is already scheduled, we've already requested
 174                  * the scheduler thread wakeup */
 175                 if (!refheld) {
 176                         /* Add a reference to the conn if we are not holding a reference
 177                          * already from the exisiting scheduler. We now use the same
 178                          * reference if we need to reschedule a conn while in a scheduler
 179                          * thread.
 180                          */
 181                         kgnilnd_conn_addref(conn);
 182                 }
 183                 LASSERTF(list_empty(&conn->gnc_schedlist), "conn %p already sched state %d\n",
 184                          conn, sched);
 185
 186                 CDEBUG(D_INFO, "scheduling conn 0x%p caller %s:%d\n", conn, caller, line);
 187
 188                 spin_lock(&dev->gnd_lock);
 189                 list_add_tail(&conn->gnc_schedlist, &dev->gnd_ready_conns);
 190                 spin_unlock(&dev->gnd_lock);
 191                 set_mb(conn->gnc_last_sched_ask, jiffies);
 192                 rc = 1;
 193         } else {
 194                 CDEBUG(D_INFO, "not scheduling conn 0x%p: %d caller %s:%d\n", conn, sched, caller, line);
 195                 rc = 0;
 196         }
 197
 198         /* make sure thread(s) going to process conns - but let it make
 199          * separate decision from conn schedule */
 200         kgnilnd_schedule_device(dev);
 201         return rc;
 202 }
 203
 204 void
 205 kgnilnd_schedule_dgram(kgn_device_t *dev)
 206 {
 207         int                  wake;
 208
 209         wake = xchg(&dev->gnd_dgram_ready, GNILND_DGRAM_SCHED);
 210         if (wake != GNILND_DGRAM_SCHED)  {
 211                 wake_up(&dev->gnd_dgram_waitq);
 212         } else {
 213                 CDEBUG(D_NETTRACE, "not waking: %d\n", wake);
 214         }
 215 }
 216
 217 void
 218 kgnilnd_free_tx(kgn_tx_t *tx)
 219 {
 220         /* taken from kgnilnd_tx_add_state_locked */
 221
 222         LASSERTF((tx->tx_list_p == NULL &&
 223                   tx->tx_list_state == GNILND_TX_ALLOCD) &&
 224                 list_empty(&tx->tx_list),
 225                 "tx %p with bad state %s (list_p %p) tx_list %s\n",
 226                 tx, kgnilnd_tx_state2str(tx->tx_list_state), tx->tx_list_p,
 227                 list_empty(&tx->tx_list) ? "empty" : "not empty");
 228
 229         atomic_dec(&kgnilnd_data.kgn_ntx);
 230
 231         /* we only allocate this if we need to */
 232         if (tx->tx_phys != NULL) {
 233                 kmem_cache_free(kgnilnd_data.kgn_tx_phys_cache, tx->tx_phys);
 234                 CDEBUG(D_MALLOC, "slab-freed 'tx_phys': %lu at %p.\n",
 235                        LNET_MAX_IOV * sizeof(gni_mem_segment_t), tx->tx_phys);
 236         }
 237
 238         /* Only free the buffer if we used it */
 239         if (tx->tx_buffer_copy != NULL) {
 240                 kgnilnd_vfree(tx->tx_buffer_copy, tx->tx_rdma_desc.length);
 241                 tx->tx_buffer_copy = NULL;
 242                 CDEBUG(D_MALLOC, "vfreed buffer2\n");
 243         }
 244 #if 0
 245         KGNILND_POISON(tx, 0x5a, sizeof(kgn_tx_t));
 246 #endif
 247         CDEBUG(D_MALLOC, "slab-freed 'tx': %lu at %p.\n", sizeof(*tx), tx);
 248         kmem_cache_free(kgnilnd_data.kgn_tx_cache, tx);
 249 }
 250
 251 kgn_tx_t *
 252 kgnilnd_alloc_tx (void)
 253 {
 254         kgn_tx_t        *tx = NULL;
 255
 256         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_ALLOC_TX))
 257                 return tx;
 258
 259         tx = kmem_cache_alloc(kgnilnd_data.kgn_tx_cache, GFP_ATOMIC);
 260         if (tx == NULL) {
 261                 CERROR("failed to allocate tx\n");
 262                 return NULL;
 263         }
 264         CDEBUG(D_MALLOC, "slab-alloced 'tx': %lu at %p.\n",
 265                sizeof(*tx), tx);
 266
 267         /* need this memset, cache alloc'd memory is not cleared */
 268         memset(tx, 0, sizeof(*tx));
 269
 270         /* setup everything here to minimize time under the lock */
 271         tx->tx_buftype = GNILND_BUF_NONE;
 272         tx->tx_msg.gnm_type = GNILND_MSG_NONE;
 273         INIT_LIST_HEAD(&tx->tx_list);
 274         INIT_LIST_HEAD(&tx->tx_map_list);
 275         tx->tx_list_state = GNILND_TX_ALLOCD;
 276
 277         atomic_inc(&kgnilnd_data.kgn_ntx);
 278
 279         return tx;
 280 }
 281
 282 /* csum_fold needs to be run on the return value before shipping over the wire */
 283 #define _kgnilnd_cksum(seed, ptr, nob)  csum_partial(ptr, nob, seed)
 284
 285 /* we don't use offset as every one is passing a buffer reference that already
 286  * includes the offset into the base address -
 287  *  see kgnilnd_setup_virt_buffer and kgnilnd_setup_immediate_buffer */
 288 static inline __u16
 289 kgnilnd_cksum(void *ptr, size_t nob)
 290 {
 291         __u16   sum;
 292
 293         sum = csum_fold(_kgnilnd_cksum(0, ptr, nob));
 294
 295         /* don't use magic 'no checksum' value */
 296         if (sum == 0)
 297                 sum = 1;
 298
 299         CDEBUG(D_INFO, "cksum 0x%x for ptr 0x%p sz %zu\n",
 300                sum, ptr, nob);
 301
 302         return sum;
 303 }
 304
 305 __u16
 306 kgnilnd_cksum_kiov(unsigned int nkiov, lnet_kiov_t *kiov,
 307                     unsigned int offset, unsigned int nob, int dump_blob)
 308 {
 309         __wsum             cksum = 0;
 310         __wsum             tmpck;
 311         __u16              retsum;
 312         void              *addr;
 313         unsigned int       fraglen;
 314         int                i, odd;
 315
 316         LASSERT(nkiov > 0);
 317         LASSERT(nob > 0);
 318
 319         CDEBUG(D_BUFFS, "calc cksum for kiov 0x%p nkiov %u offset %u nob %u, dump %d\n",
 320                kiov, nkiov, offset, nob, dump_blob);
 321
 322         /* if loops changes, please change kgnilnd_setup_phys_buffer */
 323
 324         while (offset >= kiov->kiov_len) {
 325                 offset -= kiov->kiov_len;
 326                 nkiov--;
 327                 kiov++;
 328                 LASSERT(nkiov > 0);
 329         }
 330
 331         /* ignore nob here, if nob < (kiov_len - offset), kiov == 1 */
 332         odd = (unsigned long) (kiov[0].kiov_len - offset) & 1;
 333
 334         if ((odd || *kgnilnd_tunables.kgn_vmap_cksum) && nkiov > 1) {
 335                 struct page **pages = kgnilnd_data.kgn_cksum_map_pages[get_cpu()];
 336
 337                 LASSERTF(pages != NULL, "NULL pages for cpu %d map_pages 0x%p\n",
 338                          get_cpu(), kgnilnd_data.kgn_cksum_map_pages);
 339
 340                 CDEBUG(D_BUFFS, "odd %d len %u offset %u nob %u\n",
 341                        odd, kiov[0].kiov_len, offset, nob);
 342
 343                 for (i = 0; i < nkiov; i++) {
 344                         pages[i] = kiov[i].kiov_page;
 345                 }
 346
 347                 addr = vmap(pages, nkiov, VM_MAP, PAGE_KERNEL);
 348                 if (addr == NULL) {
 349                         CNETERR("Couldn't vmap %d frags on %d bytes to avoid odd length fragment in cksum\n",
 350                                 nkiov, nob);
 351                         /* return zero to avoid killing tx - we'll just get warning on console
 352                          * when remote end sees zero checksum */
 353                         RETURN(0);
 354                 }
 355                 atomic_inc(&kgnilnd_data.kgn_nvmap_cksum);
 356
 357                 tmpck = _kgnilnd_cksum(0, (void *) addr + kiov[0].kiov_offset + offset, nob);
 358                 cksum = tmpck;
 359
 360                 if (dump_blob) {
 361                         kgnilnd_dump_blob(D_BUFFS, "flat kiov RDMA payload",
 362                                           (void *)addr + kiov[0].kiov_offset + offset, nob);
 363                 }
 364                 CDEBUG(D_BUFFS, "cksum 0x%x (+0x%x) for addr 0x%p+%u len %u offset %u\n",
 365                        cksum, tmpck, addr, kiov[0].kiov_offset, nob, offset);
 366                 vunmap(addr);
 367         } else {
 368                 do {
 369                         fraglen = min(kiov->kiov_len - offset, nob);
 370
 371                         /* make dang sure we don't send a bogus checksum if somehow we get
 372                          * an odd length fragment on anything but the last entry in a kiov  -
 373                          * we know from kgnilnd_setup_rdma_buffer that we can't have non
 374                          * PAGE_SIZE pages in the middle, so if nob < PAGE_SIZE, it is the last one */
 375                         LASSERTF(!(fraglen&1) || (nob < PAGE_SIZE),
 376                                  "odd fraglen %u on nkiov %d, nob %u kiov_len %u offset %u kiov 0x%p\n",
 377                                  fraglen, nkiov, nob, kiov->kiov_len, offset, kiov);
 378
 379                         addr = (void *)kmap(kiov->kiov_page) + kiov->kiov_offset + offset;
 380                         tmpck = _kgnilnd_cksum(cksum, addr, fraglen);
 381
 382                         CDEBUG(D_BUFFS,
 383                                "cksum 0x%x (+0x%x) for page 0x%p+%u (0x%p) len %u offset %u\n",
 384                                cksum, tmpck, kiov->kiov_page, kiov->kiov_offset, addr,
 385                                fraglen, offset);
 386
 387                         cksum = tmpck;
 388
 389                         if (dump_blob)
 390                                 kgnilnd_dump_blob(D_BUFFS, "kiov cksum", addr, fraglen);
 391
 392                         kunmap(kiov->kiov_page);
 393
 394                         kiov++;
 395                         nkiov--;
 396                         nob -= fraglen;
 397                         offset = 0;
 398
 399                         /* iov must not run out before end of data */
 400                         LASSERTF(nob == 0 || nkiov > 0, "nob %u nkiov %u\n", nob, nkiov);
 401
 402                 } while (nob > 0);
 403         }
 404
 405         retsum = csum_fold(cksum);
 406
 407         /* don't use magic 'no checksum' value */
 408         if (retsum == 0)
 409                 retsum = 1;
 410
 411         CDEBUG(D_BUFFS, "retsum 0x%x from cksum 0x%x\n", retsum, cksum);
 412
 413         return retsum;
 414 }
 415
 416 void
 417 kgnilnd_init_msg(kgn_msg_t *msg, int type, lnet_nid_t source)
 418 {
 419         msg->gnm_magic = GNILND_MSG_MAGIC;
 420         msg->gnm_version = GNILND_MSG_VERSION;
 421         msg->gnm_type = type;
 422         msg->gnm_payload_len = 0;
 423         msg->gnm_srcnid = source;
 424         /* gnm_connstamp gets set when FMA is sent */
 425         /* gnm_srcnid is set on creation via function argument
 426          * The right interface/net and nid is passed in when the message
 427          * is created.
 428          */
 429 }
 430
 431 kgn_tx_t *
 432 kgnilnd_new_tx_msg(int type, lnet_nid_t source)
 433 {
 434         kgn_tx_t *tx = kgnilnd_alloc_tx();
 435
 436         if (tx != NULL) {
 437                 kgnilnd_init_msg(&tx->tx_msg, type, source);
 438         } else {
 439                 CERROR("couldn't allocate new tx type %s!\n",
 440                        kgnilnd_msgtype2str(type));
 441         }
 442
 443         return tx;
 444 }
 445
 446 static void
 447 kgnilnd_nak_rdma(kgn_conn_t *conn, int rx_type, int error, __u64 cookie, lnet_nid_t source) {
 448         kgn_tx_t        *tx;
 449
 450         int             nak_type;
 451
 452         switch (rx_type) {
 453         case GNILND_MSG_GET_REQ:
 454         case GNILND_MSG_GET_DONE:
 455                 nak_type = GNILND_MSG_GET_NAK;
 456                 break;
 457         case GNILND_MSG_PUT_REQ:
 458         case GNILND_MSG_PUT_ACK:
 459         case GNILND_MSG_PUT_DONE:
 460                 nak_type = GNILND_MSG_PUT_NAK;
 461                 break;
 462         case GNILND_MSG_PUT_REQ_REV:
 463         case GNILND_MSG_PUT_DONE_REV:
 464                 nak_type = GNILND_MSG_PUT_NAK_REV;
 465                 break;
 466         case GNILND_MSG_GET_REQ_REV:
 467         case GNILND_MSG_GET_ACK_REV:
 468         case GNILND_MSG_GET_DONE_REV:
 469                 nak_type = GNILND_MSG_GET_NAK_REV;
 470                 break;
 471         default:
 472                 CERROR("invalid msg type %s (%d)\n",
 473                         kgnilnd_msgtype2str(rx_type), rx_type);
 474                 LBUG();
 475         }
 476         /* only allow NAK on error and truncate to zero */
 477         LASSERTF(error <= 0, "error %d conn 0x%p, cookie %llu\n",
 478                  error, conn, cookie);
 479
 480         tx = kgnilnd_new_tx_msg(nak_type, source);
 481         if (tx == NULL) {
 482                 CNETERR("can't get TX to NAK RDMA to %s\n",
 483                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
 484                 return;
 485         }
 486
 487         tx->tx_msg.gnm_u.completion.gncm_retval = error;
 488         tx->tx_msg.gnm_u.completion.gncm_cookie = cookie;
 489         kgnilnd_queue_tx(conn, tx);
 490 }
 491
 492 int
 493 kgnilnd_setup_immediate_buffer(kgn_tx_t *tx, unsigned int niov,
 494                                struct kvec *iov, lnet_kiov_t *kiov,
 495                                unsigned int offset, unsigned int nob)
 496 {
 497         kgn_msg_t       *msg = &tx->tx_msg;
 498         int              i;
 499
 500         /* To help save on MDDs for short messages, we'll vmap a kiov to allow
 501          * gni_smsg_send to send that as the payload */
 502
 503         LASSERT(tx->tx_buftype == GNILND_BUF_NONE);
 504
 505         if (nob == 0) {
 506                 tx->tx_buffer = NULL;
 507         } else if (kiov != NULL) {
 508
 509                 if ((niov > 0) && unlikely(niov > (nob/PAGE_SIZE))) {
 510                         niov = ((nob + offset + kiov->kiov_offset + PAGE_SIZE - 1) /
 511                                 PAGE_SIZE);
 512                 }
 513
 514                 LASSERTF(niov > 0 && niov < GNILND_MAX_IMMEDIATE/PAGE_SIZE,
 515                         "bad niov %d msg %p kiov %p iov %p offset %d nob%d\n",
 516                         niov, msg, kiov, iov, offset, nob);
 517
 518                 while (offset >= kiov->kiov_len) {
 519                         offset -= kiov->kiov_len;
 520                         niov--;
 521                         kiov++;
 522                         LASSERT(niov > 0);
 523                 }
 524                 for (i = 0; i < niov; i++) {
 525                         /* We can't have a kiov_offset on anything but the first entry,
 526                          * otherwise we'll have a hole at the end of the mapping as we only map
 527                          * whole pages.
 528                          * Also, if we have a kiov_len < PAGE_SIZE but we need to map more
 529                          * than kiov_len, we will also have a whole at the end of that page
 530                          * which isn't allowed */
 531                         if ((kiov[i].kiov_offset != 0 && i > 0) ||
 532                             (kiov[i].kiov_offset + kiov[i].kiov_len != PAGE_SIZE && i < niov - 1)) {
 533                                 CNETERR("Can't make payload contiguous in I/O VM:"
 534                                        "page %d, offset %u, nob %u, kiov_offset %u kiov_len %u \n",
 535                                        i, offset, nob, kiov->kiov_offset, kiov->kiov_len);
 536                                 RETURN(-EINVAL);
 537                         }
 538                         tx->tx_imm_pages[i] = kiov[i].kiov_page;
 539                 }
 540
 541                 /* hijack tx_phys for the later unmap */
 542                 if (niov == 1) {
 543                         /* tx->phyx being equal to NULL is the signal for unmap to discern between kmap and vmap */
 544                         tx->tx_phys = NULL;
 545                         tx->tx_buffer = (void *)kmap(tx->tx_imm_pages[0]) + kiov[0].kiov_offset + offset;
 546                         atomic_inc(&kgnilnd_data.kgn_nkmap_short);
 547                         GNIDBG_TX(D_NET, tx, "kmapped page for %d bytes for kiov 0x%p, buffer 0x%p",
 548                                 nob, kiov, tx->tx_buffer);
 549                 } else {
 550                         tx->tx_phys = vmap(tx->tx_imm_pages, niov, VM_MAP, PAGE_KERNEL);
 551                         if (tx->tx_phys == NULL) {
 552                                 CNETERR("Couldn't vmap %d frags on %d bytes\n", niov, nob);
 553                                 RETURN(-ENOMEM);
 554
 555                         }
 556                         atomic_inc(&kgnilnd_data.kgn_nvmap_short);
 557                         /* make sure we take into account the kiov offset as the start of the buffer */
 558                         tx->tx_buffer = (void *)tx->tx_phys + kiov[0].kiov_offset + offset;
 559                         GNIDBG_TX(D_NET, tx, "mapped %d pages for %d bytes from kiov 0x%p to 0x%p, buffer 0x%p",
 560                                 niov, nob, kiov, tx->tx_phys, tx->tx_buffer);
 561                 }
 562                 tx->tx_buftype = GNILND_BUF_IMMEDIATE_KIOV;
 563                 tx->tx_nob = nob;
 564
 565         } else {
 566                 /* For now this is almost identical to kgnilnd_setup_virt_buffer, but we
 567                  * could "flatten" the payload into a single contiguous buffer ready
 568                  * for sending direct over an FMA if we ever needed to. */
 569
 570                 LASSERT(niov > 0);
 571
 572                 while (offset >= iov->iov_len) {
 573                         offset -= iov->iov_len;
 574                         niov--;
 575                         iov++;
 576                         LASSERT(niov > 0);
 577                 }
 578
 579                 if (nob > iov->iov_len - offset) {
 580                         CERROR("Can't handle multiple vaddr fragments\n");
 581                         return -EMSGSIZE;
 582                 }
 583
 584                 tx->tx_buffer = (void *)(((unsigned long)iov->iov_base) + offset);
 585
 586                 tx->tx_buftype = GNILND_BUF_IMMEDIATE;
 587                 tx->tx_nob = nob;
 588         }
 589
 590         /* checksum payload early - it shouldn't be changing after lnd_send */
 591         if (*kgnilnd_tunables.kgn_checksum >= 2) {
 592                 msg->gnm_payload_cksum = kgnilnd_cksum(tx->tx_buffer, nob);
 593                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SMSG_CKSUM2)) {
 594                         msg->gnm_payload_cksum += 0xe00e;
 595                 }
 596                 if (*kgnilnd_tunables.kgn_checksum_dump > 1) {
 597                         kgnilnd_dump_blob(D_BUFFS, "payload checksum",
 598                                           tx->tx_buffer, nob);
 599                 }
 600         } else {
 601                 msg->gnm_payload_cksum = 0;
 602         }
 603
 604         return 0;
 605 }
 606
 607 int
 608 kgnilnd_setup_virt_buffer(kgn_tx_t *tx,
 609                           unsigned int niov, struct kvec *iov,
 610                           unsigned int offset, unsigned int nob)
 611
 612 {
 613         LASSERT(nob > 0);
 614         LASSERT(niov > 0);
 615         LASSERT(tx->tx_buftype == GNILND_BUF_NONE);
 616
 617         while (offset >= iov->iov_len) {
 618                 offset -= iov->iov_len;
 619                 niov--;
 620                 iov++;
 621                 LASSERT(niov > 0);
 622         }
 623
 624         if (nob > iov->iov_len - offset) {
 625                 CERROR("Can't handle multiple vaddr fragments\n");
 626                 return -EMSGSIZE;
 627         }
 628
 629         tx->tx_buftype = GNILND_BUF_VIRT_UNMAPPED;
 630         tx->tx_nob = nob;
 631         tx->tx_buffer = (void *)(((unsigned long)iov->iov_base) + offset);
 632         return 0;
 633 }
 634
 635 int
 636 kgnilnd_setup_phys_buffer(kgn_tx_t *tx, int nkiov, lnet_kiov_t *kiov,
 637                           unsigned int offset, unsigned int nob)
 638 {
 639         gni_mem_segment_t *phys;
 640         int             rc = 0;
 641         unsigned int    fraglen;
 642
 643         GNIDBG_TX(D_NET, tx, "niov %d kiov 0x%p offset %u nob %u", nkiov, kiov, offset, nob);
 644
 645         LASSERT(nob > 0);
 646         LASSERT(nkiov > 0);
 647         LASSERT(tx->tx_buftype == GNILND_BUF_NONE);
 648
 649         /* only allocate this if we are going to use it */
 650         tx->tx_phys = kmem_cache_alloc(kgnilnd_data.kgn_tx_phys_cache,
 651                                               GFP_ATOMIC);
 652         if (tx->tx_phys == NULL) {
 653                 CERROR("failed to allocate tx_phys\n");
 654                 rc = -ENOMEM;
 655                 GOTO(error, rc);
 656         }
 657
 658         CDEBUG(D_MALLOC, "slab-alloced 'tx->tx_phys': %lu at %p.\n",
 659                LNET_MAX_IOV * sizeof(gni_mem_segment_t), tx->tx_phys);
 660
 661         /* if loops changes, please change kgnilnd_cksum_kiov
 662          *   and kgnilnd_setup_immediate_buffer */
 663
 664         while (offset >= kiov->kiov_len) {
 665                 offset -= kiov->kiov_len;
 666                 nkiov--;
 667                 kiov++;
 668                 LASSERT(nkiov > 0);
 669         }
 670
 671         /* at this point, kiov points to the first page that we'll actually map
 672          * now that we've seeked into the koiv for offset and dropped any
 673          * leading pages that fall entirely within the offset */
 674         tx->tx_buftype = GNILND_BUF_PHYS_UNMAPPED;
 675         tx->tx_nob = nob;
 676
 677         /* kiov_offset is start of 'valid' buffer, so index offset past that */
 678         tx->tx_buffer = (void *)((unsigned long)(kiov->kiov_offset + offset));
 679         phys = tx->tx_phys;
 680
 681         CDEBUG(D_NET, "tx 0x%p buffer 0x%p map start kiov 0x%p+%u niov %d offset %u\n",
 682                tx, tx->tx_buffer, kiov, kiov->kiov_offset, nkiov, offset);
 683
 684         do {
 685                 fraglen = min(kiov->kiov_len - offset, nob);
 686
 687                 /* We can't have a kiov_offset on anything but the first entry,
 688                  * otherwise we'll have a hole at the end of the mapping as we only map
 689                  * whole pages. Only the first page is allowed to have an offset -
 690                  * we'll add that into tx->tx_buffer and that will get used when we
 691                  * map in the segments (see kgnilnd_map_buffer).
 692                  * Also, if we have a kiov_len < PAGE_SIZE but we need to map more
 693                  * than kiov_len, we will also have a whole at the end of that page
 694                  * which isn't allowed */
 695                 if ((phys != tx->tx_phys) &&
 696                     ((kiov->kiov_offset != 0) ||
 697                      ((kiov->kiov_len < PAGE_SIZE) && (nob > kiov->kiov_len)))) {
 698                         CERROR("Can't make payload contiguous in I/O VM:"
 699                                "page %d, offset %u, nob %u, kiov_offset %u kiov_len %u \n",
 700                                (int)(phys - tx->tx_phys),
 701                                offset, nob, kiov->kiov_offset, kiov->kiov_len);
 702                         rc = -EINVAL;
 703                         GOTO(error, rc);
 704                 }
 705
 706                 if ((phys - tx->tx_phys) == LNET_MAX_IOV) {
 707                         CERROR ("payload too big (%d)\n", (int)(phys - tx->tx_phys));
 708                         rc = -EMSGSIZE;
 709                         GOTO(error, rc);
 710                 }
 711
 712                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_PHYS_SETUP)) {
 713                         rc = -EINVAL;
 714                         GOTO(error, rc);
 715                 }
 716
 717                 CDEBUG(D_BUFFS, "page 0x%p kiov_offset %u kiov_len %u nob %u "
 718                                "nkiov %u offset %u\n",
 719                       kiov->kiov_page, kiov->kiov_offset, kiov->kiov_len, nob, nkiov, offset);
 720
 721                 phys->address = page_to_phys(kiov->kiov_page);
 722                 phys++;
 723                 kiov++;
 724                 nkiov--;
 725                 nob -= fraglen;
 726                 offset = 0;
 727
 728                 /* iov must not run out before end of data */
 729                 LASSERTF(nob == 0 || nkiov > 0, "nob %u nkiov %u\n", nob, nkiov);
 730
 731         } while (nob > 0);
 732
 733         tx->tx_phys_npages = phys - tx->tx_phys;
 734
 735         return 0;
 736
 737 error:
 738         if (tx->tx_phys != NULL) {
 739                 kmem_cache_free(kgnilnd_data.kgn_tx_phys_cache, tx->tx_phys);
 740                 CDEBUG(D_MALLOC, "slab-freed 'tx_phys': %lu at %p.\n",
 741                        sizeof(*tx->tx_phys), tx->tx_phys);
 742                 tx->tx_phys = NULL;
 743         }
 744         return rc;
 745 }
 746
 747 static inline int
 748 kgnilnd_setup_rdma_buffer(kgn_tx_t *tx, unsigned int niov,
 749                           struct kvec *iov, lnet_kiov_t *kiov,
 750                           unsigned int offset, unsigned int nob)
 751 {
 752         int     rc;
 753
 754         LASSERTF((iov == NULL) != (kiov == NULL), "iov 0x%p, kiov 0x%p, tx 0x%p,"
 755                                                 " offset %d, nob %d, niov %d\n"
 756                                                 , iov, kiov, tx, offset, nob, niov);
 757
 758         if (kiov != NULL) {
 759                 rc = kgnilnd_setup_phys_buffer(tx, niov, kiov, offset, nob);
 760         } else {
 761                 rc = kgnilnd_setup_virt_buffer(tx, niov, iov, offset, nob);
 762         }
 763         return rc;
 764 }
 765
 766 /* kgnilnd_parse_lnet_rdma()
 767  * lntmsg - message passed in from lnet.
 768  * niov, kiov, offset - see lnd_t in lib-types.h for descriptions.
 769  * nob - actual number of bytes to in this message.
 770  * put_len - It is possible for PUTs to have a different length than the
 771  *           length stored in lntmsg->msg_len since LNET can adjust this
 772  *           length based on it's buffer size and offset.
 773  *           lnet_try_match_md() sets the mlength that we use to do the RDMA
 774  *           transfer.
 775  */
 776 static void
 777 kgnilnd_parse_lnet_rdma(lnet_msg_t *lntmsg, unsigned int *niov,
 778                         unsigned int *offset, unsigned int *nob,
 779                         lnet_kiov_t **kiov, int put_len)
 780 {
 781         /* GETs are weird, see kgnilnd_send */
 782         if (lntmsg->msg_type == LNET_MSG_GET) {
 783                 if ((lntmsg->msg_md->md_options & LNET_MD_KIOV) == 0) {
 784                         *kiov = NULL;
 785                 } else {
 786                         *kiov = lntmsg->msg_md->md_iov.kiov;
 787                 }
 788                 *niov = lntmsg->msg_md->md_niov;
 789                 *nob = lntmsg->msg_md->md_length;
 790                 *offset = 0;
 791         } else {
 792                 *kiov = lntmsg->msg_kiov;
 793                 *niov = lntmsg->msg_niov;
 794                 *nob = put_len;
 795                 *offset = lntmsg->msg_offset;
 796         }
 797 }
 798
 799 static inline void
 800 kgnilnd_compute_rdma_cksum(kgn_tx_t *tx, int put_len)
 801 {
 802         unsigned int     niov, offset, nob;
 803         lnet_kiov_t     *kiov;
 804         lnet_msg_t      *lntmsg = tx->tx_lntmsg[0];
 805         int              dump_cksum = (*kgnilnd_tunables.kgn_checksum_dump > 1);
 806
 807         GNITX_ASSERTF(tx, ((tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE) ||
 808                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE) ||
 809                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE_REV) ||
 810                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV) ||
 811                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_ACK_REV) ||
 812                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_REQ_REV)),
 813                       "bad type %s", kgnilnd_msgtype2str(tx->tx_msg.gnm_type));
 814
 815         if ((tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE_REV) ||
 816             (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV)) {
 817                 tx->tx_msg.gnm_payload_cksum = 0;
 818                 return;
 819         }
 820         if (*kgnilnd_tunables.kgn_checksum < 3) {
 821                 tx->tx_msg.gnm_payload_cksum = 0;
 822                 return;
 823         }
 824
 825         GNITX_ASSERTF(tx, lntmsg, "no LNet message!", NULL);
 826
 827         kgnilnd_parse_lnet_rdma(lntmsg, &niov, &offset, &nob, &kiov,
 828                                 put_len);
 829
 830         if (kiov != NULL) {
 831                 tx->tx_msg.gnm_payload_cksum = kgnilnd_cksum_kiov(niov, kiov, offset, nob, dump_cksum);
 832         } else {
 833                 tx->tx_msg.gnm_payload_cksum = kgnilnd_cksum(tx->tx_buffer, nob);
 834                 if (dump_cksum) {
 835                         kgnilnd_dump_blob(D_BUFFS, "peer RDMA payload", tx->tx_buffer, nob);
 836                 }
 837         }
 838
 839         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SMSG_CKSUM3)) {
 840                 tx->tx_msg.gnm_payload_cksum += 0xd00d;
 841         }
 842 }
 843
 844 /* kgnilnd_verify_rdma_cksum()
 845  * tx - PUT_DONE/GET_DONE matched tx.
 846  * rx_cksum - received checksum to compare against.
 847  * put_len - see kgnilnd_parse_lnet_rdma comments.
 848  */
 849 static inline int
 850 kgnilnd_verify_rdma_cksum(kgn_tx_t *tx, __u16 rx_cksum, int put_len)
 851 {
 852         int              rc = 0;
 853         __u16            cksum;
 854         unsigned int     niov, offset, nob;
 855         lnet_kiov_t     *kiov;
 856         lnet_msg_t      *lntmsg = tx->tx_lntmsg[0];
 857         int dump_on_err = *kgnilnd_tunables.kgn_checksum_dump;
 858
 859         /* we can only match certain requests */
 860         GNITX_ASSERTF(tx, ((tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) ||
 861                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK) ||
 862                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_REQ_REV) ||
 863                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_ACK_REV) ||
 864                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV) ||
 865                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE_REV)),
 866                       "bad type %s", kgnilnd_msgtype2str(tx->tx_msg.gnm_type));
 867
 868         if ((tx->tx_msg.gnm_type == GNILND_MSG_PUT_REQ_REV) ||
 869             (tx->tx_msg.gnm_type == GNILND_MSG_GET_ACK_REV)) {
 870                 return 0;
 871         }
 872
 873         if (rx_cksum == 0)  {
 874                 if (*kgnilnd_tunables.kgn_checksum >= 3) {
 875                         GNIDBG_MSG(D_WARNING, &tx->tx_msg,
 876                                    "no RDMA payload checksum when enabled");
 877                 }
 878                 return 0;
 879         }
 880
 881         GNITX_ASSERTF(tx, lntmsg, "no LNet message!", NULL);
 882
 883         kgnilnd_parse_lnet_rdma(lntmsg, &niov, &offset, &nob, &kiov, put_len);
 884
 885         if (kiov != NULL) {
 886                 cksum = kgnilnd_cksum_kiov(niov, kiov, offset, nob, 0);
 887         } else {
 888                 cksum = kgnilnd_cksum(tx->tx_buffer, nob);
 889         }
 890
 891         if (cksum != rx_cksum) {
 892                 GNIDBG_MSG(D_NETERROR, &tx->tx_msg,
 893                            "Bad RDMA payload checksum (%x expected %x); "
 894                            "kiov 0x%p niov %d nob %u offset %u",
 895                             cksum, rx_cksum, kiov, niov, nob, offset);
 896                 switch (dump_on_err) {
 897                 case 2:
 898                         if (kiov != NULL) {
 899                                 kgnilnd_cksum_kiov(niov, kiov, offset, nob, 1);
 900                         } else {
 901                                 kgnilnd_dump_blob(D_BUFFS, "RDMA payload",
 902                                                   tx->tx_buffer, nob);
 903                         }
 904                         /* fall through to dump log */
 905                 case 1:
 906                         libcfs_debug_dumplog();
 907                         break;
 908                 default:
 909                         break;
 910                 }
 911                 rc = -ENOKEY;
 912                 /* kgnilnd_check_fma_rx will close conn, kill tx with error */
 913         }
 914         return rc;
 915 }
 916
 917 void
 918 kgnilnd_mem_add_map_list(kgn_device_t *dev, kgn_tx_t *tx)
 919 {
 920         int     bytes;
 921
 922         GNITX_ASSERTF(tx, list_empty(&tx->tx_map_list),
 923                 "already mapped!", NULL);
 924
 925         spin_lock(&dev->gnd_map_lock);
 926         switch (tx->tx_buftype) {
 927         default:
 928                 GNIDBG_TX(D_EMERG, tx,
 929                         "SOFTWARE BUG: invalid mapping %d", tx->tx_buftype);
 930                 spin_unlock(&dev->gnd_map_lock);
 931                 LBUG();
 932                 break;
 933
 934         case GNILND_BUF_PHYS_MAPPED:
 935                 bytes = tx->tx_phys_npages * PAGE_SIZE;
 936                 dev->gnd_map_nphys++;
 937                 dev->gnd_map_physnop += tx->tx_phys_npages;
 938                 break;
 939
 940         case GNILND_BUF_VIRT_MAPPED:
 941                 bytes = tx->tx_nob;
 942                 dev->gnd_map_nvirt++;
 943                 dev->gnd_map_virtnob += tx->tx_nob;
 944                 break;
 945         }
 946
 947         if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK ||
 948             tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
 949                 atomic64_add(bytes, &dev->gnd_rdmaq_bytes_out);
 950                 GNIDBG_TX(D_NETTRACE, tx, "rdma ++ %d to %lld",
 951                           bytes, atomic64_read(&dev->gnd_rdmaq_bytes_out));
 952         }
 953
 954         atomic_inc(&dev->gnd_n_mdd);
 955         atomic64_add(bytes, &dev->gnd_nbytes_map);
 956
 957         /* clear retrans to prevent any SMSG goofiness as that code uses the same counter */
 958         tx->tx_retrans = 0;
 959
 960         /* we only get here in the valid cases */
 961         list_add_tail(&tx->tx_map_list, &dev->gnd_map_list);
 962         dev->gnd_map_version++;
 963         spin_unlock(&dev->gnd_map_lock);
 964 }
 965
 966 void
 967 kgnilnd_mem_del_map_list(kgn_device_t *dev, kgn_tx_t *tx)
 968 {
 969         int     bytes;
 970
 971         GNITX_ASSERTF(tx, !list_empty(&tx->tx_map_list),
 972                 "not mapped!", NULL);
 973         spin_lock(&dev->gnd_map_lock);
 974
 975         switch (tx->tx_buftype) {
 976         default:
 977                 GNIDBG_TX(D_EMERG, tx,
 978                         "SOFTWARE BUG: invalid mapping %d", tx->tx_buftype);
 979                 spin_unlock(&dev->gnd_map_lock);
 980                 LBUG();
 981                 break;
 982
 983         case GNILND_BUF_PHYS_UNMAPPED:
 984                 bytes = tx->tx_phys_npages * PAGE_SIZE;
 985                 dev->gnd_map_nphys--;
 986                 dev->gnd_map_physnop -= tx->tx_phys_npages;
 987                 break;
 988
 989         case GNILND_BUF_VIRT_UNMAPPED:
 990                 bytes = tx->tx_nob;
 991                 dev->gnd_map_nvirt--;
 992                 dev->gnd_map_virtnob -= tx->tx_nob;
 993                 break;
 994         }
 995
 996         if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK ||
 997             tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
 998                 atomic64_sub(bytes, &dev->gnd_rdmaq_bytes_out);
 999                 LASSERTF(atomic64_read(&dev->gnd_rdmaq_bytes_out) >= 0,
1000                          "bytes_out negative! %ld\n", atomic64_read(&dev->gnd_rdmaq_bytes_out));
1001                 GNIDBG_TX(D_NETTRACE, tx, "rdma -- %d to %lld",
1002                           bytes, atomic64_read(&dev->gnd_rdmaq_bytes_out));
1003         }
1004
1005         atomic_dec(&dev->gnd_n_mdd);
1006         atomic64_sub(bytes, &dev->gnd_nbytes_map);
1007
1008         /* we only get here in the valid cases */
1009         list_del_init(&tx->tx_map_list);
1010         dev->gnd_map_version++;
1011         spin_unlock(&dev->gnd_map_lock);
1012 }
1013
1014 int
1015 kgnilnd_map_buffer(kgn_tx_t *tx)
1016 {
1017         kgn_conn_t       *conn = tx->tx_conn;
1018         kgn_device_t     *dev = conn->gnc_device;
1019         __u32             flags = GNI_MEM_READWRITE;
1020         gni_return_t      rrc;
1021
1022         /* The kgnilnd_mem_register(_segments) Gemini Driver functions can
1023          * be called concurrently as there are internal locks that protect
1024          * any data structures or HW resources. We just need to ensure
1025          * that our concurrency doesn't result in the kgn_device_t
1026          * getting nuked while we are in here */
1027
1028         LASSERTF(conn != NULL, "tx %p with NULL conn, someone forgot"
1029                 " to set tx_conn before calling %s\n", tx, __FUNCTION__);
1030
1031         if (unlikely(CFS_FAIL_CHECK(CFS_FAIL_GNI_MAP_TX)))
1032                 RETURN(-ENOMEM);
1033
1034         if (*kgnilnd_tunables.kgn_bte_relaxed_ordering) {
1035                 flags |= GNI_MEM_RELAXED_PI_ORDERING;
1036         }
1037
1038         switch (tx->tx_buftype) {
1039         default:
1040                 LBUG();
1041
1042         case GNILND_BUF_NONE:
1043         case GNILND_BUF_IMMEDIATE:
1044         case GNILND_BUF_IMMEDIATE_KIOV:
1045         case GNILND_BUF_PHYS_MAPPED:
1046         case GNILND_BUF_VIRT_MAPPED:
1047                 return 0;
1048
1049         case GNILND_BUF_PHYS_UNMAPPED:
1050                 GNITX_ASSERTF(tx, tx->tx_phys != NULL, "physical buffer not there!", NULL);
1051                 rrc = kgnilnd_mem_register_segments(dev->gnd_handle,
1052                         tx->tx_phys, tx->tx_phys_npages, NULL,
1053                         GNI_MEM_PHYS_SEGMENTS | flags,
1054                         &tx->tx_map_key);
1055                 /* could race with other uses of the map counts, but this is ok
1056                  * - this needs to turn into a non-fatal error soon to allow
1057                  *  GART resource, etc starvation handling */
1058                 if (rrc != GNI_RC_SUCCESS) {
1059                         GNIDBG_TX(D_NET, tx, "Can't map %d pages: dev %d "
1060                                 "phys %u pp %u, virt %u nob %llu",
1061                                 tx->tx_phys_npages, dev->gnd_id,
1062                                 dev->gnd_map_nphys, dev->gnd_map_physnop,
1063                                 dev->gnd_map_nvirt, dev->gnd_map_virtnob);
1064                         RETURN(rrc == GNI_RC_ERROR_RESOURCE ? -ENOMEM : -EINVAL);
1065                 }
1066
1067                 tx->tx_buftype = GNILND_BUF_PHYS_MAPPED;
1068                 kgnilnd_mem_add_map_list(dev, tx);
1069                 return 0;
1070
1071         case GNILND_BUF_VIRT_UNMAPPED:
1072                 rrc = kgnilnd_mem_register(dev->gnd_handle,
1073                         (__u64)tx->tx_buffer, tx->tx_nob,
1074                         NULL, flags, &tx->tx_map_key);
1075                 if (rrc != GNI_RC_SUCCESS) {
1076                         GNIDBG_TX(D_NET, tx, "Can't map %u bytes: dev %d "
1077                                 "phys %u pp %u, virt %u nob %llu",
1078                                 tx->tx_nob, dev->gnd_id,
1079                                 dev->gnd_map_nphys, dev->gnd_map_physnop,
1080                                 dev->gnd_map_nvirt, dev->gnd_map_virtnob);
1081                         RETURN(rrc == GNI_RC_ERROR_RESOURCE ? -ENOMEM : -EINVAL);
1082                 }
1083
1084                 tx->tx_buftype = GNILND_BUF_VIRT_MAPPED;
1085                 kgnilnd_mem_add_map_list(dev, tx);
1086                 if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK ||
1087                     tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
1088                         atomic64_add(tx->tx_nob, &dev->gnd_rdmaq_bytes_out);
1089                         GNIDBG_TX(D_NETTRACE, tx, "rdma ++ %d to %ld\n",
1090                                tx->tx_nob, atomic64_read(&dev->gnd_rdmaq_bytes_out));
1091                 }
1092
1093                 return 0;
1094         }
1095 }
1096
1097 void
1098 kgnilnd_add_purgatory_tx(kgn_tx_t *tx)
1099 {
1100         kgn_conn_t              *conn = tx->tx_conn;
1101         kgn_mdd_purgatory_t     *gmp;
1102
1103         LIBCFS_ALLOC(gmp, sizeof(*gmp));
1104         LASSERTF(gmp != NULL, "couldn't allocate MDD purgatory member;"
1105                 " asserting to avoid data corruption\n");
1106         if (tx->tx_buffer_copy)
1107                 gmp->gmp_map_key = tx->tx_buffer_copy_map_key;
1108         else
1109                 gmp->gmp_map_key = tx->tx_map_key;
1110
1111         atomic_inc(&conn->gnc_device->gnd_n_mdd_held);
1112
1113         /* ensure that we don't have a blank purgatory - indicating the
1114          * conn is not already on purgatory lists - we'd never recover these
1115          * MDD if that were the case */
1116         GNITX_ASSERTF(tx, conn->gnc_in_purgatory,
1117                 "conn 0x%p->%s with NULL purgatory",
1118                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
1119
1120         /* link 'er up! - only place we really need to lock for
1121          * concurrent access */
1122         spin_lock(&conn->gnc_list_lock);
1123         list_add_tail(&gmp->gmp_list, &conn->gnc_mdd_list);
1124         spin_unlock(&conn->gnc_list_lock);
1125 }
1126
1127 void
1128 kgnilnd_unmap_buffer(kgn_tx_t *tx, int error)
1129 {
1130         kgn_device_t     *dev;
1131         gni_return_t      rrc;
1132         int               hold_timeout = 0;
1133
1134         /* code below relies on +1 relationship ... */
1135         CLASSERT(GNILND_BUF_PHYS_MAPPED == (GNILND_BUF_PHYS_UNMAPPED + 1));
1136         CLASSERT(GNILND_BUF_VIRT_MAPPED == (GNILND_BUF_VIRT_UNMAPPED + 1));
1137
1138         switch (tx->tx_buftype) {
1139         default:
1140                 LBUG();
1141
1142         case GNILND_BUF_NONE:
1143         case GNILND_BUF_IMMEDIATE:
1144         case GNILND_BUF_PHYS_UNMAPPED:
1145         case GNILND_BUF_VIRT_UNMAPPED:
1146                 break;
1147         case GNILND_BUF_IMMEDIATE_KIOV:
1148                 if (tx->tx_phys != NULL) {
1149                         vunmap(tx->tx_phys);
1150                 } else if (tx->tx_phys == NULL && tx->tx_buffer != NULL) {
1151                         kunmap(tx->tx_imm_pages[0]);
1152                 }
1153                 /* clear to prevent kgnilnd_free_tx from thinking
1154                  * this is a RDMA descriptor */
1155                 tx->tx_phys = NULL;
1156                 break;
1157
1158         case GNILND_BUF_PHYS_MAPPED:
1159         case GNILND_BUF_VIRT_MAPPED:
1160                 LASSERT(tx->tx_conn != NULL);
1161
1162                 dev = tx->tx_conn->gnc_device;
1163
1164                 /* only want to hold if we are closing conn without
1165                  * verified peer notification  - the theory is that
1166                  * a TX error can be communicated in all other cases */
1167                 if (tx->tx_conn->gnc_state != GNILND_CONN_ESTABLISHED &&
1168                     error != -GNILND_NOPURG &&
1169                     kgnilnd_check_purgatory_conn(tx->tx_conn)) {
1170                         kgnilnd_add_purgatory_tx(tx);
1171
1172                         /* The timeout we give to kgni is a deadman stop only.
1173                          *  we are setting high to ensure we don't have the kgni timer
1174                          *  fire before ours fires _and_ is handled */
1175                         hold_timeout = GNILND_TIMEOUT2DEADMAN;
1176
1177                         GNIDBG_TX(D_NET, tx,
1178                                  "dev %p delaying MDD release for %dms key %#llx.%#llx",
1179                                  tx->tx_conn->gnc_device, hold_timeout,
1180                                  tx->tx_map_key.qword1, tx->tx_map_key.qword2);
1181                 }
1182                 if (tx->tx_buffer_copy != NULL) {
1183                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_buffer_copy_map_key, hold_timeout);
1184                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1185                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_map_key, 0);
1186                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1187                 } else {
1188                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_map_key, hold_timeout);
1189                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1190                 }
1191
1192                 tx->tx_buftype--;
1193                 kgnilnd_mem_del_map_list(dev, tx);
1194                 break;
1195         }
1196 }
1197
1198 void
1199 kgnilnd_tx_done(kgn_tx_t *tx, int completion)
1200 {
1201         lnet_msg_t      *lntmsg0, *lntmsg1;
1202         int             status0, status1;
1203         lnet_ni_t       *ni = NULL;
1204         kgn_conn_t      *conn = tx->tx_conn;
1205
1206         LASSERT(!in_interrupt());
1207
1208         lntmsg0 = tx->tx_lntmsg[0]; tx->tx_lntmsg[0] = NULL;
1209         lntmsg1 = tx->tx_lntmsg[1]; tx->tx_lntmsg[1] = NULL;
1210
1211         if (completion &&
1212             !(tx->tx_state & GNILND_TX_QUIET_ERROR) &&
1213             !kgnilnd_conn_clean_errno(completion)) {
1214                 GNIDBG_TOMSG(D_NETERROR, &tx->tx_msg,
1215                        "error %d on tx 0x%p->%s id %u/%d state %s age %ds",
1216                        completion, tx, conn ?
1217                        libcfs_nid2str(conn->gnc_peer->gnp_nid) : "<?>",
1218                        tx->tx_id.txe_smsg_id, tx->tx_id.txe_idx,
1219                        kgnilnd_tx_state2str(tx->tx_list_state),
1220                        cfs_duration_sec((unsigned long)jiffies - tx->tx_qtime));
1221         }
1222
1223         /* The error codes determine if we hold onto the MDD */
1224         kgnilnd_unmap_buffer(tx, completion);
1225
1226         /* we have to deliver a reply on lntmsg[1] for the GET, so make sure
1227          * we play nice with the error codes to avoid delivering a failed
1228          * REQUEST and then a REPLY event as well */
1229
1230         /* return -EIO to lnet - it is the magic value for failed sends */
1231         if (tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
1232                 status0 = 0;
1233                 status1 = completion;
1234         } else {
1235                 status0 = status1 = completion;
1236         }
1237
1238         tx->tx_buftype = GNILND_BUF_NONE;
1239         tx->tx_msg.gnm_type = GNILND_MSG_NONE;
1240
1241         /* lnet_finalize doesn't do anything with the *ni, so ok for us to
1242          * set NULL when we are a tx without a conn */
1243         if (conn != NULL) {
1244                 ni = conn->gnc_peer->gnp_net->gnn_ni;
1245
1246                 spin_lock(&conn->gnc_tx_lock);
1247
1248                 LASSERTF(test_and_clear_bit(tx->tx_id.txe_idx,
1249                         (volatile unsigned long *)&conn->gnc_tx_bits),
1250                         "conn %p tx %p bit %d already cleared\n",
1251                         conn, tx, tx->tx_id.txe_idx);
1252
1253                 LASSERTF(conn->gnc_tx_ref_table[tx->tx_id.txe_idx] != NULL,
1254                          "msg_id %d already NULL\n", tx->tx_id.txe_idx);
1255
1256                 conn->gnc_tx_ref_table[tx->tx_id.txe_idx] = NULL;
1257                 spin_unlock(&conn->gnc_tx_lock);
1258         }
1259
1260         kgnilnd_free_tx(tx);
1261
1262         /* finalize AFTER freeing lnet msgs */
1263
1264         /* warning - we should hold no locks here - calling lnet_finalize
1265          * could free up lnet credits, resulting in a call chain back into
1266          * the LND via kgnilnd_send and friends */
1267
1268         lnet_finalize(ni, lntmsg0, status0);
1269
1270         if (lntmsg1 != NULL) {
1271                 lnet_finalize(ni, lntmsg1, status1);
1272         }
1273 }
1274
1275 void
1276 kgnilnd_txlist_done(struct list_head *txlist, int error)
1277 {
1278         kgn_tx_t        *tx, *txn;
1279         int              err_printed = 0;
1280
1281         if (list_empty(txlist))
1282                 return;
1283
1284         list_for_each_entry_safe(tx, txn, txlist, tx_list) {
1285                 /* only print the first error */
1286                 if (err_printed)
1287                         tx->tx_state |= GNILND_TX_QUIET_ERROR;
1288                 list_del_init(&tx->tx_list);
1289                 kgnilnd_tx_done(tx, error);
1290                 err_printed++;
1291         }
1292 }
1293 int
1294 kgnilnd_set_tx_id(kgn_tx_t *tx, kgn_conn_t *conn)
1295 {
1296         int     id;
1297
1298         spin_lock(&conn->gnc_tx_lock);
1299
1300         /* ID zero is NOT ALLOWED!!! */
1301
1302 search_again:
1303         id = find_next_zero_bit((unsigned long *)&conn->gnc_tx_bits,
1304                                  GNILND_MAX_MSG_ID, conn->gnc_next_tx);
1305         if (id == GNILND_MAX_MSG_ID) {
1306                 if (conn->gnc_next_tx != 1) {
1307                         /* we only searched from next_tx to end and didn't find
1308                          * one, so search again from start */
1309                         conn->gnc_next_tx = 1;
1310                         goto search_again;
1311                 }
1312                 /* couldn't find one! */
1313                 spin_unlock(&conn->gnc_tx_lock);
1314                 return -E2BIG;
1315         }
1316
1317         /* bump next_tx to prevent immediate reuse */
1318         conn->gnc_next_tx = id + 1;
1319
1320         set_bit(id, (volatile unsigned long *)&conn->gnc_tx_bits);
1321         LASSERTF(conn->gnc_tx_ref_table[id] == NULL,
1322                  "tx 0x%p already at id %d\n",
1323                  conn->gnc_tx_ref_table[id], id);
1324
1325         /* delay these until we have a valid ID - prevents bad clear of the bit
1326          * in kgnilnd_tx_done */
1327         tx->tx_conn = conn;
1328         tx->tx_id.txe_cqid = conn->gnc_cqid;
1329
1330         tx->tx_id.txe_idx = id;
1331         conn->gnc_tx_ref_table[id] = tx;
1332
1333         /* Using jiffies to help differentiate against TX reuse - with
1334          * the usual minimum of a 250HZ clock, we wrap jiffies on the same TX
1335          * if we are sending to the same node faster than 256000/sec.
1336          * To help guard against this, we OR in the tx_seq - that is 32 bits */
1337
1338         tx->tx_id.txe_chips = (__u32)(jiffies | atomic_read(&conn->gnc_tx_seq));
1339
1340         GNIDBG_TX(D_NET, tx, "set cookie/id/bits", NULL);
1341
1342         spin_unlock(&conn->gnc_tx_lock);
1343         return 0;
1344 }
1345
1346 static inline int
1347 kgnilnd_tx_should_retry(kgn_conn_t *conn, kgn_tx_t *tx)
1348 {
1349         int             max_retrans = *kgnilnd_tunables.kgn_max_retransmits;
1350         int             log_retrans;
1351         int             log_retrans_level;
1352
1353         /* I need kgni credits to send this.  Replace tx at the head of the
1354          * fmaq and I'll get rescheduled when credits appear */
1355         tx->tx_state = 0;
1356         tx->tx_retrans++;
1357         conn->gnc_tx_retrans++;
1358         log_retrans = ((tx->tx_retrans < 25) || ((tx->tx_retrans % 25) == 0) ||
1359                         (tx->tx_retrans > (max_retrans / 2)));
1360         log_retrans_level = tx->tx_retrans < (max_retrans / 2) ? D_NET : D_NETERROR;
1361
1362         /* Decision time - either error, warn or just retransmit */
1363
1364         /* we don't care about TX timeout - it could be that the network is slower
1365          * or throttled. We'll keep retranmitting - so if the network is so slow
1366          * that we fill up our mailbox, we'll keep trying to resend that msg
1367          * until we exceed the max_retrans _or_ gnc_last_rx expires, indicating
1368          * that he hasn't send us any traffic in return */
1369
1370         if (tx->tx_retrans > max_retrans) {
1371                 /* this means we are not backing off the retransmits
1372                  * in a healthy manner and are likely chewing up the
1373                  * CPU cycles quite badly */
1374                 GNIDBG_TOMSG(D_ERROR, &tx->tx_msg,
1375                         "SOFTWARE BUG: too many retransmits (%d) for tx id %x "
1376                         "conn 0x%p->%s\n",
1377                         tx->tx_retrans, tx->tx_id, conn,
1378                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
1379
1380                 /* yes - double errors to help debug this condition */
1381                 GNIDBG_TOMSG(D_NETERROR, &tx->tx_msg, "connection dead. "
1382                         "unable to send to %s for %lu secs (%d tries)",
1383                         libcfs_nid2str(tx->tx_conn->gnc_peer->gnp_nid),
1384                         cfs_duration_sec(jiffies - tx->tx_cred_wait),
1385                         tx->tx_retrans);
1386
1387                 kgnilnd_close_conn(conn, -ETIMEDOUT);
1388
1389                 /* caller should terminate */
1390                 RETURN(0);
1391         } else {
1392                 /* some reasonable throttling of the debug message */
1393                 if (log_retrans) {
1394                         unsigned long now = jiffies;
1395                         /* XXX Nic: Mystical TX debug here... */
1396                         GNIDBG_SMSG_CREDS(log_retrans_level, conn);
1397                         GNIDBG_TOMSG(log_retrans_level, &tx->tx_msg,
1398                                 "NOT_DONE on conn 0x%p->%s id %x retrans %d wait %dus"
1399                                 " last_msg %uus/%uus last_cq %uus/%uus",
1400                                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
1401                                 tx->tx_id, tx->tx_retrans,
1402                                 jiffies_to_usecs(now - tx->tx_cred_wait),
1403                                 jiffies_to_usecs(now - conn->gnc_last_tx),
1404                                 jiffies_to_usecs(now - conn->gnc_last_rx),
1405                                 jiffies_to_usecs(now - conn->gnc_last_tx_cq),
1406                                 jiffies_to_usecs(now - conn->gnc_last_rx_cq));
1407                 }
1408                 /* caller should retry */
1409                 RETURN(1);
1410         }
1411 }
1412
1413 /* caller must be holding gnd_cq_mutex and not unlock it afterwards, as we need to drop it
1414  * to avoid bad ordering with state_lock */
1415
1416 static inline int
1417 kgnilnd_sendmsg_nolock(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1418                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1419 {
1420         kgn_conn_t      *conn = tx->tx_conn;
1421         kgn_msg_t       *msg = &tx->tx_msg;
1422         int              retry_send;
1423         gni_return_t     rrc;
1424         unsigned long    newest_last_rx, timeout;
1425         unsigned long    now;
1426
1427         LASSERTF((msg->gnm_type == GNILND_MSG_IMMEDIATE) ?
1428                 immediatenob <= *kgnilnd_tunables.kgn_max_immediate :
1429                 immediatenob == 0,
1430                 "msg 0x%p type %d wrong payload size %d\n",
1431                 msg, msg->gnm_type, immediatenob);
1432
1433         /* make sure we catch all the cases where we'd send on a dirty old mbox
1434          * but allow case for sending CLOSE. Since this check is within the CQ
1435          * mutex barrier and the close message is only sent through
1436          * kgnilnd_send_conn_close the last message out the door will be the
1437          * close message.
1438          */
1439         if (atomic_read(&conn->gnc_peer->gnp_dirty_eps) != 0 && msg->gnm_type != GNILND_MSG_CLOSE) {
1440                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1441                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1442                 /* Return -ETIME, we are closing the connection already so we dont want to
1443                  * have this tx hit the wire. The tx will be killed by the calling function.
1444                  * Once the EP is marked dirty the close message will be the last
1445                  * thing to hit the wire */
1446                 return -ETIME;
1447         }
1448
1449         now = jiffies;
1450         timeout = cfs_time_seconds(conn->gnc_timeout);
1451
1452         newest_last_rx = GNILND_LASTRX(conn);
1453
1454         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SEND_TIMEOUT)) {
1455                 now = now + (GNILND_TIMEOUTRX(timeout) * 2);
1456         }
1457
1458         if (time_after_eq(now, newest_last_rx + GNILND_TIMEOUTRX(timeout))) {
1459                 GNIDBG_CONN(D_NETERROR|D_CONSOLE, conn,
1460                             "Cant send to %s after timeout lapse of %lu; TO %lu\n",
1461                 libcfs_nid2str(conn->gnc_peer->gnp_nid),
1462                 cfs_duration_sec(now - newest_last_rx),
1463                 cfs_duration_sec(GNILND_TIMEOUTRX(timeout)));
1464                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1465                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1466                 return -ETIME;
1467         }
1468
1469         GNITX_ASSERTF(tx, (conn != NULL) && (tx->tx_id.txe_idx != 0), "tx id unset!", NULL);
1470         /* msg->gnm_srcnid is set when the message is initialized by whatever function is
1471          * creating the message this allows the message to contain the correct LNET NID/NET needed
1472          * instead of the one that the peer/conn uses for sending the data.
1473          */
1474         msg->gnm_connstamp = conn->gnc_my_connstamp;
1475         msg->gnm_payload_len = immediatenob;
1476         msg->gnm_seq = atomic_read(&conn->gnc_tx_seq);
1477
1478         /* always init here - kgn_checksum is a /sys module tunable
1479          * and can be flipped at any point, even between msg init and sending */
1480         msg->gnm_cksum = 0;
1481         if (*kgnilnd_tunables.kgn_checksum) {
1482                 /* We must set here and not in kgnilnd_init_msg,
1483                  * we could resend this msg many times
1484                  * (NOT_DONE from gni_smsg_send below) and wouldn't pass
1485                  * through init_msg again */
1486                 msg->gnm_cksum = kgnilnd_cksum(msg, sizeof(kgn_msg_t));
1487                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SMSG_CKSUM1)) {
1488                         msg->gnm_cksum += 0xf00f;
1489                 }
1490         }
1491
1492         GNIDBG_TOMSG(D_NET, msg, "tx 0x%p conn 0x%p->%s sending SMSG sz %u id %x/%d [%p for %u]",
1493                tx, conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
1494                sizeof(kgn_msg_t), tx->tx_id.txe_smsg_id,
1495                tx->tx_id.txe_idx, immediate, immediatenob);
1496
1497         if (unlikely(tx->tx_state & GNILND_TX_FAIL_SMSG)) {
1498                 rrc = cfs_fail_val ? cfs_fail_val : GNI_RC_NOT_DONE;
1499         } else {
1500                 rrc = kgnilnd_smsg_send(conn->gnc_ephandle,
1501                                         msg, sizeof(*msg), immediate,
1502                                         immediatenob,
1503                                         tx->tx_id.txe_smsg_id);
1504         }
1505
1506         switch (rrc) {
1507         case GNI_RC_SUCCESS:
1508                 atomic_inc(&conn->gnc_tx_seq);
1509                 conn->gnc_last_tx = jiffies;
1510                 /* no locking here as LIVE isn't a list */
1511                 kgnilnd_tx_add_state_locked(tx, NULL, conn, GNILND_TX_LIVE_FMAQ, 1);
1512
1513                 /* this needs to be checked under lock as it might be freed from a completion
1514                  * event.
1515                  */
1516                 if (msg->gnm_type == GNILND_MSG_NOOP) {
1517                         set_mb(conn->gnc_last_noop_sent, jiffies);
1518                 }
1519
1520                 /* serialize with seeing CQ events for completion on this, as well as
1521                  * tx_seq */
1522                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1523                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1524
1525                 atomic_inc(&conn->gnc_device->gnd_short_ntx);
1526                 atomic64_add(immediatenob, &conn->gnc_device->gnd_short_txbytes);
1527                 kgnilnd_peer_alive(conn->gnc_peer);
1528                 GNIDBG_SMSG_CREDS(D_NET, conn);
1529                 return 0;
1530
1531         case GNI_RC_NOT_DONE:
1532                 /* XXX Nic: We need to figure out how to track this
1533                  * - there are bound to be good reasons for it,
1534                  * but we want to know when it happens */
1535                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1536                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1537                 /* We'll handle this error inline - makes the calling logic much more
1538                  * clean */
1539
1540                 /* If no lock, caller doesn't want us to retry */
1541                 if (state_lock == NULL) {
1542                         return -EAGAIN;
1543                 }
1544
1545                 retry_send = kgnilnd_tx_should_retry(conn, tx);
1546                 if (retry_send) {
1547                         /* add to head of list for the state and retries */
1548                         spin_lock(state_lock);
1549                         kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, state, 0);
1550                         spin_unlock(state_lock);
1551
1552                         /* We only reschedule for a certain number of retries, then
1553                          * we will wait for the CQ events indicating a release of SMSG
1554                          * credits */
1555                         if (tx->tx_retrans < (*kgnilnd_tunables.kgn_max_retransmits/4)) {
1556                                 kgnilnd_schedule_conn(conn);
1557                                 return 0;
1558                         } else {
1559                                 /* CQ event coming in signifies either TX completed or
1560                                  * RX receive. Either of these *could* free up credits
1561                                  * in the SMSG mbox and we should try sending again */
1562                                 GNIDBG_TX(D_NET, tx, "waiting for CQID %u event to resend",
1563                                          tx->tx_conn->gnc_cqid);
1564                                 /* use +ve return code to let upper layers know they
1565                                  * should stop looping on sends */
1566                                 return EAGAIN;
1567                         }
1568                 } else {
1569                         return -EAGAIN;
1570                 }
1571         default:
1572                 /* handle bad retcode gracefully */
1573                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1574                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1575                 return -EIO;
1576         }
1577 }
1578
1579 /* kgnilnd_sendmsg has hard wait on gnd_cq_mutex */
1580 static inline int
1581 kgnilnd_sendmsg(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1582                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1583 {
1584         kgn_device_t    *dev = tx->tx_conn->gnc_device;
1585         unsigned long    timestamp;
1586         int              rc;
1587
1588         timestamp = jiffies;
1589         kgnilnd_gl_mutex_lock(&dev->gnd_cq_mutex);
1590         kgnilnd_conn_mutex_lock(&tx->tx_conn->gnc_smsg_mutex);
1591         /* delay in jiffies - we are really concerned only with things that
1592          * result in a schedule() or really holding this off for long times .
1593          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1594         dev->gnd_mutex_delay += (long) jiffies - timestamp;
1595
1596         rc = kgnilnd_sendmsg_nolock(tx, immediate, immediatenob, state_lock, state);
1597
1598         RETURN(rc);
1599 }
1600
1601
1602 /* returns -EAGAIN for lock miss, anything else < 0 is hard error, >=0 for success */
1603 static inline int
1604 kgnilnd_sendmsg_trylock(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1605                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1606 {
1607         kgn_conn_t      *conn = tx->tx_conn;
1608         kgn_device_t    *dev = conn->gnc_device;
1609         unsigned long    timestamp;
1610         int              rc;
1611
1612         timestamp = jiffies;
1613
1614         /* technically we are doing bad things with the read_lock on the peer_conn
1615          * table, but we shouldn't be sleeping inside here - and we don't sleep/block
1616          * for the mutex. I bet lockdep is gonna flag this one though... */
1617
1618         /* there are a few cases where we don't want the immediate send - like
1619          * when we are in the scheduler thread and it'd harm the latency of
1620          * getting messages up to LNet */
1621
1622         /* rmb for gnd_ready */
1623         smp_rmb();
1624         if (conn->gnc_device->gnd_ready == GNILND_DEV_LOOP) {
1625                 rc = 0;
1626                 atomic_inc(&conn->gnc_device->gnd_fast_block);
1627         } else if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
1628                /* dont hit HW during quiesce */
1629                 rc = 0;
1630         } else if (unlikely(atomic_read(&conn->gnc_peer->gnp_dirty_eps))) {
1631                /* dont hit HW if stale EPs and conns left to close */
1632                 rc = 0;
1633         } else {
1634                 atomic_inc(&conn->gnc_device->gnd_fast_try);
1635                 rc = kgnilnd_trylock(&conn->gnc_device->gnd_cq_mutex,
1636                                      &conn->gnc_smsg_mutex);
1637         }
1638         if (!rc) {
1639                 rc = -EAGAIN;
1640         } else {
1641                 /* we got the mutex and weren't blocked */
1642
1643                 /* delay in jiffies - we are really concerned only with things that
1644                  * result in a schedule() or really holding this off for long times .
1645                  * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1646                 dev->gnd_mutex_delay += (long) jiffies - timestamp;
1647
1648                 atomic_inc(&conn->gnc_device->gnd_fast_ok);
1649                 tx->tx_qtime = jiffies;
1650                 tx->tx_state = GNILND_TX_WAITING_COMPLETION;
1651                 rc = kgnilnd_sendmsg_nolock(tx, tx->tx_buffer, tx->tx_nob, &conn->gnc_list_lock, GNILND_TX_FMAQ);
1652                 /* _nolock unlocks the mutex for us */
1653         }
1654
1655         RETURN(rc);
1656 }
1657
1658 /* lets us know if we can push this RDMA through now */
1659 static int
1660 kgnilnd_auth_rdma_bytes(kgn_device_t *dev, kgn_tx_t *tx)
1661 {
1662         long    bytes_left;
1663
1664         bytes_left = atomic64_sub_return(tx->tx_nob, &dev->gnd_rdmaq_bytes_ok);
1665
1666         if (bytes_left < 0) {
1667                 atomic64_add(tx->tx_nob, &dev->gnd_rdmaq_bytes_ok);
1668                 atomic_inc(&dev->gnd_rdmaq_nstalls);
1669                 smp_wmb();
1670
1671                 CDEBUG(D_NET, "no bytes to send, turning on timer for %lu\n",
1672                        dev->gnd_rdmaq_deadline);
1673                 mod_timer(&dev->gnd_rdmaq_timer, dev->gnd_rdmaq_deadline);
1674                 /* we never del this timer - at worst it schedules us.. */
1675                 return -EAGAIN;
1676         } else {
1677                 return 0;
1678         }
1679 }
1680
1681 /* this adds a TX to the queue pending throttling authorization before
1682  * we allow our remote peer to launch a PUT at us */
1683 void
1684 kgnilnd_queue_rdma(kgn_conn_t *conn, kgn_tx_t *tx)
1685 {
1686         int     rc;
1687
1688         /* we cannot go into send_mapped_tx from here as we are holding locks
1689          * and mem registration might end up allocating memory in kgni.
1690          * That said, we'll push this as far as we can into the queue process */
1691         rc = kgnilnd_auth_rdma_bytes(conn->gnc_device, tx);
1692
1693         if (rc < 0) {
1694                 spin_lock(&conn->gnc_device->gnd_rdmaq_lock);
1695                 kgnilnd_tx_add_state_locked(tx, NULL, conn, GNILND_TX_RDMAQ, 0);
1696                 /* lets us know how delayed RDMA is */
1697                 tx->tx_qtime = jiffies;
1698                 spin_unlock(&conn->gnc_device->gnd_rdmaq_lock);
1699         } else {
1700                 /* we have RDMA authorized, now it just needs a MDD and to hit the wire */
1701                 spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
1702                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 0);
1703                 /* lets us know how delayed mapping is */
1704                 tx->tx_qtime = jiffies;
1705                 spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
1706         }
1707
1708         /* make sure we wake up sched to run this */
1709         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
1710 }
1711
1712 /* push TX through state machine */
1713 void
1714 kgnilnd_queue_tx(kgn_conn_t *conn, kgn_tx_t *tx)
1715 {
1716         int            rc = 0;
1717         int            add_tail = 1;
1718
1719         /* set the tx_id here, we delay it until we have an actual conn
1720          * to fiddle with
1721          * in some cases, the tx_id is already set to provide for things
1722          * like RDMA completion cookies, etc */
1723         if (tx->tx_id.txe_idx == 0) {
1724                 rc = kgnilnd_set_tx_id(tx, conn);
1725                 if (rc != 0) {
1726                         kgnilnd_tx_done(tx, rc);
1727                         return;
1728                 }
1729         }
1730
1731         CDEBUG(D_NET, "%s to conn %p for %s\n", kgnilnd_msgtype2str(tx->tx_msg.gnm_type),
1732                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
1733
1734         /* Only let NOOPs to be sent while fail loc is set, otherwise kill the tx.
1735          */
1736         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_ONLY_NOOP) && (tx->tx_msg.gnm_type != GNILND_MSG_NOOP)) {
1737                 kgnilnd_tx_done(tx, rc);
1738                 return;
1739         }
1740
1741         switch (tx->tx_msg.gnm_type) {
1742         case GNILND_MSG_PUT_ACK:
1743         case GNILND_MSG_GET_REQ:
1744         case GNILND_MSG_PUT_REQ_REV:
1745         case GNILND_MSG_GET_ACK_REV:
1746                 /* hijacking time! If this messages will authorize our peer to
1747                  * send his dirty little bytes in an RDMA, we need to get permission */
1748                 kgnilnd_queue_rdma(conn, tx);
1749                 break;
1750         case GNILND_MSG_IMMEDIATE:
1751                 /* try to send right now, can help reduce latency */
1752                 rc = kgnilnd_sendmsg_trylock(tx, tx->tx_buffer, tx->tx_nob, &conn->gnc_list_lock, GNILND_TX_FMAQ);
1753
1754                 if (rc >= 0) {
1755                         /* it was sent, break out of switch to avoid default case of queueing */
1756                         break;
1757                 }
1758                 /* needs to queue to try again, so fall through to default case */
1759         case GNILND_MSG_NOOP:
1760                 /* Just make sure this goes out first for this conn */
1761                 add_tail = 0;
1762                 /* fall through... */
1763         default:
1764                 spin_lock(&conn->gnc_list_lock);
1765                 kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, GNILND_TX_FMAQ, add_tail);
1766                 tx->tx_qtime = jiffies;
1767                 spin_unlock(&conn->gnc_list_lock);
1768                 kgnilnd_schedule_conn(conn);
1769         }
1770 }
1771
1772 void
1773 kgnilnd_launch_tx(kgn_tx_t *tx, kgn_net_t *net, lnet_process_id_t *target)
1774 {
1775         kgn_peer_t      *peer;
1776         kgn_peer_t      *new_peer = NULL;
1777         kgn_conn_t      *conn = NULL;
1778         int              rc;
1779         int              node_state;
1780
1781         ENTRY;
1782
1783         /* If I get here, I've committed to send, so I complete the tx with
1784          * failure on any problems */
1785
1786         GNITX_ASSERTF(tx, tx->tx_conn == NULL,
1787                       "tx already has connection %p", tx->tx_conn);
1788
1789         /* do all of the peer & conn searching in one swoop - this avoids
1790          * nastiness when dropping locks and needing to maintain a sane state
1791          * in the face of stack reset or something else nuking peers & conns */
1792
1793         /* I expect to find him, so only take a read lock */
1794         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
1795
1796         peer = kgnilnd_find_peer_locked(target->nid);
1797         if (peer != NULL) {
1798                 conn = kgnilnd_find_conn_locked(peer);
1799                 /* this could be NULL during quiesce */
1800                 if (conn != NULL)  {
1801                         /* Connection exists; queue message on it */
1802                         kgnilnd_queue_tx(conn, tx);
1803                         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1804                         RETURN_EXIT;
1805                 }
1806
1807                 /* don't create a connection if the peer is marked down */
1808                 if (peer->gnp_state != GNILND_PEER_UP) {
1809                         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1810                         rc = -ENETRESET;
1811                         GOTO(no_peer, rc);
1812                 }
1813         }
1814
1815         /* creating peer or conn; I'll need a write lock... */
1816         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1817
1818         CFS_RACE(CFS_FAIL_GNI_FIND_TARGET);
1819
1820         node_state = kgnilnd_get_node_state(LNET_NIDADDR(target->nid));
1821
1822         /* NB - this will not block during normal operations -
1823          * the only writer of this is in the startup/shutdown path. */
1824         rc = down_read_trylock(&kgnilnd_data.kgn_net_rw_sem);
1825         if (!rc) {
1826                 rc = -ESHUTDOWN;
1827                 GOTO(no_peer, rc);
1828         }
1829
1830         /* ignore previous peer entirely - we cycled the lock, so we
1831          * will create new peer and at worst drop it if peer is still
1832          * in the tables */
1833         rc = kgnilnd_create_peer_safe(&new_peer, target->nid, net, node_state);
1834         if (rc != 0) {
1835                 up_read(&kgnilnd_data.kgn_net_rw_sem);
1836                 GOTO(no_peer, rc);
1837         }
1838
1839         write_lock(&kgnilnd_data.kgn_peer_conn_lock);
1840         up_read(&kgnilnd_data.kgn_net_rw_sem);
1841
1842         /* search for peer again now that we have the lock
1843          * if we don't find it, add our new one to the list */
1844         kgnilnd_add_peer_locked(target->nid, new_peer, &peer);
1845
1846         /* don't create a connection if the peer is not up */
1847         if (peer->gnp_state != GNILND_PEER_UP) {
1848                 write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1849                 rc = -ENETRESET;
1850                 GOTO(no_peer, rc);
1851         }
1852
1853         conn = kgnilnd_find_or_create_conn_locked(peer);
1854
1855         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_DGRAM_DROP_TX)) {
1856                 write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1857                 GOTO(no_peer, rc);
1858         }
1859
1860         if (conn != NULL) {
1861                 /* oh hey, found a conn now... magical */
1862                 kgnilnd_queue_tx(conn, tx);
1863         } else {
1864                 /* no conn, must be trying to connect - so we queue for now */
1865                 tx->tx_qtime = jiffies;
1866                 kgnilnd_tx_add_state_locked(tx, peer, NULL, GNILND_TX_PEERQ, 1);
1867         }
1868         write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1869         RETURN_EXIT;
1870 no_peer:
1871         kgnilnd_tx_done(tx, rc);
1872         RETURN_EXIT;
1873 }
1874
1875 int
1876 kgnilnd_rdma(kgn_tx_t *tx, int type,
1877             kgn_rdma_desc_t *sink, unsigned int nob, __u64 cookie)
1878 {
1879         kgn_conn_t   *conn = tx->tx_conn;
1880         unsigned long timestamp;
1881         gni_post_type_t post_type;
1882         gni_return_t  rrc;
1883         int rc = 0;
1884         unsigned int desc_nob = nob;
1885         void *desc_buffer = tx->tx_buffer;
1886         gni_mem_handle_t desc_map_key = tx->tx_map_key;
1887         LASSERTF(kgnilnd_tx_mapped(tx),
1888                 "unmapped tx %p\n", tx);
1889         LASSERTF(conn != NULL,
1890                 "NULL conn on tx %p, naughty, naughty\n", tx);
1891         LASSERTF(nob <= sink->gnrd_nob,
1892                 "nob %u > sink->gnrd_nob %d (%p)\n",
1893                 nob, sink->gnrd_nob, sink);
1894         LASSERTF(nob <= tx->tx_nob,
1895                 "nob %d > tx(%p)->tx_nob %d\n",
1896                 nob, tx, tx->tx_nob);
1897
1898         switch (type) {
1899         case GNILND_MSG_GET_DONE:
1900         case GNILND_MSG_PUT_DONE:
1901                 post_type = GNI_POST_RDMA_PUT;
1902                 break;
1903         case GNILND_MSG_GET_DONE_REV:
1904         case GNILND_MSG_PUT_DONE_REV:
1905                 post_type = GNI_POST_RDMA_GET;
1906                 break;
1907         default:
1908                 CERROR("invalid msg type %s (%d)\n",
1909                         kgnilnd_msgtype2str(type), type);
1910                 LBUG();
1911         }
1912         if (post_type == GNI_POST_RDMA_GET) {
1913                 /* Check for remote buffer / local buffer / length alignment. All must be 4 byte
1914                  * aligned. If the local buffer is not aligned correctly using the copy buffer
1915                  * will fix that issue. If length is misaligned copy buffer will also fix the issue, we end
1916                  * up transferring extra bytes into the buffer but only copy the correct nob into the original
1917                  * buffer.  Remote offset correction is done through a combination of adjusting the offset,
1918                  * making sure the length and addr are aligned and copying the data into the correct location
1919                  * once the transfer has completed.
1920                  */
1921                 if ((((__u64)((unsigned long)tx->tx_buffer)) & 3) ||
1922                       (sink->gnrd_addr & 3) ||
1923                       (nob & 3)) {
1924
1925                         tx->tx_offset = ((__u64)((unsigned long)sink->gnrd_addr)) & 3;
1926                         if (tx->tx_offset)
1927                                 kgnilnd_admin_addref(kgnilnd_data.kgn_rev_offset);
1928
1929                         if ((nob + tx->tx_offset) & 3) {
1930                                 desc_nob = ((nob + tx->tx_offset) + (4 - ((nob + tx->tx_offset) & 3)));
1931                                 kgnilnd_admin_addref(kgnilnd_data.kgn_rev_length);
1932                         } else {
1933                                 desc_nob = (nob + tx->tx_offset);
1934                         }
1935
1936                         if (tx->tx_buffer_copy == NULL) {
1937                                 /* Allocate the largest copy buffer we will need, this will prevent us from overwriting data
1938                                  * and require at most we allocate a few extra bytes. */
1939                                 tx->tx_buffer_copy = kgnilnd_vzalloc(desc_nob);
1940
1941                                 if (!tx->tx_buffer_copy) {
1942                                         /* allocation of buffer failed nak the rdma */
1943                                         kgnilnd_nak_rdma(tx->tx_conn, tx->tx_msg.gnm_type, -EFAULT, cookie, tx->tx_msg.gnm_srcnid);
1944                                         kgnilnd_tx_done(tx, -EFAULT);
1945                                         return 0;
1946                                 }
1947                                 kgnilnd_admin_addref(kgnilnd_data.kgn_rev_copy_buff);
1948                                 rc = kgnilnd_mem_register(conn->gnc_device->gnd_handle, (__u64)tx->tx_buffer_copy, desc_nob, NULL, GNI_MEM_READWRITE, &tx->tx_buffer_copy_map_key);
1949                                 if (rc != GNI_RC_SUCCESS) {
1950                                         /* Registration Failed nak rdma and kill the tx. */
1951                                         kgnilnd_vfree(tx->tx_buffer_copy,
1952                                                       desc_nob);
1953                                         tx->tx_buffer_copy = NULL;
1954                                         kgnilnd_nak_rdma(tx->tx_conn, tx->tx_msg.gnm_type, -EFAULT, cookie, tx->tx_msg.gnm_srcnid);
1955                                         kgnilnd_tx_done(tx, -EFAULT);
1956                                         return 0;
1957                                 }
1958                         }
1959                         desc_map_key = tx->tx_buffer_copy_map_key;
1960                         desc_buffer = tx->tx_buffer_copy;
1961                 }
1962         }
1963
1964         memset(&tx->tx_rdma_desc, 0, sizeof(tx->tx_rdma_desc));
1965         tx->tx_rdma_desc.post_id = tx->tx_id.txe_cookie;
1966         tx->tx_rdma_desc.type = post_type;
1967         tx->tx_rdma_desc.cq_mode = GNI_CQMODE_GLOBAL_EVENT;
1968         tx->tx_rdma_desc.local_addr = (__u64)((unsigned long)desc_buffer);
1969         tx->tx_rdma_desc.local_mem_hndl = desc_map_key;
1970         tx->tx_rdma_desc.remote_addr = sink->gnrd_addr - tx->tx_offset;
1971         tx->tx_rdma_desc.remote_mem_hndl = sink->gnrd_key;
1972         tx->tx_rdma_desc.length = desc_nob;
1973         tx->tx_nob_rdma = nob;
1974         if (post_type == GNI_POST_RDMA_PUT && *kgnilnd_tunables.kgn_bte_put_dlvr_mode)
1975                 tx->tx_rdma_desc.dlvr_mode = *kgnilnd_tunables.kgn_bte_put_dlvr_mode;
1976         if (post_type == GNI_POST_RDMA_GET && *kgnilnd_tunables.kgn_bte_get_dlvr_mode)
1977                 tx->tx_rdma_desc.dlvr_mode = *kgnilnd_tunables.kgn_bte_get_dlvr_mode;
1978         /* prep final completion message */
1979         kgnilnd_init_msg(&tx->tx_msg, type, tx->tx_msg.gnm_srcnid);
1980         tx->tx_msg.gnm_u.completion.gncm_cookie = cookie;
1981         /* send actual size RDMA'd in retval */
1982         tx->tx_msg.gnm_u.completion.gncm_retval = nob;
1983
1984         kgnilnd_compute_rdma_cksum(tx, nob);
1985
1986         if (nob == 0) {
1987                 kgnilnd_queue_tx(conn, tx);
1988                 return 0;
1989         }
1990
1991         /* Don't lie (CLOSE == RDMA idle) */
1992         LASSERTF(!conn->gnc_close_sent, "tx %p on conn %p after close sent %d\n",
1993                  tx, conn, conn->gnc_close_sent);
1994
1995         GNIDBG_TX(D_NET, tx, "Post RDMA type 0x%02x conn %p dlvr_mode "
1996                 "0x%x cookie:%#llx",
1997                 type, conn, tx->tx_rdma_desc.dlvr_mode, cookie);
1998
1999         /* set CQ dedicated for RDMA */
2000         tx->tx_rdma_desc.src_cq_hndl = conn->gnc_device->gnd_snd_rdma_cqh;
2001
2002         timestamp = jiffies;
2003         kgnilnd_conn_mutex_lock(&conn->gnc_rdma_mutex);
2004         kgnilnd_gl_mutex_lock(&conn->gnc_device->gnd_cq_mutex);
2005         /* delay in jiffies - we are really concerned only with things that
2006          * result in a schedule() or really holding this off for long times .
2007          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
2008         conn->gnc_device->gnd_mutex_delay += (long) jiffies - timestamp;
2009
2010         rrc = kgnilnd_post_rdma(conn->gnc_ephandle, &tx->tx_rdma_desc);
2011
2012         if (rrc == GNI_RC_ERROR_RESOURCE) {
2013                 kgnilnd_conn_mutex_unlock(&conn->gnc_rdma_mutex);
2014                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
2015                 kgnilnd_unmap_buffer(tx, 0);
2016
2017                 if (tx->tx_buffer_copy != NULL) {
2018                         kgnilnd_vfree(tx->tx_buffer_copy, desc_nob);
2019                         tx->tx_buffer_copy = NULL;
2020                 }
2021
2022                 spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2023                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn,
2024                                             GNILND_TX_MAPQ, 0);
2025                 spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2026                 kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2027                 return -EAGAIN;
2028         }
2029
2030         spin_lock(&conn->gnc_list_lock);
2031         kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, GNILND_TX_LIVE_RDMAQ, 1);
2032         tx->tx_qtime = jiffies;
2033         spin_unlock(&conn->gnc_list_lock);
2034         kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
2035         kgnilnd_conn_mutex_unlock(&conn->gnc_rdma_mutex);
2036
2037         /* XXX Nic: is this a place we should handle more errors for
2038          * robustness sake */
2039         LASSERT(rrc == GNI_RC_SUCCESS);
2040         return 0;
2041 }
2042
2043 kgn_rx_t *
2044 kgnilnd_alloc_rx(void)
2045 {
2046         kgn_rx_t        *rx;
2047
2048         rx = kmem_cache_alloc(kgnilnd_data.kgn_rx_cache, GFP_ATOMIC);
2049         if (rx == NULL) {
2050                 CERROR("failed to allocate rx\n");
2051                 return NULL;
2052         }
2053         CDEBUG(D_MALLOC, "slab-alloced 'rx': %lu at %p.\n",
2054                sizeof(*rx), rx);
2055
2056         /* no memset to zero, we'll always fill all members */
2057         return rx;
2058 }
2059
2060 /* release is to just free connection resources
2061  * we use this for the eager path after copying */
2062 void
2063 kgnilnd_release_msg(kgn_conn_t *conn)
2064 {
2065         gni_return_t    rrc;
2066         unsigned long   timestamp;
2067
2068         CDEBUG(D_NET, "consuming %p\n", conn);
2069
2070         timestamp = jiffies;
2071         kgnilnd_gl_mutex_lock(&conn->gnc_device->gnd_cq_mutex);
2072         /* delay in jiffies - we are really concerned only with things that
2073          * result in a schedule() or really holding this off for long times .
2074          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
2075         conn->gnc_device->gnd_mutex_delay += (long) jiffies - timestamp;
2076
2077         rrc = kgnilnd_smsg_release(conn->gnc_ephandle);
2078         kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
2079
2080         LASSERTF(rrc == GNI_RC_SUCCESS, "bad rrc %d\n", rrc);
2081         GNIDBG_SMSG_CREDS(D_NET, conn);
2082
2083         return;
2084 }
2085
2086 void
2087 kgnilnd_consume_rx(kgn_rx_t *rx)
2088 {
2089         kgn_conn_t      *conn = rx->grx_conn;
2090         kgn_msg_t       *rxmsg = rx->grx_msg;
2091
2092         /* if we are eager, free the cache alloc'd msg */
2093         if (unlikely(rx->grx_eager)) {
2094                 LIBCFS_FREE(rxmsg, sizeof(*rxmsg) + *kgnilnd_tunables.kgn_max_immediate);
2095                 atomic_dec(&kgnilnd_data.kgn_neager_allocs);
2096
2097                 /* release ref from eager_recv */
2098                 kgnilnd_conn_decref(conn);
2099         } else {
2100                 GNIDBG_MSG(D_NET, rxmsg, "rx %p processed", rx);
2101                 kgnilnd_release_msg(conn);
2102         }
2103
2104         kmem_cache_free(kgnilnd_data.kgn_rx_cache, rx);
2105         CDEBUG(D_MALLOC, "slab-freed 'rx': %lu at %p.\n",
2106                sizeof(*rx), rx);
2107
2108         return;
2109 }
2110
2111 int
2112 kgnilnd_send(lnet_ni_t *ni, void *private, lnet_msg_t *lntmsg)
2113 {
2114         lnet_hdr_t       *hdr = &lntmsg->msg_hdr;
2115         int               type = lntmsg->msg_type;
2116         lnet_process_id_t target = lntmsg->msg_target;
2117         int               target_is_router = lntmsg->msg_target_is_router;
2118         int               routing = lntmsg->msg_routing;
2119         unsigned int      niov = lntmsg->msg_niov;
2120         struct kvec      *iov = lntmsg->msg_iov;
2121         lnet_kiov_t      *kiov = lntmsg->msg_kiov;
2122         unsigned int      offset = lntmsg->msg_offset;
2123         unsigned int      nob = lntmsg->msg_len;
2124         unsigned int      msg_vmflush = lntmsg->msg_vmflush;
2125         kgn_net_t        *net = ni->ni_data;
2126         kgn_tx_t         *tx;
2127         int               rc = 0;
2128         int               mpflag = 0;
2129         int               reverse_rdma_flag = *kgnilnd_tunables.kgn_reverse_rdma;
2130
2131         /* NB 'private' is different depending on what we're sending.... */
2132         LASSERT(!in_interrupt());
2133
2134         CDEBUG(D_NET, "sending msg type %d with %d bytes in %d frags to %s\n",
2135                type, nob, niov, libcfs_id2str(target));
2136
2137         LASSERTF(nob == 0 || niov > 0,
2138                 "lntmsg %p nob %d niov %d\n", lntmsg, nob, niov);
2139         LASSERTF(niov <= LNET_MAX_IOV,
2140                 "lntmsg %p niov %d\n", lntmsg, niov);
2141
2142         /* payload is either all vaddrs or all pages */
2143         LASSERTF(!(kiov != NULL && iov != NULL),
2144                 "lntmsg %p kiov %p iov %p\n", lntmsg, kiov, iov);
2145
2146         if (msg_vmflush)
2147                 mpflag = cfs_memory_pressure_get_and_set();
2148
2149         switch (type) {
2150         default:
2151                 CERROR("lntmsg %p with unexpected type %d\n",
2152                         lntmsg, type);
2153                 LBUG();
2154
2155         case LNET_MSG_ACK:
2156                 LASSERTF(nob == 0, "lntmsg %p nob %d\n",
2157                         lntmsg, nob);
2158                 break;
2159
2160         case LNET_MSG_GET:
2161                 LASSERT(niov == 0);
2162                 LASSERT(nob == 0);
2163
2164                 if (routing || target_is_router)
2165                         break;                  /* send IMMEDIATE */
2166
2167                 /* it is safe to do direct GET with out mapping buffer for RDMA as we
2168                  * check the eventual sink buffer here - if small enough, remote
2169                  * end is perfectly capable of returning data in short message -
2170                  * The magic is that we call lnet_parse in kgnilnd_recv with rdma_req=0
2171                  * for IMMEDIATE messages which will have it send a real reply instead
2172                  * of doing kgnilnd_recv to have the RDMA continued */
2173                 if (lntmsg->msg_md->md_length <= *kgnilnd_tunables.kgn_max_immediate)
2174                        break;
2175
2176                 if ((reverse_rdma_flag & GNILND_REVERSE_GET) == 0)
2177                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_REQ, ni->ni_nid);
2178                 else
2179                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_REQ_REV, ni->ni_nid);
2180
2181                 if (tx == NULL) {
2182                         rc = -ENOMEM;
2183                         goto out;
2184                 }
2185                 /* slightly different options as we might actually have a GET with a
2186                  * MD_KIOV set but a non-NULL md_iov.iov */
2187                 if ((lntmsg->msg_md->md_options & LNET_MD_KIOV) == 0)
2188                         rc = kgnilnd_setup_rdma_buffer(tx, lntmsg->msg_md->md_niov,
2189                                                       lntmsg->msg_md->md_iov.iov, NULL,
2190                                                       0, lntmsg->msg_md->md_length);
2191                 else
2192                         rc = kgnilnd_setup_rdma_buffer(tx, lntmsg->msg_md->md_niov,
2193                                                       NULL, lntmsg->msg_md->md_iov.kiov,
2194                                                       0, lntmsg->msg_md->md_length);
2195                 if (rc != 0) {
2196                         CERROR("unable to setup buffer: %d\n", rc);
2197                         kgnilnd_tx_done(tx, rc);
2198                         rc = -EIO;
2199                         goto out;
2200                 }
2201
2202                 tx->tx_lntmsg[1] = lnet_create_reply_msg(ni, lntmsg);
2203                 if (tx->tx_lntmsg[1] == NULL) {
2204                         CERROR("Can't create reply for GET to %s\n",
2205                                libcfs_nid2str(target.nid));
2206                         kgnilnd_tx_done(tx, rc);
2207                         rc = -EIO;
2208                         goto out;
2209                 }
2210
2211                 tx->tx_lntmsg[0] = lntmsg;
2212                 if ((reverse_rdma_flag & GNILND_REVERSE_GET) == 0)
2213                         tx->tx_msg.gnm_u.get.gngm_hdr = *hdr;
2214                 else
2215                         tx->tx_msg.gnm_u.putreq.gnprm_hdr = *hdr;
2216
2217                 /* rest of tx_msg is setup just before it is sent */
2218                 kgnilnd_launch_tx(tx, net, &target);
2219                 goto out;
2220         case LNET_MSG_REPLY:
2221         case LNET_MSG_PUT:
2222                 /* to save on MDDs, we'll handle short kiov by vmap'ing
2223                  * and sending via SMSG */
2224                 if (nob <= *kgnilnd_tunables.kgn_max_immediate)
2225                        break;
2226
2227                 if ((reverse_rdma_flag & GNILND_REVERSE_PUT) == 0)
2228                         tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_REQ, ni->ni_nid);
2229                 else
2230                         tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_REQ_REV, ni->ni_nid);
2231
2232                 if (tx == NULL) {
2233                         rc = -ENOMEM;
2234                         goto out;
2235                 }
2236
2237                 rc = kgnilnd_setup_rdma_buffer(tx, niov, iov, kiov, offset, nob);
2238                 if (rc != 0) {
2239                         kgnilnd_tx_done(tx, rc);
2240                         rc = -EIO;
2241                         goto out;
2242                 }
2243
2244                 tx->tx_lntmsg[0] = lntmsg;
2245                 if ((reverse_rdma_flag & GNILND_REVERSE_PUT) == 0)
2246                         tx->tx_msg.gnm_u.putreq.gnprm_hdr = *hdr;
2247                 else
2248                         tx->tx_msg.gnm_u.get.gngm_hdr = *hdr;
2249
2250                 /* rest of tx_msg is setup just before it is sent */
2251                 kgnilnd_launch_tx(tx, net, &target);
2252                 goto out;
2253         }
2254
2255         /* send IMMEDIATE */
2256
2257         LASSERTF(nob <= *kgnilnd_tunables.kgn_max_immediate,
2258                 "lntmsg 0x%p too large %d\n", lntmsg, nob);
2259
2260         tx = kgnilnd_new_tx_msg(GNILND_MSG_IMMEDIATE, ni->ni_nid);
2261         if (tx == NULL) {
2262                 rc = -ENOMEM;
2263                 goto out;
2264         }
2265
2266         rc = kgnilnd_setup_immediate_buffer(tx, niov, iov, kiov, offset, nob);
2267         if (rc != 0) {
2268                 kgnilnd_tx_done(tx, rc);
2269                 goto out;
2270         }
2271
2272         tx->tx_msg.gnm_u.immediate.gnim_hdr = *hdr;
2273         tx->tx_lntmsg[0] = lntmsg;
2274         kgnilnd_launch_tx(tx, net, &target);
2275
2276 out:
2277         /* use stored value as we could have already finalized lntmsg here from a failed launch */
2278         if (msg_vmflush)
2279                 cfs_memory_pressure_restore(mpflag);
2280         return rc;
2281 }
2282
2283 void
2284 kgnilnd_setup_rdma(lnet_ni_t *ni, kgn_rx_t *rx, lnet_msg_t *lntmsg, int mlen)
2285 {
2286         kgn_conn_t    *conn = rx->grx_conn;
2287         kgn_msg_t     *rxmsg = rx->grx_msg;
2288         unsigned int   niov = lntmsg->msg_niov;
2289         struct kvec   *iov = lntmsg->msg_iov;
2290         lnet_kiov_t   *kiov = lntmsg->msg_kiov;
2291         unsigned int   offset = lntmsg->msg_offset;
2292         unsigned int   nob = lntmsg->msg_len;
2293         int            done_type;
2294         kgn_tx_t      *tx;
2295         int            rc = 0;
2296
2297         switch (rxmsg->gnm_type) {
2298         case GNILND_MSG_PUT_REQ_REV:
2299                 done_type = GNILND_MSG_PUT_DONE_REV;
2300                 nob = mlen;
2301                 break;
2302         case GNILND_MSG_GET_REQ:
2303                 done_type = GNILND_MSG_GET_DONE;
2304                 break;
2305         default:
2306                 CERROR("invalid msg type %s (%d)\n",
2307                         kgnilnd_msgtype2str(rxmsg->gnm_type),
2308                         rxmsg->gnm_type);
2309                 LBUG();
2310         }
2311
2312         tx = kgnilnd_new_tx_msg(done_type, ni->ni_nid);
2313         if (tx == NULL)
2314                 goto failed_0;
2315
2316         rc = kgnilnd_set_tx_id(tx, conn);
2317         if (rc != 0)
2318                 goto failed_1;
2319
2320         rc = kgnilnd_setup_rdma_buffer(tx, niov, iov, kiov, offset, nob);
2321         if (rc != 0)
2322                 goto failed_1;
2323
2324         tx->tx_lntmsg[0] = lntmsg;
2325         tx->tx_getinfo = rxmsg->gnm_u.get;
2326
2327         /* we only queue from kgnilnd_recv - we might get called from other contexts
2328          * and we don't want to block the mutex in those cases */
2329
2330         spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2331         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2332         spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2333         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2334
2335         return;
2336
2337  failed_1:
2338         kgnilnd_tx_done(tx, rc);
2339         kgnilnd_nak_rdma(conn, done_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2340  failed_0:
2341         lnet_finalize(ni, lntmsg, rc);
2342 }
2343
2344 int
2345 kgnilnd_eager_recv(lnet_ni_t *ni, void *private, lnet_msg_t *lntmsg,
2346                    void **new_private)
2347 {
2348         kgn_rx_t        *rx = private;
2349         kgn_conn_t      *conn = rx->grx_conn;
2350         kgn_msg_t       *rxmsg = rx->grx_msg;
2351         kgn_msg_t       *eagermsg = NULL;
2352         kgn_peer_t      *peer = NULL;
2353         kgn_conn_t      *found_conn = NULL;
2354
2355         GNIDBG_MSG(D_NET, rxmsg, "eager recv for conn %p, rxmsg %p, lntmsg %p",
2356                 conn, rxmsg, lntmsg);
2357
2358         if (rxmsg->gnm_payload_len > *kgnilnd_tunables.kgn_max_immediate) {
2359                 GNIDBG_MSG(D_ERROR, rxmsg, "payload too large %d",
2360                         rxmsg->gnm_payload_len);
2361                 return -EPROTO;
2362         }
2363         /* Grab a read lock so the connection doesnt disappear on us
2364          * while we look it up
2365          */
2366         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
2367
2368         peer = kgnilnd_find_peer_locked(rxmsg->gnm_srcnid);
2369         if (peer != NULL)
2370                 found_conn = kgnilnd_find_conn_locked(peer);
2371
2372
2373         /* Verify the connection found is the same one that the message
2374          * is supposed to be using, if it is not output an error message
2375          * and return.
2376          */
2377         if (!peer || !found_conn
2378             || found_conn->gnc_peer_connstamp != rxmsg->gnm_connstamp) {
2379                 read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2380                 CERROR("Couldnt find matching peer %p or conn %p / %p\n",
2381                         peer, conn, found_conn);
2382                 if (found_conn) {
2383                         CERROR("Unexpected connstamp %#llx(%#llx expected)"
2384                                 " from %s", rxmsg->gnm_connstamp,
2385                                 found_conn->gnc_peer_connstamp,
2386                                 libcfs_nid2str(peer->gnp_nid));
2387                 }
2388                 return -ENOTCONN;
2389         }
2390
2391         /* add conn ref to ensure it doesn't go away until all eager
2392          * messages processed */
2393         kgnilnd_conn_addref(conn);
2394
2395         /* Now that we have verified the connection is valid and added a
2396          * reference we can remove the read_lock on the peer_conn_lock */
2397         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2398
2399         /* we have no credits or buffers for this message, so copy it
2400          * somewhere for a later kgnilnd_recv */
2401         if (atomic_read(&kgnilnd_data.kgn_neager_allocs) >=
2402                         *kgnilnd_tunables.kgn_eager_credits) {
2403                 CERROR("Out of eager credits to %s\n",
2404                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
2405                 return -ENOMEM;
2406         }
2407
2408         atomic_inc(&kgnilnd_data.kgn_neager_allocs);
2409
2410         LIBCFS_ALLOC(eagermsg, sizeof(*eagermsg) + *kgnilnd_tunables.kgn_max_immediate);
2411         if (eagermsg == NULL) {
2412                 kgnilnd_conn_decref(conn);
2413                 CERROR("couldn't allocate eager rx message for conn %p to %s\n",
2414                         conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
2415                 return -ENOMEM;
2416         }
2417
2418         /* copy msg and payload */
2419         memcpy(eagermsg, rxmsg, sizeof(*rxmsg) + rxmsg->gnm_payload_len);
2420         rx->grx_msg = eagermsg;
2421         rx->grx_eager = 1;
2422
2423         /* stash this for lnet_finalize on cancel-on-conn-close */
2424         rx->grx_lntmsg = lntmsg;
2425
2426         /* keep the same rx_t, it just has a new grx_msg now */
2427         *new_private = private;
2428
2429         /* release SMSG buffer */
2430         kgnilnd_release_msg(conn);
2431
2432         return 0;
2433 }
2434
2435 int
2436 kgnilnd_recv(lnet_ni_t *ni, void *private, lnet_msg_t *lntmsg,
2437              int delayed, unsigned int niov,
2438              struct kvec *iov, lnet_kiov_t *kiov,
2439              unsigned int offset, unsigned int mlen, unsigned int rlen)
2440 {
2441         kgn_rx_t    *rx = private;
2442         kgn_conn_t  *conn = rx->grx_conn;
2443         kgn_msg_t   *rxmsg = rx->grx_msg;
2444         kgn_tx_t    *tx;
2445         int          rc = 0;
2446         __u32        pload_cksum;
2447         ENTRY;
2448
2449         LASSERT(!in_interrupt());
2450         LASSERTF(mlen <= rlen, "%d <= %d\n", mlen, rlen);
2451         /* Either all pages or all vaddrs */
2452         LASSERTF(!(kiov != NULL && iov != NULL), "kiov %p iov %p\n",
2453                 kiov, iov);
2454
2455         GNIDBG_MSG(D_NET, rxmsg, "conn %p, rxmsg %p, lntmsg %p"
2456                 " niov=%d kiov=%p iov=%p offset=%d mlen=%d rlen=%d",
2457                 conn, rxmsg, lntmsg,
2458                 niov, kiov, iov, offset, mlen, rlen);
2459
2460         /* we need to lock here as recv can be called from any context */
2461         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
2462         if (rx->grx_eager && conn->gnc_state != GNILND_CONN_ESTABLISHED) {
2463                 read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2464
2465                 /* someone closed the conn after we copied this out, nuke it */
2466                 kgnilnd_consume_rx(rx);
2467                 lnet_finalize(ni, lntmsg, conn->gnc_error);
2468                 RETURN(0);
2469         }
2470         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2471
2472         switch (rxmsg->gnm_type) {
2473         default:
2474                 GNIDBG_MSG(D_NETERROR, rxmsg, "conn %p, rx %p, rxmsg %p, lntmsg %p"
2475                 " niov=%d kiov=%p iov=%p offset=%d mlen=%d rlen=%d",
2476                 conn, rx, rxmsg, lntmsg, niov, kiov, iov, offset, mlen, rlen);
2477                 LBUG();
2478
2479         case GNILND_MSG_IMMEDIATE:
2480                 if (mlen > rxmsg->gnm_payload_len) {
2481                         GNIDBG_MSG(D_ERROR, rxmsg,
2482                                 "Immediate message from %s too big: %d > %d",
2483                                 libcfs_nid2str(conn->gnc_peer->gnp_nid), mlen,
2484                                 rxmsg->gnm_payload_len);
2485                         rc = -EINVAL;
2486                         kgnilnd_consume_rx(rx);
2487                         RETURN(rc);
2488                 }
2489
2490                 /* rxmsg[1] is a pointer to the payload, sitting in the buffer
2491                  * right after the kgn_msg_t header - so just 'cute' way of saying
2492                  * rxmsg + sizeof(kgn_msg_t) */
2493
2494                 /* check payload checksum if sent */
2495
2496                 if (*kgnilnd_tunables.kgn_checksum >= 2 &&
2497                         !rxmsg->gnm_payload_cksum &&
2498                         rxmsg->gnm_payload_len != 0)
2499                         GNIDBG_MSG(D_WARNING, rxmsg, "no msg payload checksum when enabled");
2500
2501                 if (rxmsg->gnm_payload_cksum != 0) {
2502                         /* gnm_payload_len set in kgnilnd_sendmsg from tx->tx_nob,
2503                          * which is what is used to calculate the cksum on the TX side */
2504                         pload_cksum = kgnilnd_cksum(&rxmsg[1], rxmsg->gnm_payload_len);
2505
2506                         if (rxmsg->gnm_payload_cksum != pload_cksum) {
2507                                 GNIDBG_MSG(D_NETERROR, rxmsg,
2508                                            "Bad payload checksum (%x expected %x)",
2509                                             pload_cksum, rxmsg->gnm_payload_cksum);
2510                                 switch (*kgnilnd_tunables.kgn_checksum_dump) {
2511                                 case 2:
2512                                         kgnilnd_dump_blob(D_BUFFS, "bad payload checksum",
2513                                                           &rxmsg[1], rxmsg->gnm_payload_len);
2514                                         /* fall through to dump */
2515                                 case 1:
2516                                         libcfs_debug_dumplog();
2517                                         break;
2518                                 default:
2519                                         break;
2520                                 }
2521                                 rc = -ENOKEY;
2522                                 /* checksum problems are fatal, kill the conn */
2523                                 kgnilnd_consume_rx(rx);
2524                                 kgnilnd_close_conn(conn, rc);
2525                                 RETURN(rc);
2526                         }
2527                 }
2528
2529                 if (kiov != NULL)
2530                         lnet_copy_flat2kiov(
2531                                 niov, kiov, offset,
2532                                 *kgnilnd_tunables.kgn_max_immediate,
2533                                 &rxmsg[1], 0, mlen);
2534                 else
2535                         lnet_copy_flat2iov(
2536                                 niov, iov, offset,
2537                                 *kgnilnd_tunables.kgn_max_immediate,
2538                                 &rxmsg[1], 0, mlen);
2539
2540                 kgnilnd_consume_rx(rx);
2541                 lnet_finalize(ni, lntmsg, 0);
2542                 RETURN(0);
2543
2544         case GNILND_MSG_PUT_REQ:
2545                 /* LNET wants to truncate or drop transaction, sending NAK */
2546                 if (mlen == 0) {
2547                         kgnilnd_consume_rx(rx);
2548                         lnet_finalize(ni, lntmsg, 0);
2549
2550                         /* only error if lntmsg == NULL, otherwise we are just
2551                          * short circuiting the rdma process of 0 bytes */
2552                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2553                                         lntmsg == NULL ? -ENOENT : 0,
2554                                         rxmsg->gnm_u.get.gngm_cookie,
2555                                         ni->ni_nid);
2556                         RETURN(0);
2557                 }
2558                 /* sending ACK with sink buff. info */
2559                 tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_ACK, ni->ni_nid);
2560                 if (tx == NULL) {
2561                         kgnilnd_consume_rx(rx);
2562                         RETURN(-ENOMEM);
2563                 }
2564
2565                 rc = kgnilnd_set_tx_id(tx, conn);
2566                 if (rc != 0) {
2567                         GOTO(nak_put_req, rc);
2568                 }
2569
2570                 rc = kgnilnd_setup_rdma_buffer(tx, niov, iov, kiov, offset, mlen);
2571                 if (rc != 0) {
2572                         GOTO(nak_put_req, rc);
2573                 }
2574
2575                 tx->tx_msg.gnm_u.putack.gnpam_src_cookie =
2576                         rxmsg->gnm_u.putreq.gnprm_cookie;
2577                 tx->tx_msg.gnm_u.putack.gnpam_dst_cookie = tx->tx_id.txe_cookie;
2578                 tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_addr =
2579                         (__u64)((unsigned long)tx->tx_buffer);
2580                 tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_nob = mlen;
2581
2582                 tx->tx_lntmsg[0] = lntmsg; /* finalize this on RDMA_DONE */
2583                 tx->tx_qtime = jiffies;
2584                 /* we only queue from kgnilnd_recv - we might get called from other contexts
2585                  * and we don't want to block the mutex in those cases */
2586
2587                 spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2588                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2589                 spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2590                 kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2591
2592                 kgnilnd_consume_rx(rx);
2593                 RETURN(0);
2594
2595 nak_put_req:
2596                 /* make sure we send an error back when the PUT fails */
2597                 kgnilnd_nak_rdma(conn, rxmsg->gnm_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2598                 kgnilnd_tx_done(tx, rc);
2599                 kgnilnd_consume_rx(rx);
2600
2601                 /* return magic LNet network error */
2602                 RETURN(-EIO);
2603         case GNILND_MSG_GET_REQ_REV:
2604                 /* LNET wants to truncate or drop transaction, sending NAK */
2605                 if (mlen == 0) {
2606                         kgnilnd_consume_rx(rx);
2607                         lnet_finalize(ni, lntmsg, 0);
2608
2609                         /* only error if lntmsg == NULL, otherwise we are just
2610                          * short circuiting the rdma process of 0 bytes */
2611                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2612                                         lntmsg == NULL ? -ENOENT : 0,
2613                                         rxmsg->gnm_u.get.gngm_cookie,
2614                                         ni->ni_nid);
2615                         RETURN(0);
2616                 }
2617                 /* lntmsg can be null when parsing a LNET_GET */
2618                 if (lntmsg != NULL) {
2619                         /* sending ACK with sink buff. info */
2620                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_ACK_REV, ni->ni_nid);
2621                         if (tx == NULL) {
2622                                 kgnilnd_consume_rx(rx);
2623                                 RETURN(-ENOMEM);
2624                         }
2625
2626                         rc = kgnilnd_set_tx_id(tx, conn);
2627                         if (rc != 0)
2628                                 GOTO(nak_get_req_rev, rc);
2629
2630
2631                         rc = kgnilnd_setup_rdma_buffer(tx, niov, iov, kiov, offset, mlen);
2632                         if (rc != 0)
2633                                 GOTO(nak_get_req_rev, rc);
2634
2635
2636                         tx->tx_msg.gnm_u.putack.gnpam_src_cookie =
2637                                 rxmsg->gnm_u.putreq.gnprm_cookie;
2638                         tx->tx_msg.gnm_u.putack.gnpam_dst_cookie = tx->tx_id.txe_cookie;
2639                         tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_addr =
2640                                 (__u64)((unsigned long)tx->tx_buffer);
2641                         tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_nob = mlen;
2642
2643                         tx->tx_lntmsg[0] = lntmsg; /* finalize this on RDMA_DONE */
2644
2645                         /* we only queue from kgnilnd_recv - we might get called from other contexts
2646                          * and we don't want to block the mutex in those cases */
2647
2648                         spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2649                         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2650                         spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2651                         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2652                 } else {
2653                         /* No match */
2654                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2655                                         -ENOENT,
2656                                         rxmsg->gnm_u.get.gngm_cookie,
2657                                         ni->ni_nid);
2658                 }
2659
2660                 kgnilnd_consume_rx(rx);
2661                 RETURN(0);
2662
2663 nak_get_req_rev:
2664                 /* make sure we send an error back when the GET fails */
2665                 kgnilnd_nak_rdma(conn, rxmsg->gnm_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2666                 kgnilnd_tx_done(tx, rc);
2667                 kgnilnd_consume_rx(rx);
2668
2669                 /* return magic LNet network error */
2670                 RETURN(-EIO);
2671
2672
2673         case GNILND_MSG_PUT_REQ_REV:
2674                 /* LNET wants to truncate or drop transaction, sending NAK */
2675                 if (mlen == 0) {
2676                         kgnilnd_consume_rx(rx);
2677                         lnet_finalize(ni, lntmsg, 0);
2678
2679                         /* only error if lntmsg == NULL, otherwise we are just
2680                          * short circuiting the rdma process of 0 bytes */
2681                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2682                                         lntmsg == NULL ? -ENOENT : 0,
2683                                         rxmsg->gnm_u.get.gngm_cookie,
2684                                         ni->ni_nid);
2685                         RETURN(0);
2686                 }
2687
2688                 if (lntmsg != NULL) {
2689                         /* Matched! */
2690                         kgnilnd_setup_rdma(ni, rx, lntmsg, mlen);
2691                 } else {
2692                         /* No match */
2693                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2694                                         -ENOENT,
2695                                         rxmsg->gnm_u.get.gngm_cookie,
2696                                         ni->ni_nid);
2697                 }
2698                 kgnilnd_consume_rx(rx);
2699                 RETURN(0);
2700         case GNILND_MSG_GET_REQ:
2701                 if (lntmsg != NULL) {
2702                         /* Matched! */
2703                         kgnilnd_setup_rdma(ni, rx, lntmsg, mlen);
2704                 } else {
2705                         /* No match */
2706                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2707                                         -ENOENT,
2708                                         rxmsg->gnm_u.get.gngm_cookie,
2709                                         ni->ni_nid);
2710                 }
2711                 kgnilnd_consume_rx(rx);
2712                 RETURN(0);
2713         }
2714         RETURN(0);
2715 }
2716
2717 /* needs write_lock on kgn_peer_conn_lock held */
2718 int
2719 kgnilnd_check_conn_timeouts_locked(kgn_conn_t *conn)
2720 {
2721         unsigned long      timeout, keepalive;
2722         unsigned long      now = jiffies;
2723         unsigned long      newest_last_rx;
2724         kgn_tx_t          *tx;
2725
2726         /* given that we found this conn hanging off a peer, it better damned
2727          * well be connected */
2728         LASSERTF(conn->gnc_state == GNILND_CONN_ESTABLISHED,
2729                  "conn 0x%p->%s with bad state%s\n", conn,
2730                  conn->gnc_peer ? libcfs_nid2str(conn->gnc_peer->gnp_nid)
2731                                : "<?>",
2732                  kgnilnd_conn_state2str(conn));
2733
2734         CDEBUG(D_NET, "checking conn %p->%s timeout %d keepalive %d "
2735                       "rx_diff %lu tx_diff %lu\n",
2736                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
2737                 conn->gnc_timeout, GNILND_TO2KA(conn->gnc_timeout),
2738                 cfs_duration_sec(now - conn->gnc_last_rx_cq),
2739                 cfs_duration_sec(now - conn->gnc_last_tx));
2740
2741         timeout = cfs_time_seconds(conn->gnc_timeout);
2742         keepalive = cfs_time_seconds(GNILND_TO2KA(conn->gnc_timeout));
2743
2744         /* just in case our lack of RX msg processing is gumming up the works - give the
2745          * remove an extra chance */
2746
2747         newest_last_rx = GNILND_LASTRX(conn);
2748
2749         if (time_after_eq(now, newest_last_rx + timeout)) {
2750                 uint32_t level = D_CONSOLE|D_NETERROR;
2751
2752                 if (conn->gnc_peer->gnp_state == GNILND_PEER_DOWN) {
2753                         level = D_NET;
2754                 }
2755                         GNIDBG_CONN(level, conn,
2756                         "No gnilnd traffic received from %s for %lu "
2757                         "seconds, terminating connection. Is node down? ",
2758                         libcfs_nid2str(conn->gnc_peer->gnp_nid),
2759                         cfs_duration_sec(now - newest_last_rx));
2760                 return -ETIMEDOUT;
2761         }
2762
2763         /* we don't timeout on last_tx stalls - we are going to trust the
2764          * underlying network to let us know when sends are failing.
2765          * At worst, the peer will timeout our RX stamp and drop the connection
2766          * at that point. We'll then see his CLOSE or at worst his RX
2767          * stamp stop and drop the connection on our end */
2768
2769         if (time_after_eq(now, conn->gnc_last_tx + keepalive)) {
2770                 CDEBUG(D_NET, "sending NOOP -> %s (%p idle %lu(%lu)) "
2771                        "last %lu/%lu/%lu %lus/%lus/%lus\n",
2772                        libcfs_nid2str(conn->gnc_peer->gnp_nid), conn,
2773                        cfs_duration_sec(jiffies - conn->gnc_last_tx),
2774                        keepalive,
2775                        conn->gnc_last_noop_want, conn->gnc_last_noop_sent,
2776                        conn->gnc_last_noop_cq,
2777                        cfs_duration_sec(jiffies - conn->gnc_last_noop_want),
2778                        cfs_duration_sec(jiffies - conn->gnc_last_noop_sent),
2779                        cfs_duration_sec(jiffies - conn->gnc_last_noop_cq));
2780                 set_mb(conn->gnc_last_noop_want, jiffies);
2781                 atomic_inc(&conn->gnc_reaper_noop);
2782                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_NOOP_SEND))
2783                         return 0;
2784
2785                 tx = kgnilnd_new_tx_msg(GNILND_MSG_NOOP, conn->gnc_peer->gnp_net->gnn_ni->ni_nid);
2786                 if (tx == NULL)
2787                         return 0;
2788                 kgnilnd_queue_tx(conn, tx);
2789         }
2790
2791         return 0;
2792 }
2793
2794 /* needs write_lock on kgn_peer_conn_lock held */
2795 void
2796 kgnilnd_check_peer_timeouts_locked(kgn_peer_t *peer, struct list_head *todie,
2797                                     struct list_head *souls)
2798 {
2799         unsigned long           timeout;
2800         kgn_conn_t             *conn, *connN = NULL;
2801         kgn_tx_t               *tx, *txN;
2802         int                     rc = 0;
2803         int                     count = 0;
2804         int                     reconnect;
2805         int                     to_reconn;
2806         short                   releaseconn = 0;
2807         unsigned long           first_rx = 0;
2808         int                     purgatory_conn_cnt = 0;
2809
2810         CDEBUG(D_NET, "checking peer 0x%p->%s for timeouts; interval %lus\n",
2811                 peer, libcfs_nid2str(peer->gnp_nid),
2812                 peer->gnp_reconnect_interval);
2813
2814         timeout = cfs_time_seconds(MAX(*kgnilnd_tunables.kgn_timeout,
2815                                        GNILND_MIN_TIMEOUT));
2816
2817         conn = kgnilnd_find_conn_locked(peer);
2818         if (conn) {
2819                 /* if there is a valid conn, check the queues for timeouts */
2820                 rc = kgnilnd_check_conn_timeouts_locked(conn);
2821                 if (rc) {
2822                         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_RX_CLOSE_CLOSING)) {
2823                                 /* simulate a RX CLOSE after the timeout but before
2824                                  * the scheduler thread gets it */
2825                                 conn->gnc_close_recvd = GNILND_CLOSE_INJECT1;
2826                                 conn->gnc_peer_error = -ETIMEDOUT;
2827                         }
2828
2829                         if (*kgnilnd_tunables.kgn_to_reconn_disable &&
2830                             rc == -ETIMEDOUT) {
2831                                 peer->gnp_state = GNILND_PEER_TIMED_OUT;
2832                                 CDEBUG(D_WARNING, "%s conn timed out, will "
2833                                        "reconnect upon request from peer\n",
2834                                        libcfs_nid2str(conn->gnc_peer->gnp_nid));
2835                         }
2836                         /* Once we mark closed, any of the scheduler threads could
2837                          * get it and move through before we hit the fail loc code */
2838                         kgnilnd_close_conn_locked(conn, rc);
2839                 } else {
2840                         /* first_rx is used to decide when to release a conn from purgatory.
2841                          */
2842                         first_rx = conn->gnc_first_rx;
2843                 }
2844         }
2845
2846         /* now regardless of starting new conn, find tx on peer queue that
2847          * are old and smell bad - do this first so we don't trigger
2848          * reconnect on empty queue if we timeout all */
2849         list_for_each_entry_safe(tx, txN, &peer->gnp_tx_queue, tx_list) {
2850                 if (time_after_eq(jiffies, tx->tx_qtime + timeout)) {
2851                         if (count == 0) {
2852                                 LCONSOLE_INFO("could not send to %s due to connection"
2853                                        " setup failure after %lu seconds\n",
2854                                        libcfs_nid2str(peer->gnp_nid),
2855                                        cfs_duration_sec(jiffies - tx->tx_qtime));
2856                         }
2857                         kgnilnd_tx_del_state_locked(tx, peer, NULL,
2858                                                    GNILND_TX_ALLOCD);
2859                         list_add_tail(&tx->tx_list, todie);
2860                         count++;
2861                 }
2862         }
2863
2864         if (count || peer->gnp_connecting == GNILND_PEER_KILL) {
2865                 CDEBUG(D_NET, "canceling %d tx for peer 0x%p->%s\n",
2866                         count, peer, libcfs_nid2str(peer->gnp_nid));
2867                 /* if we nuked all the TX, stop peer connection attempt (if there is one..) */
2868                 if (list_empty(&peer->gnp_tx_queue) ||
2869                         peer->gnp_connecting == GNILND_PEER_KILL) {
2870                         /* we pass down todie to use a common function - but we know there are
2871                          * no TX to add */
2872                         kgnilnd_cancel_peer_connect_locked(peer, todie);
2873                 }
2874         }
2875
2876         /* Don't reconnect if we are still trying to clear out old conns.
2877          * This prevents us sending traffic on the new mbox before ensuring we are done
2878          * with the old one */
2879         reconnect = (peer->gnp_state == GNILND_PEER_UP) &&
2880                     (atomic_read(&peer->gnp_dirty_eps) == 0);
2881
2882         /* fast reconnect after a timeout */
2883         to_reconn = !conn &&
2884                     (peer->gnp_last_errno == -ETIMEDOUT) &&
2885                     *kgnilnd_tunables.kgn_fast_reconn;
2886
2887         /* if we are not connected and there are tx on the gnp_tx_queue waiting
2888          * to be sent, we'll check the reconnect interval and fire up a new
2889          * connection request */
2890
2891         if (reconnect &&
2892             (peer->gnp_connecting == GNILND_PEER_IDLE) &&
2893             (time_after_eq(jiffies, peer->gnp_reconnect_time)) &&
2894             (!list_empty(&peer->gnp_tx_queue) || to_reconn)) {
2895
2896                 CDEBUG(D_NET, "starting connect to %s\n",
2897                         libcfs_nid2str(peer->gnp_nid));
2898                 LASSERTF(peer->gnp_connecting == GNILND_PEER_IDLE, "Peer was idle and we"
2899                         "have a write_lock, state issue %d\n", peer->gnp_connecting);
2900
2901                 peer->gnp_connecting = GNILND_PEER_CONNECT;
2902                 kgnilnd_peer_addref(peer); /* extra ref for connd */
2903
2904                 spin_lock(&peer->gnp_net->gnn_dev->gnd_connd_lock);
2905                 list_add_tail(&peer->gnp_connd_list,
2906                               &peer->gnp_net->gnn_dev->gnd_connd_peers);
2907                 spin_unlock(&peer->gnp_net->gnn_dev->gnd_connd_lock);
2908
2909                 kgnilnd_schedule_dgram(peer->gnp_net->gnn_dev);
2910         }
2911
2912         /* fail_loc to allow us to delay release of purgatory */
2913         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_PURG_REL_DELAY))
2914                 return;
2915
2916         /* This check allows us to verify that the new conn is actually being used. This allows us to
2917          * pull the old conns out of purgatory if they have actually seen traffic.
2918          * We only release a conn from purgatory during stack reset, admin command, or when a peer reconnects
2919          */
2920         if (first_rx &&
2921                 time_after(jiffies, first_rx + cfs_time_seconds(*kgnilnd_tunables.kgn_hardware_timeout))) {
2922                 CDEBUG(D_INFO, "We can release peer %s conn's from purgatory %lu\n",
2923                         libcfs_nid2str(peer->gnp_nid), first_rx + cfs_time_seconds(*kgnilnd_tunables.kgn_hardware_timeout));
2924                 releaseconn = 1;
2925         }
2926
2927         list_for_each_entry_safe (conn, connN, &peer->gnp_conns, gnc_list) {
2928         /* check for purgatory timeouts */
2929                 if (conn->gnc_in_purgatory) {
2930                         /* We cannot detach this conn from purgatory if it has not been closed so we reschedule it
2931                          * that way the next time we check it we can detach it from purgatory
2932                          */
2933
2934                         if (conn->gnc_state != GNILND_CONN_DONE) {
2935                                 /* Skip over conns that are currently not DONE. If they arent already scheduled
2936                                  * for completion something in the state machine is broken.
2937                                  */
2938                                 continue;
2939                         }
2940
2941                         /* We only detach a conn that is in purgatory if we have received a close message,
2942                          * we have a new valid connection that has successfully received data, or an admin
2943                          * command tells us we need to detach.
2944                          */
2945
2946                         if (conn->gnc_close_recvd || releaseconn || conn->gnc_needs_detach) {
2947                                 unsigned long   waiting;
2948
2949                                 waiting = (long) jiffies - conn->gnc_last_rx_cq;
2950
2951                                 /* C.E: The remote peer is expected to close the
2952                                  * connection (see kgnilnd_check_conn_timeouts)
2953                                  * via the reaper thread and nuke out the MDD and
2954                                  * FMA resources after conn->gnc_timeout has expired
2955                                  * without an FMA RX */
2956                                 CDEBUG(D_NET, "Reconnected to %s in %lds or admin forced detach, dropping "
2957                                         " held resources\n",
2958                                         libcfs_nid2str(conn->gnc_peer->gnp_nid),
2959                                         cfs_duration_sec(waiting));
2960
2961                                 kgnilnd_detach_purgatory_locked(conn, souls);
2962                         } else {
2963                                 purgatory_conn_cnt++;
2964                         }
2965                 }
2966         }
2967
2968         /* If we have too many connections in purgatory we could run out of
2969          * resources. Limit the number of connections to a tunable number,
2970          * clean up to the minimum all in one fell swoop... there are
2971          * situations where dvs will retry tx's and we can eat up several
2972          * hundread connection requests at once.
2973          */
2974         if (purgatory_conn_cnt > *kgnilnd_tunables.kgn_max_purgatory) {
2975                 list_for_each_entry_safe(conn, connN, &peer->gnp_conns,
2976                                          gnc_list) {
2977                         if (conn->gnc_in_purgatory &&
2978                             conn->gnc_state == GNILND_CONN_DONE) {
2979                                 CDEBUG(D_NET, "Dropping Held resource due to"
2980                                               " resource limits being hit\n");
2981                                 kgnilnd_detach_purgatory_locked(conn, souls);
2982
2983                                 if (purgatory_conn_cnt-- <
2984                                     *kgnilnd_tunables.kgn_max_purgatory)
2985                                         break;
2986                         }
2987                 }
2988         }
2989
2990         return;
2991 }
2992
2993 void
2994 kgnilnd_reaper_check(int idx)
2995 {
2996         struct list_head  *peers = &kgnilnd_data.kgn_peers[idx];
2997         struct list_head  *ctmp, *ctmpN;
2998         struct list_head   geriatrics;
2999         struct list_head   souls;
3000
3001         INIT_LIST_HEAD(&geriatrics);
3002         INIT_LIST_HEAD(&souls);
3003
3004         write_lock(&kgnilnd_data.kgn_peer_conn_lock);
3005
3006         list_for_each_safe(ctmp, ctmpN, peers) {
3007                 kgn_peer_t        *peer = NULL;
3008
3009                 /* don't timeout stuff if the network is mucked or shutting down */
3010                 if (kgnilnd_check_hw_quiesce()) {
3011                         break;
3012                 }
3013                 peer = list_entry(ctmp, kgn_peer_t, gnp_list);
3014
3015                 kgnilnd_check_peer_timeouts_locked(peer, &geriatrics, &souls);
3016         }
3017
3018         write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
3019
3020         kgnilnd_txlist_done(&geriatrics, -EHOSTUNREACH);
3021         kgnilnd_release_purgatory_list(&souls);
3022 }
3023
3024 void
3025 kgnilnd_update_reaper_timeout(long timeout)
3026 {
3027         LASSERT(timeout > 0);
3028
3029         spin_lock(&kgnilnd_data.kgn_reaper_lock);
3030
3031         if (timeout < kgnilnd_data.kgn_new_min_timeout)
3032                 kgnilnd_data.kgn_new_min_timeout = timeout;
3033
3034         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
3035 }
3036
3037 static void
3038 kgnilnd_reaper_poke_with_stick(unsigned long arg)
3039 {
3040         wake_up(&kgnilnd_data.kgn_reaper_waitq);
3041 }
3042
3043 int
3044 kgnilnd_reaper(void *arg)
3045 {
3046         long               timeout;
3047         int                i;
3048         int                hash_index = 0;
3049         unsigned long      next_check_time = jiffies;
3050         long               current_min_timeout = MAX_SCHEDULE_TIMEOUT;
3051         struct timer_list  timer;
3052         DEFINE_WAIT(wait);
3053
3054         cfs_block_allsigs();
3055
3056         /* all gnilnd threads need to run fairly urgently */
3057         set_user_nice(current, *kgnilnd_tunables.kgn_nice);
3058         spin_lock(&kgnilnd_data.kgn_reaper_lock);
3059
3060         while (!kgnilnd_data.kgn_shutdown) {
3061                 /* I wake up every 'p' seconds to check for timeouts on some
3062                  * more peers.  I try to check every connection 'n' times
3063                  * within the global minimum of all keepalive and timeout
3064                  * intervals, to ensure I attend to every connection within
3065                  * (n+1)/n times its timeout intervals. */
3066                 const int     p = GNILND_REAPER_THREAD_WAKE;
3067                 const int     n = GNILND_REAPER_NCHECKS;
3068                 int           chunk;
3069                 /* to quiesce or to not quiesce, that is the question */
3070                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
3071                         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
3072                         KGNILND_SPIN_QUIESCE;
3073                         spin_lock(&kgnilnd_data.kgn_reaper_lock);
3074                 }
3075
3076                 /* careful with the jiffy wrap... */
3077                 timeout = (long)(next_check_time - jiffies);
3078
3079                 if (timeout > 0) {
3080                         prepare_to_wait(&kgnilnd_data.kgn_reaper_waitq, &wait,
3081                                         TASK_INTERRUPTIBLE);
3082                         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
3083                         setup_timer(&timer, kgnilnd_reaper_poke_with_stick,
3084                                     next_check_time);
3085                         mod_timer(&timer, (long) jiffies + timeout);
3086
3087                         /* check flag variables before committing */
3088                         if (!kgnilnd_data.kgn_shutdown &&
3089                             !kgnilnd_data.kgn_quiesce_trigger) {
3090                                 CDEBUG(D_INFO, "schedule timeout %ld (%lu sec)\n",
3091                                        timeout, cfs_duration_sec(timeout));
3092                                 schedule();
3093                                 CDEBUG(D_INFO, "awake after schedule\n");
3094                         }
3095
3096                         del_singleshot_timer_sync(&timer);
3097                         spin_lock(&kgnilnd_data.kgn_reaper_lock);
3098                         finish_wait(&kgnilnd_data.kgn_reaper_waitq, &wait);
3099                         continue;
3100                 }
3101
3102                 /* new_min_timeout is set from the conn timeouts and keepalive
3103                  * this should end up with a min timeout of
3104                  * GNILND_TIMEOUT2KEEPALIVE(t) or roughly LND_TIMEOUT/2 */
3105                 if (kgnilnd_data.kgn_new_min_timeout < current_min_timeout) {
3106                         current_min_timeout = kgnilnd_data.kgn_new_min_timeout;
3107                         CDEBUG(D_NET, "Set new min timeout %ld\n",
3108                                current_min_timeout);
3109                 }
3110
3111                 spin_unlock(&kgnilnd_data.kgn_reaper_lock);
3112
3113                 /* Compute how many table entries to check now so I get round
3114                  * the whole table fast enough given that I do this at fixed
3115                  * intervals of 'p' seconds) */
3116                 chunk = *kgnilnd_tunables.kgn_peer_hash_size;
3117                 if (kgnilnd_data.kgn_new_min_timeout > n * p)
3118                         chunk = (chunk * n * p) /
3119                                 kgnilnd_data.kgn_new_min_timeout;
3120                 if (chunk == 0)
3121                         chunk = 1;
3122                 for (i = 0; i < chunk; i++) {
3123                         kgnilnd_reaper_check(hash_index);
3124                         hash_index = (hash_index + 1) %
3125                                 *kgnilnd_tunables.kgn_peer_hash_size;
3126                 }
3127                 next_check_time = (long) jiffies + cfs_time_seconds(p);
3128                 CDEBUG(D_INFO, "next check at %lu or in %d sec\n", next_check_time, p);
3129
3130                 spin_lock(&kgnilnd_data.kgn_reaper_lock);
3131         }
3132
3133         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
3134
3135         kgnilnd_thread_fini();
3136         return 0;
3137 }
3138
3139 int
3140 kgnilnd_recv_bte_get(kgn_tx_t *tx) {
3141         unsigned niov, offset, nob;
3142         lnet_kiov_t     *kiov;
3143         lnet_msg_t *lntmsg = tx->tx_lntmsg[0];
3144         kgnilnd_parse_lnet_rdma(lntmsg, &niov, &offset, &nob, &kiov, tx->tx_nob_rdma);
3145
3146         if (kiov != NULL) {
3147                 lnet_copy_flat2kiov(
3148                         niov, kiov, offset,
3149                         nob,
3150                         tx->tx_buffer_copy + tx->tx_offset, 0, nob);
3151         } else {
3152                 memcpy(tx->tx_buffer, tx->tx_buffer_copy + tx->tx_offset, nob);
3153         }
3154         return 0;
3155 }
3156
3157
3158 int
3159 kgnilnd_check_rdma_cq(kgn_device_t *dev)
3160 {
3161         gni_return_t           rrc;
3162         gni_post_descriptor_t *desc;
3163         __u64                  event_data;
3164         kgn_tx_ev_id_t         ev_id;
3165         char                   err_str[256];
3166         int                    should_retry, rc;
3167         long                   num_processed = 0;
3168         kgn_conn_t            *conn = NULL;
3169         kgn_tx_t              *tx = NULL;
3170         kgn_rdma_desc_t       *rdesc;
3171         unsigned int           rnob;
3172         __u64                  rcookie;
3173
3174         for (;;) {
3175                 /* make sure we don't keep looping if we need to reset */
3176                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
3177                         return num_processed;
3178                 }
3179                 rc = kgnilnd_mutex_trylock(&dev->gnd_cq_mutex);
3180                 if (!rc) {
3181                         /* we didn't get the mutex, so return that there is still work
3182                          * to be done */
3183                         return 1;
3184                 }
3185                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_DELAY_RDMA)) {
3186                         /* a bit gross - but we need a good way to test for
3187                          * delayed RDMA completions and the easiest way to do
3188                          * that is to delay the RDMA CQ events */
3189                         rrc = GNI_RC_NOT_DONE;
3190                 } else {
3191                         rrc = kgnilnd_cq_get_event(dev->gnd_snd_rdma_cqh, &event_data);
3192                 }
3193
3194                 if (rrc == GNI_RC_NOT_DONE) {
3195                         kgnilnd_gl_mutex_unlock(&dev->gnd_cq_mutex);
3196                         CDEBUG(D_INFO, "SEND RDMA CQ %d empty processed %ld\n",
3197                                dev->gnd_id, num_processed);
3198                         return num_processed;
3199                 }
3200                 dev->gnd_sched_alive = jiffies;
3201                 num_processed++;
3202
3203                 LASSERTF(!GNI_CQ_OVERRUN(event_data),
3204                         "this is bad, somehow our credits didn't protect us"
3205                         " from CQ overrun\n");
3206                 LASSERTF(GNI_CQ_GET_TYPE(event_data) == GNI_CQ_EVENT_TYPE_POST,
3207                         "rrc %d, GNI_CQ_GET_TYPE(%#llx) = %#llx\n", rrc,
3208                         event_data, GNI_CQ_GET_TYPE(event_data));
3209
3210                 rrc = kgnilnd_get_completed(dev->gnd_snd_rdma_cqh, event_data,
3211                                             &desc);
3212                 kgnilnd_gl_mutex_unlock(&dev->gnd_cq_mutex);
3213
3214                 /* XXX Nic: Need better error handling here... */
3215                 LASSERTF((rrc == GNI_RC_SUCCESS) ||
3216                           (rrc == GNI_RC_TRANSACTION_ERROR),
3217                          "rrc %d\n", rrc);
3218
3219                 ev_id.txe_cookie = desc->post_id;
3220
3221                 kgnilnd_validate_tx_ev_id(&ev_id, &tx, &conn);
3222
3223                 if (conn == NULL || tx == NULL) {
3224                         /* either conn or tx was already nuked and this is a "late"
3225                          * completion, so drop it */
3226                         continue;
3227                 }
3228
3229                 GNITX_ASSERTF(tx, tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE ||
3230                         tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE ||
3231                         tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE_REV ||
3232                         tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV,
3233                         "tx %p with type %d\n", tx, tx->tx_msg.gnm_type);
3234
3235                 GNIDBG_TX(D_NET, tx, "RDMA completion for %d bytes", tx->tx_nob);
3236
3237                 if (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV) {
3238                         lnet_set_reply_msg_len(NULL, tx->tx_lntmsg[1],
3239                                                tx->tx_msg.gnm_u.completion.gncm_retval);
3240                 }
3241
3242                 rc = 0;
3243                 if (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV && desc->status == GNI_RC_SUCCESS) {
3244                         if (tx->tx_buffer_copy != NULL)
3245                                 kgnilnd_recv_bte_get(tx);
3246                         rc = kgnilnd_verify_rdma_cksum(tx, tx->tx_putinfo.gnpam_payload_cksum, tx->tx_nob_rdma);
3247                 }
3248
3249                 if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE_REV && desc->status == GNI_RC_SUCCESS) {
3250                         if (tx->tx_buffer_copy != NULL)
3251                                 kgnilnd_recv_bte_get(tx);
3252                         rc = kgnilnd_verify_rdma_cksum(tx, tx->tx_getinfo.gngm_payload_cksum, tx->tx_nob_rdma);
3253                 }
3254
3255                 /* remove from rdmaq */
3256                 kgnilnd_conn_mutex_lock(&conn->gnc_rdma_mutex);
3257                 spin_lock(&conn->gnc_list_lock);
3258                 kgnilnd_tx_del_state_locked(tx, NULL, conn, GNILND_TX_ALLOCD);
3259                 spin_unlock(&conn->gnc_list_lock);
3260                 kgnilnd_conn_mutex_unlock(&conn->gnc_rdma_mutex);
3261
3262                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_RDMA_CQ_ERROR)) {
3263                         event_data = 1LL << 48;
3264                         rc = 1;
3265                 }
3266
3267                 if (likely(desc->status == GNI_RC_SUCCESS) && rc == 0) {
3268                         atomic_inc(&dev->gnd_rdma_ntx);
3269                         atomic64_add(tx->tx_nob, &dev->gnd_rdma_txbytes);
3270                         /* transaction succeeded, add into fmaq */
3271                         kgnilnd_queue_tx(conn, tx);
3272                         kgnilnd_peer_alive(conn->gnc_peer);
3273
3274                         /* drop ref from kgnilnd_validate_tx_ev_id */
3275                         kgnilnd_admin_decref(conn->gnc_tx_in_use);
3276                         kgnilnd_conn_decref(conn);
3277
3278                         continue;
3279                 }
3280
3281                 /* fall through to the TRANSACTION_ERROR case */
3282                 tx->tx_retrans++;
3283
3284                 /* get stringified version for log messages */
3285                 kgnilnd_cq_error_str(event_data, &err_str, 256);
3286                 kgnilnd_cq_error_recoverable(event_data, &should_retry);
3287
3288                 /* make sure we are not off in the weeds with this tx */
3289                 if (tx->tx_retrans >
3290                         *kgnilnd_tunables.kgn_max_retransmits) {
3291                         GNIDBG_TX(D_NETERROR, tx,
3292                                "giving up on TX, too many retries", NULL);
3293                         should_retry = 0;
3294                 }
3295
3296                 GNIDBG_TX(D_NETERROR, tx, "RDMA %s error (%s)",
3297                         should_retry ? "transient" : "unrecoverable", err_str);
3298
3299                 if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE ||
3300                     tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV) {
3301                         rdesc    = &tx->tx_putinfo.gnpam_desc;
3302                         rnob     = tx->tx_putinfo.gnpam_desc.gnrd_nob;
3303                         rcookie  = tx->tx_putinfo.gnpam_dst_cookie;
3304                 } else {
3305                         rdesc    = &tx->tx_getinfo.gngm_desc;
3306                         rnob     = tx->tx_lntmsg[0]->msg_len;
3307                         rcookie  = tx->tx_getinfo.gngm_cookie;
3308                 }
3309
3310                 if (should_retry) {
3311                         kgnilnd_rdma(tx,
3312                                      tx->tx_msg.gnm_type,
3313                                      rdesc,
3314                                      rnob, rcookie);
3315                 } else {
3316                         kgnilnd_nak_rdma(conn,
3317                                          tx->tx_msg.gnm_type,
3318                                          -EFAULT,
3319                                          rcookie,
3320                                          tx->tx_msg.gnm_srcnid);
3321                         kgnilnd_tx_done(tx, -GNILND_NOPURG);
3322                         kgnilnd_close_conn(conn, -ECOMM);
3323                 }
3324
3325                 /* drop ref from kgnilnd_validate_tx_ev_id */
3326                 kgnilnd_admin_decref(conn->gnc_tx_in_use);
3327                 kgnilnd_conn_decref(conn);
3328         }
3329 }
3330
3331 int
3332 kgnilnd_check_fma_send_cq(kgn_device_t *dev)
3333 {
3334         gni_return_t           rrc;
3335         __u64                  event_data;
3336         kgn_tx_ev_id_t         ev_id;
3337         kgn_tx_t              *tx = NULL;
3338         kgn_conn_t            *conn = NULL;
3339         int                    queued_fma, saw_reply, rc;
3340         long                   num_processed = 0;
3341
3342         for (;;) {
3343                 /* make sure we don't keep looping if we need to reset */
3344                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
3345                         return num_processed;
3346                 }
3347
3348                 rc = kgnilnd_mutex_trylock(&dev->gnd_cq_mutex);
3349                 if (!rc) {
3350                         /* we didn't get the mutex, so return that there is still work
3351                          * to be done */
3352                         return 1;
3353                 }
3354
3355                 rrc = kgnilnd_cq_get_event(dev->gnd_snd_fma_cqh, &event_data);
3356                 kgnilnd_gl_mutex_unlock(&dev->gnd_cq_mutex);
3357
3358                 if (rrc == GNI_RC_NOT_DONE) {
3359                         CDEBUG(D_INFO,
3360                                "SMSG send CQ %d not ready (data %#llx) "
3361                                "processed %ld\n", dev->gnd_id, event_data,
3362                                num_processed);
3363                         return num_processed;
3364                 }
3365
3366                 dev->gnd_sched_alive = jiffies;
3367                 num_processed++;
3368
3369                 LASSERTF(!GNI_CQ_OVERRUN(event_data),
3370                         "this is bad, somehow our credits didn't "
3371                         "protect us from CQ overrun\n");
3372                 LASSERTF(GNI_CQ_GET_TYPE(event_data) == GNI_CQ_EVENT_TYPE_SMSG,
3373                         "rrc %d, GNI_CQ_GET_TYPE(%#llx) = %#llx\n", rrc,
3374                         event_data, GNI_CQ_GET_TYPE(event_data));
3375
3376                 /* if SMSG couldn't handle an error, time for conn to die */
3377                 if (unlikely(rrc == GNI_RC_TRANSACTION_ERROR)) {
3378                         char            err_str[256];
3379
3380                         /* need to take the write_lock to ensure atomicity
3381                          * on the conn state if we need to close it */
3382                         write_lock(&kgnilnd_data.kgn_peer_conn_lock);
3383                         conn = kgnilnd_cqid2conn_locked(GNI_CQ_GET_INST_ID(event_data));
3384                         if (conn == NULL) {
3385                                 /* Conn was destroyed? */
3386                                 CDEBUG(D_NET,
3387                                         "SMSG CQID lookup %#llx failed\n",
3388                                         GNI_CQ_GET_INST_ID(event_data));
3389                                 write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
3390                                 continue;
3391                         }
3392
3393                         kgnilnd_cq_error_str(event_data, &err_str, 256);
3394                         CNETERR("SMSG send error to %s: rc %d (%s)\n",
3395                                libcfs_nid2str(conn->gnc_peer->gnp_nid),
3396                                rrc, err_str);
3397                         kgnilnd_close_conn_locked(conn, -ECOMM);
3398
3399                         write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
3400
3401                         /* no need to process rest of this tx -
3402                          * it is getting canceled */
3403                         continue;
3404                 }
3405
3406                 /* fall through to GNI_RC_SUCCESS case */
3407                 ev_id.txe_smsg_id = GNI_CQ_GET_MSG_ID(event_data);
3408
3409                 kgnilnd_validate_tx_ev_id(&ev_id, &tx, &conn);
3410                 if (conn == NULL || tx == NULL) {
3411                         /* either conn or tx was already nuked and this is a "late"
3412                          * completion, so drop it */
3413                         continue;
3414                 }
3415
3416                 tx->tx_conn->gnc_last_tx_cq = jiffies;
3417                 if (tx->tx_msg.gnm_type == GNILND_MSG_NOOP) {
3418                         set_mb(conn->gnc_last_noop_cq, jiffies);
3419                 }
3420
3421                 /* lock tx_list_state and tx_state */
3422                 kgnilnd_conn_mutex_lock(&conn->gnc_smsg_mutex);
3423                 spin_lock(&tx->tx_conn->gnc_list_lock);
3424
3425                 GNITX_ASSERTF(tx, tx->tx_list_state == GNILND_TX_LIVE_FMAQ,
3426                                 "state not GNILND_TX_LIVE_FMAQ", NULL);
3427                 GNITX_ASSERTF(tx, tx->tx_state & GNILND_TX_WAITING_COMPLETION,
3428                         "not waiting for completion", NULL);
3429
3430                 GNIDBG_TX(D_NET, tx, "SMSG complete tx_state %x rc %d",
3431                         tx->tx_state, rrc);
3432
3433                 tx->tx_state &= ~GNILND_TX_WAITING_COMPLETION;
3434
3435                 /* This will trigger other FMA sends that were
3436                  * pending this completion */
3437                 queued_fma = !list_empty(&tx->tx_conn->gnc_fmaq);
3438
3439                 /* we either did not expect reply or we already got it */
3440                 saw_reply = !(tx->tx_state & GNILND_TX_WAITING_REPLY);
3441
3442                 spin_unlock(&tx->tx_conn->gnc_list_lock);
3443                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
3444
3445                 if (queued_fma) {
3446                         CDEBUG(D_NET, "scheduling conn 0x%p->%s for fmaq\n",
3447                                conn,
3448                                libcfs_nid2str(conn->gnc_peer->gnp_nid));
3449                         kgnilnd_schedule_conn(conn);
3450                 }
3451
3452                 /* If saw_reply is false as soon as gnc_list_lock is dropped the tx could be nuked
3453                  * If saw_reply is true we know that the tx is safe to use as the other thread
3454                  * is already finished with it.
3455                  */
3456
3457                 if (saw_reply) {
3458                         /* no longer need to track on the live_fmaq */
3459                         kgnilnd_tx_del_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_ALLOCD);
3460
3461                         if (tx->tx_state & GNILND_TX_PENDING_RDMA) {
3462                                 /* we already got reply & were waiting for
3463                                  * completion of initial send */
3464                                 /* to initiate RDMA transaction */
3465                                 GNIDBG_TX(D_NET, tx,
3466                                          "Pending RDMA 0x%p type 0x%02x",
3467                                          tx->tx_msg.gnm_type);
3468                                 tx->tx_state &= ~GNILND_TX_PENDING_RDMA;
3469                                 rc = kgnilnd_send_mapped_tx(tx, 0);
3470                                 GNITX_ASSERTF(tx, rc == 0, "RDMA send failed: %d\n", rc);
3471                         } else {
3472                                 /* we are done with this tx */
3473                                 GNIDBG_TX(D_NET, tx,
3474                                          "Done with tx type 0x%02x",
3475                                          tx->tx_msg.gnm_type);
3476                                 kgnilnd_tx_done(tx, tx->tx_rc);
3477                         }
3478                 }
3479
3480                 /* drop ref from kgnilnd_validate_tx_ev_id */
3481                 kgnilnd_admin_decref(conn->gnc_tx_in_use);
3482                 kgnilnd_conn_decref(conn);
3483
3484                 /* if we are waiting for a REPLY, we'll handle the tx then */
3485         } /* end for loop */
3486 }
3487
3488 int
3489 kgnilnd_check_fma_rcv_cq(kgn_device_t *dev)
3490 {
3491         kgn_conn_t         *conn;
3492         gni_return_t        rrc;
3493         __u64               event_data;
3494         long                num_processed = 0;
3495         struct list_head   *conns;
3496         struct list_head   *tmp;
3497         int                 rc;
3498
3499         for (;;) {
3500                 /* make sure we don't keep looping if we need to reset */
3501                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
3502                         return num_processed;
3503                 }
3504
3505                 rc = kgnilnd_mutex_trylock(&dev->gnd_cq_mutex);
3506                 if (!rc) {
3507                         /* we didn't get the mutex, so return that there is still work
3508                          * to be done */
3509                         return 1;
3510                 }
3511                 rrc = kgnilnd_cq_get_event(dev->gnd_rcv_fma_cqh, &event_data);
3512                 kgnilnd_gl_mutex_unlock(&dev->gnd_cq_mutex);
3513
3514                 if (rrc == GNI_RC_NOT_DONE) {
3515                         CDEBUG(D_INFO, "SMSG RX CQ %d empty data %#llx "
3516                                 "processed %ld\n",
3517                                 dev->gnd_id, event_data, num_processed);
3518                         return num_processed;
3519                 }
3520                 dev->gnd_sched_alive = jiffies;
3521                 num_processed++;
3522
3523                 /* this is the only CQ that can really handle transient
3524                  * CQ errors */
3525                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_CQ_GET_EVENT)) {
3526                         rrc = cfs_fail_val ? cfs_fail_val
3527                                            : GNI_RC_ERROR_RESOURCE;
3528                         if (rrc == GNI_RC_ERROR_RESOURCE) {
3529                                 /* set overrun too */
3530                                 event_data |= (1UL << 63);
3531                                 LASSERTF(GNI_CQ_OVERRUN(event_data),
3532                                          "(1UL << 63) is no longer the bit to"
3533                                          "set to indicate CQ_OVERRUN\n");
3534                         }
3535                 }
3536                 /* sender should get error event too and take care
3537                 of failed transaction by re-transmitting */
3538                 if (rrc == GNI_RC_TRANSACTION_ERROR) {
3539                         CDEBUG(D_NET, "SMSG RX CQ error %#llx\n", event_data);
3540                         continue;
3541                 }
3542
3543                 if (likely(!GNI_CQ_OVERRUN(event_data))) {
3544                         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
3545                         conn = kgnilnd_cqid2conn_locked(
3546                                                  GNI_CQ_GET_INST_ID(event_data));
3547                         if (conn == NULL) {
3548                                 CDEBUG(D_NET, "SMSG RX CQID lookup %llu "
3549                                         "failed, dropping event %#llx\n",
3550                                         GNI_CQ_GET_INST_ID(event_data),
3551                                         event_data);
3552                         } else {
3553                                 CDEBUG(D_NET, "SMSG RX: CQID %llu "
3554                                        "conn %p->%s\n",
3555                                         GNI_CQ_GET_INST_ID(event_data),
3556                                         conn, conn->gnc_peer ?
3557                                         libcfs_nid2str(conn->gnc_peer->gnp_nid) :
3558                                         "<?>");
3559
3560                                 conn->gnc_last_rx_cq = jiffies;
3561
3562                                 /* stash first rx so we can clear out purgatory.
3563                                  */
3564                                 if (conn->gnc_first_rx == 0) {
3565                                         conn->gnc_first_rx = jiffies;
3566                                 }
3567                                 kgnilnd_peer_alive(conn->gnc_peer);
3568                                 kgnilnd_schedule_conn(conn);
3569                         }
3570                         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
3571                         continue;
3572                 }
3573
3574                 /* FMA CQ has overflowed: check ALL conns */
3575                 CNETERR("SMSG RX CQ overflow: scheduling ALL "
3576                        "conns on device %d\n", dev->gnd_id);
3577
3578                 for (rc = 0; rc < *kgnilnd_tunables.kgn_peer_hash_size; rc++) {
3579
3580                         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
3581                         conns = &kgnilnd_data.kgn_conns[rc];
3582
3583                         list_for_each(tmp, conns) {
3584                                 conn = list_entry(tmp, kgn_conn_t,
3585                                                   gnc_hashlist);
3586
3587                                 if (conn->gnc_device == dev) {
3588                                         kgnilnd_schedule_conn(conn);
3589                                         conn->gnc_last_rx_cq = jiffies;
3590                                 }
3591                         }
3592
3593                         /* don't block write lockers for too long... */
3594                         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
3595                 }
3596         }
3597 }
3598
3599 /* try_map_if_full should only be used when processing TX from list of
3600  * backlog TX waiting on mappings to free up
3601  *
3602  * Return Codes:
3603  *  try_map_if_full = 0: 0 (sent or queued), (-|+)errno failure of kgnilnd_sendmsg
3604  *  try_map_if_full = 1: 0 (sent), -ENOMEM for caller to requeue, (-|+)errno failure of kgnilnd_sendmsg */
3605
3606 int
3607 kgnilnd_send_mapped_tx(kgn_tx_t *tx, int try_map_if_full)
3608 {
3609         /* slight bit of race if multiple people calling, but at worst we'll have
3610          * order altered just a bit... which would not be determenistic anyways */
3611         int     rc = atomic_read(&tx->tx_conn->gnc_device->gnd_nq_map);
3612
3613         GNIDBG_TX(D_NET, tx, "try %d nq_map %d", try_map_if_full, rc);
3614
3615         /* We know that we have a GART reservation that should guarantee forward progress.
3616          * This means we don't need to take any extraordinary efforts if we are failing
3617          * mappings here - even if we are holding a very small number of these. */
3618
3619         if (try_map_if_full || (rc == 0)) {
3620                 rc = kgnilnd_map_buffer(tx);
3621         }
3622
3623         /* rc should be 0 if we mapped successfully here, if non-zero
3624          * we are queueing */
3625         if (rc != 0) {
3626                 /* if try_map_if_full set, they handle requeuing */
3627                 if (unlikely(try_map_if_full)) {
3628                         RETURN(rc);
3629                 } else {
3630                         spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
3631                         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
3632                         spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
3633                         /* make sure we wake up sched to run this */
3634                         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
3635                         /* return 0 as this is now queued for later sending */
3636                         RETURN(0);
3637                 }
3638         }
3639
3640         switch (tx->tx_msg.gnm_type) {
3641         default:
3642                 LBUG();
3643                 break;
3644         /* GET_REQ and PUT_ACK are outbound messages sending our mapping key to
3645          * remote node where the RDMA will be started
3646          * Special case -EAGAIN logic - this should just queued as if the mapping couldn't
3647          * be satisified. The rest of the errors are "hard" errors that require
3648          * upper layers to handle themselves.
3649          * If kgnilnd_post_rdma returns a resource error, kgnilnd_rdma will put
3650          * the tx back on the TX_MAPQ. When this tx is pulled back off the MAPQ,
3651          * it's gnm_type will now be GNILND_MSG_PUT_DONE or
3652          * GNILND_MSG_GET_DONE_REV.
3653          */
3654         case GNILND_MSG_GET_REQ:
3655                 tx->tx_msg.gnm_u.get.gngm_desc.gnrd_key = tx->tx_map_key;
3656                 tx->tx_msg.gnm_u.get.gngm_cookie = tx->tx_id.txe_cookie;
3657                 tx->tx_msg.gnm_u.get.gngm_desc.gnrd_addr = (__u64)((unsigned long)tx->tx_buffer);
3658                 tx->tx_msg.gnm_u.get.gngm_desc.gnrd_nob = tx->tx_nob;
3659                 tx->tx_state = GNILND_TX_WAITING_COMPLETION | GNILND_TX_WAITING_REPLY;
3660                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_GET_REQ_AGAIN)) {
3661                         tx->tx_state |= GNILND_TX_FAIL_SMSG;
3662                 }
3663                 /* redirect to FMAQ on failure, no need to infinite loop here in MAPQ */
3664                 rc = kgnilnd_sendmsg(tx, NULL, 0, &tx->tx_conn->gnc_list_lock, GNILND_TX_FMAQ);
3665                 break;
3666         case GNILND_MSG_PUT_ACK:
3667                 tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_key = tx->tx_map_key;
3668                 tx->tx_state = GNILND_TX_WAITING_COMPLETION | GNILND_TX_WAITING_REPLY;
3669                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_PUT_ACK_AGAIN)) {
3670                         tx->tx_state |= GNILND_TX_FAIL_SMSG;
3671                 }
3672                 /* redirect to FMAQ on failure, no need to infinite loop here in MAPQ */
3673                 rc = kgnilnd_sendmsg(tx, NULL, 0, &tx->tx_conn->gnc_list_lock, GNILND_TX_FMAQ);
3674                 break;
3675
3676         /* PUT_REQ and GET_DONE are where we do the actual RDMA */
3677         case GNILND_MSG_PUT_DONE:
3678         case GNILND_MSG_PUT_REQ:
3679                 rc = kgnilnd_rdma(tx, GNILND_MSG_PUT_DONE,
3680                              &tx->tx_putinfo.gnpam_desc,
3681                              tx->tx_putinfo.gnpam_desc.gnrd_nob,
3682                              tx->tx_putinfo.gnpam_dst_cookie);
3683                 RETURN(try_map_if_full ? rc : 0);
3684                 break;
3685         case GNILND_MSG_GET_DONE:
3686                 rc = kgnilnd_rdma(tx, GNILND_MSG_GET_DONE,
3687                              &tx->tx_getinfo.gngm_desc,
3688                              tx->tx_lntmsg[0]->msg_len,
3689                              tx->tx_getinfo.gngm_cookie);
3690                 RETURN(try_map_if_full ? rc : 0);
3691                 break;
3692         case GNILND_MSG_PUT_REQ_REV:
3693                 tx->tx_msg.gnm_u.get.gngm_desc.gnrd_key = tx->tx_map_key;
3694                 tx->tx_msg.gnm_u.get.gngm_cookie = tx->tx_id.txe_cookie;
3695                 tx->tx_msg.gnm_u.get.gngm_desc.gnrd_addr = (__u64)((unsigned long)tx->tx_buffer);
3696                 tx->tx_msg.gnm_u.get.gngm_desc.gnrd_nob = tx->tx_nob;
3697                 tx->tx_state = GNILND_TX_WAITING_COMPLETION | GNILND_TX_WAITING_REPLY;
3698                 kgnilnd_compute_rdma_cksum(tx, tx->tx_nob);
3699                 tx->tx_msg.gnm_u.get.gngm_payload_cksum = tx->tx_msg.gnm_payload_cksum;
3700
3701                 rc = kgnilnd_sendmsg(tx, NULL, 0, &tx->tx_conn->gnc_list_lock, GNILND_TX_FMAQ);
3702                 break;
3703         case GNILND_MSG_PUT_DONE_REV:
3704                 rc = kgnilnd_rdma(tx, GNILND_MSG_PUT_DONE_REV,
3705                              &tx->tx_getinfo.gngm_desc,
3706                              tx->tx_nob,
3707                              tx->tx_getinfo.gngm_cookie);
3708                 RETURN(try_map_if_full ? rc : 0);
3709                 break;
3710         case GNILND_MSG_GET_ACK_REV:
3711                 tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_key = tx->tx_map_key;
3712                 tx->tx_state = GNILND_TX_WAITING_COMPLETION | GNILND_TX_WAITING_REPLY;
3713                 /* LNET_GETS are a special case for parse */
3714                 kgnilnd_compute_rdma_cksum(tx, tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_nob);
3715                 tx->tx_msg.gnm_u.putack.gnpam_payload_cksum = tx->tx_msg.gnm_payload_cksum;
3716
3717                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_PUT_ACK_AGAIN))
3718                         tx->tx_state |= GNILND_TX_FAIL_SMSG;
3719
3720                 /* redirect to FMAQ on failure, no need to infinite loop here in MAPQ */
3721                 rc = kgnilnd_sendmsg(tx, NULL, 0, &tx->tx_conn->gnc_list_lock, GNILND_TX_FMAQ);
3722                 break;
3723         case GNILND_MSG_GET_DONE_REV:
3724         case GNILND_MSG_GET_REQ_REV:
3725                 rc = kgnilnd_rdma(tx, GNILND_MSG_GET_DONE_REV,
3726                                 &tx->tx_putinfo.gnpam_desc,
3727                                 tx->tx_putinfo.gnpam_desc.gnrd_nob,
3728                                 tx->tx_putinfo.gnpam_dst_cookie);
3729                 RETURN(try_map_if_full ? rc : 0);
3730                 break;
3731         }
3732
3733         RETURN(rc);
3734 }
3735
3736 void
3737 kgnilnd_process_fmaq(kgn_conn_t *conn)
3738 {
3739         int           more_to_do = 0;
3740         kgn_tx_t     *tx = NULL;
3741         void         *buffer = NULL;
3742         unsigned int  nob = 0;
3743         int           rc;
3744
3745         /* NB 1. kgnilnd_sendmsg() may fail if I'm out of credits right now.
3746          *       However I will be rescheduled by an FMA completion event
3747          *       when I eventually get some.
3748          * NB 2. Sampling gnc_state here races with setting it elsewhere.
3749          *       But it doesn't matter if I try to send a "real" message just
3750          *       as I start closing because I'll get scheduled to send the
3751          *       close anyway. */
3752
3753         /* Short circuit if the ep_handle is null we cant send anyway. */
3754         if (conn->gnc_ephandle == NULL)
3755                 return;
3756
3757         LASSERTF(!conn->gnc_close_sent, "Conn %p close was sent\n", conn);
3758
3759         spin_lock(&conn->gnc_list_lock);
3760
3761         if (list_empty(&conn->gnc_fmaq)) {
3762                 int     keepalive = GNILND_TO2KA(conn->gnc_timeout);
3763
3764                 spin_unlock(&conn->gnc_list_lock);
3765
3766                 if (time_after_eq(jiffies, conn->gnc_last_tx + cfs_time_seconds(keepalive))) {
3767                         CDEBUG(D_NET, "sending NOOP -> %s (%p idle %lu(%d)) "
3768                                "last %lu/%lu/%lu %lus/%lus/%lus\n",
3769                                libcfs_nid2str(conn->gnc_peer->gnp_nid), conn,
3770                                cfs_duration_sec(jiffies - conn->gnc_last_tx),
3771                                keepalive,
3772                                conn->gnc_last_noop_want, conn->gnc_last_noop_sent,
3773                                conn->gnc_last_noop_cq,
3774                                cfs_duration_sec(jiffies - conn->gnc_last_noop_want),
3775                                cfs_duration_sec(jiffies - conn->gnc_last_noop_sent),
3776                                cfs_duration_sec(jiffies - conn->gnc_last_noop_cq));
3777                         atomic_inc(&conn->gnc_sched_noop);
3778                         set_mb(conn->gnc_last_noop_want, jiffies);
3779
3780                         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_NOOP_SEND))
3781                                 return;
3782
3783                         tx = kgnilnd_new_tx_msg(GNILND_MSG_NOOP, conn->gnc_peer->gnp_net->gnn_ni->ni_nid);
3784                         if (tx != NULL) {
3785                                 int     rc;
3786
3787                                 rc = kgnilnd_set_tx_id(tx, conn);
3788                                 if (rc != 0) {
3789                                         kgnilnd_tx_done(tx, rc);
3790                                         return;
3791                                 }
3792                         }
3793                 }
3794         } else {
3795                 tx = list_first_entry(&conn->gnc_fmaq, kgn_tx_t, tx_list);
3796                 /* move from fmaq to allocd, kgnilnd_sendmsg will move to live_fmaq */
3797                 kgnilnd_tx_del_state_locked(tx, NULL, conn, GNILND_TX_ALLOCD);
3798                 more_to_do = !list_empty(&conn->gnc_fmaq);
3799                 spin_unlock(&conn->gnc_list_lock);
3800         }
3801
3802         /* if there is no real TX or no NOOP to send, bail */
3803         if (tx == NULL) {
3804                 return;
3805         }
3806
3807         if (!tx->tx_retrans)
3808                 tx->tx_cred_wait = jiffies;
3809
3810         GNITX_ASSERTF(tx, tx->tx_id.txe_smsg_id != 0,
3811                       "tx with zero id", NULL);
3812
3813         CDEBUG(D_NET, "sending regular msg: %p, type %s(0x%02x), cookie %#llx\n",
3814                tx, kgnilnd_msgtype2str(tx->tx_msg.gnm_type),
3815                tx->tx_msg.gnm_type, tx->tx_id.txe_cookie);
3816
3817         rc = 0;
3818
3819         switch (tx->tx_msg.gnm_type) {
3820         default:
3821                 LBUG();
3822
3823         case GNILND_MSG_NOOP:
3824         case GNILND_MSG_CLOSE:
3825         case GNILND_MSG_IMMEDIATE:
3826                 tx->tx_state = GNILND_TX_WAITING_COMPLETION;
3827                 buffer = tx->tx_buffer;
3828                 nob = tx->tx_nob;
3829                 break;
3830
3831         case GNILND_MSG_GET_DONE:
3832         case GNILND_MSG_PUT_DONE:
3833         case GNILND_MSG_PUT_DONE_REV:
3834         case GNILND_MSG_GET_DONE_REV:
3835         case GNILND_MSG_PUT_NAK:
3836         case GNILND_MSG_GET_NAK:
3837         case GNILND_MSG_GET_NAK_REV:
3838         case GNILND_MSG_PUT_NAK_REV:
3839                 tx->tx_state = GNILND_TX_WAITING_COMPLETION;
3840                 break;
3841
3842         case GNILND_MSG_PUT_REQ:
3843         case GNILND_MSG_GET_REQ_REV:
3844                 tx->tx_msg.gnm_u.putreq.gnprm_cookie = tx->tx_id.txe_cookie;
3845
3846         case GNILND_MSG_PUT_ACK:
3847         case GNILND_MSG_PUT_REQ_REV:
3848         case GNILND_MSG_GET_ACK_REV:
3849         case GNILND_MSG_GET_REQ:
3850                 /* This is really only to handle the retransmit of SMSG once these
3851                  * two messages are setup in send_mapped_tx */
3852                 tx->tx_state = GNILND_TX_WAITING_COMPLETION | GNILND_TX_WAITING_REPLY;
3853                 break;
3854         }
3855
3856         if (likely(rc == 0)) {
3857                 rc = kgnilnd_sendmsg(tx, buffer, nob, &conn->gnc_list_lock, GNILND_TX_FMAQ);
3858         }
3859
3860         if (rc > 0) {
3861                 /* don't explicitly reschedule here - we are short credits and will rely on
3862                  * kgnilnd_sendmsg to resched the conn if need be */
3863                 more_to_do = 0;
3864         } else if (rc < 0) {
3865                 /* bail: it wasn't sent and we didn't get EAGAIN indicating we should retrans
3866                  * almost certainly a software bug, but lets play nice with the other kids */
3867                 kgnilnd_tx_done(tx, rc);
3868                 /* just for fun, kick peer in arse - resetting conn might help to correct
3869                  * this almost certainly buggy software caused return code */
3870                 kgnilnd_close_conn(conn, rc);
3871         }
3872
3873         if (more_to_do) {
3874                 CDEBUG(D_NET, "Rescheduling %p (more to do)\n", conn);
3875                 kgnilnd_schedule_conn(conn);
3876         }
3877 }
3878
3879 int
3880 kgnilnd_process_rdmaq(kgn_device_t *dev)
3881 {
3882         int               found_work = 0;
3883         kgn_tx_t         *tx;
3884
3885         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_DELAY_RDMAQ)) {
3886                 RETURN(found_work);
3887         }
3888
3889         if (time_after_eq(jiffies, dev->gnd_rdmaq_deadline)) {
3890                 unsigned long           dead_bump;
3891                 long                    new_ok;
3892
3893                 /* if we think we need to adjust, take lock to serialize and recheck */
3894                 spin_lock(&dev->gnd_rdmaq_lock);
3895                 if (time_after_eq(jiffies, dev->gnd_rdmaq_deadline)) {
3896                         del_singleshot_timer_sync(&dev->gnd_rdmaq_timer);
3897
3898                         dead_bump = cfs_time_seconds(1) / *kgnilnd_tunables.kgn_rdmaq_intervals;
3899
3900                         /* roll the bucket forward */
3901                         dev->gnd_rdmaq_deadline = jiffies + dead_bump;
3902
3903                         if (kgnilnd_data.kgn_rdmaq_override &&
3904                                 (*kgnilnd_tunables.kgn_rdmaq_intervals != 0)) {
3905                                 new_ok = kgnilnd_data.kgn_rdmaq_override / *kgnilnd_tunables.kgn_rdmaq_intervals;
3906                         }  else {
3907                                 new_ok = ~0UL >> 1;
3908                         }
3909
3910                         /* roll current outstanding forward to make sure we carry outstanding
3911                          * committment forward
3912                          * new_ok starts out as the whole interval value
3913                          *  - first subtract bytes_out from last interval, as that would push us over
3914                          *    strict limits for this interval
3915                          *  - second, set bytes_ok to new_ok to ensure it doesn't exceed the current auth
3916                          *
3917                          * there is a small race here if someone is actively processing mappings and
3918                          * adding to rdmaq_bytes_out, but it should be small as the mappings are triggered
3919                          * quite quickly after kgnilnd_auth_rdma_bytes gives us the go-ahead
3920                          * - if this gives us problems in the future, we could use a read/write lock
3921                          * to protect the resetting of these values */
3922                         new_ok -= atomic64_read(&dev->gnd_rdmaq_bytes_out);
3923                         atomic64_set(&dev->gnd_rdmaq_bytes_ok, new_ok);
3924
3925                         CDEBUG(D_NET, "resetting rdmaq bytes to %ld, deadline +%lu -> %lu, "
3926                                        "current out %ld\n",
3927                                atomic64_read(&dev->gnd_rdmaq_bytes_ok), dead_bump, dev->gnd_rdmaq_deadline,
3928                                atomic64_read(&dev->gnd_rdmaq_bytes_out));
3929                 }
3930                 spin_unlock(&dev->gnd_rdmaq_lock);
3931         }
3932
3933         spin_lock(&dev->gnd_rdmaq_lock);
3934         while (!list_empty(&dev->gnd_rdmaq)) {
3935                 int     rc;
3936
3937                 /* make sure we break out early on quiesce */
3938                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
3939                         /* always break with lock held - we unlock outside loop */
3940                         break;
3941                 }
3942
3943                 tx = list_first_entry(&dev->gnd_rdmaq, kgn_tx_t, tx_list);
3944                 kgnilnd_tx_del_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_ALLOCD);
3945                 found_work++;
3946
3947                 /* sample with lock held, serializing with kgnilnd_complete_closed_conn */
3948                 if (tx->tx_conn->gnc_state != GNILND_CONN_ESTABLISHED) {
3949                         /* if conn is dying, mark tx in tx_ref_table for
3950                          * kgnilnd_complete_closed_conn to finish up */
3951                         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_DYING, 1);
3952
3953                         /* tx was moved to DYING, get next */
3954                         continue;
3955                 }
3956                 spin_unlock(&dev->gnd_rdmaq_lock);
3957
3958                 rc = kgnilnd_auth_rdma_bytes(dev, tx);
3959                 spin_lock(&dev->gnd_rdmaq_lock);
3960
3961                 if (rc < 0) {
3962                         /* no ticket! add back to head */
3963                         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_RDMAQ, 0);
3964                         /* clear found_work so scheduler threads wait for timer */
3965                         found_work = 0;
3966                         break;
3967                 } else {
3968                         /* TX is GO for launch */
3969                         tx->tx_qtime = jiffies;
3970                         kgnilnd_send_mapped_tx(tx, 0);
3971                         found_work++;
3972                 }
3973         }
3974         spin_unlock(&dev->gnd_rdmaq_lock);
3975
3976         RETURN(found_work);
3977 }
3978
3979 static inline void
3980 kgnilnd_swab_rdma_desc(kgn_rdma_desc_t *d)
3981 {
3982         __swab64s(&d->gnrd_key.qword1);
3983         __swab64s(&d->gnrd_key.qword2);
3984         __swab64s(&d->gnrd_addr);
3985         __swab32s(&d->gnrd_nob);
3986 }
3987
3988 #define kgnilnd_match_reply_either(w, x, y, z) _kgnilnd_match_reply(w, x, y, z)
3989 #define kgnilnd_match_reply(x, y, z) _kgnilnd_match_reply(x, y, GNILND_MSG_NONE, z)
3990
3991 kgn_tx_t *
3992 _kgnilnd_match_reply(kgn_conn_t *conn, int type1, int type2, __u64 cookie)
3993 {
3994         kgn_tx_ev_id_t    ev_id;
3995         kgn_tx_t         *tx;
3996
3997         /* we use the cookie from the original TX, so we can find the match
3998          * by parsing that and using the txe_idx */
3999         ev_id.txe_cookie = cookie;
4000
4001         tx = conn->gnc_tx_ref_table[ev_id.txe_idx];
4002
4003         if (tx != NULL) {
4004                 /* check tx to make sure kgni didn't eat it */
4005                 GNITX_ASSERTF(tx, tx->tx_msg.gnm_magic == GNILND_MSG_MAGIC,
4006                               "came back from kgni with bad magic %x\n", tx->tx_msg.gnm_magic);
4007
4008                 GNITX_ASSERTF(tx, ((tx->tx_id.txe_idx == ev_id.txe_idx) &&
4009                                   (tx->tx_id.txe_cookie = cookie)),
4010                               "conn 0x%p->%s tx_ref_table hosed: wanted "
4011                               "txe_cookie %#llx txe_idx %d "
4012                               "found tx %p cookie %#llx txe_idx %d\n",
4013                               conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
4014                               cookie, ev_id.txe_idx,
4015                               tx, tx->tx_id.txe_cookie, tx->tx_id.txe_idx);
4016
4017                 LASSERTF((((tx->tx_msg.gnm_type == type1) || (tx->tx_msg.gnm_type == type2)) &&
4018                         (tx->tx_state & GNILND_TX_WAITING_REPLY)),
4019                         "Unexpected TX type (%x, %x or %x) "
4020                         "or state (%x, expected +%x) "
4021                         "matched reply from %s\n",
4022                         tx->tx_msg.gnm_type, type1, type2,
4023                         tx->tx_state, GNILND_TX_WAITING_REPLY,
4024                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
4025         } else {
4026                 CWARN("Unmatched reply %02x, or %02x/%#llx from %s\n",
4027                       type1, type2, cookie, libcfs_nid2str(conn->gnc_peer->gnp_nid));
4028         }
4029         return tx;
4030 }
4031
4032 static inline void
4033 kgnilnd_complete_tx(kgn_tx_t *tx, int rc)
4034 {
4035         int             complete = 0;
4036         kgn_conn_t      *conn = tx->tx_conn;
4037         __u64 nob = tx->tx_nob;
4038         __u32 physnop = tx->tx_phys_npages;
4039         int   id = tx->tx_id.txe_smsg_id;
4040         int buftype = tx->tx_buftype;
4041         gni_mem_handle_t hndl;
4042         hndl.qword1 = tx->tx_map_key.qword1;
4043         hndl.qword2 = tx->tx_map_key.qword2;
4044
4045         spin_lock(&conn->gnc_list_lock);
4046
4047         GNITX_ASSERTF(tx, tx->tx_state & GNILND_TX_WAITING_REPLY,
4048                 "not waiting for reply", NULL);
4049
4050         tx->tx_rc = rc;
4051         tx->tx_state &= ~GNILND_TX_WAITING_REPLY;
4052
4053         if (rc == -EFAULT) {
4054                 CDEBUG(D_NETERROR, "Error %d TX data: TX %p tx_id %x nob %16llu physnop %8d buffertype %#8x MemHandle %#llx.%#llxx\n",
4055                         rc, tx, id, nob, physnop, buftype, hndl.qword1, hndl.qword2);
4056
4057                 if(*kgnilnd_tunables.kgn_efault_lbug) {
4058                         GNIDBG_TOMSG(D_NETERROR, &tx->tx_msg,
4059                         "error %d on tx 0x%p->%s id %u/%d state %s age %ds",
4060                         rc, tx, conn ?
4061                         libcfs_nid2str(conn->gnc_peer->gnp_nid) : "<?>",
4062                         tx->tx_id.txe_smsg_id, tx->tx_id.txe_idx,
4063                         kgnilnd_tx_state2str(tx->tx_list_state),
4064                         cfs_duration_sec((unsigned long) jiffies - tx->tx_qtime));
4065                         LBUG();
4066                 }
4067         }
4068
4069         if (!(tx->tx_state & GNILND_TX_WAITING_COMPLETION)) {
4070                 kgnilnd_tx_del_state_locked(tx, NULL, conn, GNILND_TX_ALLOCD);
4071                 /* sample under lock as follow on steps require gnc_list_lock
4072                  * - or call kgnilnd_tx_done which requires no locks held over
4073                  *   call to lnet_finalize */
4074                 complete = 1;
4075         }
4076         spin_unlock(&conn->gnc_list_lock);
4077
4078         if (complete) {
4079                 kgnilnd_tx_done(tx, tx->tx_rc);
4080         }
4081 }
4082
4083 static inline void
4084 kgnilnd_finalize_rx_done(kgn_tx_t *tx, kgn_msg_t *msg)
4085 {
4086         int              rc;
4087         kgn_conn_t      *conn = tx->tx_conn;
4088
4089         atomic_inc(&conn->gnc_device->gnd_rdma_nrx);
4090         atomic64_add(tx->tx_nob, &conn->gnc_device->gnd_rdma_rxbytes);
4091
4092         /* the gncm_retval is passed in for PUTs */
4093         rc = kgnilnd_verify_rdma_cksum(tx, msg->gnm_payload_cksum,
4094                                        msg->gnm_u.completion.gncm_retval);
4095
4096         kgnilnd_complete_tx(tx, rc);
4097 }
4098
4099 void
4100 kgnilnd_check_fma_rx(kgn_conn_t *conn)
4101 {
4102         __u32         seq;
4103         kgn_tx_t     *tx;
4104         kgn_rx_t     *rx;
4105         kgn_msg_t    *msg;
4106         void         *prefix;
4107         gni_return_t  rrc;
4108         kgn_peer_t   *peer = conn->gnc_peer;
4109         kgn_net_t    *net;
4110         int           rc = 0;
4111         __u16         tmp_cksum = 0, msg_cksum = 0;
4112         int           repost = 1, saw_complete;
4113         unsigned long timestamp, newest_last_rx, timeout;
4114         int           last_seq;
4115         ENTRY;
4116
4117         /* Short circuit if the ep_handle is null.
4118          * It's likely that its about to be closed as stale.
4119          */
4120         if (conn->gnc_ephandle == NULL)
4121                 RETURN_EXIT;
4122
4123         timestamp = jiffies;
4124         kgnilnd_gl_mutex_lock(&conn->gnc_device->gnd_cq_mutex);
4125         /* delay in jiffies - we are really concerned only with things that
4126          * result in a schedule() or really holding this off for long times .
4127          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
4128         conn->gnc_device->gnd_mutex_delay += (long) jiffies - timestamp;
4129
4130         /* Resample current time as we have no idea how long it took to get the mutex */
4131         timestamp = jiffies;
4132
4133         /* We check here when the last time we received an rx, we do this before
4134          * we call getnext in case the thread has been blocked for a while. If we
4135          * havent received an rx since our timeout value we close the connection
4136          * as we should assume the other side has closed the connection. This will
4137          * stop us from sending replies to a mailbox that is already in purgatory.
4138          */
4139
4140         timeout = cfs_time_seconds(conn->gnc_timeout);
4141         newest_last_rx = GNILND_LASTRX(conn);
4142
4143         /* Error injection to validate that timestamp checking works and closing the conn */
4144         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_RECV_TIMEOUT)) {
4145                 timestamp = timestamp + (GNILND_TIMEOUTRX(timeout) * 2);
4146         }
4147
4148         if (time_after_eq(timestamp, newest_last_rx + (GNILND_TIMEOUTRX(timeout)))) {
4149                 GNIDBG_CONN(D_NETERROR|D_CONSOLE, conn, "Cant receive from %s after timeout lapse of %lu; TO %lu",
4150                 libcfs_nid2str(conn->gnc_peer->gnp_nid),
4151                 cfs_duration_sec(timestamp - newest_last_rx),
4152                 cfs_duration_sec(GNILND_TIMEOUTRX(timeout)));
4153                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
4154                 rc = -ETIME;
4155                 kgnilnd_close_conn(conn, rc);
4156                 RETURN_EXIT;
4157         }
4158
4159         rrc = kgnilnd_smsg_getnext(conn->gnc_ephandle, &prefix);
4160
4161         if (rrc == GNI_RC_NOT_DONE) {
4162                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
4163                 CDEBUG(D_INFO, "SMSG RX empty conn 0x%p\n", conn);
4164                 RETURN_EXIT;
4165         }
4166
4167         /* Instead of asserting when we get mailbox corruption lets attempt to
4168          * close the conn and recover. We can put the conn/mailbox into
4169          * purgatory and let purgatory deal with the problem. If we see
4170          * this NETTERROR reported on production systems in large amounts
4171          * we will need to revisit the state machine to see if we can tighten
4172          * it up further to improve data protection.
4173          */
4174
4175         if (rrc == GNI_RC_INVALID_STATE) {
4176                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
4177                 GNIDBG_CONN(D_NETERROR | D_CONSOLE, conn, "Mailbox corruption "
4178                         "detected closing conn %p from peer %s\n", conn,
4179                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
4180                 rc = -EIO;
4181                 kgnilnd_close_conn(conn, rc);
4182                 RETURN_EXIT;
4183         }
4184
4185         LASSERTF(rrc == GNI_RC_SUCCESS,
4186                 "bad rc %d on conn %p from peer %s\n",
4187                 rrc, conn, libcfs_nid2str(peer->gnp_nid));
4188
4189         msg = (kgn_msg_t *)prefix;
4190
4191         rx = kgnilnd_alloc_rx();
4192         if (rx == NULL) {
4193                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
4194                 kgnilnd_release_msg(conn);
4195                 GNIDBG_MSG(D_NETERROR, msg, "Dropping SMSG RX from 0x%p->%s, no RX memory",
4196                            conn, libcfs_nid2str(peer->gnp_nid));
4197                 RETURN_EXIT;
4198         }
4199
4200         GNIDBG_MSG(D_INFO, msg, "SMSG RX on %p", conn);
4201
4202         timestamp = conn->gnc_last_rx;
4203         seq = last_seq = atomic_read(&conn->gnc_rx_seq);
4204         atomic_inc(&conn->gnc_rx_seq);
4205
4206         conn->gnc_last_rx = jiffies;
4207         /* stash first rx so we can clear out purgatory
4208          */
4209         if (conn->gnc_first_rx == 0)
4210                 conn->gnc_first_rx = jiffies;
4211
4212         /* needs to linger to protect gnc_rx_seq like we do with gnc_tx_seq */
4213         kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
4214         kgnilnd_peer_alive(conn->gnc_peer);
4215
4216         rx->grx_msg = msg;
4217         rx->grx_conn = conn;
4218         rx->grx_eager = 0;
4219         rx->grx_received = current_kernel_time();
4220
4221         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_NET_LOOKUP)) {
4222                 rc = -ENONET;
4223         } else {
4224                 rc = kgnilnd_find_net(msg->gnm_srcnid, &net);
4225         }
4226
4227         if (rc < 0) {
4228                 GOTO(out, rc);
4229         } else {
4230                 kgnilnd_net_decref(net);
4231         }
4232
4233         if (*kgnilnd_tunables.kgn_checksum && !msg->gnm_cksum)
4234                 GNIDBG_MSG(D_WARNING, msg, "no msg header checksum when enabled");
4235
4236         /* XXX Nic: Do we need to swab cksum */
4237         if (msg->gnm_cksum != 0) {
4238                 msg_cksum = msg->gnm_cksum;
4239                 msg->gnm_cksum = 0;
4240                 tmp_cksum = kgnilnd_cksum(msg, sizeof(kgn_msg_t));
4241
4242                 if (tmp_cksum != msg_cksum) {
4243                         GNIDBG_MSG(D_NETERROR, msg, "Bad hdr checksum (%x expected %x)",
4244                                         tmp_cksum, msg_cksum);
4245                         kgnilnd_dump_msg(D_BUFFS, msg);
4246                         rc = -ENOKEY;
4247                         GOTO(out, rc);
4248                 }
4249         }
4250         /* restore checksum for future debug messages */
4251         msg->gnm_cksum = tmp_cksum;
4252
4253         if (msg->gnm_magic != GNILND_MSG_MAGIC) {
4254                 if (__swab32(msg->gnm_magic) != GNILND_MSG_MAGIC) {
4255                         GNIDBG_MSG(D_NETERROR, msg, "Unexpected magic %08x from %s",
4256                                msg->gnm_magic, libcfs_nid2str(peer->gnp_nid));
4257                         rc = -EPROTO;
4258                         GOTO(out, rc);
4259                 }
4260
4261                 __swab32s(&msg->gnm_magic);
4262                 __swab16s(&msg->gnm_version);
4263                 __swab16s(&msg->gnm_type);
4264                 __swab64s(&msg->gnm_srcnid);
4265                 __swab64s(&msg->gnm_connstamp);
4266                 __swab32s(&msg->gnm_seq);
4267
4268                 /* NB message type checked below; NOT here... */
4269                 switch (msg->gnm_type) {
4270                 case GNILND_MSG_GET_ACK_REV:
4271                 case GNILND_MSG_PUT_ACK:
4272                         kgnilnd_swab_rdma_desc(&msg->gnm_u.putack.gnpam_desc);
4273                         break;
4274
4275                 case GNILND_MSG_PUT_REQ_REV:
4276                 case GNILND_MSG_GET_REQ:
4277                         kgnilnd_swab_rdma_desc(&msg->gnm_u.get.gngm_desc);
4278                         break;
4279
4280                 default:
4281                         break;
4282                 }
4283         }
4284
4285         if (msg->gnm_version != GNILND_MSG_VERSION) {
4286                 GNIDBG_MSG(D_NETERROR, msg, "Unexpected protocol version %d from %s",
4287                        msg->gnm_version, libcfs_nid2str(peer->gnp_nid));
4288                 rc = -EPROTO;
4289                 GOTO(out, rc);
4290         }
4291
4292         if (LNET_NIDADDR(msg->gnm_srcnid) != LNET_NIDADDR(peer->gnp_nid)) {
4293                 GNIDBG_MSG(D_NETERROR, msg, "Unexpected peer %s from %s",
4294                        libcfs_nid2str(msg->gnm_srcnid),
4295                        libcfs_nid2str(peer->gnp_nid));
4296                 rc = -EPROTO;
4297                 GOTO(out, rc);
4298         }
4299
4300         if (msg->gnm_connstamp != conn->gnc_peer_connstamp) {
4301                 GNIDBG_MSG(D_NETERROR, msg, "Unexpected connstamp %#llx(%#llx"
4302                        " expected) from %s",
4303                        msg->gnm_connstamp, conn->gnc_peer_connstamp,
4304                        libcfs_nid2str(peer->gnp_nid));
4305                 rc = -EPROTO;
4306                 GOTO(out, rc);
4307         }
4308
4309         if (msg->gnm_seq != seq) {
4310                 GNIDBG_MSG(D_NETERROR, msg, "Unexpected sequence number %d(%d expected) from %s",
4311                        msg->gnm_seq, seq, libcfs_nid2str(peer->gnp_nid));
4312                 rc = -EPROTO;
4313                 GOTO(out, rc);
4314         }
4315
4316         atomic_inc(&conn->gnc_device->gnd_short_nrx);
4317
4318         if (msg->gnm_type == GNILND_MSG_CLOSE) {
4319                 CDEBUG(D_NETTRACE, "%s sent us CLOSE msg\n",
4320                               libcfs_nid2str(conn->gnc_peer->gnp_nid));
4321                 write_lock(&kgnilnd_data.kgn_peer_conn_lock);
4322                 conn->gnc_close_recvd = GNILND_CLOSE_RX;
4323                 conn->gnc_peer_error = msg->gnm_u.completion.gncm_retval;
4324                 /* double check state with lock held */
4325                 if (conn->gnc_state == GNILND_CONN_ESTABLISHED) {
4326                         /* only error if we are not already closing */
4327                         if (conn->gnc_peer_error == -ETIMEDOUT) {
4328                                 unsigned long           now = jiffies;
4329                                 CNETERR("peer 0x%p->%s closed connection 0x%p due to timeout. "
4330                                        "Is node down? "
4331                                        "RX %d @ %lus/%lus; TX %d @ %lus/%lus; "
4332                                        "NOOP %lus/%lus/%lus; sched %lus/%lus/%lus ago\n",
4333                                        conn->gnc_peer, libcfs_nid2str(conn->gnc_peer->gnp_nid),
4334                                        conn, last_seq,
4335                                        cfs_duration_sec(now - timestamp),
4336                                        cfs_duration_sec(now - conn->gnc_last_rx_cq),
4337                                        atomic_read(&conn->gnc_tx_seq),
4338                                        cfs_duration_sec(now - conn->gnc_last_tx),
4339                                        cfs_duration_sec(now - conn->gnc_last_tx_cq),
4340                                        cfs_duration_sec(now - conn->gnc_last_noop_want),
4341                                        cfs_duration_sec(now - conn->gnc_last_noop_sent),
4342                                        cfs_duration_sec(now - conn->gnc_last_noop_cq),
4343                                        cfs_duration_sec(now - conn->gnc_last_sched_ask),
4344                                        cfs_duration_sec(now - conn->gnc_last_sched_do),
4345                                        cfs_duration_sec(now - conn->gnc_device->gnd_sched_alive));
4346                         }
4347                         kgnilnd_close_conn_locked(conn, -ECONNRESET);
4348                 }
4349                 write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
4350                 GOTO(out, rc);
4351         }
4352
4353         if (conn->gnc_close_recvd) {
4354                 GNIDBG_MSG(D_NETERROR, msg, "Unexpected message %s(%d/%d) after CLOSE from %s",
4355                        kgnilnd_msgtype2str(msg->gnm_type),
4356                        msg->gnm_type, conn->gnc_close_recvd,
4357                        libcfs_nid2str(conn->gnc_peer->gnp_nid));
4358                 rc = -EPROTO;
4359                 GOTO(out, rc);
4360         }
4361
4362         if (conn->gnc_state != GNILND_CONN_ESTABLISHED) {
4363                 /* XXX Nic: log message received on bad connection state */
4364                 GOTO(out, rc);
4365         }
4366
4367         switch (msg->gnm_type) {
4368         case GNILND_MSG_NOOP:
4369                 /* Nothing to do; just a keepalive */
4370                 break;
4371
4372         case GNILND_MSG_IMMEDIATE:
4373                 /* only get SMSG payload for IMMEDIATE */
4374                 atomic64_add(msg->gnm_payload_len, &conn->gnc_device->gnd_short_rxbytes);
4375                 rc = lnet_parse(net->gnn_ni, &msg->gnm_u.immediate.gnim_hdr,
4376                                 msg->gnm_srcnid, rx, 0);
4377                 repost = rc < 0;
4378                 break;
4379         case GNILND_MSG_GET_REQ_REV:
4380         case GNILND_MSG_PUT_REQ:
4381                 rc = lnet_parse(net->gnn_ni, &msg->gnm_u.putreq.gnprm_hdr,
4382                                 msg->gnm_srcnid, rx, 1);
4383                 repost = rc < 0;
4384                 break;
4385         case GNILND_MSG_GET_NAK_REV:
4386                 tx = kgnilnd_match_reply_either(conn, GNILND_MSG_GET_REQ_REV, GNILND_MSG_GET_ACK_REV,
4387                                         msg->gnm_u.completion.gncm_cookie);
4388                 if (tx == NULL)
4389                         break;
4390
4391                 kgnilnd_complete_tx(tx, msg->gnm_u.completion.gncm_retval);
4392                 break;
4393         case GNILND_MSG_PUT_NAK:
4394                 tx = kgnilnd_match_reply_either(conn, GNILND_MSG_PUT_REQ, GNILND_MSG_PUT_ACK,
4395                                         msg->gnm_u.completion.gncm_cookie);
4396                 if (tx == NULL)
4397                         break;
4398
4399                 kgnilnd_complete_tx(tx, msg->gnm_u.completion.gncm_retval);
4400                 break;
4401         case GNILND_MSG_PUT_ACK:
4402                 tx = kgnilnd_match_reply(conn, GNILND_MSG_PUT_REQ,
4403                                         msg->gnm_u.putack.gnpam_src_cookie);
4404                 if (tx == NULL)
4405                         break;
4406
4407                 /* store putack data for later: deferred rdma or re-try */
4408                 tx->tx_putinfo = msg->gnm_u.putack;
4409
4410                 saw_complete = 0;
4411                 spin_lock(&tx->tx_conn->gnc_list_lock);
4412
4413                 GNITX_ASSERTF(tx, tx->tx_state & GNILND_TX_WAITING_REPLY,
4414                         "not waiting for reply", NULL);
4415
4416                 tx->tx_state &= ~GNILND_TX_WAITING_REPLY;
4417
4418                 if (likely(!(tx->tx_state & GNILND_TX_WAITING_COMPLETION))) {
4419                         kgnilnd_tx_del_state_locked(tx, NULL, conn, GNILND_TX_ALLOCD);
4420                         /* sample under lock as follow on steps require gnc_list_lock
4421                          * - or call kgnilnd_tx_done which requires no locks held over
4422                          *   call to lnet_finalize */
4423                         saw_complete = 1;
4424                 } else {
4425                         /* cannot launch rdma if still waiting for fma-msg completion */
4426                         CDEBUG(D_NET, "tx 0x%p type 0x%02x will need to "
4427                                        "wait for SMSG completion\n", tx, tx->tx_msg.gnm_type);
4428                         tx->tx_state |= GNILND_TX_PENDING_RDMA;
4429                 }
4430                 spin_unlock(&tx->tx_conn->gnc_list_lock);
4431
4432                 if (saw_complete) {
4433                         rc = kgnilnd_send_mapped_tx(tx, 0);
4434                         if (rc < 0)
4435                                 kgnilnd_tx_done(tx, rc);
4436                 }
4437                 break;
4438         case GNILND_MSG_GET_ACK_REV:
4439                 tx = kgnilnd_match_reply(conn, GNILND_MSG_GET_REQ_REV,
4440                                         msg->gnm_u.putack.gnpam_src_cookie);
4441                 if (tx == NULL)
4442                         break;
4443
4444                 /* store putack data for later: deferred rdma or re-try */
4445                 tx->tx_putinfo = msg->gnm_u.putack;
4446                 saw_complete = 0;
4447                 spin_lock(&tx->tx_conn->gnc_list_lock);
4448
4449                 GNITX_ASSERTF(tx, tx->tx_state & GNILND_TX_WAITING_REPLY,
4450                         "not waiting for reply", NULL);
4451
4452                 tx->tx_state &= ~GNILND_TX_WAITING_REPLY;
4453
4454                 if (likely(!(tx->tx_state & GNILND_TX_WAITING_COMPLETION))) {
4455                         kgnilnd_tx_del_state_locked(tx, NULL, conn, GNILND_TX_ALLOCD);
4456                         /* sample under lock as follow on steps require gnc_list_lock
4457                          * - or call kgnilnd_tx_done which requires no locks held over
4458                          *   call to lnet_finalize */
4459                         saw_complete = 1;
4460                 } else {
4461                         /* cannot launch rdma if still waiting for fma-msg completion */
4462                         CDEBUG(D_NET, "tx 0x%p type 0x%02x will need to "
4463                                         "wait for SMSG completion\n", tx, tx->tx_msg.gnm_type);
4464                         tx->tx_state |= GNILND_TX_PENDING_RDMA;
4465                 }
4466                 spin_unlock(&tx->tx_conn->gnc_list_lock);
4467
4468                 if (saw_complete) {
4469                         rc = kgnilnd_send_mapped_tx(tx, 0);
4470                         if (rc < 0)
4471                                 kgnilnd_tx_done(tx, rc);
4472                 }
4473                 break;
4474         case GNILND_MSG_PUT_DONE:
4475                 tx = kgnilnd_match_reply(conn, GNILND_MSG_PUT_ACK,
4476                                         msg->gnm_u.completion.gncm_cookie);
4477                 if (tx == NULL)
4478                         break;
4479
4480                 GNITX_ASSERTF(tx, tx->tx_buftype == GNILND_BUF_PHYS_MAPPED ||
4481                                tx->tx_buftype == GNILND_BUF_VIRT_MAPPED,
4482                                "bad tx buftype %d", tx->tx_buftype);
4483
4484                 kgnilnd_finalize_rx_done(tx, msg);
4485                 break;
4486         case GNILND_MSG_PUT_REQ_REV:
4487         case GNILND_MSG_GET_REQ:
4488                 rc = lnet_parse(net->gnn_ni, &msg->gnm_u.get.gngm_hdr,
4489                                 msg->gnm_srcnid, rx, 1);
4490                 repost = rc < 0;
4491                 break;
4492
4493         case GNILND_MSG_GET_NAK:
4494                 tx = kgnilnd_match_reply(conn, GNILND_MSG_GET_REQ,
4495                                         msg->gnm_u.completion.gncm_cookie);
4496                 if (tx == NULL)
4497                         break;
4498
4499                 GNITX_ASSERTF(tx, tx->tx_buftype == GNILND_BUF_PHYS_MAPPED ||
4500                                tx->tx_buftype == GNILND_BUF_VIRT_MAPPED,
4501                                "bad tx buftype %d", tx->tx_buftype);
4502
4503                 kgnilnd_complete_tx(tx, msg->gnm_u.completion.gncm_retval);
4504                 break;
4505
4506         case GNILND_MSG_GET_DONE:
4507                 tx = kgnilnd_match_reply(conn, GNILND_MSG_GET_REQ,
4508                                         msg->gnm_u.completion.gncm_cookie);
4509                 if (tx == NULL)
4510                         break;
4511
4512                 GNITX_ASSERTF(tx, tx->tx_buftype == GNILND_BUF_PHYS_MAPPED ||
4513                                tx->tx_buftype == GNILND_BUF_VIRT_MAPPED,
4514                                "bad tx buftype %d", tx->tx_buftype);
4515
4516                 lnet_set_reply_msg_len(net->gnn_ni, tx->tx_lntmsg[1],
4517                                        msg->gnm_u.completion.gncm_retval);
4518
4519                 kgnilnd_finalize_rx_done(tx, msg);
4520                 break;
4521         case GNILND_MSG_GET_DONE_REV:
4522                 tx = kgnilnd_match_reply(conn, GNILND_MSG_GET_ACK_REV,
4523                                         msg->gnm_u.completion.gncm_cookie);
4524                 if (tx == NULL)
4525                         break;
4526
4527                 GNITX_ASSERTF(tx, tx->tx_buftype == GNILND_BUF_PHYS_MAPPED ||
4528                                 tx->tx_buftype == GNILND_BUF_VIRT_MAPPED,
4529                                 "bad tx buftype %d", tx->tx_buftype);
4530
4531                 kgnilnd_finalize_rx_done(tx, msg);
4532                 break;
4533
4534         case GNILND_MSG_PUT_DONE_REV:
4535                 tx = kgnilnd_match_reply(conn, GNILND_MSG_PUT_REQ_REV,
4536                                         msg->gnm_u.completion.gncm_cookie);
4537
4538                 if (tx == NULL)
4539                         break;
4540
4541                 GNITX_ASSERTF(tx, tx->tx_buftype == GNILND_BUF_PHYS_MAPPED ||
4542                                tx->tx_buftype == GNILND_BUF_VIRT_MAPPED,
4543                                "bad tx buftype %d", tx->tx_buftype);
4544
4545                 kgnilnd_finalize_rx_done(tx, msg);
4546                 break;
4547         case GNILND_MSG_PUT_NAK_REV:
4548                 tx = kgnilnd_match_reply(conn, GNILND_MSG_PUT_REQ_REV,
4549                                         msg->gnm_u.completion.gncm_cookie);
4550
4551                 if (tx == NULL)
4552                         break;
4553
4554                 GNITX_ASSERTF(tx, tx->tx_buftype == GNILND_BUF_PHYS_MAPPED ||
4555                                tx->tx_buftype == GNILND_BUF_VIRT_MAPPED,
4556                                 "bad tx buftype %d", tx->tx_buftype);
4557
4558                 kgnilnd_complete_tx(tx, msg->gnm_u.completion.gncm_retval);
4559                 break;
4560         }
4561
4562  out:
4563         if (rc < 0)                             /* protocol/comms error */
4564                 kgnilnd_close_conn(conn, rc);
4565
4566         if (repost && rx != NULL) {
4567                 kgnilnd_consume_rx(rx);
4568         }
4569
4570         /* we got an event so assume more there and call for reschedule */
4571         if (rc >= 0)
4572                 kgnilnd_schedule_conn(conn);
4573         EXIT;
4574 }
4575
4576 /* Do the failure injections that we need to affect conn processing in the following function.
4577  * When writing tests that use this function make sure to use a fail_loc with a fail mask.
4578  * If you dont you can cause the scheduler threads to spin on the conn without it leaving
4579  * process_conns.
4580  *
4581  * intent is used to signal the calling function whether or not the conn needs to be rescheduled.
4582  */
4583
4584 static inline int
4585 kgnilnd_check_conn_fail_loc(kgn_device_t *dev, kgn_conn_t *conn, int *intent)
4586 {
4587         int     rc = 0;
4588
4589         /* short circuit out when not set */
4590         if (likely(!cfs_fail_loc)) {
4591                 RETURN(rc);
4592         }
4593
4594         /* failure injection to test for stack reset clean ups */
4595         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_DROP_CLOSING)) {
4596                 /* we can't rely on busy loops being nice enough to get the
4597                  *  stack reset triggered - it'd just spin on this conn */
4598                 CFS_RACE(CFS_FAIL_GNI_DROP_CLOSING);
4599                 rc = 1;
4600                 *intent = 1;
4601                 GOTO(did_fail_loc, rc);
4602         }
4603
4604         if (conn->gnc_state == GNILND_CONN_DESTROY_EP) {
4605                 /* DESTROY_EP set in kgnilnd_conn_decref on gnc_refcount = 1 */
4606
4607                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_DROP_DESTROY_EP)) {
4608                         CFS_RACE(CFS_FAIL_GNI_DROP_DESTROY_EP);
4609                         rc = 1;
4610                         *intent = 1;
4611                         GOTO(did_fail_loc, rc);
4612                 }
4613         }
4614
4615         /* CFS_FAIL_GNI_FINISH_PURG2 is used to stop a connection from fully closing. This scheduler
4616          * will spin on the CFS_FAIL_TIMEOUT until the fail_loc is cleared at which time the connection
4617          * will be closed by kgnilnd_complete_closed_conn.
4618          */
4619         if ((conn->gnc_state == GNILND_CONN_CLOSED) && CFS_FAIL_CHECK(CFS_FAIL_GNI_FINISH_PURG2)) {
4620                 while (CFS_FAIL_TIMEOUT(CFS_FAIL_GNI_FINISH_PURG2, 1)) {};
4621                 rc = 1;
4622                 *intent = 1;
4623                 GOTO(did_fail_loc, rc);
4624         }
4625
4626         /* this one is a bit gross - we can't hold the mutex from process_conns
4627          * across a CFS_RACE here - it'd block the conn threads from doing an ep_bind
4628          * and moving onto finish_connect
4629          * so, we'll just set the rc - kgnilnd_process_conns will clear
4630          * found_work on a fail_loc, getting the scheduler thread to call schedule()
4631          * and effectively getting this thread to sleep */
4632         if ((conn->gnc_state == GNILND_CONN_CLOSED) && CFS_FAIL_CHECK(CFS_FAIL_GNI_FINISH_PURG)) {
4633                 rc = 1;
4634                 *intent = 1;
4635                 GOTO(did_fail_loc, rc);
4636         }
4637
4638 did_fail_loc:
4639         RETURN(rc);
4640 }
4641
4642 static inline void
4643 kgnilnd_send_conn_close(kgn_conn_t *conn)
4644 {
4645         kgn_tx_t        *tx;
4646
4647         /* we are closing the conn - we will try to send the CLOSE msg
4648          * but will not wait for anything else to flush */
4649
4650         /* send the close if not already done so or received one */
4651         if (!conn->gnc_close_sent && !conn->gnc_close_recvd) {
4652                 /* set close_sent regardless of the success of the
4653                  * CLOSE message. We are going to try once and then
4654                  * kick him out of the sandbox */
4655                 conn->gnc_close_sent = 1;
4656                 mb();
4657
4658                 /* EP might be null already if remote side initiated a new connection.
4659                  * kgnilnd_finish_connect destroys existing ep_handles before wiring up the new connection,
4660                  * so this check is here to make sure we dont attempt to send with a null ep_handle.
4661                  */
4662                 if (conn->gnc_ephandle != NULL) {
4663                         int rc = 0;
4664
4665                         tx = kgnilnd_new_tx_msg(GNILND_MSG_CLOSE, conn->gnc_peer->gnp_net->gnn_ni->ni_nid);
4666                         if (tx != NULL) {
4667                                 tx->tx_msg.gnm_u.completion.gncm_retval = conn->gnc_error;
4668                                 tx->tx_state = GNILND_TX_WAITING_COMPLETION;
4669                                 tx->tx_qtime = jiffies;
4670
4671                                 if (tx->tx_id.txe_idx == 0) {
4672                                         rc = kgnilnd_set_tx_id(tx, conn);
4673                                         if (rc != 0) {
4674                                                 kgnilnd_tx_done(tx, rc);
4675                                         }
4676                                 }
4677
4678                                 CDEBUG(D_NETTRACE, "sending close with errno %d\n",
4679                                                 conn->gnc_error);
4680
4681                                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_CLOSE_SEND)) {
4682                                         kgnilnd_tx_done(tx, -EAGAIN);
4683                                 } else if (!rc) {
4684                                         rc = kgnilnd_sendmsg(tx, NULL, 0, NULL, GNILND_TX_FMAQ);
4685                                         if (rc) {
4686                                                 /* It wasnt sent and we dont care. */
4687                                                 kgnilnd_tx_done(tx, rc);
4688                                         }
4689                                 }
4690
4691                         }
4692                 }
4693         }
4694
4695         /* When changing gnc_state we need to take the kgn_peer_conn_lock */
4696         write_lock(&kgnilnd_data.kgn_peer_conn_lock);
4697         conn->gnc_state = GNILND_CONN_CLOSED;
4698         write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
4699         /* mark this conn as CLOSED now that we processed it
4700          * do after TX, so we can use CLOSING in asserts */
4701
4702         mb();
4703
4704         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_RX_CLOSE_CLOSED)) {
4705                 /* simulate a RX CLOSE after the timeout but before
4706                  * the scheduler thread gets it */
4707                 conn->gnc_close_recvd = GNILND_CLOSE_INJECT2;
4708                 conn->gnc_peer_error = -ETIMEDOUT;
4709         }
4710         /* schedule to allow potential CLOSE and get the complete phase run */
4711         kgnilnd_schedule_conn(conn);
4712 }
4713
4714 int
4715 kgnilnd_process_mapped_tx(kgn_device_t *dev)
4716 {
4717         int             found_work = 0;
4718         int             rc = 0;
4719         kgn_tx_t        *tx;
4720         int              fast_remaps = GNILND_FAST_MAPPING_TRY;
4721         int             log_retrans, log_retrans_level;
4722         static int      last_map_version;
4723         ENTRY;
4724
4725         spin_lock(&dev->gnd_lock);
4726         if (list_empty(&dev->gnd_map_tx)) {
4727                 /* if the list is empty make sure we dont have a timer running */
4728                 del_singleshot_timer_sync(&dev->gnd_map_timer);
4729                 spin_unlock(&dev->gnd_lock);
4730                 RETURN(0);
4731         }
4732
4733         dev->gnd_sched_alive = jiffies;
4734
4735         /* we'll retry as fast as possible up to 25% of the limit, then we start
4736          * backing off until our map version changes - indicating we unmapped
4737          * something */
4738         tx = list_first_entry(&dev->gnd_map_tx, kgn_tx_t, tx_list);
4739         if (likely(dev->gnd_map_attempt == 0) ||
4740                 time_after_eq(jiffies, dev->gnd_next_map) ||
4741                 last_map_version != dev->gnd_map_version) {
4742
4743                 /* if this is our first attempt at mapping set last mapped to current
4744                  * jiffies so we can timeout our attempt correctly.
4745                  */
4746                 if (dev->gnd_map_attempt == 0)
4747                         dev->gnd_last_map = jiffies;
4748         } else {
4749                 GNIDBG_TX(D_NET, tx, "waiting for mapping event event to retry", NULL);
4750                 spin_unlock(&dev->gnd_lock);
4751                 RETURN(0);
4752         }
4753
4754         /* delete the previous timer if it exists */
4755         del_singleshot_timer_sync(&dev->gnd_map_timer);
4756         /* stash the last map version to let us know when a good one was seen */
4757         last_map_version = dev->gnd_map_version;
4758
4759         /* we need to to take the lock and continually refresh the head of the list as
4760          * kgnilnd_complete_closed_conn might be nuking stuff and we are cycling the lock
4761          * allowing them to squeeze in */
4762
4763         while (!list_empty(&dev->gnd_map_tx)) {
4764                 /* make sure we break out early on quiesce */
4765                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
4766                         /* always break with lock held - we unlock outside loop */
4767                         break;
4768                 }
4769
4770                 tx = list_first_entry(&dev->gnd_map_tx, kgn_tx_t, tx_list);
4771
4772                 kgnilnd_tx_del_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_ALLOCD);
4773                 found_work++;
4774
4775                 /* sample with lock held, serializing with kgnilnd_complete_closed_conn */
4776                 if (tx->tx_conn->gnc_state != GNILND_CONN_ESTABLISHED) {
4777                         /* if conn is dying, mark tx in tx_ref_table for
4778                          * kgnilnd_complete_closed_conn to finish up */
4779                         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_DYING, 1);
4780                         found_work++;
4781
4782                         /* tx was moved to DYING, get next */
4783                         continue;
4784                 }
4785
4786                 spin_unlock(&dev->gnd_lock);
4787                 rc = kgnilnd_send_mapped_tx(tx, 1);
4788
4789                 /* We made it! skip error handling.. */
4790                 if (rc >= 0) {
4791                         /* OK to continue on +ve errors as it won't get seen until
4792                          * this function is called again - we operate on a copy of the original
4793                          * list and not the live list */
4794                         spin_lock(&dev->gnd_lock);
4795                         /* reset map attempts back to zero we successfully
4796                          * mapped so we can reset our timers */
4797                         dev->gnd_map_attempt = 0;
4798                         continue;
4799                 } else if (rc == -EAGAIN) {
4800                         spin_lock(&dev->gnd_lock);
4801                         mod_timer(&dev->gnd_map_timer, dev->gnd_next_map);
4802                         spin_unlock(&dev->gnd_lock);
4803                         GOTO(get_out_mapped, rc);
4804                 } else if (rc != -ENOMEM) {
4805                         /* carp, failure we can't handle */
4806                         kgnilnd_tx_done(tx, rc);
4807                         spin_lock(&dev->gnd_lock);
4808                         /* reset map attempts back to zero we dont know what happened but it
4809                          * wasnt a failed mapping
4810                          */
4811                         dev->gnd_map_attempt = 0;
4812                         continue;
4813                 }
4814
4815                 /* time to handle the retry cases..  lock so we dont have 2 threads
4816                  * mucking with gnd_map_attempt, or gnd_next_map at the same time.
4817                  */
4818                 spin_lock(&dev->gnd_lock);
4819                 dev->gnd_map_attempt++;
4820                 if (dev->gnd_map_attempt < fast_remaps) {
4821                         /* do nothing we just want it to go as fast as possible.
4822                          * just set gnd_next_map to current jiffies so it will process
4823                          * as fast as possible.
4824                          */
4825                         dev->gnd_next_map = jiffies;
4826                 } else {
4827                         /* Retry based on GNILND_MAP_RETRY_RATE */
4828                         dev->gnd_next_map = jiffies + GNILND_MAP_RETRY_RATE;
4829                 }
4830
4831                 /* only log occasionally once we've retried fast_remaps */
4832                 log_retrans = (dev->gnd_map_attempt >= fast_remaps) &&
4833                               ((dev->gnd_map_attempt % fast_remaps) == 0);
4834                 log_retrans_level = log_retrans ? D_NETERROR : D_NET;
4835
4836                 /* make sure we are not off in the weeds with this tx */
4837                 if (time_after(jiffies, dev->gnd_last_map + GNILND_MAP_TIMEOUT)) {
4838                        GNIDBG_TX(D_NETERROR, tx,
4839                                "giving up on TX, too many retries", NULL);
4840                        spin_unlock(&dev->gnd_lock);
4841                        if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_REQ ||
4842                            tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ_REV) {
4843                                kgnilnd_nak_rdma(tx->tx_conn, tx->tx_msg.gnm_type,
4844                                                 -ENOMEM,
4845                                                 tx->tx_putinfo.gnpam_dst_cookie,
4846                                                 tx->tx_msg.gnm_srcnid);
4847                         } else {
4848                                 kgnilnd_nak_rdma(tx->tx_conn, tx->tx_msg.gnm_type,
4849                                                 -ENOMEM,
4850                                                 tx->tx_getinfo.gngm_cookie,
4851                                                 tx->tx_msg.gnm_srcnid);
4852                         }
4853                        kgnilnd_tx_done(tx, -ENOMEM);
4854                        GOTO(get_out_mapped, rc);
4855                 } else {
4856                        GNIDBG_TX(log_retrans_level, tx,
4857                                 "transient map failure #%d %d pages/%d bytes phys %u@%u "
4858                                 "virt %u@%llu "
4859                                 "nq_map %d mdd# %d/%d GART %ld",
4860                                 dev->gnd_map_attempt, tx->tx_phys_npages, tx->tx_nob,
4861                                 dev->gnd_map_nphys, dev->gnd_map_physnop * PAGE_SIZE,
4862                                 dev->gnd_map_nvirt, dev->gnd_map_virtnob,
4863                                 atomic_read(&dev->gnd_nq_map),
4864                                 atomic_read(&dev->gnd_n_mdd), atomic_read(&dev->gnd_n_mdd_held),
4865                                 atomic64_read(&dev->gnd_nbytes_map));
4866                 }
4867
4868                 /* we need to stop processing the rest of the list, so add it back in */
4869                 /* set timer to wake device when we need to schedule this tx */
4870                 mod_timer(&dev->gnd_map_timer, dev->gnd_next_map);
4871                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 0);
4872                 spin_unlock(&dev->gnd_lock);
4873                 GOTO(get_out_mapped, rc);
4874         }
4875         spin_unlock(&dev->gnd_lock);
4876 get_out_mapped:
4877         RETURN(found_work);
4878 }
4879
4880 int
4881 kgnilnd_process_conns(kgn_device_t *dev, unsigned long deadline)
4882 {
4883         int              found_work = 0;
4884         int              conn_sched;
4885         int              intent = 0;
4886         int              error_inject = 0;
4887         int              rc = 0;
4888         kgn_conn_t      *conn;
4889
4890         spin_lock(&dev->gnd_lock);
4891         while (!list_empty(&dev->gnd_ready_conns) && time_before(jiffies, deadline)) {
4892                 dev->gnd_sched_alive = jiffies;
4893                 error_inject = 0;
4894                 rc = 0;
4895
4896                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
4897                         /* break with lock held */
4898                         break;
4899                 }
4900
4901                 conn = list_first_entry(&dev->gnd_ready_conns, kgn_conn_t, gnc_schedlist);
4902                 list_del_init(&conn->gnc_schedlist);
4903                 spin_unlock(&dev->gnd_lock);
4904
4905                 conn_sched = xchg(&conn->gnc_scheduled, GNILND_CONN_PROCESS);
4906
4907                 LASSERTF(conn_sched != GNILND_CONN_IDLE &&
4908                          conn_sched != GNILND_CONN_PROCESS,
4909                          "conn %p on ready list but in bad state: %d\n",
4910                          conn, conn_sched);
4911
4912                 CDEBUG(D_INFO, "conn %p@%s for processing\n",
4913                         conn, kgnilnd_conn_state2str(conn));
4914
4915                 found_work++;
4916                 set_mb(conn->gnc_last_sched_do, jiffies);
4917
4918                 if (kgnilnd_check_conn_fail_loc(dev, conn, &intent)) {
4919
4920                         /* based on intent see if we should run again. */
4921                         rc = kgnilnd_schedule_process_conn(conn, intent);
4922                         error_inject = 1;
4923                         /* drop ref from gnd_ready_conns */
4924                         if (atomic_read(&conn->gnc_refcount) == 1 && rc != 1) {
4925                                 down_write(&dev->gnd_conn_sem);
4926                                 kgnilnd_conn_decref(conn);
4927                                 up_write(&dev->gnd_conn_sem);
4928                         } else if (rc != 1) {
4929                         kgnilnd_conn_decref(conn);
4930                         }
4931                         /* clear this so that scheduler thread doesn't spin */
4932                         found_work = 0;
4933                         /* break with lock held... */
4934                         spin_lock(&dev->gnd_lock);
4935                         break;
4936                 }
4937
4938                 if (unlikely(conn->gnc_state == GNILND_CONN_CLOSED)) {
4939                         down_write(&dev->gnd_conn_sem);
4940
4941                         /* CONN_CLOSED set in procces_fmaq when CLOSE is sent */
4942                         if (unlikely(atomic_read(&conn->gnc_tx_in_use))) {
4943                                 /* If there are tx's currently in use in another
4944                                  * thread we dont want to complete the close
4945                                  * yet. Cycle this conn back through
4946                                  * the scheduler. */
4947                                 kgnilnd_schedule_conn(conn);
4948                         } else {
4949                                 kgnilnd_complete_closed_conn(conn);
4950                         }
4951                         up_write(&dev->gnd_conn_sem);
4952                 } else if (unlikely(conn->gnc_state == GNILND_CONN_DESTROY_EP)) {
4953                         /* DESTROY_EP set in kgnilnd_conn_decref on gnc_refcount = 1 */
4954                         /* serialize SMSG CQs with ep_bind and smsg_release */
4955                         down_write(&dev->gnd_conn_sem);
4956                         kgnilnd_destroy_conn_ep(conn);
4957                         up_write(&dev->gnd_conn_sem);
4958                 } else if (unlikely(conn->gnc_state == GNILND_CONN_CLOSING)) {
4959                        /* if we need to do some CLOSE sending, etc done here do it */
4960                         down_write(&dev->gnd_conn_sem);
4961                         kgnilnd_send_conn_close(conn);
4962                         kgnilnd_check_fma_rx(conn);
4963                         up_write(&dev->gnd_conn_sem);
4964                 } else if (atomic_read(&conn->gnc_peer->gnp_dirty_eps) == 0) {
4965                         /* start moving traffic if the old conns are cleared out */
4966                         down_read(&dev->gnd_conn_sem);
4967                         kgnilnd_check_fma_rx(conn);
4968                         kgnilnd_process_fmaq(conn);
4969                         up_read(&dev->gnd_conn_sem);
4970                 }
4971
4972                 rc = kgnilnd_schedule_process_conn(conn, 0);
4973
4974                 /* drop ref from gnd_ready_conns */
4975                 if (atomic_read(&conn->gnc_refcount) == 1 && rc != 1) {
4976                         down_write(&dev->gnd_conn_sem);
4977                         kgnilnd_conn_decref(conn);
4978                         up_write(&dev->gnd_conn_sem);
4979                 } else if (rc != 1) {
4980                 kgnilnd_conn_decref(conn);
4981                 }
4982
4983                 /* check list again with lock held */
4984                 spin_lock(&dev->gnd_lock);
4985         }
4986
4987         /* If we are short circuiting due to timing we want to be scheduled
4988          * as soon as possible.
4989          */
4990         if (!list_empty(&dev->gnd_ready_conns) && !error_inject)
4991                 found_work++;
4992
4993         spin_unlock(&dev->gnd_lock);
4994
4995         RETURN(found_work);
4996 }
4997
4998 int
4999 kgnilnd_scheduler(void *arg)
5000 {
5001         int               threadno = (long)arg;
5002         kgn_device_t            *dev;
5003         int                     busy_loops = 0;
5004         unsigned long     deadline = 0;
5005         DEFINE_WAIT(wait);
5006
5007         dev = &kgnilnd_data.kgn_devices[(threadno + 1) % kgnilnd_data.kgn_ndevs];
5008
5009         cfs_block_allsigs();
5010
5011         /* all gnilnd threads need to run fairly urgently */
5012         set_user_nice(current, *kgnilnd_tunables.kgn_sched_nice);
5013         deadline = jiffies + cfs_time_seconds(*kgnilnd_tunables.kgn_sched_timeout);
5014         while (!kgnilnd_data.kgn_shutdown) {
5015                 int     found_work = 0;
5016                 /* Safe: kgn_shutdown only set when quiescent */
5017
5018                 /* to quiesce or to not quiesce, that is the question */
5019
5020                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
5021                         KGNILND_SPIN_QUIESCE;
5022                 }
5023
5024                 /* tracking for when thread goes AWOL */
5025                 dev->gnd_sched_alive = jiffies;
5026
5027                 CFS_FAIL_TIMEOUT(CFS_FAIL_GNI_SCHED_DEADLINE,
5028                         (*kgnilnd_tunables.kgn_sched_timeout + 1));
5029                 /* let folks know we are up and kicking
5030                  * - they can use this for latency savings, etc
5031                  * - only change if IRQ, if IDLE leave alone as that
5032                  *   schedule_device calls to put us back to IRQ */
5033                 (void)cmpxchg(&dev->gnd_ready, GNILND_DEV_IRQ, GNILND_DEV_LOOP);
5034
5035                 down_read(&dev->gnd_conn_sem);
5036                 /* always check these - they are super low cost  */
5037                 found_work += kgnilnd_check_fma_send_cq(dev);
5038                 found_work += kgnilnd_check_fma_rcv_cq(dev);
5039
5040                 /* rdma CQ doesn't care about eps */
5041                 found_work += kgnilnd_check_rdma_cq(dev);
5042
5043                 /* move some RDMA ? */
5044                 found_work += kgnilnd_process_rdmaq(dev);
5045
5046                 /* map some pending RDMA requests ? */
5047                 found_work += kgnilnd_process_mapped_tx(dev);
5048
5049                 /* the EP for a conn is not destroyed until all the references
5050                  * to it are gone, so these checks should be safe
5051                  * even if run in parallel with the CQ checking functions
5052                  * _AND_ a thread that processes the CLOSED->DONE
5053                  * transistion
5054                  * ...should.... */
5055
5056                 up_read(&dev->gnd_conn_sem);
5057
5058                 /* process all conns ready now */
5059                 found_work += kgnilnd_process_conns(dev, deadline);
5060
5061                 /* do an eager check to avoid the IRQ disabling in
5062                  * prepare_to_wait and friends */
5063
5064                 if (found_work &&
5065                    (busy_loops++ < *kgnilnd_tunables.kgn_loops) &&
5066                    time_before(jiffies, deadline)) {
5067                         found_work = 0;
5068                         if ((busy_loops % 10) == 0) {
5069                                 /* tickle heartbeat and watchdog to ensure our
5070                                  * piggishness doesn't turn into heartbeat failure */
5071                                 touch_nmi_watchdog();
5072                                 kgnilnd_hw_hb();
5073                         }
5074                         continue;
5075                 }
5076
5077                 /* if we got here, found_work was zero or busy_loops means we
5078                  * need to take a break. We'll clear gnd_ready but we'll check
5079                  * one last time if there is an IRQ that needs processing */
5080
5081                 prepare_to_wait(&dev->gnd_waitq, &wait, TASK_INTERRUPTIBLE);
5082
5083                 /* the first time this will go LOOP -> IDLE and let us do one final check
5084                  * during which we might get an IRQ, then IDLE->IDLE and schedule()
5085                  * - this might allow other threads to block us for a bit if they
5086                  *   try to get the mutex, but that is good as we'd need to wake
5087                  *   up soon to handle the CQ or other processing anyways */
5088
5089                 found_work += xchg(&dev->gnd_ready, GNILND_DEV_IDLE);
5090
5091                 if ((busy_loops >= *kgnilnd_tunables.kgn_loops) ||
5092                    time_after_eq(jiffies, deadline)) {
5093                         CDEBUG(D_INFO,
5094                                "yeilding: found_work %d busy_loops %d\n",
5095                                found_work, busy_loops);
5096                         busy_loops = 0;
5097                         /* use yield if we are bailing due to busy_loops
5098                          * - this will ensure we wake up soonish. This closes
5099                          * a race with kgnilnd_device_callback - where it'd
5100                          * not call wake_up() because gnd_ready == 1, but then
5101                          * we come down and schedule() because of busy_loops.
5102                          * We'd not be woken up until something poked our waitq
5103                          * again. yield() ensures we wake up without another
5104                          * waitq poke in that case */
5105                         atomic_inc(&dev->gnd_n_yield);
5106                         kgnilnd_data.kgn_last_condresched = jiffies;
5107                         yield();
5108                         CDEBUG(D_INFO, "awake after yeild\n");
5109                         deadline = jiffies + cfs_time_seconds(*kgnilnd_tunables.kgn_sched_timeout);
5110                 } else if (found_work == GNILND_DEV_IDLE) {
5111                         /* busy_loops is low and there is nothing to do,
5112                          * go to sleep and wait for a waitq poke */
5113                         CDEBUG(D_INFO,
5114                                "scheduling: found_work %d busy_loops %d\n",
5115                                found_work, busy_loops);
5116                         atomic_inc(&dev->gnd_n_schedule);
5117                         kgnilnd_data.kgn_last_scheduled = jiffies;
5118                         schedule();
5119                         CDEBUG(D_INFO, "awake after schedule\n");
5120                         deadline = jiffies + cfs_time_seconds(*kgnilnd_tunables.kgn_sched_timeout);
5121                 }
5122                 finish_wait(&dev->gnd_waitq, &wait);
5123         }
5124
5125         kgnilnd_thread_fini();
5126         return 0;
5127 }