Whamcloud - gitweb
LU-13004 gnilnd: remove support for GNILND_BUF_VIRT_*
[fs/lustre-release.git] / 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
909         if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK ||
910             tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
911                 atomic64_add(bytes, &dev->gnd_rdmaq_bytes_out);
912                 GNIDBG_TX(D_NETTRACE, tx, "rdma ++ %d to %lld",
913                           bytes, atomic64_read(&dev->gnd_rdmaq_bytes_out));
914         }
915
916         atomic_inc(&dev->gnd_n_mdd);
917         atomic64_add(bytes, &dev->gnd_nbytes_map);
918
919         /* clear retrans to prevent any SMSG goofiness as that code uses the same counter */
920         tx->tx_retrans = 0;
921
922         /* we only get here in the valid cases */
923         list_add_tail(&tx->tx_map_list, &dev->gnd_map_list);
924         dev->gnd_map_version++;
925         spin_unlock(&dev->gnd_map_lock);
926 }
927
928 void
929 kgnilnd_mem_del_map_list(kgn_device_t *dev, kgn_tx_t *tx)
930 {
931         int     bytes;
932
933         GNITX_ASSERTF(tx, !list_empty(&tx->tx_map_list),
934                 "not mapped!", NULL);
935         spin_lock(&dev->gnd_map_lock);
936
937         switch (tx->tx_buftype) {
938         default:
939                 GNIDBG_TX(D_EMERG, tx,
940                         "SOFTWARE BUG: invalid mapping %d", tx->tx_buftype);
941                 spin_unlock(&dev->gnd_map_lock);
942                 LBUG();
943                 break;
944
945         case GNILND_BUF_PHYS_UNMAPPED:
946                 bytes = tx->tx_phys_npages * PAGE_SIZE;
947                 dev->gnd_map_nphys--;
948                 dev->gnd_map_physnop -= tx->tx_phys_npages;
949                 break;
950         }
951
952         if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK ||
953             tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
954                 atomic64_sub(bytes, &dev->gnd_rdmaq_bytes_out);
955                 LASSERTF(atomic64_read(&dev->gnd_rdmaq_bytes_out) >= 0,
956                          "bytes_out negative! %ld\n", atomic64_read(&dev->gnd_rdmaq_bytes_out));
957                 GNIDBG_TX(D_NETTRACE, tx, "rdma -- %d to %lld",
958                           bytes, atomic64_read(&dev->gnd_rdmaq_bytes_out));
959         }
960
961         atomic_dec(&dev->gnd_n_mdd);
962         atomic64_sub(bytes, &dev->gnd_nbytes_map);
963
964         /* we only get here in the valid cases */
965         list_del_init(&tx->tx_map_list);
966         dev->gnd_map_version++;
967         spin_unlock(&dev->gnd_map_lock);
968 }
969
970 int
971 kgnilnd_map_buffer(kgn_tx_t *tx)
972 {
973         kgn_conn_t       *conn = tx->tx_conn;
974         kgn_device_t     *dev = conn->gnc_device;
975         __u32             flags = GNI_MEM_READWRITE;
976         gni_return_t      rrc;
977
978         /* The kgnilnd_mem_register(_segments) Gemini Driver functions can
979          * be called concurrently as there are internal locks that protect
980          * any data structures or HW resources. We just need to ensure
981          * that our concurrency doesn't result in the kgn_device_t
982          * getting nuked while we are in here */
983
984         LASSERTF(conn != NULL, "tx %p with NULL conn, someone forgot"
985                 " to set tx_conn before calling %s\n", tx, __FUNCTION__);
986
987         if (unlikely(CFS_FAIL_CHECK(CFS_FAIL_GNI_MAP_TX)))
988                 RETURN(-ENOMEM);
989
990         if (*kgnilnd_tunables.kgn_bte_relaxed_ordering) {
991                 flags |= GNI_MEM_RELAXED_PI_ORDERING;
992         }
993
994         switch (tx->tx_buftype) {
995         default:
996                 LBUG();
997
998         case GNILND_BUF_NONE:
999         case GNILND_BUF_IMMEDIATE:
1000         case GNILND_BUF_IMMEDIATE_KIOV:
1001         case GNILND_BUF_PHYS_MAPPED:
1002                 return 0;
1003
1004         case GNILND_BUF_PHYS_UNMAPPED:
1005                 GNITX_ASSERTF(tx, tx->tx_phys != NULL, "physical buffer not there!", NULL);
1006                 rrc = kgnilnd_mem_register_segments(dev->gnd_handle,
1007                         tx->tx_phys, tx->tx_phys_npages, NULL,
1008                         GNI_MEM_PHYS_SEGMENTS | flags,
1009                         &tx->tx_map_key);
1010                 /* could race with other uses of the map counts, but this is ok
1011                  * - this needs to turn into a non-fatal error soon to allow
1012                  *  GART resource, etc starvation handling */
1013                 if (rrc != GNI_RC_SUCCESS) {
1014                         GNIDBG_TX(D_NET, tx,
1015                                   "Can't map %d pages: dev %d phys %u pp %u",
1016                                 tx->tx_phys_npages, dev->gnd_id,
1017                                 dev->gnd_map_nphys, dev->gnd_map_physnop);
1018                         RETURN(rrc == GNI_RC_ERROR_RESOURCE ? -ENOMEM : -EINVAL);
1019                 }
1020
1021                 tx->tx_buftype = GNILND_BUF_PHYS_MAPPED;
1022                 kgnilnd_mem_add_map_list(dev, tx);
1023                 return 0;
1024         }
1025 }
1026
1027 void
1028 kgnilnd_add_purgatory_tx(kgn_tx_t *tx)
1029 {
1030         kgn_conn_t              *conn = tx->tx_conn;
1031         kgn_mdd_purgatory_t     *gmp;
1032
1033         LIBCFS_ALLOC(gmp, sizeof(*gmp));
1034         LASSERTF(gmp != NULL, "couldn't allocate MDD purgatory member;"
1035                 " asserting to avoid data corruption\n");
1036         if (tx->tx_buffer_copy)
1037                 gmp->gmp_map_key = tx->tx_buffer_copy_map_key;
1038         else
1039                 gmp->gmp_map_key = tx->tx_map_key;
1040
1041         atomic_inc(&conn->gnc_device->gnd_n_mdd_held);
1042
1043         /* ensure that we don't have a blank purgatory - indicating the
1044          * conn is not already on purgatory lists - we'd never recover these
1045          * MDD if that were the case */
1046         GNITX_ASSERTF(tx, conn->gnc_in_purgatory,
1047                 "conn 0x%p->%s with NULL purgatory",
1048                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
1049
1050         /* link 'er up! - only place we really need to lock for
1051          * concurrent access */
1052         spin_lock(&conn->gnc_list_lock);
1053         list_add_tail(&gmp->gmp_list, &conn->gnc_mdd_list);
1054         spin_unlock(&conn->gnc_list_lock);
1055 }
1056
1057 void
1058 kgnilnd_unmap_buffer(kgn_tx_t *tx, int error)
1059 {
1060         kgn_device_t     *dev;
1061         gni_return_t      rrc;
1062         int               hold_timeout = 0;
1063
1064         /* code below relies on +1 relationship ... */
1065         BUILD_BUG_ON(GNILND_BUF_PHYS_MAPPED !=
1066                      (GNILND_BUF_PHYS_UNMAPPED + 1));
1067
1068         switch (tx->tx_buftype) {
1069         default:
1070                 LBUG();
1071
1072         case GNILND_BUF_NONE:
1073         case GNILND_BUF_IMMEDIATE:
1074         case GNILND_BUF_PHYS_UNMAPPED:
1075                 break;
1076         case GNILND_BUF_IMMEDIATE_KIOV:
1077                 if (tx->tx_phys != NULL) {
1078                         vunmap(tx->tx_phys);
1079                 } else if (tx->tx_phys == NULL && tx->tx_buffer != NULL) {
1080                         kunmap(tx->tx_imm_pages[0]);
1081                 }
1082                 /* clear to prevent kgnilnd_free_tx from thinking
1083                  * this is a RDMA descriptor */
1084                 tx->tx_phys = NULL;
1085                 break;
1086
1087         case GNILND_BUF_PHYS_MAPPED:
1088                 LASSERT(tx->tx_conn != NULL);
1089
1090                 dev = tx->tx_conn->gnc_device;
1091
1092                 /* only want to hold if we are closing conn without
1093                  * verified peer notification  - the theory is that
1094                  * a TX error can be communicated in all other cases */
1095                 if (tx->tx_conn->gnc_state != GNILND_CONN_ESTABLISHED &&
1096                     error != -GNILND_NOPURG &&
1097                     kgnilnd_check_purgatory_conn(tx->tx_conn)) {
1098                         kgnilnd_add_purgatory_tx(tx);
1099
1100                         /* The timeout we give to kgni is a deadman stop only.
1101                          *  we are setting high to ensure we don't have the kgni timer
1102                          *  fire before ours fires _and_ is handled */
1103                         hold_timeout = GNILND_TIMEOUT2DEADMAN;
1104
1105                         GNIDBG_TX(D_NET, tx,
1106                                  "dev %p delaying MDD release for %dms key %#llx.%#llx",
1107                                  tx->tx_conn->gnc_device, hold_timeout,
1108                                  tx->tx_map_key.qword1, tx->tx_map_key.qword2);
1109                 }
1110                 if (tx->tx_buffer_copy != NULL) {
1111                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_buffer_copy_map_key, hold_timeout);
1112                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1113                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_map_key, 0);
1114                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1115                 } else {
1116                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_map_key, hold_timeout);
1117                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1118                 }
1119
1120                 tx->tx_buftype--;
1121                 kgnilnd_mem_del_map_list(dev, tx);
1122                 break;
1123         }
1124 }
1125
1126 void
1127 kgnilnd_tx_done(kgn_tx_t *tx, int completion)
1128 {
1129         struct lnet_msg      *lntmsg0, *lntmsg1;
1130         int             status0, status1;
1131         struct lnet_ni       *ni = NULL;
1132         kgn_conn_t      *conn = tx->tx_conn;
1133
1134         LASSERT(!in_interrupt());
1135
1136         lntmsg0 = tx->tx_lntmsg[0]; tx->tx_lntmsg[0] = NULL;
1137         lntmsg1 = tx->tx_lntmsg[1]; tx->tx_lntmsg[1] = NULL;
1138
1139         if (completion &&
1140             !(tx->tx_state & GNILND_TX_QUIET_ERROR) &&
1141             !kgnilnd_conn_clean_errno(completion)) {
1142                 GNIDBG_TOMSG(D_NETERROR, &tx->tx_msg,
1143                        "error %d on tx 0x%p->%s id %u/%d state %s age %ds",
1144                        completion, tx, conn ?
1145                        libcfs_nid2str(conn->gnc_peer->gnp_nid) : "<?>",
1146                        tx->tx_id.txe_smsg_id, tx->tx_id.txe_idx,
1147                        kgnilnd_tx_state2str(tx->tx_list_state),
1148                        cfs_duration_sec((unsigned long)jiffies - tx->tx_qtime));
1149         }
1150
1151         /* The error codes determine if we hold onto the MDD */
1152         kgnilnd_unmap_buffer(tx, completion);
1153
1154         /* we have to deliver a reply on lntmsg[1] for the GET, so make sure
1155          * we play nice with the error codes to avoid delivering a failed
1156          * REQUEST and then a REPLY event as well */
1157
1158         /* return -EIO to lnet - it is the magic value for failed sends */
1159         if (tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
1160                 status0 = 0;
1161                 status1 = completion;
1162         } else {
1163                 status0 = status1 = completion;
1164         }
1165
1166         tx->tx_buftype = GNILND_BUF_NONE;
1167         tx->tx_msg.gnm_type = GNILND_MSG_NONE;
1168
1169         /* lnet_finalize doesn't do anything with the *ni, so ok for us to
1170          * set NULL when we are a tx without a conn */
1171         if (conn != NULL) {
1172                 ni = conn->gnc_peer->gnp_net->gnn_ni;
1173
1174                 spin_lock(&conn->gnc_tx_lock);
1175
1176                 LASSERTF(test_and_clear_bit(tx->tx_id.txe_idx,
1177                         (volatile unsigned long *)&conn->gnc_tx_bits),
1178                         "conn %p tx %p bit %d already cleared\n",
1179                         conn, tx, tx->tx_id.txe_idx);
1180
1181                 LASSERTF(conn->gnc_tx_ref_table[tx->tx_id.txe_idx] != NULL,
1182                          "msg_id %d already NULL\n", tx->tx_id.txe_idx);
1183
1184                 conn->gnc_tx_ref_table[tx->tx_id.txe_idx] = NULL;
1185                 spin_unlock(&conn->gnc_tx_lock);
1186         }
1187
1188         kgnilnd_free_tx(tx);
1189
1190         /* finalize AFTER freeing lnet msgs */
1191
1192         /* warning - we should hold no locks here - calling lnet_finalize
1193          * could free up lnet credits, resulting in a call chain back into
1194          * the LND via kgnilnd_send and friends */
1195
1196         lnet_finalize(lntmsg0, status0);
1197
1198         if (lntmsg1 != NULL) {
1199                 lnet_finalize(lntmsg1, status1);
1200         }
1201 }
1202
1203 void
1204 kgnilnd_txlist_done(struct list_head *txlist, int error)
1205 {
1206         kgn_tx_t        *tx, *txn;
1207         int              err_printed = 0;
1208
1209         if (list_empty(txlist))
1210                 return;
1211
1212         list_for_each_entry_safe(tx, txn, txlist, tx_list) {
1213                 /* only print the first error */
1214                 if (err_printed)
1215                         tx->tx_state |= GNILND_TX_QUIET_ERROR;
1216                 list_del_init(&tx->tx_list);
1217                 kgnilnd_tx_done(tx, error);
1218                 err_printed++;
1219         }
1220 }
1221 int
1222 kgnilnd_set_tx_id(kgn_tx_t *tx, kgn_conn_t *conn)
1223 {
1224         int     id;
1225
1226         spin_lock(&conn->gnc_tx_lock);
1227
1228         /* ID zero is NOT ALLOWED!!! */
1229
1230 search_again:
1231         id = find_next_zero_bit((unsigned long *)&conn->gnc_tx_bits,
1232                                  GNILND_MAX_MSG_ID, conn->gnc_next_tx);
1233         if (id == GNILND_MAX_MSG_ID) {
1234                 if (conn->gnc_next_tx != 1) {
1235                         /* we only searched from next_tx to end and didn't find
1236                          * one, so search again from start */
1237                         conn->gnc_next_tx = 1;
1238                         goto search_again;
1239                 }
1240                 /* couldn't find one! */
1241                 spin_unlock(&conn->gnc_tx_lock);
1242                 return -E2BIG;
1243         }
1244
1245         /* bump next_tx to prevent immediate reuse */
1246         conn->gnc_next_tx = id + 1;
1247
1248         set_bit(id, (volatile unsigned long *)&conn->gnc_tx_bits);
1249         LASSERTF(conn->gnc_tx_ref_table[id] == NULL,
1250                  "tx 0x%p already at id %d\n",
1251                  conn->gnc_tx_ref_table[id], id);
1252
1253         /* delay these until we have a valid ID - prevents bad clear of the bit
1254          * in kgnilnd_tx_done */
1255         tx->tx_conn = conn;
1256         tx->tx_id.txe_cqid = conn->gnc_cqid;
1257
1258         tx->tx_id.txe_idx = id;
1259         conn->gnc_tx_ref_table[id] = tx;
1260
1261         /* Using jiffies to help differentiate against TX reuse - with
1262          * the usual minimum of a 250HZ clock, we wrap jiffies on the same TX
1263          * if we are sending to the same node faster than 256000/sec.
1264          * To help guard against this, we OR in the tx_seq - that is 32 bits */
1265
1266         tx->tx_id.txe_chips = (__u32)(jiffies | atomic_read(&conn->gnc_tx_seq));
1267
1268         GNIDBG_TX(D_NET, tx, "set cookie/id/bits", NULL);
1269
1270         spin_unlock(&conn->gnc_tx_lock);
1271         return 0;
1272 }
1273
1274 static inline void
1275 kgnilnd_tx_log_retrans(kgn_conn_t *conn, kgn_tx_t *tx)
1276 {
1277         int             log_retrans;
1278
1279         log_retrans = ((tx->tx_retrans < 25) || ((tx->tx_retrans % 25) == 0));
1280
1281         /* we don't care about TX timeout - it could be that the network is slower
1282          * or throttled. We'll keep retranmitting - so if the network is so slow
1283          * that we fill up our mailbox, we'll keep trying to resend that msg
1284          * until we exceed the max_retrans _or_ gnc_last_rx expires, indicating
1285          * that he hasn't send us any traffic in return */
1286         
1287         /* some reasonable throttling of the debug message */
1288         if (log_retrans) {
1289                 unsigned long now = jiffies;
1290                 /* XXX Nic: Mystical TX debug here... */
1291                 /* We expect retransmissions so only log when D_NET is enabled */
1292                 GNIDBG_SMSG_CREDS(D_NET, conn);
1293                 GNIDBG_TOMSG(D_NET, &tx->tx_msg,
1294                         "NOT_DONE on conn 0x%p->%s id %x retrans %d wait %dus"
1295                         " last_msg %uus/%uus last_cq %uus/%uus",
1296                         conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
1297                         tx->tx_id, tx->tx_retrans,
1298                         jiffies_to_usecs(now - tx->tx_cred_wait),
1299                         jiffies_to_usecs(now - conn->gnc_last_tx),
1300                         jiffies_to_usecs(now - conn->gnc_last_rx),
1301                         jiffies_to_usecs(now - conn->gnc_last_tx_cq),
1302                         jiffies_to_usecs(now - conn->gnc_last_rx_cq));
1303         }
1304 }
1305
1306 /* caller must be holding gnd_cq_mutex and not unlock it afterwards, as we need to drop it
1307  * to avoid bad ordering with state_lock */
1308
1309 static inline int
1310 kgnilnd_sendmsg_nolock(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1311                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1312 {
1313         kgn_conn_t      *conn = tx->tx_conn;
1314         kgn_msg_t       *msg = &tx->tx_msg;
1315         gni_return_t     rrc;
1316         unsigned long    newest_last_rx, timeout;
1317         unsigned long    now;
1318
1319         LASSERTF((msg->gnm_type == GNILND_MSG_IMMEDIATE) ?
1320                 immediatenob <= *kgnilnd_tunables.kgn_max_immediate :
1321                 immediatenob == 0,
1322                 "msg 0x%p type %d wrong payload size %d\n",
1323                 msg, msg->gnm_type, immediatenob);
1324
1325         /* make sure we catch all the cases where we'd send on a dirty old mbox
1326          * but allow case for sending CLOSE. Since this check is within the CQ
1327          * mutex barrier and the close message is only sent through
1328          * kgnilnd_send_conn_close the last message out the door will be the
1329          * close message.
1330          */
1331         if (atomic_read(&conn->gnc_peer->gnp_dirty_eps) != 0 && msg->gnm_type != GNILND_MSG_CLOSE) {
1332                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1333                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1334                 /* Return -ETIME, we are closing the connection already so we dont want to
1335                  * have this tx hit the wire. The tx will be killed by the calling function.
1336                  * Once the EP is marked dirty the close message will be the last
1337                  * thing to hit the wire */
1338                 return -ETIME;
1339         }
1340
1341         now = jiffies;
1342         timeout = cfs_time_seconds(conn->gnc_timeout);
1343
1344         newest_last_rx = GNILND_LASTRX(conn);
1345
1346         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SEND_TIMEOUT)) {
1347                 now = now + (GNILND_TIMEOUTRX(timeout) * 2);
1348         }
1349
1350         if (time_after_eq(now, newest_last_rx + GNILND_TIMEOUTRX(timeout))) {
1351                 GNIDBG_CONN(D_NETERROR|D_CONSOLE, conn,
1352                             "Cant send to %s after timeout lapse of %lu; TO %lu\n",
1353                 libcfs_nid2str(conn->gnc_peer->gnp_nid),
1354                 cfs_duration_sec(now - newest_last_rx),
1355                 cfs_duration_sec(GNILND_TIMEOUTRX(timeout)));
1356                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1357                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1358                 return -ETIME;
1359         }
1360
1361         GNITX_ASSERTF(tx, (conn != NULL) && (tx->tx_id.txe_idx != 0), "tx id unset!", NULL);
1362         /* msg->gnm_srcnid is set when the message is initialized by whatever function is
1363          * creating the message this allows the message to contain the correct LNET NID/NET needed
1364          * instead of the one that the peer/conn uses for sending the data.
1365          */
1366         msg->gnm_connstamp = conn->gnc_my_connstamp;
1367         msg->gnm_payload_len = immediatenob;
1368         msg->gnm_seq = atomic_read(&conn->gnc_tx_seq);
1369
1370         /* always init here - kgn_checksum is a /sys module tunable
1371          * and can be flipped at any point, even between msg init and sending */
1372         msg->gnm_cksum = 0;
1373         if (*kgnilnd_tunables.kgn_checksum) {
1374                 /* We must set here and not in kgnilnd_init_msg,
1375                  * we could resend this msg many times
1376                  * (NOT_DONE from gni_smsg_send below) and wouldn't pass
1377                  * through init_msg again */
1378                 msg->gnm_cksum = kgnilnd_cksum(msg, sizeof(kgn_msg_t));
1379                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SMSG_CKSUM1)) {
1380                         msg->gnm_cksum += 0xf00f;
1381                 }
1382         }
1383
1384         GNIDBG_TOMSG(D_NET, msg, "tx 0x%p conn 0x%p->%s sending SMSG sz %u id %x/%d [%p for %u]",
1385                tx, conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
1386                sizeof(kgn_msg_t), tx->tx_id.txe_smsg_id,
1387                tx->tx_id.txe_idx, immediate, immediatenob);
1388
1389         if (unlikely(tx->tx_state & GNILND_TX_FAIL_SMSG)) {
1390                 rrc = cfs_fail_val ? cfs_fail_val : GNI_RC_NOT_DONE;
1391         } else {
1392                 rrc = kgnilnd_smsg_send(conn->gnc_ephandle,
1393                                         msg, sizeof(*msg), immediate,
1394                                         immediatenob,
1395                                         tx->tx_id.txe_smsg_id);
1396         }
1397
1398         switch (rrc) {
1399         case GNI_RC_SUCCESS:
1400                 atomic_inc(&conn->gnc_tx_seq);
1401                 conn->gnc_last_tx = jiffies;
1402                 /* no locking here as LIVE isn't a list */
1403                 kgnilnd_tx_add_state_locked(tx, NULL, conn, GNILND_TX_LIVE_FMAQ, 1);
1404
1405                 /* this needs to be checked under lock as it might be freed from a completion
1406                  * event.
1407                  */
1408                 if (msg->gnm_type == GNILND_MSG_NOOP) {
1409                         set_mb(conn->gnc_last_noop_sent, jiffies);
1410                 }
1411
1412                 /* serialize with seeing CQ events for completion on this, as well as
1413                  * tx_seq */
1414                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1415                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1416
1417                 atomic_inc(&conn->gnc_device->gnd_short_ntx);
1418                 atomic64_add(immediatenob, &conn->gnc_device->gnd_short_txbytes);
1419                 kgnilnd_peer_alive(conn->gnc_peer);
1420                 GNIDBG_SMSG_CREDS(D_NET, conn);
1421                 return 0;
1422
1423         case GNI_RC_NOT_DONE:
1424                 /* Jshimek: We can get GNI_RC_NOT_DONE for 3 reasons currently
1425                  * 1: out of mbox credits
1426                  * 2: out of mbox payload credits
1427                  * 3: On Aries out of dla credits
1428                  */
1429                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1430                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1431                 /* We'll handle this error inline - makes the calling logic much more
1432                  * clean */
1433
1434                 /* If no lock, caller doesn't want us to retry */
1435                 if (state_lock == NULL) {
1436                         return -EAGAIN;
1437                 }
1438
1439                 /* I need kgni credits to send this.  Replace tx at the head of the
1440                  * fmaq and I'll get rescheduled when credits appear. Reset the tx_state
1441                  * and bump retrans counts since we are requeueing the tx.
1442                  */
1443                 tx->tx_state = 0;
1444                 tx->tx_retrans++;
1445                 conn->gnc_tx_retrans++;
1446
1447                 kgnilnd_tx_log_retrans(conn, tx);
1448                 /* add to head of list for the state and retries */
1449                 spin_lock(state_lock);
1450                 kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, state, 0);
1451                 spin_unlock(state_lock);
1452
1453                 /* We only reschedule for a certain number of retries, then
1454                  * we will wait for the CQ events indicating a release of SMSG
1455                  * credits */
1456                 if (tx->tx_retrans < *kgnilnd_tunables.kgn_max_retransmits) {
1457                         kgnilnd_schedule_conn(conn);
1458                         return 0;
1459                 } else {
1460                         /* CQ event coming in signifies either TX completed or
1461                          * RX receive. Either of these *could* free up credits
1462                          * in the SMSG mbox and we should try sending again */
1463                         GNIDBG_TX(D_NET, tx, "waiting for CQID %u event to resend",
1464                                  tx->tx_conn->gnc_cqid);
1465                         kgnilnd_schedule_delay_conn(conn);
1466                         /* use +ve return code to let upper layers know they
1467                          * should stop looping on sends */
1468                         return EAGAIN;
1469                 }
1470         default:
1471                 /* handle bad retcode gracefully */
1472                 kgnilnd_conn_mutex_unlock(&conn->gnc_smsg_mutex);
1473                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1474                 return -EIO;
1475         }
1476 }
1477
1478 /* kgnilnd_sendmsg has hard wait on gnd_cq_mutex */
1479 static inline int
1480 kgnilnd_sendmsg(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1481                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1482 {
1483         kgn_device_t    *dev = tx->tx_conn->gnc_device;
1484         unsigned long    timestamp;
1485         int              rc;
1486
1487         timestamp = jiffies;
1488         kgnilnd_gl_mutex_lock(&dev->gnd_cq_mutex);
1489         kgnilnd_conn_mutex_lock(&tx->tx_conn->gnc_smsg_mutex);
1490         /* delay in jiffies - we are really concerned only with things that
1491          * result in a schedule() or really holding this off for long times .
1492          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1493         dev->gnd_mutex_delay += (long) jiffies - timestamp;
1494
1495         rc = kgnilnd_sendmsg_nolock(tx, immediate, immediatenob, state_lock, state);
1496
1497         RETURN(rc);
1498 }
1499
1500
1501 /* returns -EAGAIN for lock miss, anything else < 0 is hard error, >=0 for success */
1502 static inline int
1503 kgnilnd_sendmsg_trylock(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1504                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1505 {
1506         kgn_conn_t      *conn = tx->tx_conn;
1507         kgn_device_t    *dev = conn->gnc_device;
1508         unsigned long    timestamp;
1509         int              rc;
1510
1511         timestamp = jiffies;
1512
1513         /* technically we are doing bad things with the read_lock on the peer_conn
1514          * table, but we shouldn't be sleeping inside here - and we don't sleep/block
1515          * for the mutex. I bet lockdep is gonna flag this one though... */
1516
1517         /* there are a few cases where we don't want the immediate send - like
1518          * when we are in the scheduler thread and it'd harm the latency of
1519          * getting messages up to LNet */
1520
1521         /* rmb for gnd_ready */
1522         smp_rmb();
1523         if (conn->gnc_device->gnd_ready == GNILND_DEV_LOOP) {
1524                 rc = 0;
1525                 atomic_inc(&conn->gnc_device->gnd_fast_block);
1526         } else if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
1527                /* dont hit HW during quiesce */
1528                 rc = 0;
1529         } else if (unlikely(atomic_read(&conn->gnc_peer->gnp_dirty_eps))) {
1530                /* dont hit HW if stale EPs and conns left to close */
1531                 rc = 0;
1532         } else {
1533                 atomic_inc(&conn->gnc_device->gnd_fast_try);
1534                 rc = kgnilnd_trylock(&conn->gnc_device->gnd_cq_mutex,
1535                                      &conn->gnc_smsg_mutex);
1536         }
1537         if (!rc) {
1538                 rc = -EAGAIN;
1539         } else {
1540                 /* we got the mutex and weren't blocked */
1541
1542                 /* delay in jiffies - we are really concerned only with things that
1543                  * result in a schedule() or really holding this off for long times .
1544                  * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1545                 dev->gnd_mutex_delay += (long) jiffies - timestamp;
1546
1547                 atomic_inc(&conn->gnc_device->gnd_fast_ok);
1548                 tx->tx_qtime = jiffies;
1549                 tx->tx_state = GNILND_TX_WAITING_COMPLETION;
1550                 rc = kgnilnd_sendmsg_nolock(tx, tx->tx_buffer, tx->tx_nob, &conn->gnc_list_lock, GNILND_TX_FMAQ);
1551                 /* _nolock unlocks the mutex for us */
1552         }
1553
1554         RETURN(rc);
1555 }
1556
1557 /* lets us know if we can push this RDMA through now */
1558 static int
1559 kgnilnd_auth_rdma_bytes(kgn_device_t *dev, kgn_tx_t *tx)
1560 {
1561         long    bytes_left;
1562
1563         bytes_left = atomic64_sub_return(tx->tx_nob, &dev->gnd_rdmaq_bytes_ok);
1564
1565         if (bytes_left < 0) {
1566                 atomic64_add(tx->tx_nob, &dev->gnd_rdmaq_bytes_ok);
1567                 atomic_inc(&dev->gnd_rdmaq_nstalls);
1568                 smp_wmb();
1569
1570                 CDEBUG(D_NET, "no bytes to send, turning on timer for %lu\n",
1571                        dev->gnd_rdmaq_deadline);
1572                 mod_timer(&dev->gnd_rdmaq_timer, dev->gnd_rdmaq_deadline);
1573                 /* we never del this timer - at worst it schedules us.. */
1574                 return -EAGAIN;
1575         } else {
1576                 return 0;
1577         }
1578 }
1579
1580 /* this adds a TX to the queue pending throttling authorization before
1581  * we allow our remote peer to launch a PUT at us */
1582 void
1583 kgnilnd_queue_rdma(kgn_conn_t *conn, kgn_tx_t *tx)
1584 {
1585         int     rc;
1586
1587         /* we cannot go into send_mapped_tx from here as we are holding locks
1588          * and mem registration might end up allocating memory in kgni.
1589          * That said, we'll push this as far as we can into the queue process */
1590         rc = kgnilnd_auth_rdma_bytes(conn->gnc_device, tx);
1591
1592         if (rc < 0) {
1593                 spin_lock(&conn->gnc_device->gnd_rdmaq_lock);
1594                 kgnilnd_tx_add_state_locked(tx, NULL, conn, GNILND_TX_RDMAQ, 0);
1595                 /* lets us know how delayed RDMA is */
1596                 tx->tx_qtime = jiffies;
1597                 spin_unlock(&conn->gnc_device->gnd_rdmaq_lock);
1598         } else {
1599                 /* we have RDMA authorized, now it just needs a MDD and to hit the wire */
1600                 spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
1601                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 0);
1602                 /* lets us know how delayed mapping is */
1603                 tx->tx_qtime = jiffies;
1604                 spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
1605         }
1606
1607         /* make sure we wake up sched to run this */
1608         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
1609 }
1610
1611 /* push TX through state machine */
1612 void
1613 kgnilnd_queue_tx(kgn_conn_t *conn, kgn_tx_t *tx)
1614 {
1615         int            rc = 0;
1616         int            add_tail = 1;
1617
1618         /* set the tx_id here, we delay it until we have an actual conn
1619          * to fiddle with
1620          * in some cases, the tx_id is already set to provide for things
1621          * like RDMA completion cookies, etc */
1622         if (tx->tx_id.txe_idx == 0) {
1623                 rc = kgnilnd_set_tx_id(tx, conn);
1624                 if (rc != 0) {
1625                         kgnilnd_tx_done(tx, rc);
1626                         return;
1627                 }
1628         }
1629
1630         CDEBUG(D_NET, "%s to conn %p for %s\n", kgnilnd_msgtype2str(tx->tx_msg.gnm_type),
1631                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
1632
1633         /* Only let NOOPs to be sent while fail loc is set, otherwise kill the tx.
1634          */
1635         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_ONLY_NOOP) && (tx->tx_msg.gnm_type != GNILND_MSG_NOOP)) {
1636                 kgnilnd_tx_done(tx, rc);
1637                 return;
1638         }
1639
1640         switch (tx->tx_msg.gnm_type) {
1641         case GNILND_MSG_PUT_ACK:
1642         case GNILND_MSG_GET_REQ:
1643         case GNILND_MSG_PUT_REQ_REV:
1644         case GNILND_MSG_GET_ACK_REV:
1645                 /* hijacking time! If this messages will authorize our peer to
1646                  * send his dirty little bytes in an RDMA, we need to get permission */
1647                 kgnilnd_queue_rdma(conn, tx);
1648                 break;
1649         case GNILND_MSG_IMMEDIATE:
1650                 /* try to send right now, can help reduce latency */
1651                 rc = kgnilnd_sendmsg_trylock(tx, tx->tx_buffer, tx->tx_nob, &conn->gnc_list_lock, GNILND_TX_FMAQ);
1652
1653                 if (rc >= 0) {
1654                         /* it was sent, break out of switch to avoid default case of queueing */
1655                         break;
1656                 }
1657                 /* needs to queue to try again, so fall through to default case */
1658         case GNILND_MSG_NOOP:
1659                 /* Just make sure this goes out first for this conn */
1660                 add_tail = 0;
1661                 /* fall through... */
1662         default:
1663                 spin_lock(&conn->gnc_list_lock);
1664                 kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, GNILND_TX_FMAQ, add_tail);
1665                 tx->tx_qtime = jiffies;
1666                 spin_unlock(&conn->gnc_list_lock);
1667                 kgnilnd_schedule_conn(conn);
1668         }
1669 }
1670
1671 void
1672 kgnilnd_launch_tx(kgn_tx_t *tx, kgn_net_t *net, struct lnet_process_id *target)
1673 {
1674         kgn_peer_t      *peer;
1675         kgn_peer_t      *new_peer = NULL;
1676         kgn_conn_t      *conn = NULL;
1677         int              rc;
1678         int              node_state;
1679
1680         ENTRY;
1681
1682         /* If I get here, I've committed to send, so I complete the tx with
1683          * failure on any problems */
1684
1685         GNITX_ASSERTF(tx, tx->tx_conn == NULL,
1686                       "tx already has connection %p", tx->tx_conn);
1687
1688         /* do all of the peer & conn searching in one swoop - this avoids
1689          * nastiness when dropping locks and needing to maintain a sane state
1690          * in the face of stack reset or something else nuking peers & conns */
1691
1692         /* I expect to find him, so only take a read lock */
1693         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
1694
1695         peer = kgnilnd_find_peer_locked(target->nid);
1696         if (peer != NULL) {
1697                 conn = kgnilnd_find_conn_locked(peer);
1698                 /* this could be NULL during quiesce */
1699                 if (conn != NULL)  {
1700                         /* Connection exists; queue message on it */
1701                         kgnilnd_queue_tx(conn, tx);
1702                         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1703                         RETURN_EXIT;
1704                 }
1705
1706                 /* don't create a connection if the peer is marked down */
1707                 if (peer->gnp_state != GNILND_PEER_UP) {
1708                         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1709                         rc = -ENETRESET;
1710                         GOTO(no_peer, rc);
1711                 }
1712         }
1713
1714         /* creating peer or conn; I'll need a write lock... */
1715         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1716
1717         CFS_RACE(CFS_FAIL_GNI_FIND_TARGET);
1718
1719         node_state = kgnilnd_get_node_state(LNET_NIDADDR(target->nid));
1720
1721         /* NB - this will not block during normal operations -
1722          * the only writer of this is in the startup/shutdown path. */
1723         rc = down_read_trylock(&kgnilnd_data.kgn_net_rw_sem);
1724         if (!rc) {
1725                 rc = -ESHUTDOWN;
1726                 GOTO(no_peer, rc);
1727         }
1728
1729         /* ignore previous peer entirely - we cycled the lock, so we
1730          * will create new peer and at worst drop it if peer is still
1731          * in the tables */
1732         rc = kgnilnd_create_peer_safe(&new_peer, target->nid, net, node_state);
1733         if (rc != 0) {
1734                 up_read(&kgnilnd_data.kgn_net_rw_sem);
1735                 GOTO(no_peer, rc);
1736         }
1737
1738         write_lock(&kgnilnd_data.kgn_peer_conn_lock);
1739         up_read(&kgnilnd_data.kgn_net_rw_sem);
1740
1741         /* search for peer again now that we have the lock
1742          * if we don't find it, add our new one to the list */
1743         kgnilnd_add_peer_locked(target->nid, new_peer, &peer);
1744
1745         /* don't create a connection if the peer is not up */
1746         if (peer->gnp_state != GNILND_PEER_UP) {
1747                 write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1748                 rc = -ENETRESET;
1749                 GOTO(no_peer, rc);
1750         }
1751
1752         conn = kgnilnd_find_or_create_conn_locked(peer);
1753
1754         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_DGRAM_DROP_TX)) {
1755                 write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1756                 GOTO(no_peer, rc);
1757         }
1758
1759         if (conn != NULL) {
1760                 /* oh hey, found a conn now... magical */
1761                 kgnilnd_queue_tx(conn, tx);
1762         } else {
1763                 /* no conn, must be trying to connect - so we queue for now */
1764                 tx->tx_qtime = jiffies;
1765                 kgnilnd_tx_add_state_locked(tx, peer, NULL, GNILND_TX_PEERQ, 1);
1766         }
1767         write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1768         RETURN_EXIT;
1769 no_peer:
1770         kgnilnd_tx_done(tx, rc);
1771         RETURN_EXIT;
1772 }
1773
1774 int
1775 kgnilnd_rdma(kgn_tx_t *tx, int type,
1776             kgn_rdma_desc_t *sink, unsigned int nob, __u64 cookie)
1777 {
1778         kgn_conn_t   *conn = tx->tx_conn;
1779         unsigned long timestamp;
1780         gni_post_type_t post_type;
1781         gni_return_t  rrc;
1782         int rc = 0;
1783         unsigned int desc_nob = nob;
1784         void *desc_buffer = tx->tx_buffer;
1785         gni_mem_handle_t desc_map_key = tx->tx_map_key;
1786         LASSERTF(kgnilnd_tx_mapped(tx),
1787                 "unmapped tx %p\n", tx);
1788         LASSERTF(conn != NULL,
1789                 "NULL conn on tx %p, naughty, naughty\n", tx);
1790         LASSERTF(nob <= sink->gnrd_nob,
1791                 "nob %u > sink->gnrd_nob %d (%p)\n",
1792                 nob, sink->gnrd_nob, sink);
1793         LASSERTF(nob <= tx->tx_nob,
1794                 "nob %d > tx(%p)->tx_nob %d\n",
1795                 nob, tx, tx->tx_nob);
1796
1797         switch (type) {
1798         case GNILND_MSG_GET_DONE:
1799         case GNILND_MSG_PUT_DONE:
1800                 post_type = GNI_POST_RDMA_PUT;
1801                 break;
1802         case GNILND_MSG_GET_DONE_REV:
1803         case GNILND_MSG_PUT_DONE_REV:
1804                 post_type = GNI_POST_RDMA_GET;
1805                 break;
1806         default:
1807                 CERROR("invalid msg type %s (%d)\n",
1808                         kgnilnd_msgtype2str(type), type);
1809                 LBUG();
1810         }
1811         if (post_type == GNI_POST_RDMA_GET) {
1812                 /* Check for remote buffer / local buffer / length alignment. All must be 4 byte
1813                  * aligned. If the local buffer is not aligned correctly using the copy buffer
1814                  * will fix that issue. If length is misaligned copy buffer will also fix the issue, we end
1815                  * up transferring extra bytes into the buffer but only copy the correct nob into the original
1816                  * buffer.  Remote offset correction is done through a combination of adjusting the offset,
1817                  * making sure the length and addr are aligned and copying the data into the correct location
1818                  * once the transfer has completed.
1819                  */
1820                 if ((((__u64)((unsigned long)tx->tx_buffer)) & 3) ||
1821                       (sink->gnrd_addr & 3) ||
1822                       (nob & 3)) {
1823
1824                         tx->tx_offset = ((__u64)((unsigned long)sink->gnrd_addr)) & 3;
1825                         if (tx->tx_offset)
1826                                 atomic_inc(&kgnilnd_data.kgn_rev_offset);
1827
1828                         if ((nob + tx->tx_offset) & 3) {
1829                                 desc_nob = ((nob + tx->tx_offset) + (4 - ((nob + tx->tx_offset) & 3)));
1830                                 atomic_inc(&kgnilnd_data.kgn_rev_length);
1831                         } else {
1832                                 desc_nob = (nob + tx->tx_offset);
1833                         }
1834
1835                         if (tx->tx_buffer_copy == NULL) {
1836                                 /* Allocate the largest copy buffer we will need, this will prevent us from overwriting data
1837                                  * and require at most we allocate a few extra bytes. */
1838                                 tx->tx_buffer_copy = kgnilnd_vzalloc(desc_nob);
1839
1840                                 if (!tx->tx_buffer_copy) {
1841                                         /* allocation of buffer failed nak the rdma */
1842                                         kgnilnd_nak_rdma(tx->tx_conn, tx->tx_msg.gnm_type, -EFAULT, cookie, tx->tx_msg.gnm_srcnid);
1843                                         kgnilnd_tx_done(tx, -EFAULT);
1844                                         return 0;
1845                                 }
1846                                 atomic_inc(&kgnilnd_data.kgn_rev_copy_buff);
1847                                 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);
1848                                 if (rc != GNI_RC_SUCCESS) {
1849                                         /* Registration Failed nak rdma and kill the tx. */
1850                                         kgnilnd_vfree(tx->tx_buffer_copy,
1851                                                       desc_nob);
1852                                         tx->tx_buffer_copy = NULL;
1853                                         kgnilnd_nak_rdma(tx->tx_conn, tx->tx_msg.gnm_type, -EFAULT, cookie, tx->tx_msg.gnm_srcnid);
1854                                         kgnilnd_tx_done(tx, -EFAULT);
1855                                         return 0;
1856                                 }
1857                         }
1858                         desc_map_key = tx->tx_buffer_copy_map_key;
1859                         desc_buffer = tx->tx_buffer_copy;
1860                 }
1861         }
1862
1863         memset(&tx->tx_rdma_desc, 0, sizeof(tx->tx_rdma_desc));
1864         tx->tx_rdma_desc.post_id = tx->tx_id.txe_cookie;
1865         tx->tx_rdma_desc.type = post_type;
1866         tx->tx_rdma_desc.cq_mode = GNI_CQMODE_GLOBAL_EVENT;
1867         tx->tx_rdma_desc.local_addr = (__u64)((unsigned long)desc_buffer);
1868         tx->tx_rdma_desc.local_mem_hndl = desc_map_key;
1869         tx->tx_rdma_desc.remote_addr = sink->gnrd_addr - tx->tx_offset;
1870         tx->tx_rdma_desc.remote_mem_hndl = sink->gnrd_key;
1871         tx->tx_rdma_desc.length = desc_nob;
1872         tx->tx_nob_rdma = nob;
1873         if (post_type == GNI_POST_RDMA_PUT && *kgnilnd_tunables.kgn_bte_put_dlvr_mode)
1874                 tx->tx_rdma_desc.dlvr_mode = *kgnilnd_tunables.kgn_bte_put_dlvr_mode;
1875         if (post_type == GNI_POST_RDMA_GET && *kgnilnd_tunables.kgn_bte_get_dlvr_mode)
1876                 tx->tx_rdma_desc.dlvr_mode = *kgnilnd_tunables.kgn_bte_get_dlvr_mode;
1877         /* prep final completion message */
1878         kgnilnd_init_msg(&tx->tx_msg, type, tx->tx_msg.gnm_srcnid);
1879         tx->tx_msg.gnm_u.completion.gncm_cookie = cookie;
1880         /* send actual size RDMA'd in retval */
1881         tx->tx_msg.gnm_u.completion.gncm_retval = nob;
1882
1883         kgnilnd_compute_rdma_cksum(tx, nob);
1884
1885         if (nob == 0) {
1886                 kgnilnd_queue_tx(conn, tx);
1887                 return 0;
1888         }
1889
1890         /* Don't lie (CLOSE == RDMA idle) */
1891         LASSERTF(!conn->gnc_close_sent, "tx %p on conn %p after close sent %d\n",
1892                  tx, conn, conn->gnc_close_sent);
1893
1894         GNIDBG_TX(D_NET, tx, "Post RDMA type 0x%02x conn %p dlvr_mode "
1895                 "0x%x cookie:%#llx",
1896                 type, conn, tx->tx_rdma_desc.dlvr_mode, cookie);
1897
1898         /* set CQ dedicated for RDMA */
1899         tx->tx_rdma_desc.src_cq_hndl = conn->gnc_device->gnd_snd_rdma_cqh;
1900
1901         timestamp = jiffies;
1902         kgnilnd_conn_mutex_lock(&conn->gnc_rdma_mutex);
1903         kgnilnd_gl_mutex_lock(&conn->gnc_device->gnd_cq_mutex);
1904         /* delay in jiffies - we are really concerned only with things that
1905          * result in a schedule() or really holding this off for long times .
1906          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1907         conn->gnc_device->gnd_mutex_delay += (long) jiffies - timestamp;
1908
1909         rrc = kgnilnd_post_rdma(conn->gnc_ephandle, &tx->tx_rdma_desc);
1910
1911         if (rrc == GNI_RC_ERROR_RESOURCE) {
1912                 kgnilnd_conn_mutex_unlock(&conn->gnc_rdma_mutex);
1913                 kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1914                 kgnilnd_unmap_buffer(tx, 0);
1915
1916                 if (tx->tx_buffer_copy != NULL) {
1917                         kgnilnd_vfree(tx->tx_buffer_copy, desc_nob);
1918                         tx->tx_buffer_copy = NULL;
1919                 }
1920
1921                 spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
1922                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn,
1923                                             GNILND_TX_MAPQ, 0);
1924                 spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
1925                 kgnilnd_schedule_device(tx->tx_conn->gnc_device);
1926                 return -EAGAIN;
1927         }
1928
1929         spin_lock(&conn->gnc_list_lock);
1930         kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, GNILND_TX_LIVE_RDMAQ, 1);
1931         tx->tx_qtime = jiffies;
1932         spin_unlock(&conn->gnc_list_lock);
1933         kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1934         kgnilnd_conn_mutex_unlock(&conn->gnc_rdma_mutex);
1935
1936         /* XXX Nic: is this a place we should handle more errors for
1937          * robustness sake */
1938         LASSERT(rrc == GNI_RC_SUCCESS);
1939         return 0;
1940 }
1941
1942 kgn_rx_t *
1943 kgnilnd_alloc_rx(void)
1944 {
1945         kgn_rx_t        *rx;
1946
1947         rx = kmem_cache_alloc(kgnilnd_data.kgn_rx_cache, GFP_ATOMIC);
1948         if (rx == NULL) {
1949                 CERROR("failed to allocate rx\n");
1950                 return NULL;
1951         }
1952         CDEBUG(D_MALLOC, "slab-alloced 'rx': %lu at %p.\n",
1953                sizeof(*rx), rx);
1954
1955         /* no memset to zero, we'll always fill all members */
1956         return rx;
1957 }
1958
1959 /* release is to just free connection resources
1960  * we use this for the eager path after copying */
1961 void
1962 kgnilnd_release_msg(kgn_conn_t *conn)
1963 {
1964         gni_return_t    rrc;
1965         unsigned long   timestamp;
1966
1967         CDEBUG(D_NET, "consuming %p\n", conn);
1968
1969         timestamp = jiffies;
1970         kgnilnd_gl_mutex_lock(&conn->gnc_device->gnd_cq_mutex);
1971         /* delay in jiffies - we are really concerned only with things that
1972          * result in a schedule() or really holding this off for long times .
1973          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1974         conn->gnc_device->gnd_mutex_delay += (long) jiffies - timestamp;
1975
1976         rrc = kgnilnd_smsg_release(conn->gnc_ephandle);
1977         kgnilnd_gl_mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1978
1979         LASSERTF(rrc == GNI_RC_SUCCESS, "bad rrc %d\n", rrc);
1980         GNIDBG_SMSG_CREDS(D_NET, conn);
1981
1982         kgnilnd_schedule_conn(conn);
1983 }
1984
1985 void
1986 kgnilnd_consume_rx(kgn_rx_t *rx)
1987 {
1988         kgn_conn_t      *conn = rx->grx_conn;
1989         kgn_msg_t       *rxmsg = rx->grx_msg;
1990
1991         /* if we are eager, free the cache alloc'd msg */
1992         if (unlikely(rx->grx_eager)) {
1993                 LIBCFS_FREE(rxmsg, sizeof(*rxmsg) + *kgnilnd_tunables.kgn_max_immediate);
1994                 atomic_dec(&kgnilnd_data.kgn_neager_allocs);
1995
1996                 /* release ref from eager_recv */
1997                 kgnilnd_conn_decref(conn);
1998         } else {
1999                 GNIDBG_MSG(D_NET, rxmsg, "rx %p processed", rx);
2000                 kgnilnd_release_msg(conn);
2001         }
2002
2003         kmem_cache_free(kgnilnd_data.kgn_rx_cache, rx);
2004         CDEBUG(D_MALLOC, "slab-freed 'rx': %lu at %p.\n",
2005                sizeof(*rx), rx);
2006 }
2007
2008 int
2009 kgnilnd_send(struct lnet_ni *ni, void *private, struct lnet_msg *lntmsg)
2010 {
2011         struct lnet_hdr  *hdr = &lntmsg->msg_hdr;
2012         int               type = lntmsg->msg_type;
2013         struct lnet_process_id target = lntmsg->msg_target;
2014         int               target_is_router = lntmsg->msg_target_is_router;
2015         int               routing = lntmsg->msg_routing;
2016         unsigned int      niov = lntmsg->msg_niov;
2017         struct bio_vec   *kiov = lntmsg->msg_kiov;
2018         unsigned int      offset = lntmsg->msg_offset;
2019         unsigned int      nob = lntmsg->msg_len;
2020         unsigned int      msg_vmflush = lntmsg->msg_vmflush;
2021         kgn_net_t        *net = ni->ni_data;
2022         kgn_tx_t         *tx;
2023         int               rc = 0;
2024         /* '1' for consistency with code that checks !mpflag to restore */
2025         unsigned int mpflag = 1;
2026         int               reverse_rdma_flag = *kgnilnd_tunables.kgn_reverse_rdma;
2027
2028         /* NB 'private' is different depending on what we're sending.... */
2029         LASSERT(!in_interrupt());
2030
2031         CDEBUG(D_NET, "sending msg type %d with %d bytes in %d frags to %s\n",
2032                type, nob, niov, libcfs_id2str(target));
2033
2034         LASSERTF(nob == 0 || niov > 0,
2035                 "lntmsg %p nob %d niov %d\n", lntmsg, nob, niov);
2036         LASSERTF(niov <= LNET_MAX_IOV,
2037                 "lntmsg %p niov %d\n", lntmsg, niov);
2038
2039         if (msg_vmflush)
2040                 mpflag = memalloc_noreclaim_save();
2041
2042         switch (type) {
2043         default:
2044                 CERROR("lntmsg %p with unexpected type %d\n",
2045                         lntmsg, type);
2046                 LBUG();
2047
2048         case LNET_MSG_ACK:
2049                 LASSERTF(nob == 0, "lntmsg %p nob %d\n",
2050                         lntmsg, nob);
2051                 break;
2052
2053         case LNET_MSG_GET:
2054                 LASSERT(niov == 0);
2055                 LASSERT(nob == 0);
2056
2057                 if (routing || target_is_router)
2058                         break;                  /* send IMMEDIATE */
2059
2060                 /* it is safe to do direct GET with out mapping buffer for RDMA as we
2061                  * check the eventual sink buffer here - if small enough, remote
2062                  * end is perfectly capable of returning data in short message -
2063                  * The magic is that we call lnet_parse in kgnilnd_recv with rdma_req=0
2064                  * for IMMEDIATE messages which will have it send a real reply instead
2065                  * of doing kgnilnd_recv to have the RDMA continued */
2066                 if (lntmsg->msg_md->md_length <= *kgnilnd_tunables.kgn_max_immediate)
2067                        break;
2068
2069                 if ((reverse_rdma_flag & GNILND_REVERSE_GET) == 0)
2070                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_REQ, ni->ni_nid);
2071                 else
2072                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_REQ_REV, ni->ni_nid);
2073
2074                 if (tx == NULL) {
2075                         rc = -ENOMEM;
2076                         goto out;
2077                 }
2078                 rc = kgnilnd_setup_rdma_buffer(tx, lntmsg->msg_md->md_niov,
2079                                                lntmsg->msg_md->md_kiov,
2080                                                0, lntmsg->msg_md->md_length);
2081                 if (rc != 0) {
2082                         CERROR("unable to setup buffer: %d\n", rc);
2083                         kgnilnd_tx_done(tx, rc);
2084                         rc = -EIO;
2085                         goto out;
2086                 }
2087
2088                 tx->tx_lntmsg[1] = lnet_create_reply_msg(ni, lntmsg);
2089                 if (tx->tx_lntmsg[1] == NULL) {
2090                         CERROR("Can't create reply for GET to %s\n",
2091                                libcfs_nid2str(target.nid));
2092                         kgnilnd_tx_done(tx, rc);
2093                         rc = -EIO;
2094                         goto out;
2095                 }
2096
2097                 tx->tx_lntmsg[0] = lntmsg;
2098                 if ((reverse_rdma_flag & GNILND_REVERSE_GET) == 0)
2099                         tx->tx_msg.gnm_u.get.gngm_hdr = *hdr;
2100                 else
2101                         tx->tx_msg.gnm_u.putreq.gnprm_hdr = *hdr;
2102
2103                 /* rest of tx_msg is setup just before it is sent */
2104                 kgnilnd_launch_tx(tx, net, &target);
2105                 goto out;
2106         case LNET_MSG_REPLY:
2107         case LNET_MSG_PUT:
2108                 /* to save on MDDs, we'll handle short kiov by vmap'ing
2109                  * and sending via SMSG */
2110                 if (nob <= *kgnilnd_tunables.kgn_max_immediate)
2111                        break;
2112
2113                 if ((reverse_rdma_flag & GNILND_REVERSE_PUT) == 0)
2114                         tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_REQ, ni->ni_nid);
2115                 else
2116                         tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_REQ_REV, ni->ni_nid);
2117
2118                 if (tx == NULL) {
2119                         rc = -ENOMEM;
2120                         goto out;
2121                 }
2122
2123                 rc = kgnilnd_setup_rdma_buffer(tx, niov,
2124                                                kiov, offset, nob);
2125                 if (rc != 0) {
2126                         kgnilnd_tx_done(tx, rc);
2127                         rc = -EIO;
2128                         goto out;
2129                 }
2130
2131                 tx->tx_lntmsg[0] = lntmsg;
2132                 if ((reverse_rdma_flag & GNILND_REVERSE_PUT) == 0)
2133                         tx->tx_msg.gnm_u.putreq.gnprm_hdr = *hdr;
2134                 else
2135                         tx->tx_msg.gnm_u.get.gngm_hdr = *hdr;
2136
2137                 /* rest of tx_msg is setup just before it is sent */
2138                 kgnilnd_launch_tx(tx, net, &target);
2139                 goto out;
2140         }
2141
2142         /* send IMMEDIATE */
2143
2144         LASSERTF(nob <= *kgnilnd_tunables.kgn_max_immediate,
2145                 "lntmsg 0x%p too large %d\n", lntmsg, nob);
2146
2147         tx = kgnilnd_new_tx_msg(GNILND_MSG_IMMEDIATE, ni->ni_nid);
2148         if (tx == NULL) {
2149                 rc = -ENOMEM;
2150                 goto out;
2151         }
2152
2153         rc = kgnilnd_setup_immediate_buffer(tx, niov, NULL, kiov, offset, nob);
2154         if (rc != 0) {
2155                 kgnilnd_tx_done(tx, rc);
2156                 goto out;
2157         }
2158
2159         tx->tx_msg.gnm_u.immediate.gnim_hdr = *hdr;
2160         tx->tx_lntmsg[0] = lntmsg;
2161         kgnilnd_launch_tx(tx, net, &target);
2162
2163 out:
2164         /* use stored value as we could have already finalized lntmsg here from a failed launch */
2165         if (msg_vmflush)
2166                 memalloc_noreclaim_restore(mpflag);
2167         return rc;
2168 }
2169
2170 void
2171 kgnilnd_setup_rdma(struct lnet_ni *ni, kgn_rx_t *rx, struct lnet_msg *lntmsg, int mlen)
2172 {
2173         kgn_conn_t    *conn = rx->grx_conn;
2174         kgn_msg_t     *rxmsg = rx->grx_msg;
2175         unsigned int   niov = lntmsg->msg_niov;
2176         struct bio_vec *kiov = lntmsg->msg_kiov;
2177         unsigned int   offset = lntmsg->msg_offset;
2178         unsigned int   nob = lntmsg->msg_len;
2179         int            done_type;
2180         kgn_tx_t      *tx;
2181         int            rc = 0;
2182
2183         switch (rxmsg->gnm_type) {
2184         case GNILND_MSG_PUT_REQ_REV:
2185                 done_type = GNILND_MSG_PUT_DONE_REV;
2186                 nob = mlen;
2187                 break;
2188         case GNILND_MSG_GET_REQ:
2189                 done_type = GNILND_MSG_GET_DONE;
2190                 break;
2191         default:
2192                 CERROR("invalid msg type %s (%d)\n",
2193                         kgnilnd_msgtype2str(rxmsg->gnm_type),
2194                         rxmsg->gnm_type);
2195                 LBUG();
2196         }
2197
2198         tx = kgnilnd_new_tx_msg(done_type, ni->ni_nid);
2199         if (tx == NULL)
2200                 goto failed_0;
2201
2202         rc = kgnilnd_set_tx_id(tx, conn);
2203         if (rc != 0)
2204                 goto failed_1;
2205
2206         rc = kgnilnd_setup_rdma_buffer(tx, niov, kiov, offset, nob);
2207         if (rc != 0)
2208                 goto failed_1;
2209
2210         tx->tx_lntmsg[0] = lntmsg;
2211         tx->tx_getinfo = rxmsg->gnm_u.get;
2212
2213         /* we only queue from kgnilnd_recv - we might get called from other contexts
2214          * and we don't want to block the mutex in those cases */
2215
2216         spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2217         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2218         spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2219         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2220
2221         return;
2222
2223  failed_1:
2224         kgnilnd_tx_done(tx, rc);
2225         kgnilnd_nak_rdma(conn, done_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2226  failed_0:
2227         lnet_finalize(lntmsg, rc);
2228 }
2229
2230 int
2231 kgnilnd_eager_recv(struct lnet_ni *ni, void *private, struct lnet_msg *lntmsg,
2232                    void **new_private)
2233 {
2234         kgn_rx_t        *rx = private;
2235         kgn_conn_t      *conn = rx->grx_conn;
2236         kgn_msg_t       *rxmsg = rx->grx_msg;
2237         kgn_msg_t       *eagermsg = NULL;
2238         kgn_peer_t      *peer = NULL;
2239         kgn_conn_t      *found_conn = NULL;
2240
2241         GNIDBG_MSG(D_NET, rxmsg, "eager recv for conn %p, rxmsg %p, lntmsg %p",
2242                 conn, rxmsg, lntmsg);
2243
2244         if (rxmsg->gnm_payload_len > *kgnilnd_tunables.kgn_max_immediate) {
2245                 GNIDBG_MSG(D_ERROR, rxmsg, "payload too large %d",
2246                         rxmsg->gnm_payload_len);
2247                 return -EPROTO;
2248         }
2249         /* Grab a read lock so the connection doesnt disappear on us
2250          * while we look it up
2251          */
2252         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
2253
2254         peer = kgnilnd_find_peer_locked(rxmsg->gnm_srcnid);
2255         if (peer != NULL)
2256                 found_conn = kgnilnd_find_conn_locked(peer);
2257
2258
2259         /* Verify the connection found is the same one that the message
2260          * is supposed to be using, if it is not output an error message
2261          * and return.
2262          */
2263         if (!peer || !found_conn
2264             || found_conn->gnc_peer_connstamp != rxmsg->gnm_connstamp) {
2265                 read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2266                 CERROR("Couldnt find matching peer %p or conn %p / %p\n",
2267                         peer, conn, found_conn);
2268                 if (found_conn) {
2269                         CERROR("Unexpected connstamp %#llx(%#llx expected)"
2270                                 " from %s", rxmsg->gnm_connstamp,
2271                                 found_conn->gnc_peer_connstamp,
2272                                 libcfs_nid2str(peer->gnp_nid));
2273                 }
2274                 return -ENOTCONN;
2275         }
2276
2277         /* add conn ref to ensure it doesn't go away until all eager
2278          * messages processed */
2279         kgnilnd_conn_addref(conn);
2280
2281         /* Now that we have verified the connection is valid and added a
2282          * reference we can remove the read_lock on the peer_conn_lock */
2283         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2284
2285         /* we have no credits or buffers for this message, so copy it
2286          * somewhere for a later kgnilnd_recv */
2287         if (atomic_read(&kgnilnd_data.kgn_neager_allocs) >=
2288                         *kgnilnd_tunables.kgn_eager_credits) {
2289                 CERROR("Out of eager credits to %s\n",
2290                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
2291                 return -ENOMEM;
2292         }
2293
2294         atomic_inc(&kgnilnd_data.kgn_neager_allocs);
2295
2296         LIBCFS_ALLOC(eagermsg, sizeof(*eagermsg) + *kgnilnd_tunables.kgn_max_immediate);
2297         if (eagermsg == NULL) {
2298                 kgnilnd_conn_decref(conn);
2299                 CERROR("couldn't allocate eager rx message for conn %p to %s\n",
2300                         conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
2301                 return -ENOMEM;
2302         }
2303
2304         /* copy msg and payload */
2305         memcpy(eagermsg, rxmsg, sizeof(*rxmsg) + rxmsg->gnm_payload_len);
2306         rx->grx_msg = eagermsg;
2307         rx->grx_eager = 1;
2308
2309         /* stash this for lnet_finalize on cancel-on-conn-close */
2310         rx->grx_lntmsg = lntmsg;
2311
2312         /* keep the same rx_t, it just has a new grx_msg now */
2313         *new_private = private;
2314
2315         /* release SMSG buffer */
2316         kgnilnd_release_msg(conn);
2317
2318         return 0;
2319 }
2320
2321 int
2322 kgnilnd_recv(struct lnet_ni *ni, void *private, struct lnet_msg *lntmsg,
2323              int delayed, unsigned int niov,
2324              struct bio_vec *kiov,
2325              unsigned int offset, unsigned int mlen, unsigned int rlen)
2326 {
2327         kgn_rx_t    *rx = private;
2328         kgn_conn_t  *conn = rx->grx_conn;
2329         kgn_msg_t   *rxmsg = rx->grx_msg;
2330         kgn_tx_t    *tx;
2331         int          rc = 0;
2332         __u32        pload_cksum;
2333         ENTRY;
2334
2335         LASSERT(!in_interrupt());
2336         LASSERTF(mlen <= rlen, "%d <= %d\n", mlen, rlen);
2337
2338         GNIDBG_MSG(D_NET, rxmsg, "conn %p, rxmsg %p, lntmsg %p"
2339                 " niov=%d kiov=%p offset=%d mlen=%d rlen=%d",
2340                 conn, rxmsg, lntmsg,
2341                 niov, kiov, offset, mlen, rlen);
2342
2343         /* we need to lock here as recv can be called from any context */
2344         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
2345         if (rx->grx_eager && conn->gnc_state != GNILND_CONN_ESTABLISHED) {
2346                 read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2347
2348                 /* someone closed the conn after we copied this out, nuke it */
2349                 kgnilnd_consume_rx(rx);
2350                 lnet_finalize(lntmsg, conn->gnc_error);
2351                 RETURN(0);
2352         }
2353         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2354
2355         switch (rxmsg->gnm_type) {
2356         default:
2357                 GNIDBG_MSG(D_NETERROR, rxmsg, "conn %p, rx %p, rxmsg %p, lntmsg %p"
2358                 " niov=%d kiov=%p offset=%d mlen=%d rlen=%d",
2359                 conn, rx, rxmsg, lntmsg, niov, kiov, offset, mlen, rlen);
2360                 LBUG();
2361
2362         case GNILND_MSG_IMMEDIATE:
2363                 if (mlen > rxmsg->gnm_payload_len) {
2364                         GNIDBG_MSG(D_ERROR, rxmsg,
2365                                 "Immediate message from %s too big: %d > %d",
2366                                 libcfs_nid2str(conn->gnc_peer->gnp_nid), mlen,
2367                                 rxmsg->gnm_payload_len);
2368                         rc = -EINVAL;
2369                         kgnilnd_consume_rx(rx);
2370                         RETURN(rc);
2371                 }
2372
2373                 /* rxmsg[1] is a pointer to the payload, sitting in the buffer
2374                  * right after the kgn_msg_t header - so just 'cute' way of saying
2375                  * rxmsg + sizeof(kgn_msg_t) */
2376
2377                 /* check payload checksum if sent */
2378
2379                 if (*kgnilnd_tunables.kgn_checksum >= 2 &&
2380                         !rxmsg->gnm_payload_cksum &&
2381                         rxmsg->gnm_payload_len != 0)
2382                         GNIDBG_MSG(D_WARNING, rxmsg, "no msg payload checksum when enabled");
2383
2384                 if (rxmsg->gnm_payload_cksum != 0) {
2385                         /* gnm_payload_len set in kgnilnd_sendmsg from tx->tx_nob,
2386                          * which is what is used to calculate the cksum on the TX side */
2387                         pload_cksum = kgnilnd_cksum(&rxmsg[1], rxmsg->gnm_payload_len);
2388
2389                         if (rxmsg->gnm_payload_cksum != pload_cksum) {
2390                                 GNIDBG_MSG(D_NETERROR, rxmsg,
2391                                            "Bad payload checksum (%x expected %x)",
2392                                             pload_cksum, rxmsg->gnm_payload_cksum);
2393                                 switch (*kgnilnd_tunables.kgn_checksum_dump) {
2394                                 case 2:
2395                                         kgnilnd_dump_blob(D_BUFFS, "bad payload checksum",
2396                                                           &rxmsg[1], rxmsg->gnm_payload_len);
2397                                         /* fall through to dump */
2398                                 case 1:
2399                                         libcfs_debug_dumplog();
2400                                         break;
2401                                 default:
2402                                         break;
2403                                 }
2404                                 rc = -ENOKEY;
2405                                 /* checksum problems are fatal, kill the conn */
2406                                 kgnilnd_consume_rx(rx);
2407                                 kgnilnd_close_conn(conn, rc);
2408                                 RETURN(rc);
2409                         }
2410                 }
2411
2412                 lnet_copy_flat2kiov(
2413                         niov, kiov, offset,
2414                         *kgnilnd_tunables.kgn_max_immediate,
2415                         &rxmsg[1], 0, mlen);
2416
2417                 kgnilnd_consume_rx(rx);
2418                 lnet_finalize(lntmsg, 0);
2419                 RETURN(0);
2420
2421         case GNILND_MSG_PUT_REQ:
2422                 /* LNET wants to truncate or drop transaction, sending NAK */
2423                 if (mlen == 0) {
2424                         kgnilnd_consume_rx(rx);
2425                         lnet_finalize(lntmsg, 0);
2426
2427                         /* only error if lntmsg == NULL, otherwise we are just
2428                          * short circuiting the rdma process of 0 bytes */
2429                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2430                                         lntmsg == NULL ? -ENOENT : 0,
2431                                         rxmsg->gnm_u.get.gngm_cookie,
2432                                         ni->ni_nid);
2433                         RETURN(0);
2434                 }
2435                 /* sending ACK with sink buff. info */
2436                 tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_ACK, ni->ni_nid);
2437                 if (tx == NULL) {
2438                         kgnilnd_consume_rx(rx);
2439                         RETURN(-ENOMEM);
2440                 }
2441
2442                 rc = kgnilnd_set_tx_id(tx, conn);
2443                 if (rc != 0) {
2444                         GOTO(nak_put_req, rc);
2445                 }
2446
2447                 rc = kgnilnd_setup_rdma_buffer(tx, niov,
2448                                                kiov, offset, mlen);
2449                 if (rc != 0) {
2450                         GOTO(nak_put_req, rc);
2451                 }
2452
2453                 tx->tx_msg.gnm_u.putack.gnpam_src_cookie =
2454                         rxmsg->gnm_u.putreq.gnprm_cookie;
2455                 tx->tx_msg.gnm_u.putack.gnpam_dst_cookie = tx->tx_id.txe_cookie;
2456                 tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_addr =
2457                         (__u64)((unsigned long)tx->tx_buffer);
2458                 tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_nob = mlen;
2459
2460                 tx->tx_lntmsg[0] = lntmsg; /* finalize this on RDMA_DONE */
2461                 tx->tx_qtime = jiffies;
2462                 /* we only queue from kgnilnd_recv - we might get called from other contexts
2463                  * and we don't want to block the mutex in those cases */
2464
2465                 spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2466                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2467                 spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2468                 kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2469
2470                 kgnilnd_consume_rx(rx);
2471                 RETURN(0);
2472
2473 nak_put_req:
2474                 /* make sure we send an error back when the PUT fails */
2475                 kgnilnd_nak_rdma(conn, rxmsg->gnm_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2476                 kgnilnd_tx_done(tx, rc);
2477                 kgnilnd_consume_rx(rx);
2478
2479                 /* return magic LNet network error */
2480                 RETURN(-EIO);
2481         case GNILND_MSG_GET_REQ_REV:
2482                 /* LNET wants to truncate or drop transaction, sending NAK */
2483                 if (mlen == 0) {
2484                         kgnilnd_consume_rx(rx);
2485                         lnet_finalize(lntmsg, 0);
2486
2487                         /* only error if lntmsg == NULL, otherwise we are just
2488                          * short circuiting the rdma process of 0 bytes */
2489                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2490                                         lntmsg == NULL ? -ENOENT : 0,
2491                                         rxmsg->gnm_u.get.gngm_cookie,
2492                                         ni->ni_nid);
2493                         RETURN(0);
2494                 }
2495                 /* lntmsg can be null when parsing a LNET_GET */
2496                 if (lntmsg != NULL) {
2497                         /* sending ACK with sink buff. info */
2498                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_ACK_REV, ni->ni_nid);
2499                         if (tx == NULL) {
2500                                 kgnilnd_consume_rx(rx);
2501                                 RETURN(-ENOMEM);
2502                         }
2503
2504                         rc = kgnilnd_set_tx_id(tx, conn);
2505                         if (rc != 0)
2506                                 GOTO(nak_get_req_rev, rc);
2507
2508                         rc = kgnilnd_setup_rdma_buffer(tx, niov,
2509                                                        kiov, offset, mlen);
2510                         if (rc != 0)
2511                                 GOTO(nak_get_req_rev, rc);
2512
2513                         tx->tx_msg.gnm_u.putack.gnpam_src_cookie =
2514                                 rxmsg->gnm_u.putreq.gnprm_cookie;
2515                         tx->tx_msg.gnm_u.putack.gnpam_dst_cookie = tx->tx_id.txe_cookie;
2516                         tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_addr =
2517                                 (__u64)((unsigned long)tx->tx_buffer);
2518                         tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_nob = mlen;
2519
2520                         tx->tx_lntmsg[0] = lntmsg; /* finalize this on RDMA_DONE */
2521
2522                         /* we only queue from kgnilnd_recv - we might get called from other contexts
2523                          * and we don't want to block the mutex in those cases */
2524
2525                         spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2526                         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2527                         spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2528                         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2529                 } else {
2530                         /* No match */
2531                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2532                                         -ENOENT,
2533                                         rxmsg->gnm_u.get.gngm_cookie,
2534                                         ni->ni_nid);
2535                 }
2536
2537                 kgnilnd_consume_rx(rx);
2538                 RETURN(0);
2539
2540 nak_get_req_rev:
2541                 /* make sure we send an error back when the GET fails */
2542                 kgnilnd_nak_rdma(conn, rxmsg->gnm_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2543                 kgnilnd_tx_done(tx, rc);
2544                 kgnilnd_consume_rx(rx);
2545
2546                 /* return magic LNet network error */
2547                 RETURN(-EIO);
2548
2549
2550         case GNILND_MSG_PUT_REQ_REV:
2551                 /* LNET wants to truncate or drop transaction, sending NAK */
2552                 if (mlen == 0) {
2553                         kgnilnd_consume_rx(rx);
2554                         lnet_finalize(lntmsg, 0);
2555
2556                         /* only error if lntmsg == NULL, otherwise we are just
2557                          * short circuiting the rdma process of 0 bytes */
2558                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2559                                         lntmsg == NULL ? -ENOENT : 0,
2560                                         rxmsg->gnm_u.get.gngm_cookie,
2561                                         ni->ni_nid);
2562                         RETURN(0);
2563                 }
2564
2565                 if (lntmsg != NULL) {
2566                         /* Matched! */
2567                         kgnilnd_setup_rdma(ni, rx, lntmsg, mlen);
2568                 } else {
2569                         /* No match */
2570                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2571                                         -ENOENT,
2572                                         rxmsg->gnm_u.get.gngm_cookie,
2573                                         ni->ni_nid);
2574                 }
2575                 kgnilnd_consume_rx(rx);
2576                 RETURN(0);
2577         case GNILND_MSG_GET_REQ:
2578                 if (lntmsg != NULL) {
2579                         /* Matched! */
2580                         kgnilnd_setup_rdma(ni, rx, lntmsg, mlen);
2581                 } else {
2582                         /* No match */
2583                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2584                                         -ENOENT,
2585                                         rxmsg->gnm_u.get.gngm_cookie,
2586                                         ni->ni_nid);
2587                 }
2588                 kgnilnd_consume_rx(rx);
2589                 RETURN(0);
2590         }
2591         RETURN(0);
2592 }
2593
2594 /* needs write_lock on kgn_peer_conn_lock held */
2595 int
2596 kgnilnd_check_conn_timeouts_locked(kgn_conn_t *conn)
2597 {
2598         unsigned long      timeout, keepalive;
2599         unsigned long      now = jiffies;
2600         unsigned long      newest_last_rx;
2601         kgn_tx_t          *tx;
2602
2603         /* given that we found this conn hanging off a peer, it better damned
2604          * well be connected */
2605         LASSERTF(conn->gnc_state == GNILND_CONN_ESTABLISHED,
2606                  "conn 0x%p->%s with bad state%s\n", conn,
2607                  conn->gnc_peer ? libcfs_nid2str(conn->gnc_peer->gnp_nid)
2608                                : "<?>",
2609                  kgnilnd_conn_state2str(conn));
2610
2611         CDEBUG(D_NET, "checking conn %p->%s timeout %d keepalive %d "
2612                       "rx_diff %lu tx_diff %lu\n",
2613                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
2614                 conn->gnc_timeout, GNILND_TO2KA(conn->gnc_timeout),
2615                 cfs_duration_sec(now - conn->gnc_last_rx_cq),
2616                 cfs_duration_sec(now - conn->gnc_last_tx));
2617
2618         timeout = cfs_time_seconds(conn->gnc_timeout);
2619         keepalive = cfs_time_seconds(GNILND_TO2KA(conn->gnc_timeout));
2620
2621         /* just in case our lack of RX msg processing is gumming up the works - give the
2622          * remove an extra chance */
2623
2624         newest_last_rx = GNILND_LASTRX(conn);
2625
2626         if (time_after_eq(now, newest_last_rx + timeout)) {
2627                 uint32_t level = D_CONSOLE|D_NETERROR;
2628
2629                 if (conn->gnc_peer->gnp_state == GNILND_PEER_DOWN) {
2630                         level = D_NET;
2631                 }
2632                         GNIDBG_CONN(level, conn,
2633                         "No gnilnd traffic received from %s for %lu "
2634                         "seconds, terminating connection. Is node down? ",
2635                         libcfs_nid2str(conn->gnc_peer->gnp_nid),
2636                         cfs_duration_sec(now - newest_last_rx));
2637                 return -ETIMEDOUT;
2638         }
2639
2640         /* we don't timeout on last_tx stalls - we are going to trust the
2641          * underlying network to let us know when sends are failing.
2642          * At worst, the peer will timeout our RX stamp and drop the connection
2643          * at that point. We'll then see his CLOSE or at worst his RX
2644          * stamp stop and drop the connection on our end */
2645
2646         if (time_after_eq(now, conn->gnc_last_tx + keepalive)) {
2647                 CDEBUG(D_NET, "sending NOOP -> %s (%p idle %lu(%lu)) "
2648                        "last %lu/%lu/%lu %lus/%lus/%lus\n",
2649                        libcfs_nid2str(conn->gnc_peer->gnp_nid), conn,
2650                        cfs_duration_sec(jiffies - conn->gnc_last_tx),
2651                        keepalive,
2652                        conn->gnc_last_noop_want, conn->gnc_last_noop_sent,
2653                        conn->gnc_last_noop_cq,
2654                        cfs_duration_sec(jiffies - conn->gnc_last_noop_want),
2655                        cfs_duration_sec(jiffies - conn->gnc_last_noop_sent),
2656                        cfs_duration_sec(jiffies - conn->gnc_last_noop_cq));
2657                 set_mb(conn->gnc_last_noop_want, jiffies);
2658                 atomic_inc(&conn->gnc_reaper_noop);
2659                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_NOOP_SEND))
2660                         return 0;
2661
2662                 tx = kgnilnd_new_tx_msg(GNILND_MSG_NOOP, conn->gnc_peer->gnp_net->gnn_ni->ni_nid);
2663                 if (tx == NULL)
2664                         return 0;
2665                 kgnilnd_queue_tx(conn, tx);
2666         }
2667
2668         return 0;
2669 }
2670
2671 /* needs write_lock on kgn_peer_conn_lock held */
2672 void
2673 kgnilnd_check_peer_timeouts_locked(kgn_peer_t *peer, struct list_head *todie,
2674                                     struct list_head *souls)
2675 {
2676         unsigned long           timeout;
2677         kgn_conn_t             *conn, *connN = NULL;
2678         kgn_tx_t               *tx, *txN;
2679         int                     rc = 0;
2680         int                     count = 0;
2681         int                     reconnect;
2682         int                     to_reconn;
2683         short                   releaseconn = 0;
2684         unsigned long           first_rx = 0;
2685         int                     purgatory_conn_cnt = 0;
2686
2687         CDEBUG(D_NET, "checking peer 0x%p->%s for timeouts; interval %lus\n",
2688                 peer, libcfs_nid2str(peer->gnp_nid),
2689                 peer->gnp_reconnect_interval);
2690
2691         timeout = cfs_time_seconds(max(*kgnilnd_tunables.kgn_timeout,
2692                                        GNILND_MIN_TIMEOUT));
2693
2694         conn = kgnilnd_find_conn_locked(peer);
2695         if (conn) {
2696                 /* if there is a valid conn, check the queues for timeouts */
2697                 rc = kgnilnd_check_conn_timeouts_locked(conn);
2698                 if (rc) {
2699                         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_RX_CLOSE_CLOSING)) {
2700                                 /* simulate a RX CLOSE after the timeout but before
2701                                  * the scheduler thread gets it */
2702                                 conn->gnc_close_recvd = GNILND_CLOSE_INJECT1;
2703                                 conn->gnc_peer_error = -ETIMEDOUT;
2704                         }
2705
2706                         if (*kgnilnd_tunables.kgn_to_reconn_disable &&
2707                             rc == -ETIMEDOUT) {
2708                                 peer->gnp_state = GNILND_PEER_TIMED_OUT;
2709                                 CDEBUG(D_WARNING, "%s conn timed out, will "
2710                                        "reconnect upon request from peer\n",
2711                                        libcfs_nid2str(conn->gnc_peer->gnp_nid));
2712                         }
2713                         /* Once we mark closed, any of the scheduler threads could
2714                          * get it and move through before we hit the fail loc code */
2715                         kgnilnd_close_conn_locked(conn, rc);
2716                 } else {
2717                         /* first_rx is used to decide when to release a conn from purgatory.
2718                          */
2719                         first_rx = conn->gnc_first_rx;
2720                 }
2721         }
2722
2723         /* now regardless of starting new conn, find tx on peer queue that
2724          * are old and smell bad - do this first so we don't trigger
2725          * reconnect on empty queue if we timeout all */
2726         list_for_each_entry_safe(tx, txN, &peer->gnp_tx_queue, tx_list) {
2727                 if (time_after_eq(jiffies, tx->tx_qtime + timeout)) {
2728                         if (count == 0) {
2729                                 LCONSOLE_INFO("could not send to %s due to connection"
2730                                        " setup failure after %lu seconds\n",
2731                                        libcfs_nid2str(peer->gnp_nid),
2732                                        cfs_duration_sec(jiffies - tx->tx_qtime));
2733                         }
2734                         kgnilnd_tx_del_state_locked(tx, peer, NULL,
2735                                                    GNILND_TX_ALLOCD);
2736                         list_add_tail(&tx->tx_list, todie);
2737                         count++;
2738                 }
2739         }
2740
2741         if (count || peer->gnp_connecting == GNILND_PEER_KILL) {
2742                 CDEBUG(D_NET, "canceling %d tx for peer 0x%p->%s\n",
2743                         count, peer, libcfs_nid2str(peer->gnp_nid));
2744                 /* if we nuked all the TX, stop peer connection attempt (if there is one..) */
2745                 if (list_empty(&peer->gnp_tx_queue) ||
2746                         peer->gnp_connecting == GNILND_PEER_KILL) {
2747                         /* we pass down todie to use a common function - but we know there are
2748                          * no TX to add */
2749                         kgnilnd_cancel_peer_connect_locked(peer, todie);
2750                 }
2751         }
2752
2753         /* Don't reconnect if we are still trying to clear out old conns.
2754          * This prevents us sending traffic on the new mbox before ensuring we are done
2755          * with the old one */
2756         reconnect = (peer->gnp_state == GNILND_PEER_UP) &&
2757                     (atomic_read(&peer->gnp_dirty_eps) == 0);
2758
2759         /* fast reconnect after a timeout */
2760         to_reconn = !conn &&
2761                     (peer->gnp_last_errno == -ETIMEDOUT) &&
2762                     *kgnilnd_tunables.kgn_fast_reconn;
2763
2764         /* if we are not connected and there are tx on the gnp_tx_queue waiting
2765          * to be sent, we'll check the reconnect interval and fire up a new
2766          * connection request */
2767
2768         if (reconnect &&
2769             (peer->gnp_connecting == GNILND_PEER_IDLE) &&
2770             (time_after_eq(jiffies, peer->gnp_reconnect_time)) &&
2771             (!list_empty(&peer->gnp_tx_queue) || to_reconn)) {
2772
2773                 CDEBUG(D_NET, "starting connect to %s\n",
2774                         libcfs_nid2str(peer->gnp_nid));
2775                 LASSERTF(peer->gnp_connecting == GNILND_PEER_IDLE,
2776                          "Peer was idle and we have a write_lock, state issue %d\n",
2777                          peer->gnp_connecting);
2778
2779                 peer->gnp_connecting = GNILND_PEER_CONNECT;
2780                 kgnilnd_peer_addref(peer); /* extra ref for connd */
2781
2782                 spin_lock(&peer->gnp_net->gnn_dev->gnd_connd_lock);
2783                 list_add_tail(&peer->gnp_connd_list,
2784                               &peer->gnp_net->gnn_dev->gnd_connd_peers);
2785                 spin_unlock(&peer->gnp_net->gnn_dev->gnd_connd_lock);
2786
2787                 kgnilnd_schedule_dgram(peer->gnp_net->gnn_dev);
2788         }
2789
2790         /* fail_loc to allow us to delay release of purgatory */
2791         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_PURG_REL_DELAY))
2792                 return;
2793
2794         /* This check allows us to verify that the new conn is actually being used. This allows us to
2795          * pull the old conns out of purgatory if they have actually seen traffic.
2796          * We only release a conn from purgatory during stack reset, admin command, or when a peer reconnects
2797          */
2798         if (first_rx &&
2799                 time_after(jiffies, first_rx + cfs_time_seconds(*kgnilnd_tunables.kgn_hardware_timeout))) {
2800                 CDEBUG(D_INFO, "We can release peer %s conn's from purgatory %lu\n",
2801                         libcfs_nid2str(peer->gnp_nid), first_rx + cfs_time_seconds(*kgnilnd_tunables.kgn_hardware_timeout));
2802                 releaseconn = 1;
2803         }
2804
2805         list_for_each_entry_safe (conn, connN, &peer->gnp_conns, gnc_list) {
2806         /* check for purgatory timeouts */
2807                 if (conn->gnc_in_purgatory) {
2808                         /* We cannot detach this conn from purgatory if it has not been closed so we reschedule it
2809                          * that way the next time we check it we can detach it from purgatory
2810                          */
2811
2812                         if (conn->gnc_state != GNILND_CONN_DONE) {
2813                                 /* Skip over conns that are currently not DONE. If they arent already scheduled
2814                                  * for completion something in the state machine is broken.
2815                                  */
2816                                 continue;
2817                         }
2818
2819                         /* We only detach a conn that is in purgatory if we have received a close message,
2820                          * we have a new valid connection that has successfully received data, or an admin
2821                          * command tells us we need to detach.
2822                          */
2823
2824                         if (conn->gnc_close_recvd || releaseconn || conn->gnc_needs_detach) {
2825                                 unsigned long   waiting;
2826
2827                                 waiting = (long) jiffies - conn->gnc_last_rx_cq;
2828
2829                                 /* C.E: The remote peer is expected to close the
2830                                  * connection (see kgnilnd_check_conn_timeouts)
2831                                  * via the reaper thread and nuke out the MDD and
2832                                  * FMA resources after conn->gnc_timeout has expired
2833                                  * without an FMA RX */
2834                                 CDEBUG(D_NET, "Reconnected to %s in %lds or admin forced detach, dropping "
2835                                         " held resources\n",
2836                                         libcfs_nid2str(conn->gnc_peer->gnp_nid),
2837                                         cfs_duration_sec(waiting));
2838
2839                                 kgnilnd_detach_purgatory_locked(conn, souls);
2840                         } else {
2841                                 purgatory_conn_cnt++;
2842                         }
2843                 }
2844         }
2845
2846         /* If we have too many connections in purgatory we could run out of
2847          * resources. Limit the number of connections to a tunable number,
2848          * clean up to the minimum all in one fell swoop... there are
2849          * situations where dvs will retry tx's and we can eat up several
2850          * hundread connection requests at once.
2851          */
2852         if (purgatory_conn_cnt > *kgnilnd_tunables.kgn_max_purgatory) {
2853                 list_for_each_entry_safe(conn, connN, &peer->gnp_conns,
2854                                          gnc_list) {
2855                         if (conn->gnc_in_purgatory &&
2856                             conn->gnc_state == GNILND_CONN_DONE) {
2857                                 CDEBUG(D_NET, "Dropping Held resource due to"
2858                                               " resource limits being hit\n");
2859                                 kgnilnd_detach_purgatory_locked(conn, souls);
2860
2861                                 if (purgatory_conn_cnt-- <
2862                                     *kgnilnd_tunables.kgn_max_purgatory)
2863                                         break;
2864                         }
2865                 }
2866         }
2867 }
2868
2869 void
2870 kgnilnd_reaper_check(int idx)
2871 {
2872         struct list_head  *peers = &kgnilnd_data.kgn_peers[idx];
2873         struct list_head  *ctmp, *ctmpN;
2874         LIST_HEAD(geriatrics);
2875         LIST_HEAD(souls);
2876
2877         write_lock(&kgnilnd_data.kgn_peer_conn_lock);
2878
2879         list_for_each_safe(ctmp, ctmpN, peers) {
2880                 kgn_peer_t        *peer = NULL;
2881
2882                 /* don't timeout stuff if the network is mucked or shutting down */
2883                 if (kgnilnd_check_hw_quiesce()) {
2884                         break;
2885                 }
2886                 peer = list_entry(ctmp, kgn_peer_t, gnp_list);
2887
2888                 kgnilnd_check_peer_timeouts_locked(peer, &geriatrics, &souls);
2889         }
2890
2891         write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2892
2893         kgnilnd_txlist_done(&geriatrics, -EHOSTUNREACH);
2894         kgnilnd_release_purgatory_list(&souls);
2895 }
2896
2897 void
2898 kgnilnd_update_reaper_timeout(long timeout)
2899 {
2900         LASSERT(timeout > 0);
2901
2902         spin_lock(&kgnilnd_data.kgn_reaper_lock);
2903
2904         if (timeout < kgnilnd_data.kgn_new_min_timeout)
2905                 kgnilnd_data.kgn_new_min_timeout = timeout;
2906
2907         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
2908 }
2909
2910 static void
2911 kgnilnd_reaper_poke_with_stick(unsigned long arg)
2912 {
2913         wake_up(&kgnilnd_data.kgn_reaper_waitq);
2914 }
2915
2916 int
2917 kgnilnd_reaper(void *arg)
2918 {
2919         long               timeout;
2920         int                i;
2921         int                hash_index = 0;
2922         unsigned long      next_check_time = jiffies;
2923         long               current_min_timeout = MAX_SCHEDULE_TIMEOUT;
2924         struct timer_list  timer;
2925         DEFINE_WAIT(wait);
2926
2927         /* all gnilnd threads need to run fairly urgently */
2928         set_user_nice(current, *kgnilnd_tunables.kgn_nice);
2929         spin_lock(&kgnilnd_data.kgn_reaper_lock);
2930
2931         while (!kgnilnd_data.kgn_shutdown) {
2932                 /* I wake up every 'p' seconds to check for timeouts on some
2933                  * more peers.  I try to check every connection 'n' times
2934                  * within the global minimum of all keepalive and timeout
2935                  * intervals, to ensure I attend to every connection within
2936                  * (n+1)/n times its timeout intervals. */
2937                 const int     p = GNILND_REAPER_THREAD_WAKE;
2938                 const int     n = GNILND_REAPER_NCHECKS;
2939                 int           chunk;
2940                 /* to quiesce or to not quiesce, that is the question */
2941                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
2942                         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
2943                         KGNILND_SPIN_QUIESCE;
2944                         spin_lock(&kgnilnd_data.kgn_reaper_lock);
2945                 }
2946
2947                 /* careful with the jiffy wrap... */
2948                 timeout = (long)(next_check_time - jiffies);
2949
2950                 if (timeout > 0) {
2951                         prepare_to_wait(&kgnilnd_data.kgn_reaper_waitq, &wait,
2952                                         TASK_INTERRUPTIBLE);
2953                         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
2954                         setup_timer(&timer, kgnilnd_reaper_poke_with_stick,
2955                                     next_check_time);
2956                         mod_timer(&timer, (long) jiffies + timeout);
2957
2958                         /* check flag variables before committing */
2959                         if (!kgnilnd_data.kgn_shutdown &&
2960                             !kgnilnd_data.kgn_quiesce_trigger) {
2961                                 CDEBUG(D_INFO, "schedule timeout %ld (%lu sec)\n",
2962                                        timeout, cfs_duration_sec(timeout));
2963                                 schedule();
2964                                 CDEBUG(D_INFO, "awake after schedule\n");
2965                         }
2966
2967                         del_singleshot_timer_sync(&timer);
2968                         spin_lock(&kgnilnd_data.kgn_reaper_lock);
2969                         finish_wait(&kgnilnd_data.kgn_reaper_waitq, &wait);
2970                         continue;
2971                 }
2972
2973                 /* new_min_timeout is set from the conn timeouts and keepalive
2974                  * this should end up with a min timeout of
2975                  * GNILND_TIMEOUT2KEEPALIVE(t) or roughly LND_TIMEOUT/2 */
2976                 if (kgnilnd_data.kgn_new_min_timeout < current_min_timeout) {
2977                         current_min_timeout = kgnilnd_data.kgn_new_min_timeout;
2978                         CDEBUG(D_NET, "Set new min timeout %ld\n",
2979                                current_min_timeout);
2980                 }
2981
2982                 spin_unlock(&kgnilnd_data.kgn_reaper_lock);
2983
2984                 /* Compute how many table entries to check now so I get round
2985                  * the whole table fast enough given that I do this at fixed
2986                  * intervals of 'p' seconds) */
2987                 chunk = *kgnilnd_tunables.kgn_peer_hash_size;
2988                 if (kgnilnd_data.kgn_new_min_timeout > n * p)
2989                         chunk = (chunk * n * p) /
2990                                 kgnilnd_data.kgn_new_min_timeout;
2991                 if (chunk == 0)
2992                         chunk = 1;
2993                 for (i = 0; i < chunk; i++) {
2994                         kgnilnd_reaper_check(hash_index);
2995                         hash_index = (hash_index + 1) %
2996                                 *kgnilnd_tunables.kgn_peer_hash_size;
2997                 }
2998                 next_check_time = (long) jiffies + cfs_time_seconds(p);
2999                 CDEBUG(D_INFO, "next check at %lu or in %d sec\n", next_check_time, p);
3000
3001                 spin_lock(&kgnilnd_data.kgn_reaper_lock);
3002         }
3003
3004         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
3005
3006         kgnilnd_thread_fini();
3007         return 0;
3008 }
3009
3010 int
3011 kgnilnd_recv_bte_get(kgn_tx_t *tx) {
3012         unsigned niov, offset, nob;
3013         struct bio_vec *kiov;
3014         struct lnet_msg *lntmsg = tx->tx_lntmsg[0];
3015         kgnilnd_parse_lnet_rdma(lntmsg, &niov, &offset, &nob, &kiov, tx->tx_nob_rdma);
3016
3017         if (kiov != NULL) {
3018                 lnet_copy_flat2kiov(
3019                         niov, kiov, offset,
3020                         nob,
3021                         tx->tx_buffer_copy + tx->tx_offset, 0, nob);
3022         } else {
3023                 memcpy(tx->tx_buffer, tx->tx_buffer_copy + tx->tx_offset, nob);
3024         }
3025         return 0;
3026 }
3027
3028
3029 int
3030 kgnilnd_check_rdma_cq(kgn_device_t *dev)
3031 {
3032         gni_return_t           rrc;
3033         gni_post_descriptor_t *desc;
3034         __u64                  event_data;
3035         kgn_tx_ev_id_t         ev_id;
3036         char                   err_str[256];
3037         int                    should_retry, rc;
3038       &