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