Whamcloud - gitweb
LU-2675 lnet: remove lnet/include/lnet/linux/
[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: Nic Henke <nic@cray.com>
8  *
9  *   This file is part of Lustre, http://www.lustre.org.
10  *
11  *   Lustre is free software; you can redistribute it and/or
12  *   modify it under the terms of version 2 of the GNU General Public
13  *   License as published by the Free Software Foundation.
14  *
15  *   Lustre is distributed in the hope that it will be useful,
16  *   but WITHOUT ANY WARRANTY; without even the implied warranty of
17  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  *   GNU General Public License for more details.
19  *
20  *   You should have received a copy of the GNU General Public License
21  *   along with Lustre; if not, write to the Free Software
22  *   Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
23  *
24  */
25
26 #include <asm/page.h>
27 #include <linux/nmi.h>
28 #include "gnilnd.h"
29
30 /* this is useful when needed to debug wire corruption. */
31 static void
32 kgnilnd_dump_blob(int level, char *prefix, void *buf, int len) {
33         __u64 *ptr;
34
35         ptr = (__u64 *) buf;
36
37         while (len > 0) {
38                 if (len >= 32) {
39                         CDEBUG(level,
40                                "%s 0x%p: 0x%16.16llx 0x%16.16llx 0x%16.16llx 0x%16.16llx\n",
41                                prefix, ptr, *(ptr), *(ptr + 1), *(ptr + 2), *(ptr + 3));
42                         ptr += 4;
43                         len -= 32;
44                 } else if (len >= 16) {
45                         CDEBUG(level,
46                                "%s 0x%p: 0x%16.16llx 0x%16.16llx\n",
47                                prefix, ptr, *(ptr), *(ptr + 1));
48                         ptr += 2;
49                         len -= 16;
50                 } else {
51                         CDEBUG(level, "%s 0x%p: 0x%16.16llx\n",
52                                prefix, ptr, *(ptr));
53                         ptr++;
54                         len -= 8;
55                 }
56         }
57 }
58
59 static void
60 kgnilnd_dump_msg(int mask, kgn_msg_t *msg)
61 {
62         CDEBUG(mask, "0x%8.8x 0x%4.4x 0x%4.4x 0x%16.16llx"
63                 " 0x%16.16llx 0x%8.8x 0x%4.4x 0x%4.4x 0x%8.8x\n",
64                 msg->gnm_magic, msg->gnm_version,
65                 msg->gnm_type, msg->gnm_srcnid,
66                 msg->gnm_connstamp, msg->gnm_seq,
67                 msg->gnm_cksum, msg->gnm_payload_cksum,
68                 msg->gnm_payload_len);
69 }
70
71 void
72 kgnilnd_schedule_device(kgn_device_t *dev)
73 {
74         short         already_live = 0;
75
76         /* we'll only want to wake if the scheduler thread
77          * has come around and set ready to zero */
78         already_live = cmpxchg(&dev->gnd_ready, GNILND_DEV_IDLE, GNILND_DEV_IRQ);
79
80         if (!already_live) {
81                 wake_up_all(&dev->gnd_waitq);
82         }
83         return;
84 }
85
86 void kgnilnd_schedule_device_timer(unsigned long arg)
87 {
88         kgn_device_t *dev = (kgn_device_t *) arg;
89
90         kgnilnd_schedule_device(dev);
91 }
92
93 void
94 kgnilnd_device_callback(__u32 devid, __u64 arg)
95 {
96         kgn_device_t *dev;
97         int           index = (int) arg;
98
99         if (index >= kgnilnd_data.kgn_ndevs) {
100                 /* use _EMERG instead of an LBUG to prevent LBUG'ing in
101                  * interrupt context. */
102                 LCONSOLE_EMERG("callback for unknown device %d->%d\n",
103                                 devid, index);
104                 return;
105         }
106
107         dev = &kgnilnd_data.kgn_devices[index];
108         /* just basic sanity */
109         if (dev->gnd_id == devid) {
110                 kgnilnd_schedule_device(dev);
111         } else {
112                 LCONSOLE_EMERG("callback for bad device %d devid %d\n",
113                                 dev->gnd_id, devid);
114         }
115 }
116
117 /* sched_intent values:
118  * < 0 : do not reschedule under any circumstances
119  * == 0: reschedule if someone marked him WANTS_SCHED
120  * > 0 : force a reschedule */
121 /* Return code 0 means it did not schedule the conn, 1
122  *  means it succesfully scheduled the conn.
123  */
124
125 int
126 kgnilnd_schedule_process_conn(kgn_conn_t *conn, int sched_intent)
127 {
128         int     conn_sched;
129
130         /* move back to IDLE but save previous state.
131          * if we see WANTS_SCHED, we'll call kgnilnd_schedule_conn and
132          * let the xchg there handle any racing callers to get it
133          * onto gnd_ready_conns */
134
135         conn_sched = xchg(&conn->gnc_scheduled, GNILND_CONN_IDLE);
136         LASSERTF(conn_sched == GNILND_CONN_WANTS_SCHED ||
137                  conn_sched == GNILND_CONN_PROCESS,
138                  "conn %p after process in bad state: %d\n",
139                  conn, conn_sched);
140
141         if (sched_intent >= 0) {
142                 if ((sched_intent > 0 || (conn_sched == GNILND_CONN_WANTS_SCHED))) {
143                         return kgnilnd_schedule_conn_refheld(conn, 1);
144                 }
145         }
146         return 0;
147 }
148
149 /* Return of 0 for conn not scheduled, 1 returned if conn was scheduled or marked
150  * as scheduled */
151
152 int
153 _kgnilnd_schedule_conn(kgn_conn_t *conn, const char *caller, int line, int refheld)
154 {
155         kgn_device_t        *dev = conn->gnc_device;
156         int                  sched;
157         int                  rc;
158
159         sched = xchg(&conn->gnc_scheduled, GNILND_CONN_WANTS_SCHED);
160         /* we only care about the last person who marked want_sched since they
161          * are most likely the culprit
162          */
163         memcpy(conn->gnc_sched_caller, caller, sizeof(conn->gnc_sched_caller));
164         conn->gnc_sched_line = line;
165         /* if we are IDLE, add to list - only one guy sees IDLE and "wins"
166          * the chance to put it onto gnd_ready_conns.
167          * otherwise, leave marked as WANTS_SCHED and the thread that "owns"
168          *  the conn in process_conns will take care of moving it back to
169          *  SCHED when it is done processing */
170
171         if (sched == GNILND_CONN_IDLE) {
172                 /* if the conn is already scheduled, we've already requested
173                  * the scheduler thread wakeup */
174                 if (!refheld) {
175                         /* Add a reference to the conn if we are not holding a reference
176                          * already from the exisiting scheduler. We now use the same
177                          * reference if we need to reschedule a conn while in a scheduler
178                          * thread.
179                          */
180                         kgnilnd_conn_addref(conn);
181                 }
182                 LASSERTF(list_empty(&conn->gnc_schedlist), "conn %p already sched state %d\n",
183                          conn, sched);
184
185                 CDEBUG(D_INFO, "scheduling conn 0x%p caller %s:%d\n", conn, caller, line);
186
187                 spin_lock(&dev->gnd_lock);
188                 list_add_tail(&conn->gnc_schedlist, &dev->gnd_ready_conns);
189                 spin_unlock(&dev->gnd_lock);
190                 set_mb(conn->gnc_last_sched_ask, jiffies);
191                 rc = 1;
192         } else {
193                 CDEBUG(D_INFO, "not scheduling conn 0x%p: %d caller %s:%d\n", conn, sched, caller, line);
194                 rc = 0;
195         }
196
197         /* make sure thread(s) going to process conns - but let it make
198          * separate decision from conn schedule */
199         kgnilnd_schedule_device(dev);
200         return rc;
201 }
202
203 void
204 kgnilnd_schedule_dgram(kgn_device_t *dev)
205 {
206         int                  wake;
207
208         wake = xchg(&dev->gnd_dgram_ready, GNILND_DGRAM_SCHED);
209         if (wake != GNILND_DGRAM_SCHED)  {
210                 wake_up(&dev->gnd_dgram_waitq);
211         } else {
212                 CDEBUG(D_NETTRACE, "not waking: %d\n", wake);
213         }
214 }
215
216 void
217 kgnilnd_free_tx(kgn_tx_t *tx)
218 {
219         /* taken from kgnilnd_tx_add_state_locked */
220
221         LASSERTF((tx->tx_list_p == NULL &&
222                   tx->tx_list_state == GNILND_TX_ALLOCD) &&
223                 list_empty(&tx->tx_list),
224                 "tx %p with bad state %s (list_p %p) tx_list %s\n",
225                 tx, kgnilnd_tx_state2str(tx->tx_list_state), tx->tx_list_p,
226                 list_empty(&tx->tx_list) ? "empty" : "not empty");
227
228         atomic_dec(&kgnilnd_data.kgn_ntx);
229
230         /* we only allocate this if we need to */
231         if (tx->tx_phys != NULL) {
232                 kmem_cache_free(kgnilnd_data.kgn_tx_phys_cache, tx->tx_phys);
233                 CDEBUG(D_MALLOC, "slab-freed 'tx_phys': %lu at %p.\n",
234                        LNET_MAX_IOV * sizeof(gni_mem_segment_t), tx->tx_phys);
235         }
236
237         /* Only free the buffer if we used it */
238         if (tx->tx_buffer_copy != NULL) {
239                 vfree(tx->tx_buffer_copy);
240                 tx->tx_buffer_copy = NULL;
241                 CDEBUG(D_MALLOC, "vfreed buffer2\n");
242         }
243 #if 0
244         KGNILND_POISON(tx, 0x5a, sizeof(kgn_tx_t));
245 #endif
246         CDEBUG(D_MALLOC, "slab-freed 'tx': %lu at %p.\n", sizeof(*tx), tx);
247         kmem_cache_free(kgnilnd_data.kgn_tx_cache, tx);
248 }
249
250 kgn_tx_t *
251 kgnilnd_alloc_tx (void)
252 {
253         kgn_tx_t        *tx = NULL;
254
255         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_ALLOC_TX))
256                 return tx;
257
258         tx = kmem_cache_alloc(kgnilnd_data.kgn_tx_cache, GFP_ATOMIC);
259         if (tx == NULL) {
260                 CERROR("failed to allocate tx\n");
261                 return NULL;
262         }
263         CDEBUG(D_MALLOC, "slab-alloced 'tx': %lu at %p.\n",
264                sizeof(*tx), tx);
265
266         /* need this memset, cache alloc'd memory is not cleared */
267         memset(tx, 0, sizeof(*tx));
268
269         /* setup everything here to minimize time under the lock */
270         tx->tx_buftype = GNILND_BUF_NONE;
271         tx->tx_msg.gnm_type = GNILND_MSG_NONE;
272         INIT_LIST_HEAD(&tx->tx_list);
273         INIT_LIST_HEAD(&tx->tx_map_list);
274         tx->tx_list_state = GNILND_TX_ALLOCD;
275
276         atomic_inc(&kgnilnd_data.kgn_ntx);
277
278         return tx;
279 }
280
281 /* csum_fold needs to be run on the return value before shipping over the wire */
282 #define _kgnilnd_cksum(seed, ptr, nob)  csum_partial(ptr, nob, seed)
283
284 /* we don't use offset as every one is passing a buffer reference that already
285  * includes the offset into the base address -
286  *  see kgnilnd_setup_virt_buffer and kgnilnd_setup_immediate_buffer */
287 static inline __u16
288 kgnilnd_cksum(void *ptr, size_t nob)
289 {
290         __u16   sum;
291
292         sum = csum_fold(_kgnilnd_cksum(0, ptr, nob));
293
294         /* don't use magic 'no checksum' value */
295         if (sum == 0)
296                 sum = 1;
297
298         CDEBUG(D_INFO, "cksum 0x%x for ptr 0x%p sz %zu\n",
299                sum, ptr, nob);
300
301         return sum;
302 }
303
304 inline __u16
305 kgnilnd_cksum_kiov(unsigned int nkiov, lnet_kiov_t *kiov,
306                     unsigned int offset, unsigned int nob, int dump_blob)
307 {
308         __wsum             cksum = 0;
309         __wsum             tmpck;
310         __u16              retsum;
311         void              *addr;
312         unsigned int       fraglen;
313         int                i, odd;
314
315         LASSERT(nkiov > 0);
316         LASSERT(nob > 0);
317
318         CDEBUG(D_BUFFS, "calc cksum for kiov 0x%p nkiov %u offset %u nob %u, dump %d\n",
319                kiov, nkiov, offset, nob, dump_blob);
320
321         /* if loops changes, please change kgnilnd_setup_phys_buffer */
322
323         while (offset >= kiov->kiov_len) {
324                 offset -= kiov->kiov_len;
325                 nkiov--;
326                 kiov++;
327                 LASSERT(nkiov > 0);
328         }
329
330         /* ignore nob here, if nob < (kiov_len - offset), kiov == 1 */
331         odd = (unsigned long) (kiov[0].kiov_len - offset) & 1;
332
333         if ((odd || *kgnilnd_tunables.kgn_vmap_cksum) && nkiov > 1) {
334                 struct page **pages = kgnilnd_data.kgn_cksum_map_pages[get_cpu()];
335
336                 LASSERTF(pages != NULL, "NULL pages for cpu %d map_pages 0x%p\n",
337                          get_cpu(), kgnilnd_data.kgn_cksum_map_pages);
338
339                 CDEBUG(D_BUFFS, "odd %d len %u offset %u nob %u\n",
340                        odd, kiov[0].kiov_len, offset, nob);
341
342                 for (i = 0; i < nkiov; i++) {
343                         pages[i] = kiov[i].kiov_page;
344                 }
345
346                 addr = vmap(pages, nkiov, VM_MAP, PAGE_KERNEL);
347                 if (addr == NULL) {
348                         CNETERR("Couldn't vmap %d frags on %d bytes to avoid odd length fragment in cksum\n",
349                                 nkiov, nob);
350                         /* return zero to avoid killing tx - we'll just get warning on console
351                          * when remote end sees zero checksum */
352                         RETURN(0);
353                 }
354                 atomic_inc(&kgnilnd_data.kgn_nvmap_cksum);
355
356                 tmpck = _kgnilnd_cksum(0, (void *) addr + kiov[0].kiov_offset + offset, nob);
357                 cksum = tmpck;
358
359                 if (dump_blob) {
360                         kgnilnd_dump_blob(D_BUFFS, "flat kiov RDMA payload",
361                                           (void *)addr + kiov[0].kiov_offset + offset, nob);
362                 }
363                 CDEBUG(D_BUFFS, "cksum 0x%x (+0x%x) for addr 0x%p+%u len %u offset %u\n",
364                        cksum, tmpck, addr, kiov[0].kiov_offset, nob, offset);
365                 vunmap(addr);
366         } else {
367                 do {
368                         fraglen = min(kiov->kiov_len - offset, nob);
369
370                         /* make dang sure we don't send a bogus checksum if somehow we get
371                          * an odd length fragment on anything but the last entry in a kiov  -
372                          * we know from kgnilnd_setup_rdma_buffer that we can't have non
373                          * PAGE_SIZE pages in the middle, so if nob < PAGE_SIZE, it is the last one */
374                         LASSERTF(!(fraglen&1) || (nob < PAGE_SIZE),
375                                  "odd fraglen %u on nkiov %d, nob %u kiov_len %u offset %u kiov 0x%p\n",
376                                  fraglen, nkiov, nob, kiov->kiov_len, offset, kiov);
377
378                         addr = (void *)kmap(kiov->kiov_page) + kiov->kiov_offset + offset;
379                         tmpck = _kgnilnd_cksum(cksum, addr, fraglen);
380
381                         CDEBUG(D_BUFFS,
382                                "cksum 0x%x (+0x%x) for page 0x%p+%u (0x%p) len %u offset %u\n",
383                                cksum, tmpck, kiov->kiov_page, kiov->kiov_offset, addr,
384                                fraglen, offset);
385
386                         cksum = tmpck;
387
388                         if (dump_blob)
389                                 kgnilnd_dump_blob(D_BUFFS, "kiov cksum", addr, fraglen);
390
391                         kunmap(kiov->kiov_page);
392
393                         kiov++;
394                         nkiov--;
395                         nob -= fraglen;
396                         offset = 0;
397
398                         /* iov must not run out before end of data */
399                         LASSERTF(nob == 0 || nkiov > 0, "nob %u nkiov %u\n", nob, nkiov);
400
401                 } while (nob > 0);
402         }
403
404         retsum = csum_fold(cksum);
405
406         /* don't use magic 'no checksum' value */
407         if (retsum == 0)
408                 retsum = 1;
409
410         CDEBUG(D_BUFFS, "retsum 0x%x from cksum 0x%x\n", retsum, cksum);
411
412         return retsum;
413 }
414
415 void
416 kgnilnd_init_msg(kgn_msg_t *msg, int type, lnet_nid_t source)
417 {
418         msg->gnm_magic = GNILND_MSG_MAGIC;
419         msg->gnm_version = GNILND_MSG_VERSION;
420         msg->gnm_type = type;
421         msg->gnm_payload_len = 0;
422         msg->gnm_srcnid = source;
423         /* gnm_connstamp gets set when FMA is sent */
424         /* gnm_srcnid is set on creation via function argument
425          * The right interface/net and nid is passed in when the message
426          * is created.
427          */
428 }
429
430 kgn_tx_t *
431 kgnilnd_new_tx_msg(int type, lnet_nid_t source)
432 {
433         kgn_tx_t *tx = kgnilnd_alloc_tx();
434
435         if (tx != NULL) {
436                 kgnilnd_init_msg(&tx->tx_msg, type, source);
437         } else {
438                 CERROR("couldn't allocate new tx type %s!\n",
439                        kgnilnd_msgtype2str(type));
440         }
441
442         return tx;
443 }
444
445 static void
446 kgnilnd_nak_rdma(kgn_conn_t *conn, int rx_type, int error, __u64 cookie, lnet_nid_t source) {
447         kgn_tx_t        *tx;
448
449         int             nak_type;
450
451         switch (rx_type) {
452         case GNILND_MSG_GET_REQ:
453         case GNILND_MSG_GET_DONE:
454                 nak_type = GNILND_MSG_GET_NAK;
455                 break;
456         case GNILND_MSG_PUT_REQ:
457         case GNILND_MSG_PUT_ACK:
458         case GNILND_MSG_PUT_DONE:
459                 nak_type = GNILND_MSG_PUT_NAK;
460                 break;
461         case GNILND_MSG_PUT_REQ_REV:
462         case GNILND_MSG_PUT_DONE_REV:
463                 nak_type = GNILND_MSG_PUT_NAK_REV;
464                 break;
465         case GNILND_MSG_GET_REQ_REV:
466         case GNILND_MSG_GET_ACK_REV:
467         case GNILND_MSG_GET_DONE_REV:
468                 nak_type = GNILND_MSG_GET_NAK_REV;
469                 break;
470         default:
471                 CERROR("invalid msg type %s (%d)\n",
472                         kgnilnd_msgtype2str(rx_type), rx_type);
473                 LBUG();
474         }
475         /* only allow NAK on error and truncate to zero */
476         LASSERTF(error <= 0, "error %d conn 0x%p, cookie "LPU64"\n",
477                  error, conn, cookie);
478
479         tx = kgnilnd_new_tx_msg(nak_type, source);
480         if (tx == NULL) {
481                 CNETERR("can't get TX to NAK RDMA to %s\n",
482                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
483                 return;
484         }
485
486         tx->tx_msg.gnm_u.completion.gncm_retval = error;
487         tx->tx_msg.gnm_u.completion.gncm_cookie = cookie;
488         kgnilnd_queue_tx(conn, tx);
489 }
490
491 int
492 kgnilnd_setup_immediate_buffer(kgn_tx_t *tx, unsigned int niov, struct iovec *iov,
493                                lnet_kiov_t *kiov, unsigned int offset, unsigned int nob)
494
495 {
496         kgn_msg_t       *msg = &tx->tx_msg;
497         int              i;
498
499         /* To help save on MDDs for short messages, we'll vmap a kiov to allow
500          * gni_smsg_send to send that as the payload */
501
502         LASSERT(tx->tx_buftype == GNILND_BUF_NONE);
503
504         if (nob == 0) {
505                 tx->tx_buffer = NULL;
506         } else if (kiov != NULL) {
507                 LASSERTF(niov > 0 && niov < GNILND_MAX_IMMEDIATE/PAGE_SIZE,
508                          "bad niov %d\n", niov);
509
510                 while (offset >= kiov->kiov_len) {
511                         offset -= kiov->kiov_len;
512                         niov--;
513                         kiov++;
514                         LASSERT(niov > 0);
515                 }
516                 for (i = 0; i < niov; i++) {
517                         /* We can't have a kiov_offset on anything but the first entry,
518                          * otherwise we'll have a hole at the end of the mapping as we only map
519                          * whole pages.
520                          * Also, if we have a kiov_len < PAGE_SIZE but we need to map more
521                          * than kiov_len, we will also have a whole at the end of that page
522                          * which isn't allowed */
523                         if ((kiov[i].kiov_offset != 0 && i > 0) ||
524                             (kiov[i].kiov_offset + kiov[i].kiov_len != PAGE_SIZE && i < niov - 1)) {
525                                 CNETERR("Can't make payload contiguous in I/O VM:"
526                                        "page %d, offset %u, nob %u, kiov_offset %u kiov_len %u \n",
527                                        i, offset, nob, kiov->kiov_offset, kiov->kiov_len);
528                                 RETURN(-EINVAL);
529                         }
530                         tx->tx_imm_pages[i] = kiov[i].kiov_page;
531                 }
532
533                 /* hijack tx_phys for the later unmap */
534                 if (niov == 1) {
535                         /* tx->phyx being equal to NULL is the signal for unmap to discern between kmap and vmap */
536                         tx->tx_phys = NULL;
537                         tx->tx_buffer = (void *)kmap(tx->tx_imm_pages[0]) + kiov[0].kiov_offset + offset;
538                         atomic_inc(&kgnilnd_data.kgn_nkmap_short);
539                         GNIDBG_TX(D_NET, tx, "kmapped page for %d bytes for kiov 0x%p, buffer 0x%p",
540                                 nob, kiov, tx->tx_buffer);
541                 } else {
542                         tx->tx_phys = vmap(tx->tx_imm_pages, niov, VM_MAP, PAGE_KERNEL);
543                         if (tx->tx_phys == NULL) {
544                                 CNETERR("Couldn't vmap %d frags on %d bytes\n", niov, nob);
545                                 RETURN(-ENOMEM);
546
547                         }
548                         atomic_inc(&kgnilnd_data.kgn_nvmap_short);
549                         /* make sure we take into account the kiov offset as the start of the buffer */
550                         tx->tx_buffer = (void *)tx->tx_phys + kiov[0].kiov_offset + offset;
551                         GNIDBG_TX(D_NET, tx, "mapped %d pages for %d bytes from kiov 0x%p to 0x%p, buffer 0x%p",
552                                 niov, nob, kiov, tx->tx_phys, tx->tx_buffer);
553                 }
554                 tx->tx_buftype = GNILND_BUF_IMMEDIATE_KIOV;
555                 tx->tx_nob = nob;
556
557         } else {
558                 /* For now this is almost identical to kgnilnd_setup_virt_buffer, but we
559                  * could "flatten" the payload into a single contiguous buffer ready
560                  * for sending direct over an FMA if we ever needed to. */
561
562                 LASSERT(niov > 0);
563
564                 while (offset >= iov->iov_len) {
565                         offset -= iov->iov_len;
566                         niov--;
567                         iov++;
568                         LASSERT(niov > 0);
569                 }
570
571                 if (nob > iov->iov_len - offset) {
572                         CERROR("Can't handle multiple vaddr fragments\n");
573                         return -EMSGSIZE;
574                 }
575
576                 tx->tx_buffer = (void *)(((unsigned long)iov->iov_base) + offset);
577
578                 tx->tx_buftype = GNILND_BUF_IMMEDIATE;
579                 tx->tx_nob = nob;
580         }
581
582         /* checksum payload early - it shouldn't be changing after lnd_send */
583         if (*kgnilnd_tunables.kgn_checksum >= 2) {
584                 msg->gnm_payload_cksum = kgnilnd_cksum(tx->tx_buffer, nob);
585                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SMSG_CKSUM2)) {
586                         msg->gnm_payload_cksum += 0xe00e;
587                 }
588                 if (*kgnilnd_tunables.kgn_checksum_dump > 1) {
589                         kgnilnd_dump_blob(D_BUFFS, "payload checksum",
590                                           tx->tx_buffer, nob);
591                 }
592         } else {
593                 msg->gnm_payload_cksum = 0;
594         }
595
596         return 0;
597 }
598
599 int
600 kgnilnd_setup_virt_buffer(kgn_tx_t *tx,
601                           unsigned int niov, struct iovec *iov,
602                           unsigned int offset, unsigned int nob)
603
604 {
605         LASSERT(nob > 0);
606         LASSERT(niov > 0);
607         LASSERT(tx->tx_buftype == GNILND_BUF_NONE);
608
609         while (offset >= iov->iov_len) {
610                 offset -= iov->iov_len;
611                 niov--;
612                 iov++;
613                 LASSERT(niov > 0);
614         }
615
616         if (nob > iov->iov_len - offset) {
617                 CERROR("Can't handle multiple vaddr fragments\n");
618                 return -EMSGSIZE;
619         }
620
621         tx->tx_buftype = GNILND_BUF_VIRT_UNMAPPED;
622         tx->tx_nob = nob;
623         tx->tx_buffer = (void *)(((unsigned long)iov->iov_base) + offset);
624         return 0;
625 }
626
627 int
628 kgnilnd_setup_phys_buffer(kgn_tx_t *tx, int nkiov, lnet_kiov_t *kiov,
629                           unsigned int offset, unsigned int nob)
630 {
631         gni_mem_segment_t *phys;
632         int             rc = 0;
633         unsigned int    fraglen;
634
635         GNIDBG_TX(D_NET, tx, "niov %d kiov 0x%p offset %u nob %u", nkiov, kiov, offset, nob);
636
637         LASSERT(nob > 0);
638         LASSERT(nkiov > 0);
639         LASSERT(tx->tx_buftype == GNILND_BUF_NONE);
640
641         /* only allocate this if we are going to use it */
642         tx->tx_phys = kmem_cache_alloc(kgnilnd_data.kgn_tx_phys_cache,
643                                               GFP_ATOMIC);
644         if (tx->tx_phys == NULL) {
645                 CERROR("failed to allocate tx_phys\n");
646                 rc = -ENOMEM;
647                 GOTO(error, rc);
648         }
649
650         CDEBUG(D_MALLOC, "slab-alloced 'tx->tx_phys': %lu at %p.\n",
651                LNET_MAX_IOV * sizeof(gni_mem_segment_t), tx->tx_phys);
652
653         /* if loops changes, please change kgnilnd_cksum_kiov
654          *   and kgnilnd_setup_immediate_buffer */
655
656         while (offset >= kiov->kiov_len) {
657                 offset -= kiov->kiov_len;
658                 nkiov--;
659                 kiov++;
660                 LASSERT(nkiov > 0);
661         }
662
663         /* at this point, kiov points to the first page that we'll actually map
664          * now that we've seeked into the koiv for offset and dropped any
665          * leading pages that fall entirely within the offset */
666         tx->tx_buftype = GNILND_BUF_PHYS_UNMAPPED;
667         tx->tx_nob = nob;
668
669         /* kiov_offset is start of 'valid' buffer, so index offset past that */
670         tx->tx_buffer = (void *)((unsigned long)(kiov->kiov_offset + offset));
671         phys = tx->tx_phys;
672
673         CDEBUG(D_NET, "tx 0x%p buffer 0x%p map start kiov 0x%p+%u niov %d offset %u\n",
674                tx, tx->tx_buffer, kiov, kiov->kiov_offset, nkiov, offset);
675
676         do {
677                 fraglen = min(kiov->kiov_len - offset, nob);
678
679                 /* We can't have a kiov_offset on anything but the first entry,
680                  * otherwise we'll have a hole at the end of the mapping as we only map
681                  * whole pages. Only the first page is allowed to have an offset -
682                  * we'll add that into tx->tx_buffer and that will get used when we
683                  * map in the segments (see kgnilnd_map_buffer).
684                  * Also, if we have a kiov_len < PAGE_SIZE but we need to map more
685                  * than kiov_len, we will also have a whole at the end of that page
686                  * which isn't allowed */
687                 if ((phys != tx->tx_phys) &&
688                     ((kiov->kiov_offset != 0) ||
689                      ((kiov->kiov_len < PAGE_SIZE) && (nob > kiov->kiov_len)))) {
690                         CERROR("Can't make payload contiguous in I/O VM:"
691                                "page %d, offset %u, nob %u, kiov_offset %u kiov_len %u \n",
692                                (int)(phys - tx->tx_phys),
693                                offset, nob, kiov->kiov_offset, kiov->kiov_len);
694                         rc = -EINVAL;
695                         GOTO(error, rc);
696                 }
697
698                 if ((phys - tx->tx_phys) == LNET_MAX_IOV) {
699                         CERROR ("payload too big (%d)\n", (int)(phys - tx->tx_phys));
700                         rc = -EMSGSIZE;
701                         GOTO(error, rc);
702                 }
703
704                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_PHYS_SETUP)) {
705                         rc = -EINVAL;
706                         GOTO(error, rc);
707                 }
708
709                 CDEBUG(D_BUFFS, "page 0x%p kiov_offset %u kiov_len %u nob %u "
710                                "nkiov %u offset %u\n",
711                       kiov->kiov_page, kiov->kiov_offset, kiov->kiov_len, nob, nkiov, offset);
712
713                 phys->address = page_to_phys(kiov->kiov_page);
714                 phys++;
715                 kiov++;
716                 nkiov--;
717                 nob -= fraglen;
718                 offset = 0;
719
720                 /* iov must not run out before end of data */
721                 LASSERTF(nob == 0 || nkiov > 0, "nob %u nkiov %u\n", nob, nkiov);
722
723         } while (nob > 0);
724
725         tx->tx_phys_npages = phys - tx->tx_phys;
726
727         return 0;
728
729 error:
730         if (tx->tx_phys != NULL) {
731                 kmem_cache_free(kgnilnd_data.kgn_tx_phys_cache, tx->tx_phys);
732                 CDEBUG(D_MALLOC, "slab-freed 'tx_phys': %lu at %p.\n",
733                        sizeof(*tx->tx_phys), tx->tx_phys);
734                 tx->tx_phys = NULL;
735         }
736         return rc;
737 }
738
739 static inline int
740 kgnilnd_setup_rdma_buffer(kgn_tx_t *tx, unsigned int niov,
741                           struct iovec *iov, lnet_kiov_t *kiov,
742                           unsigned int offset, unsigned int nob)
743 {
744         int     rc;
745
746         LASSERTF((iov == NULL) != (kiov == NULL), "iov 0x%p, kiov 0x%p, tx 0x%p,"
747                                                 " offset %d, nob %d, niov %d\n"
748                                                 , iov, kiov, tx, offset, nob, niov);
749
750         if (kiov != NULL) {
751                 rc = kgnilnd_setup_phys_buffer(tx, niov, kiov, offset, nob);
752         } else {
753                 rc = kgnilnd_setup_virt_buffer(tx, niov, iov, offset, nob);
754         }
755         return rc;
756 }
757
758 /* kgnilnd_parse_lnet_rdma()
759  * lntmsg - message passed in from lnet.
760  * niov, kiov, offset - see lnd_t in lib-types.h for descriptions.
761  * nob - actual number of bytes to in this message.
762  * put_len - It is possible for PUTs to have a different length than the
763  *           length stored in lntmsg->msg_len since LNET can adjust this
764  *           length based on it's buffer size and offset.
765  *           lnet_try_match_md() sets the mlength that we use to do the RDMA
766  *           transfer.
767  */
768 static void
769 kgnilnd_parse_lnet_rdma(lnet_msg_t *lntmsg, unsigned int *niov,
770                         unsigned int *offset, unsigned int *nob,
771                         lnet_kiov_t **kiov, int put_len)
772 {
773         /* GETs are weird, see kgnilnd_send */
774         if (lntmsg->msg_type == LNET_MSG_GET) {
775                 if ((lntmsg->msg_md->md_options & LNET_MD_KIOV) == 0) {
776                         *kiov = NULL;
777                 } else {
778                         *kiov = lntmsg->msg_md->md_iov.kiov;
779                 }
780                 *niov = lntmsg->msg_md->md_niov;
781                 *nob = lntmsg->msg_md->md_length;
782                 *offset = 0;
783         } else {
784                 *kiov = lntmsg->msg_kiov;
785                 *niov = lntmsg->msg_niov;
786                 *nob = put_len;
787                 *offset = lntmsg->msg_offset;
788         }
789 }
790
791 static inline void
792 kgnilnd_compute_rdma_cksum(kgn_tx_t *tx, int put_len)
793 {
794         unsigned int     niov, offset, nob;
795         lnet_kiov_t     *kiov;
796         lnet_msg_t      *lntmsg = tx->tx_lntmsg[0];
797         int              dump_cksum = (*kgnilnd_tunables.kgn_checksum_dump > 1);
798
799         GNITX_ASSERTF(tx, ((tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE) ||
800                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE) ||
801                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE_REV) ||
802                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV) ||
803                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_ACK_REV) ||
804                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_REQ_REV)),
805                       "bad type %s", kgnilnd_msgtype2str(tx->tx_msg.gnm_type));
806
807         if ((tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE_REV) ||
808             (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV)) {
809                 tx->tx_msg.gnm_payload_cksum = 0;
810                 return;
811         }
812         if (*kgnilnd_tunables.kgn_checksum < 3) {
813                 tx->tx_msg.gnm_payload_cksum = 0;
814                 return;
815         }
816
817         GNITX_ASSERTF(tx, lntmsg, "no LNet message!", NULL);
818
819         kgnilnd_parse_lnet_rdma(lntmsg, &niov, &offset, &nob, &kiov,
820                                 put_len);
821
822         if (kiov != NULL) {
823                 tx->tx_msg.gnm_payload_cksum = kgnilnd_cksum_kiov(niov, kiov, offset, nob, dump_cksum);
824         } else {
825                 tx->tx_msg.gnm_payload_cksum = kgnilnd_cksum(tx->tx_buffer, nob);
826                 if (dump_cksum) {
827                         kgnilnd_dump_blob(D_BUFFS, "peer RDMA payload", tx->tx_buffer, nob);
828                 }
829         }
830
831         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SMSG_CKSUM3)) {
832                 tx->tx_msg.gnm_payload_cksum += 0xd00d;
833         }
834 }
835
836 /* kgnilnd_verify_rdma_cksum()
837  * tx - PUT_DONE/GET_DONE matched tx.
838  * rx_cksum - received checksum to compare against.
839  * put_len - see kgnilnd_parse_lnet_rdma comments.
840  */
841 static inline int
842 kgnilnd_verify_rdma_cksum(kgn_tx_t *tx, __u16 rx_cksum, int put_len)
843 {
844         int              rc = 0;
845         __u16            cksum;
846         unsigned int     niov, offset, nob;
847         lnet_kiov_t     *kiov;
848         lnet_msg_t      *lntmsg = tx->tx_lntmsg[0];
849         int dump_on_err = *kgnilnd_tunables.kgn_checksum_dump;
850
851         /* we can only match certain requests */
852         GNITX_ASSERTF(tx, ((tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) ||
853                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK) ||
854                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_REQ_REV) ||
855                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_ACK_REV) ||
856                            (tx->tx_msg.gnm_type == GNILND_MSG_GET_DONE_REV) ||
857                            (tx->tx_msg.gnm_type == GNILND_MSG_PUT_DONE_REV)),
858                       "bad type %s", kgnilnd_msgtype2str(tx->tx_msg.gnm_type));
859
860         if ((tx->tx_msg.gnm_type == GNILND_MSG_PUT_REQ_REV) ||
861             (tx->tx_msg.gnm_type == GNILND_MSG_GET_ACK_REV)) {
862                 return 0;
863         }
864
865         if (rx_cksum == 0)  {
866                 if (*kgnilnd_tunables.kgn_checksum >= 3) {
867                         GNIDBG_MSG(D_WARNING, &tx->tx_msg,
868                                    "no RDMA payload checksum when enabled");
869                 }
870                 return 0;
871         }
872
873         GNITX_ASSERTF(tx, lntmsg, "no LNet message!", NULL);
874
875         kgnilnd_parse_lnet_rdma(lntmsg, &niov, &offset, &nob, &kiov, put_len);
876
877         if (kiov != NULL) {
878                 cksum = kgnilnd_cksum_kiov(niov, kiov, offset, nob, 0);
879         } else {
880                 cksum = kgnilnd_cksum(tx->tx_buffer, nob);
881         }
882
883         if (cksum != rx_cksum) {
884                 GNIDBG_MSG(D_NETERROR, &tx->tx_msg,
885                            "Bad RDMA payload checksum (%x expected %x); "
886                            "kiov 0x%p niov %d nob %u offset %u",
887                             cksum, rx_cksum, kiov, niov, nob, offset);
888                 switch (dump_on_err) {
889                 case 2:
890                         if (kiov != NULL) {
891                                 kgnilnd_cksum_kiov(niov, kiov, offset, nob, 1);
892                         } else {
893                                 kgnilnd_dump_blob(D_BUFFS, "RDMA payload",
894                                                   tx->tx_buffer, nob);
895                         }
896                         /* fall through to dump log */
897                 case 1:
898                         libcfs_debug_dumplog();
899                         break;
900                 default:
901                         break;
902                 }
903                 rc = -ENOKEY;
904                 /* kgnilnd_check_fma_rx will close conn, kill tx with error */
905         }
906         return rc;
907 }
908
909 void
910 kgnilnd_mem_add_map_list(kgn_device_t *dev, kgn_tx_t *tx)
911 {
912         int     bytes;
913
914         GNITX_ASSERTF(tx, list_empty(&tx->tx_map_list),
915                 "already mapped!", NULL);
916
917         spin_lock(&dev->gnd_map_lock);
918         switch (tx->tx_buftype) {
919         default:
920                 GNIDBG_TX(D_EMERG, tx,
921                         "SOFTWARE BUG: invalid mapping %d", tx->tx_buftype);
922                 spin_unlock(&dev->gnd_map_lock);
923                 LBUG();
924                 break;
925
926         case GNILND_BUF_PHYS_MAPPED:
927                 bytes = tx->tx_phys_npages * PAGE_SIZE;
928                 dev->gnd_map_nphys++;
929                 dev->gnd_map_physnop += tx->tx_phys_npages;
930                 break;
931
932         case GNILND_BUF_VIRT_MAPPED:
933                 bytes = tx->tx_nob;
934                 dev->gnd_map_nvirt++;
935                 dev->gnd_map_virtnob += tx->tx_nob;
936                 break;
937         }
938
939         if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK ||
940             tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
941                 atomic64_add(bytes, &dev->gnd_rdmaq_bytes_out);
942                 GNIDBG_TX(D_NETTRACE, tx, "rdma ++ %d to "LPD64"",
943                           bytes, atomic64_read(&dev->gnd_rdmaq_bytes_out));
944         }
945
946         atomic_inc(&dev->gnd_n_mdd);
947         atomic64_add(bytes, &dev->gnd_nbytes_map);
948
949         /* clear retrans to prevent any SMSG goofiness as that code uses the same counter */
950         tx->tx_retrans = 0;
951
952         /* we only get here in the valid cases */
953         list_add_tail(&tx->tx_map_list, &dev->gnd_map_list);
954         dev->gnd_map_version++;
955         spin_unlock(&dev->gnd_map_lock);
956 }
957
958 void
959 kgnilnd_mem_del_map_list(kgn_device_t *dev, kgn_tx_t *tx)
960 {
961         int     bytes;
962
963         GNITX_ASSERTF(tx, !list_empty(&tx->tx_map_list),
964                 "not mapped!", NULL);
965         spin_lock(&dev->gnd_map_lock);
966
967         switch (tx->tx_buftype) {
968         default:
969                 GNIDBG_TX(D_EMERG, tx,
970                         "SOFTWARE BUG: invalid mapping %d", tx->tx_buftype);
971                 spin_unlock(&dev->gnd_map_lock);
972                 LBUG();
973                 break;
974
975         case GNILND_BUF_PHYS_UNMAPPED:
976                 bytes = tx->tx_phys_npages * PAGE_SIZE;
977                 dev->gnd_map_nphys--;
978                 dev->gnd_map_physnop -= tx->tx_phys_npages;
979                 break;
980
981         case GNILND_BUF_VIRT_UNMAPPED:
982                 bytes = tx->tx_nob;
983                 dev->gnd_map_nvirt--;
984                 dev->gnd_map_virtnob -= tx->tx_nob;
985                 break;
986         }
987
988         if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK ||
989             tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
990                 atomic64_sub(bytes, &dev->gnd_rdmaq_bytes_out);
991                 LASSERTF(atomic64_read(&dev->gnd_rdmaq_bytes_out) >= 0,
992                          "bytes_out negative! %ld\n", atomic64_read(&dev->gnd_rdmaq_bytes_out));
993                 GNIDBG_TX(D_NETTRACE, tx, "rdma -- %d to "LPD64"",
994                           bytes, atomic64_read(&dev->gnd_rdmaq_bytes_out));
995         }
996
997         atomic_dec(&dev->gnd_n_mdd);
998         atomic64_sub(bytes, &dev->gnd_nbytes_map);
999
1000         /* we only get here in the valid cases */
1001         list_del_init(&tx->tx_map_list);
1002         dev->gnd_map_version++;
1003         spin_unlock(&dev->gnd_map_lock);
1004 }
1005
1006 int
1007 kgnilnd_map_buffer(kgn_tx_t *tx)
1008 {
1009         kgn_conn_t       *conn = tx->tx_conn;
1010         kgn_device_t     *dev = conn->gnc_device;
1011         __u32             flags = GNI_MEM_READWRITE;
1012         gni_return_t      rrc;
1013
1014         /* The kgnilnd_mem_register(_segments) Gemini Driver functions can
1015          * be called concurrently as there are internal locks that protect
1016          * any data structures or HW resources. We just need to ensure
1017          * that our concurrency doesn't result in the kgn_device_t
1018          * getting nuked while we are in here */
1019
1020         LASSERTF(conn != NULL, "tx %p with NULL conn, someone forgot"
1021                 " to set tx_conn before calling %s\n", tx, __FUNCTION__);
1022
1023         if (unlikely(CFS_FAIL_CHECK(CFS_FAIL_GNI_MAP_TX)))
1024                 RETURN(-ENOMEM);
1025
1026         if (*kgnilnd_tunables.kgn_bte_relaxed_ordering) {
1027                 flags |= GNI_MEM_RELAXED_PI_ORDERING;
1028         }
1029
1030         switch (tx->tx_buftype) {
1031         default:
1032                 LBUG();
1033
1034         case GNILND_BUF_NONE:
1035         case GNILND_BUF_IMMEDIATE:
1036         case GNILND_BUF_IMMEDIATE_KIOV:
1037         case GNILND_BUF_PHYS_MAPPED:
1038         case GNILND_BUF_VIRT_MAPPED:
1039                 return 0;
1040
1041         case GNILND_BUF_PHYS_UNMAPPED:
1042                 GNITX_ASSERTF(tx, tx->tx_phys != NULL, "physical buffer not there!", NULL);
1043                 rrc = kgnilnd_mem_register_segments(dev->gnd_handle,
1044                         tx->tx_phys, tx->tx_phys_npages, NULL,
1045                         GNI_MEM_PHYS_SEGMENTS | flags,
1046                         &tx->tx_map_key);
1047                 /* could race with other uses of the map counts, but this is ok
1048                  * - this needs to turn into a non-fatal error soon to allow
1049                  *  GART resource, etc starvation handling */
1050                 if (rrc != GNI_RC_SUCCESS) {
1051                         GNIDBG_TX(D_NET, tx, "Can't map %d pages: dev %d "
1052                                 "phys %u pp %u, virt %u nob "LPU64"",
1053                                 tx->tx_phys_npages, dev->gnd_id,
1054                                 dev->gnd_map_nphys, dev->gnd_map_physnop,
1055                                 dev->gnd_map_nvirt, dev->gnd_map_virtnob);
1056                         RETURN(rrc == GNI_RC_ERROR_RESOURCE ? -ENOMEM : -EINVAL);
1057                 }
1058
1059                 tx->tx_buftype = GNILND_BUF_PHYS_MAPPED;
1060                 kgnilnd_mem_add_map_list(dev, tx);
1061                 return 0;
1062
1063         case GNILND_BUF_VIRT_UNMAPPED:
1064                 rrc = kgnilnd_mem_register(dev->gnd_handle,
1065                         (__u64)tx->tx_buffer, tx->tx_nob,
1066                         NULL, flags, &tx->tx_map_key);
1067                 if (rrc != GNI_RC_SUCCESS) {
1068                         GNIDBG_TX(D_NET, tx, "Can't map %u bytes: dev %d "
1069                                 "phys %u pp %u, virt %u nob "LPU64"",
1070                                 tx->tx_nob, dev->gnd_id,
1071                                 dev->gnd_map_nphys, dev->gnd_map_physnop,
1072                                 dev->gnd_map_nvirt, dev->gnd_map_virtnob);
1073                         RETURN(rrc == GNI_RC_ERROR_RESOURCE ? -ENOMEM : -EINVAL);
1074                 }
1075
1076                 tx->tx_buftype = GNILND_BUF_VIRT_MAPPED;
1077                 kgnilnd_mem_add_map_list(dev, tx);
1078                 if (tx->tx_msg.gnm_type == GNILND_MSG_PUT_ACK ||
1079                     tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
1080                         atomic64_add(tx->tx_nob, &dev->gnd_rdmaq_bytes_out);
1081                         GNIDBG_TX(D_NETTRACE, tx, "rdma ++ %d to %ld\n",
1082                                tx->tx_nob, atomic64_read(&dev->gnd_rdmaq_bytes_out));
1083                 }
1084
1085                 return 0;
1086         }
1087 }
1088
1089 void
1090 kgnilnd_add_purgatory_tx(kgn_tx_t *tx)
1091 {
1092         kgn_conn_t              *conn = tx->tx_conn;
1093         kgn_mdd_purgatory_t     *gmp;
1094
1095         LIBCFS_ALLOC(gmp, sizeof(*gmp));
1096         LASSERTF(gmp != NULL, "couldn't allocate MDD purgatory member;"
1097                 " asserting to avoid data corruption\n");
1098         if (tx->tx_buffer_copy)
1099                 gmp->gmp_map_key = tx->tx_buffer_copy_map_key;
1100         else
1101                 gmp->gmp_map_key = tx->tx_map_key;
1102
1103         atomic_inc(&conn->gnc_device->gnd_n_mdd_held);
1104
1105         /* ensure that we don't have a blank purgatory - indicating the
1106          * conn is not already on purgatory lists - we'd never recover these
1107          * MDD if that were the case */
1108         GNITX_ASSERTF(tx, conn->gnc_in_purgatory,
1109                 "conn 0x%p->%s with NULL purgatory",
1110                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
1111
1112         /* link 'er up! - only place we really need to lock for
1113          * concurrent access */
1114         spin_lock(&conn->gnc_list_lock);
1115         list_add_tail(&gmp->gmp_list, &conn->gnc_mdd_list);
1116         spin_unlock(&conn->gnc_list_lock);
1117 }
1118
1119 void
1120 kgnilnd_unmap_buffer(kgn_tx_t *tx, int error)
1121 {
1122         kgn_device_t     *dev;
1123         gni_return_t      rrc;
1124         int               hold_timeout = 0;
1125
1126         /* code below relies on +1 relationship ... */
1127         CLASSERT(GNILND_BUF_PHYS_MAPPED == (GNILND_BUF_PHYS_UNMAPPED + 1));
1128         CLASSERT(GNILND_BUF_VIRT_MAPPED == (GNILND_BUF_VIRT_UNMAPPED + 1));
1129
1130         switch (tx->tx_buftype) {
1131         default:
1132                 LBUG();
1133
1134         case GNILND_BUF_NONE:
1135         case GNILND_BUF_IMMEDIATE:
1136         case GNILND_BUF_PHYS_UNMAPPED:
1137         case GNILND_BUF_VIRT_UNMAPPED:
1138                 break;
1139         case GNILND_BUF_IMMEDIATE_KIOV:
1140                 if (tx->tx_phys != NULL) {
1141                         vunmap(tx->tx_phys);
1142                 } else if (tx->tx_phys == NULL && tx->tx_buffer != NULL) {
1143                         kunmap(tx->tx_imm_pages[0]);
1144                 }
1145                 /* clear to prevent kgnilnd_free_tx from thinking
1146                  * this is a RDMA descriptor */
1147                 tx->tx_phys = NULL;
1148                 break;
1149
1150         case GNILND_BUF_PHYS_MAPPED:
1151         case GNILND_BUF_VIRT_MAPPED:
1152                 LASSERT(tx->tx_conn != NULL);
1153
1154                 dev = tx->tx_conn->gnc_device;
1155
1156                 /* only want to hold if we are closing conn without
1157                  * verified peer notification  - the theory is that
1158                  * a TX error can be communicated in all other cases */
1159                 if (tx->tx_conn->gnc_state != GNILND_CONN_ESTABLISHED &&
1160                     kgnilnd_check_purgatory_conn(tx->tx_conn)) {
1161                         kgnilnd_add_purgatory_tx(tx);
1162
1163                         /* The timeout we give to kgni is a deadman stop only.
1164                          *  we are setting high to ensure we don't have the kgni timer
1165                          *  fire before ours fires _and_ is handled */
1166                         hold_timeout = GNILND_TIMEOUT2DEADMAN;
1167
1168                         GNIDBG_TX(D_NET, tx,
1169                                  "dev %p delaying MDD release for %dms key "LPX64"."LPX64"",
1170                                  tx->tx_conn->gnc_device, hold_timeout,
1171                                  tx->tx_map_key.qword1, tx->tx_map_key.qword2);
1172                 }
1173                 if (tx->tx_buffer_copy != NULL) {
1174                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_buffer_copy_map_key, hold_timeout);
1175                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1176                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_map_key, 0);
1177                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1178                 } else {
1179                         rrc = kgnilnd_mem_deregister(dev->gnd_handle, &tx->tx_map_key, hold_timeout);
1180                         LASSERTF(rrc == GNI_RC_SUCCESS, "rrc %d\n", rrc);
1181                 }
1182
1183                 tx->tx_buftype--;
1184                 kgnilnd_mem_del_map_list(dev, tx);
1185                 break;
1186         }
1187 }
1188
1189 void
1190 kgnilnd_tx_done(kgn_tx_t *tx, int completion)
1191 {
1192         lnet_msg_t      *lntmsg0, *lntmsg1;
1193         int             status0, status1;
1194         lnet_ni_t       *ni = NULL;
1195         kgn_conn_t      *conn = tx->tx_conn;
1196
1197         LASSERT(!in_interrupt());
1198
1199         lntmsg0 = tx->tx_lntmsg[0]; tx->tx_lntmsg[0] = NULL;
1200         lntmsg1 = tx->tx_lntmsg[1]; tx->tx_lntmsg[1] = NULL;
1201
1202         if (completion &&
1203             !(tx->tx_state & GNILND_TX_QUIET_ERROR) &&
1204             !kgnilnd_conn_clean_errno(completion)) {
1205                 GNIDBG_TOMSG(D_NETERROR, &tx->tx_msg,
1206                        "error %d on tx 0x%p->%s id %u/%d state %s age %ds",
1207                        completion, tx, conn ?
1208                        libcfs_nid2str(conn->gnc_peer->gnp_nid) : "<?>",
1209                        tx->tx_id.txe_smsg_id, tx->tx_id.txe_idx,
1210                        kgnilnd_tx_state2str(tx->tx_list_state),
1211                        cfs_duration_sec((unsigned long)jiffies - tx->tx_qtime));
1212         }
1213
1214         /* The error codes determine if we hold onto the MDD */
1215         kgnilnd_unmap_buffer(tx, completion);
1216
1217         /* we have to deliver a reply on lntmsg[1] for the GET, so make sure
1218          * we play nice with the error codes to avoid delivering a failed
1219          * REQUEST and then a REPLY event as well */
1220
1221         /* return -EIO to lnet - it is the magic value for failed sends */
1222         if (tx->tx_msg.gnm_type == GNILND_MSG_GET_REQ) {
1223                 status0 = 0;
1224                 status1 = completion;
1225         } else {
1226                 status0 = status1 = completion;
1227         }
1228
1229         tx->tx_buftype = GNILND_BUF_NONE;
1230         tx->tx_msg.gnm_type = GNILND_MSG_NONE;
1231
1232         /* lnet_finalize doesn't do anything with the *ni, so ok for us to
1233          * set NULL when we are a tx without a conn */
1234         if (conn != NULL) {
1235                 ni = conn->gnc_peer->gnp_net->gnn_ni;
1236
1237                 spin_lock(&conn->gnc_tx_lock);
1238
1239                 LASSERTF(test_and_clear_bit(tx->tx_id.txe_idx,
1240                         (volatile unsigned long *)&conn->gnc_tx_bits),
1241                         "conn %p tx %p bit %d already cleared\n",
1242                         conn, tx, tx->tx_id.txe_idx);
1243
1244                 LASSERTF(conn->gnc_tx_ref_table[tx->tx_id.txe_idx] != NULL,
1245                          "msg_id %d already NULL\n", tx->tx_id.txe_idx);
1246
1247                 conn->gnc_tx_ref_table[tx->tx_id.txe_idx] = NULL;
1248                 spin_unlock(&conn->gnc_tx_lock);
1249         }
1250
1251         kgnilnd_free_tx(tx);
1252
1253         /* finalize AFTER freeing lnet msgs */
1254
1255         /* warning - we should hold no locks here - calling lnet_finalize
1256          * could free up lnet credits, resulting in a call chain back into
1257          * the LND via kgnilnd_send and friends */
1258
1259         lnet_finalize(ni, lntmsg0, status0);
1260
1261         if (lntmsg1 != NULL) {
1262                 lnet_finalize(ni, lntmsg1, status1);
1263         }
1264 }
1265
1266 void
1267 kgnilnd_txlist_done(struct list_head *txlist, int error)
1268 {
1269         kgn_tx_t        *tx, *txn;
1270         int              err_printed = 0;
1271
1272         if (list_empty(txlist))
1273                 return;
1274
1275         list_for_each_entry_safe(tx, txn, txlist, tx_list) {
1276                 /* only print the first error */
1277                 if (err_printed)
1278                         tx->tx_state |= GNILND_TX_QUIET_ERROR;
1279                 list_del_init(&tx->tx_list);
1280                 kgnilnd_tx_done(tx, error);
1281                 err_printed++;
1282         }
1283 }
1284 int
1285 kgnilnd_set_tx_id(kgn_tx_t *tx, kgn_conn_t *conn)
1286 {
1287         int     id;
1288
1289         spin_lock(&conn->gnc_tx_lock);
1290
1291         /* ID zero is NOT ALLOWED!!! */
1292
1293 search_again:
1294         id = find_next_zero_bit((unsigned long *)&conn->gnc_tx_bits,
1295                                  GNILND_MAX_MSG_ID, conn->gnc_next_tx);
1296         if (id == GNILND_MAX_MSG_ID) {
1297                 if (conn->gnc_next_tx != 1) {
1298                         /* we only searched from next_tx to end and didn't find
1299                          * one, so search again from start */
1300                         conn->gnc_next_tx = 1;
1301                         goto search_again;
1302                 }
1303                 /* couldn't find one! */
1304                 spin_unlock(&conn->gnc_tx_lock);
1305                 return -E2BIG;
1306         }
1307
1308         /* bump next_tx to prevent immediate reuse */
1309         conn->gnc_next_tx = id + 1;
1310
1311         set_bit(id, (volatile unsigned long *)&conn->gnc_tx_bits);
1312         LASSERTF(conn->gnc_tx_ref_table[id] == NULL,
1313                  "tx 0x%p already at id %d\n",
1314                  conn->gnc_tx_ref_table[id], id);
1315
1316         /* delay these until we have a valid ID - prevents bad clear of the bit
1317          * in kgnilnd_tx_done */
1318         tx->tx_conn = conn;
1319         tx->tx_id.txe_cqid = conn->gnc_cqid;
1320
1321         tx->tx_id.txe_idx = id;
1322         conn->gnc_tx_ref_table[id] = tx;
1323
1324         /* Using jiffies to help differentiate against TX reuse - with
1325          * the usual minimum of a 250HZ clock, we wrap jiffies on the same TX
1326          * if we are sending to the same node faster than 256000/sec.
1327          * To help guard against this, we OR in the tx_seq - that is 32 bits */
1328
1329         tx->tx_id.txe_chips = (__u32)(jiffies | conn->gnc_tx_seq);
1330
1331         GNIDBG_TX(D_NET, tx, "set cookie/id/bits", NULL);
1332
1333         spin_unlock(&conn->gnc_tx_lock);
1334         return 0;
1335 }
1336
1337 static inline int
1338 kgnilnd_tx_should_retry(kgn_conn_t *conn, kgn_tx_t *tx)
1339 {
1340         int             max_retrans = *kgnilnd_tunables.kgn_max_retransmits;
1341         int             log_retrans;
1342         int             log_retrans_level;
1343
1344         /* I need kgni credits to send this.  Replace tx at the head of the
1345          * fmaq and I'll get rescheduled when credits appear */
1346         tx->tx_state = 0;
1347         tx->tx_retrans++;
1348         conn->gnc_tx_retrans++;
1349         log_retrans = ((tx->tx_retrans < 25) || ((tx->tx_retrans % 25) == 0) ||
1350                         (tx->tx_retrans > (max_retrans / 2)));
1351         log_retrans_level = tx->tx_retrans < (max_retrans / 2) ? D_NET : D_NETERROR;
1352
1353         /* Decision time - either error, warn or just retransmit */
1354
1355         /* we don't care about TX timeout - it could be that the network is slower
1356          * or throttled. We'll keep retranmitting - so if the network is so slow
1357          * that we fill up our mailbox, we'll keep trying to resend that msg
1358          * until we exceed the max_retrans _or_ gnc_last_rx expires, indicating
1359          * that he hasn't send us any traffic in return */
1360
1361         if (tx->tx_retrans > max_retrans) {
1362                 /* this means we are not backing off the retransmits
1363                  * in a healthy manner and are likely chewing up the
1364                  * CPU cycles quite badly */
1365                 GNIDBG_TOMSG(D_ERROR, &tx->tx_msg,
1366                         "SOFTWARE BUG: too many retransmits (%d) for tx id %x "
1367                         "conn 0x%p->%s\n",
1368                         tx->tx_retrans, tx->tx_id, conn,
1369                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
1370
1371                 /* yes - double errors to help debug this condition */
1372                 GNIDBG_TOMSG(D_NETERROR, &tx->tx_msg, "connection dead. "
1373                         "unable to send to %s for %lu secs (%d tries)",
1374                         libcfs_nid2str(tx->tx_conn->gnc_peer->gnp_nid),
1375                         cfs_duration_sec(jiffies - tx->tx_cred_wait),
1376                         tx->tx_retrans);
1377
1378                 kgnilnd_close_conn(conn, -ETIMEDOUT);
1379
1380                 /* caller should terminate */
1381                 RETURN(0);
1382         } else {
1383                 /* some reasonable throttling of the debug message */
1384                 if (log_retrans) {
1385                         unsigned long now = jiffies;
1386                         /* XXX Nic: Mystical TX debug here... */
1387                         GNIDBG_SMSG_CREDS(log_retrans_level, conn);
1388                         GNIDBG_TOMSG(log_retrans_level, &tx->tx_msg,
1389                                 "NOT_DONE on conn 0x%p->%s id %x retrans %d wait %dus"
1390                                 " last_msg %uus/%uus last_cq %uus/%uus",
1391                                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
1392                                 tx->tx_id, tx->tx_retrans,
1393                                 jiffies_to_usecs(now - tx->tx_cred_wait),
1394                                 jiffies_to_usecs(now - conn->gnc_last_tx),
1395                                 jiffies_to_usecs(now - conn->gnc_last_rx),
1396                                 jiffies_to_usecs(now - conn->gnc_last_tx_cq),
1397                                 jiffies_to_usecs(now - conn->gnc_last_rx_cq));
1398                 }
1399                 /* caller should retry */
1400                 RETURN(1);
1401         }
1402 }
1403
1404 /* caller must be holding gnd_cq_mutex and not unlock it afterwards, as we need to drop it
1405  * to avoid bad ordering with state_lock */
1406
1407 static inline int
1408 kgnilnd_sendmsg_nolock(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1409                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1410 {
1411         kgn_conn_t      *conn = tx->tx_conn;
1412         kgn_msg_t       *msg = &tx->tx_msg;
1413         int              retry_send;
1414         gni_return_t     rrc;
1415         unsigned long    newest_last_rx, timeout;
1416         unsigned long    now;
1417
1418         LASSERTF((msg->gnm_type == GNILND_MSG_IMMEDIATE) ?
1419                 immediatenob <= *kgnilnd_tunables.kgn_max_immediate :
1420                 immediatenob == 0,
1421                 "msg 0x%p type %d wrong payload size %d\n",
1422                 msg, msg->gnm_type, immediatenob);
1423
1424         /* make sure we catch all the cases where we'd send on a dirty old mbox
1425          * but allow case for sending CLOSE. Since this check is within the CQ
1426          * mutex barrier and the close message is only sent through
1427          * kgnilnd_send_conn_close the last message out the door will be the
1428          * close message.
1429          */
1430         if (atomic_read(&conn->gnc_peer->gnp_dirty_eps) != 0 && msg->gnm_type != GNILND_MSG_CLOSE) {
1431                 mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1432                 /* Return -ETIME, we are closing the connection already so we dont want to
1433                  * have this tx hit the wire. The tx will be killed by the calling function.
1434                  * Once the EP is marked dirty the close message will be the last
1435                  * thing to hit the wire */
1436                 return -ETIME;
1437         }
1438
1439         now = jiffies;
1440         timeout = cfs_time_seconds(conn->gnc_timeout);
1441
1442         newest_last_rx = GNILND_LASTRX(conn);
1443
1444         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SEND_TIMEOUT)) {
1445                 now = now + (GNILND_TIMEOUTRX(timeout) * 2);
1446         }
1447
1448         if (time_after_eq(now, newest_last_rx + GNILND_TIMEOUTRX(timeout))) {
1449                 GNIDBG_CONN(D_NETERROR|D_CONSOLE, conn, "Cant send to %s after timeout lapse of %lu; TO %lu",
1450                 libcfs_nid2str(conn->gnc_peer->gnp_nid),
1451                 cfs_duration_sec(now - newest_last_rx),
1452                 cfs_duration_sec(GNILND_TIMEOUTRX(timeout)));
1453                 mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1454                 return -ETIME;
1455         }
1456
1457         GNITX_ASSERTF(tx, (conn != NULL) && (tx->tx_id.txe_idx != 0), "tx id unset!", NULL);
1458         /* msg->gnm_srcnid is set when the message is initialized by whatever function is
1459          * creating the message this allows the message to contain the correct LNET NID/NET needed
1460          * instead of the one that the peer/conn uses for sending the data.
1461          */
1462         msg->gnm_connstamp = conn->gnc_my_connstamp;
1463         msg->gnm_payload_len = immediatenob;
1464         msg->gnm_seq = conn->gnc_tx_seq;
1465
1466         /* always init here - kgn_checksum is a /sys module tunable
1467          * and can be flipped at any point, even between msg init and sending */
1468         msg->gnm_cksum = 0;
1469         if (*kgnilnd_tunables.kgn_checksum) {
1470                 /* We must set here and not in kgnilnd_init_msg,
1471                  * we could resend this msg many times
1472                  * (NOT_DONE from gni_smsg_send below) and wouldn't pass
1473                  * through init_msg again */
1474                 msg->gnm_cksum = kgnilnd_cksum(msg, sizeof(kgn_msg_t));
1475                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_SMSG_CKSUM1)) {
1476                         msg->gnm_cksum += 0xf00f;
1477                 }
1478         }
1479
1480         GNIDBG_TOMSG(D_NET, msg, "tx 0x%p conn 0x%p->%s sending SMSG sz %u id %x/%d [%p for %u]",
1481                tx, conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
1482                sizeof(kgn_msg_t), tx->tx_id.txe_smsg_id,
1483                tx->tx_id.txe_idx, immediate, immediatenob);
1484
1485         if (unlikely(tx->tx_state & GNILND_TX_FAIL_SMSG)) {
1486                 rrc = cfs_fail_val ? cfs_fail_val : GNI_RC_NOT_DONE;
1487         } else {
1488         rrc = kgnilnd_smsg_send(conn->gnc_ephandle,
1489                                     msg, sizeof(*msg), immediate, immediatenob,
1490                             tx->tx_id.txe_smsg_id);
1491         }
1492
1493         switch (rrc) {
1494         case GNI_RC_SUCCESS:
1495                 conn->gnc_tx_seq++;
1496                 conn->gnc_last_tx = jiffies;
1497                 /* no locking here as LIVE isn't a list */
1498                 kgnilnd_tx_add_state_locked(tx, NULL, conn, GNILND_TX_LIVE_FMAQ, 1);
1499
1500                 /* this needs to be checked under lock as it might be freed from a completion
1501                  * event.
1502                  */
1503                 if (msg->gnm_type == GNILND_MSG_NOOP) {
1504                         set_mb(conn->gnc_last_noop_sent, jiffies);
1505                 }
1506
1507                 /* serialize with seeing CQ events for completion on this, as well as
1508                  * tx_seq */
1509                 mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1510
1511                 atomic_inc(&conn->gnc_device->gnd_short_ntx);
1512                 atomic64_add(immediatenob, &conn->gnc_device->gnd_short_txbytes);
1513                 kgnilnd_peer_alive(conn->gnc_peer);
1514                 GNIDBG_SMSG_CREDS(D_NET, conn);
1515                 return 0;
1516
1517         case GNI_RC_NOT_DONE:
1518                 /* XXX Nic: We need to figure out how to track this
1519                  * - there are bound to be good reasons for it,
1520                  * but we want to know when it happens */
1521
1522                 mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1523                 /* We'll handle this error inline - makes the calling logic much more
1524                  * clean */
1525
1526                 /* If no lock, caller doesn't want us to retry */
1527                 if (state_lock == NULL) {
1528                         return -EAGAIN;
1529                 }
1530
1531                 retry_send = kgnilnd_tx_should_retry(conn, tx);
1532                 if (retry_send) {
1533                         /* add to head of list for the state and retries */
1534                         spin_lock(state_lock);
1535                         kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, state, 0);
1536                         spin_unlock(state_lock);
1537
1538                         /* We only reschedule for a certain number of retries, then
1539                          * we will wait for the CQ events indicating a release of SMSG
1540                          * credits */
1541                         if (tx->tx_retrans < (*kgnilnd_tunables.kgn_max_retransmits/4)) {
1542                                 kgnilnd_schedule_conn(conn);
1543                                 return 0;
1544                         } else {
1545                                 /* CQ event coming in signifies either TX completed or
1546                                  * RX receive. Either of these *could* free up credits
1547                                  * in the SMSG mbox and we should try sending again */
1548                                 GNIDBG_TX(D_NET, tx, "waiting for CQID %u event to resend",
1549                                          tx->tx_conn->gnc_cqid);
1550                                 /* use +ve return code to let upper layers know they
1551                                  * should stop looping on sends */
1552                                 return EAGAIN;
1553                         }
1554                 } else {
1555                         return -EAGAIN;
1556                 }
1557         default:
1558                 /* handle bad retcode gracefully */
1559                 mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1560                 return -EIO;
1561         }
1562 }
1563
1564 /* kgnilnd_sendmsg has hard wait on gnd_cq_mutex */
1565 static inline int
1566 kgnilnd_sendmsg(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1567                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1568 {
1569         kgn_device_t    *dev = tx->tx_conn->gnc_device;
1570         unsigned long    timestamp;
1571         int              rc;
1572
1573         timestamp = jiffies;
1574         mutex_lock(&dev->gnd_cq_mutex);
1575         /* delay in jiffies - we are really concerned only with things that
1576          * result in a schedule() or really holding this off for long times .
1577          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1578         dev->gnd_mutex_delay += (long) jiffies - timestamp;
1579
1580         rc = kgnilnd_sendmsg_nolock(tx, immediate, immediatenob, state_lock, state);
1581
1582         RETURN(rc);
1583 }
1584
1585
1586 /* returns -EAGAIN for lock miss, anything else < 0 is hard error, >=0 for success */
1587 static inline int
1588 kgnilnd_sendmsg_trylock(kgn_tx_t *tx, void *immediate, unsigned int immediatenob,
1589                 spinlock_t *state_lock, kgn_tx_list_state_t state)
1590 {
1591         kgn_conn_t      *conn = tx->tx_conn;
1592         kgn_device_t    *dev = conn->gnc_device;
1593         unsigned long    timestamp;
1594         int              rc;
1595
1596         timestamp = jiffies;
1597
1598         /* technically we are doing bad things with the read_lock on the peer_conn
1599          * table, but we shouldn't be sleeping inside here - and we don't sleep/block
1600          * for the mutex. I bet lockdep is gonna flag this one though... */
1601
1602         /* there are a few cases where we don't want the immediate send - like
1603          * when we are in the scheduler thread and it'd harm the latency of
1604          * getting messages up to LNet */
1605
1606         /* rmb for gnd_ready */
1607         smp_rmb();
1608         if (conn->gnc_device->gnd_ready == GNILND_DEV_LOOP) {
1609                 rc = 0;
1610                 atomic_inc(&conn->gnc_device->gnd_fast_block);
1611         } else if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
1612                /* dont hit HW during quiesce */
1613                 rc = 0;
1614         } else if (unlikely(atomic_read(&conn->gnc_peer->gnp_dirty_eps))) {
1615                /* dont hit HW if stale EPs and conns left to close */
1616                 rc = 0;
1617         } else {
1618                 atomic_inc(&conn->gnc_device->gnd_fast_try);
1619                 rc = mutex_trylock(&conn->gnc_device->gnd_cq_mutex);
1620         }
1621         if (!rc) {
1622                 rc = -EAGAIN;
1623         } else {
1624                 /* we got the mutex and weren't blocked */
1625
1626                 /* delay in jiffies - we are really concerned only with things that
1627                  * result in a schedule() or really holding this off for long times .
1628                  * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1629                 dev->gnd_mutex_delay += (long) jiffies - timestamp;
1630
1631                 atomic_inc(&conn->gnc_device->gnd_fast_ok);
1632                 tx->tx_qtime = jiffies;
1633                 tx->tx_state = GNILND_TX_WAITING_COMPLETION;
1634                 rc = kgnilnd_sendmsg_nolock(tx, tx->tx_buffer, tx->tx_nob, &conn->gnc_list_lock, GNILND_TX_FMAQ);
1635                 /* _nolock unlocks the mutex for us */
1636         }
1637
1638         RETURN(rc);
1639 }
1640
1641 /* lets us know if we can push this RDMA through now */
1642 inline int
1643 kgnilnd_auth_rdma_bytes(kgn_device_t *dev, kgn_tx_t *tx)
1644 {
1645         long    bytes_left;
1646
1647         bytes_left = atomic64_sub_return(tx->tx_nob, &dev->gnd_rdmaq_bytes_ok);
1648
1649         if (bytes_left < 0) {
1650                 atomic64_add(tx->tx_nob, &dev->gnd_rdmaq_bytes_ok);
1651                 atomic_inc(&dev->gnd_rdmaq_nstalls);
1652                 smp_wmb();
1653
1654                 CDEBUG(D_NET, "no bytes to send, turning on timer for %lu\n",
1655                        dev->gnd_rdmaq_deadline);
1656                 mod_timer(&dev->gnd_rdmaq_timer, dev->gnd_rdmaq_deadline);
1657                 /* we never del this timer - at worst it schedules us.. */
1658                 return -EAGAIN;
1659         } else {
1660                 return 0;
1661         }
1662 }
1663
1664 /* this adds a TX to the queue pending throttling authorization before
1665  * we allow our remote peer to launch a PUT at us */
1666 void
1667 kgnilnd_queue_rdma(kgn_conn_t *conn, kgn_tx_t *tx)
1668 {
1669         int     rc;
1670
1671         /* we cannot go into send_mapped_tx from here as we are holding locks
1672          * and mem registration might end up allocating memory in kgni.
1673          * That said, we'll push this as far as we can into the queue process */
1674         rc = kgnilnd_auth_rdma_bytes(conn->gnc_device, tx);
1675
1676         if (rc < 0) {
1677                 spin_lock(&conn->gnc_device->gnd_rdmaq_lock);
1678                 kgnilnd_tx_add_state_locked(tx, NULL, conn, GNILND_TX_RDMAQ, 0);
1679                 /* lets us know how delayed RDMA is */
1680                 tx->tx_qtime = jiffies;
1681                 spin_unlock(&conn->gnc_device->gnd_rdmaq_lock);
1682         } else {
1683                 /* we have RDMA authorized, now it just needs a MDD and to hit the wire */
1684                 spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
1685                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 0);
1686                 /* lets us know how delayed mapping is */
1687                 tx->tx_qtime = jiffies;
1688                 spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
1689         }
1690
1691         /* make sure we wake up sched to run this */
1692         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
1693 }
1694
1695 /* push TX through state machine */
1696 void
1697 kgnilnd_queue_tx(kgn_conn_t *conn, kgn_tx_t *tx)
1698 {
1699         int            rc = 0;
1700         int            add_tail = 1;
1701
1702         /* set the tx_id here, we delay it until we have an actual conn
1703          * to fiddle with
1704          * in some cases, the tx_id is already set to provide for things
1705          * like RDMA completion cookies, etc */
1706         if (tx->tx_id.txe_idx == 0) {
1707                 rc = kgnilnd_set_tx_id(tx, conn);
1708                 if (rc != 0) {
1709                         kgnilnd_tx_done(tx, rc);
1710                         return;
1711                 }
1712         }
1713
1714         CDEBUG(D_NET, "%s to conn %p for %s\n", kgnilnd_msgtype2str(tx->tx_msg.gnm_type),
1715                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
1716
1717         /* Only let NOOPs to be sent while fail loc is set, otherwise kill the tx.
1718          */
1719         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_ONLY_NOOP) && (tx->tx_msg.gnm_type != GNILND_MSG_NOOP)) {
1720                 kgnilnd_tx_done(tx, rc);
1721                 return;
1722         }
1723
1724         switch (tx->tx_msg.gnm_type) {
1725         case GNILND_MSG_PUT_ACK:
1726         case GNILND_MSG_GET_REQ:
1727         case GNILND_MSG_PUT_REQ_REV:
1728         case GNILND_MSG_GET_ACK_REV:
1729                 /* hijacking time! If this messages will authorize our peer to
1730                  * send his dirty little bytes in an RDMA, we need to get permission */
1731                 kgnilnd_queue_rdma(conn, tx);
1732                 break;
1733         case GNILND_MSG_IMMEDIATE:
1734                 /* try to send right now, can help reduce latency */
1735                 rc = kgnilnd_sendmsg_trylock(tx, tx->tx_buffer, tx->tx_nob, &conn->gnc_list_lock, GNILND_TX_FMAQ);
1736
1737                 if (rc >= 0) {
1738                         /* it was sent, break out of switch to avoid default case of queueing */
1739                         break;
1740                 }
1741                 /* needs to queue to try again, so fall through to default case */
1742         case GNILND_MSG_NOOP:
1743                 /* Just make sure this goes out first for this conn */
1744                 add_tail = 0;
1745                 /* fall through... */
1746         default:
1747                 spin_lock(&conn->gnc_list_lock);
1748                 kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, GNILND_TX_FMAQ, add_tail);
1749                 tx->tx_qtime = jiffies;
1750                 spin_unlock(&conn->gnc_list_lock);
1751                 kgnilnd_schedule_conn(conn);
1752         }
1753 }
1754
1755 void
1756 kgnilnd_launch_tx(kgn_tx_t *tx, kgn_net_t *net, lnet_process_id_t *target)
1757 {
1758         kgn_peer_t      *peer;
1759         kgn_peer_t      *new_peer = NULL;
1760         kgn_conn_t      *conn = NULL;
1761         int              rc;
1762         int              node_state;
1763
1764         ENTRY;
1765
1766         /* If I get here, I've committed to send, so I complete the tx with
1767          * failure on any problems */
1768
1769         GNITX_ASSERTF(tx, tx->tx_conn == NULL,
1770                       "tx already has connection %p", tx->tx_conn);
1771
1772         /* do all of the peer & conn searching in one swoop - this avoids
1773          * nastiness when dropping locks and needing to maintain a sane state
1774          * in the face of stack reset or something else nuking peers & conns */
1775
1776         /* I expect to find him, so only take a read lock */
1777         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
1778
1779         peer = kgnilnd_find_peer_locked(target->nid);
1780         if (peer != NULL) {
1781                 conn = kgnilnd_find_conn_locked(peer);
1782                 /* this could be NULL during quiesce */
1783                 if (conn != NULL)  {
1784                         /* Connection exists; queue message on it */
1785                         kgnilnd_queue_tx(conn, tx);
1786                         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1787                         RETURN_EXIT;
1788                 }
1789
1790                 /* don't create a connection if the peer is marked down */
1791                 if (peer->gnp_down == GNILND_RCA_NODE_DOWN) {
1792                         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1793                         rc = -ENETRESET;
1794                         GOTO(no_peer, rc);
1795                 }
1796         }
1797
1798         /* creating peer or conn; I'll need a write lock... */
1799         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1800
1801         CFS_RACE(CFS_FAIL_GNI_FIND_TARGET);
1802
1803         node_state = kgnilnd_get_node_state(LNET_NIDADDR(target->nid));
1804
1805         /* NB - this will not block during normal operations -
1806          * the only writer of this is in the startup/shutdown path. */
1807         rc = down_read_trylock(&kgnilnd_data.kgn_net_rw_sem);
1808         if (!rc) {
1809                 rc = -ESHUTDOWN;
1810                 GOTO(no_peer, rc);
1811         }
1812
1813         /* ignore previous peer entirely - we cycled the lock, so we
1814          * will create new peer and at worst drop it if peer is still
1815          * in the tables */
1816         rc = kgnilnd_create_peer_safe(&new_peer, target->nid, net, node_state);
1817         if (rc != 0) {
1818                 up_read(&kgnilnd_data.kgn_net_rw_sem);
1819                 GOTO(no_peer, rc);
1820         }
1821
1822         write_lock(&kgnilnd_data.kgn_peer_conn_lock);
1823         up_read(&kgnilnd_data.kgn_net_rw_sem);
1824
1825         /* search for peer again now that we have the lock
1826          * if we don't find it, add our new one to the list */
1827         kgnilnd_add_peer_locked(target->nid, new_peer, &peer);
1828
1829         /* don't create a connection if the peer is not up */
1830         if (peer->gnp_down != GNILND_RCA_NODE_UP) {
1831                 write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1832                 rc = -ENETRESET;
1833                 GOTO(no_peer, rc);
1834         }
1835
1836         conn = kgnilnd_find_or_create_conn_locked(peer);
1837
1838         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_DGRAM_DROP_TX)) {
1839                 write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1840                 GOTO(no_peer, rc);
1841         }
1842
1843         if (conn != NULL) {
1844                 /* oh hey, found a conn now... magical */
1845                 kgnilnd_queue_tx(conn, tx);
1846         } else {
1847                 /* no conn, must be trying to connect - so we queue for now */
1848                 tx->tx_qtime = jiffies;
1849                 kgnilnd_tx_add_state_locked(tx, peer, NULL, GNILND_TX_PEERQ, 1);
1850         }
1851         write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
1852         RETURN_EXIT;
1853 no_peer:
1854         kgnilnd_tx_done(tx, rc);
1855         RETURN_EXIT;
1856 }
1857
1858 void
1859 kgnilnd_rdma(kgn_tx_t *tx, int type,
1860             kgn_rdma_desc_t *sink, unsigned int nob, __u64 cookie)
1861 {
1862         kgn_conn_t   *conn = tx->tx_conn;
1863         unsigned long timestamp;
1864         gni_post_type_t post_type;
1865         gni_return_t  rrc;
1866         int rc = 0;
1867         unsigned int desc_nob = nob;
1868         void *desc_buffer = tx->tx_buffer;
1869         gni_mem_handle_t desc_map_key = tx->tx_map_key;
1870         LASSERTF(kgnilnd_tx_mapped(tx),
1871                 "unmapped tx %p\n", tx);
1872         LASSERTF(conn != NULL,
1873                 "NULL conn on tx %p, naughty, naughty\n", tx);
1874         LASSERTF(nob <= sink->gnrd_nob,
1875                 "nob %u > sink->gnrd_nob %d (%p)\n",
1876                 nob, sink->gnrd_nob, sink);
1877         LASSERTF(nob <= tx->tx_nob,
1878                 "nob %d > tx(%p)->tx_nob %d\n",
1879                 nob, tx, tx->tx_nob);
1880
1881         switch (type) {
1882         case GNILND_MSG_GET_DONE:
1883         case GNILND_MSG_PUT_DONE:
1884                 post_type = GNI_POST_RDMA_PUT;
1885                 break;
1886         case GNILND_MSG_GET_DONE_REV:
1887         case GNILND_MSG_PUT_DONE_REV:
1888                 post_type = GNI_POST_RDMA_GET;
1889                 break;
1890         default:
1891                 CERROR("invalid msg type %s (%d)\n",
1892                         kgnilnd_msgtype2str(type), type);
1893                 LBUG();
1894         }
1895         if (post_type == GNI_POST_RDMA_GET) {
1896                 /* Check for remote buffer / local buffer / length alignment. All must be 4 byte
1897                  * aligned. If the local buffer is not aligned correctly using the copy buffer
1898                  * will fix that issue. If length is misaligned copy buffer will also fix the issue, we end
1899                  * up transferring extra bytes into the buffer but only copy the correct nob into the original
1900                  * buffer.  Remote offset correction is done through a combination of adjusting the offset,
1901                  * making sure the length and addr are aligned and copying the data into the correct location
1902                  * once the transfer has completed.
1903                  */
1904                 if ((((__u64)((unsigned long)tx->tx_buffer)) & 3) ||
1905                       (sink->gnrd_addr & 3) ||
1906                       (nob & 3)) {
1907
1908                         tx->tx_offset = ((__u64)((unsigned long)sink->gnrd_addr)) & 3;
1909                         if (tx->tx_offset)
1910                                 kgnilnd_admin_addref(kgnilnd_data.kgn_rev_offset);
1911
1912                         if ((nob + tx->tx_offset) & 3) {
1913                                 desc_nob = ((nob + tx->tx_offset) + (4 - ((nob + tx->tx_offset) & 3)));
1914                                 kgnilnd_admin_addref(kgnilnd_data.kgn_rev_length);
1915                         } else {
1916                                 desc_nob = (nob + tx->tx_offset);
1917                         }
1918
1919                         if (tx->tx_buffer_copy == NULL) {
1920                                 /* Allocate the largest copy buffer we will need, this will prevent us from overwriting data
1921                                  * and require at most we allocate a few extra bytes. */
1922                                 tx->tx_buffer_copy = vmalloc(desc_nob);
1923
1924                                 if (!tx->tx_buffer_copy) {
1925                                         /* allocation of buffer failed nak the rdma */
1926                                         kgnilnd_nak_rdma(tx->tx_conn, tx->tx_msg.gnm_type, -EFAULT, cookie, tx->tx_msg.gnm_srcnid);
1927                                         kgnilnd_tx_done(tx, -EFAULT);
1928                                         return;
1929                                 }
1930                                 kgnilnd_admin_addref(kgnilnd_data.kgn_rev_copy_buff);
1931                                 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);
1932                                 if (rc != GNI_RC_SUCCESS) {
1933                                         /* Registration Failed nak rdma and kill the tx. */
1934                                         vfree(tx->tx_buffer_copy);
1935                                         tx->tx_buffer_copy = NULL;
1936                                         kgnilnd_nak_rdma(tx->tx_conn, tx->tx_msg.gnm_type, -EFAULT, cookie, tx->tx_msg.gnm_srcnid);
1937                                         kgnilnd_tx_done(tx, -EFAULT);
1938                                         return;
1939                                 }
1940                         }
1941                         desc_map_key = tx->tx_buffer_copy_map_key;
1942                         desc_buffer = tx->tx_buffer_copy;
1943                 }
1944         }
1945
1946         memset(&tx->tx_rdma_desc, 0, sizeof(tx->tx_rdma_desc));
1947         tx->tx_rdma_desc.post_id = tx->tx_id.txe_cookie;
1948         tx->tx_rdma_desc.type = post_type;
1949         tx->tx_rdma_desc.cq_mode = GNI_CQMODE_GLOBAL_EVENT;
1950         tx->tx_rdma_desc.local_addr = (__u64)((unsigned long)desc_buffer);
1951         tx->tx_rdma_desc.local_mem_hndl = desc_map_key;
1952         tx->tx_rdma_desc.remote_addr = sink->gnrd_addr - tx->tx_offset;
1953         tx->tx_rdma_desc.remote_mem_hndl = sink->gnrd_key;
1954         tx->tx_rdma_desc.length = desc_nob;
1955         tx->tx_nob_rdma = nob;
1956         if (*kgnilnd_tunables.kgn_bte_dlvr_mode)
1957                 tx->tx_rdma_desc.dlvr_mode = *kgnilnd_tunables.kgn_bte_dlvr_mode;
1958         /* prep final completion message */
1959         kgnilnd_init_msg(&tx->tx_msg, type, tx->tx_msg.gnm_srcnid);
1960         tx->tx_msg.gnm_u.completion.gncm_cookie = cookie;
1961         /* send actual size RDMA'd in retval */
1962         tx->tx_msg.gnm_u.completion.gncm_retval = nob;
1963
1964         kgnilnd_compute_rdma_cksum(tx, nob);
1965
1966         if (nob == 0) {
1967                 kgnilnd_queue_tx(conn, tx);
1968                 return;
1969         }
1970
1971         /* Don't lie (CLOSE == RDMA idle) */
1972         LASSERTF(!conn->gnc_close_sent, "tx %p on conn %p after close sent %d\n",
1973                  tx, conn, conn->gnc_close_sent);
1974
1975         GNIDBG_TX(D_NET, tx, "Post RDMA type 0x%02x dlvr_mode 0x%x cookie:"LPX64,
1976                 type, tx->tx_rdma_desc.dlvr_mode, cookie);
1977
1978         /* set CQ dedicated for RDMA */
1979         tx->tx_rdma_desc.src_cq_hndl = conn->gnc_device->gnd_snd_rdma_cqh;
1980
1981         timestamp = jiffies;
1982         mutex_lock(&conn->gnc_device->gnd_cq_mutex);
1983         /* delay in jiffies - we are really concerned only with things that
1984          * result in a schedule() or really holding this off for long times .
1985          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
1986         conn->gnc_device->gnd_mutex_delay += (long) jiffies - timestamp;
1987
1988         rrc = kgnilnd_post_rdma(conn->gnc_ephandle, &tx->tx_rdma_desc);
1989
1990         spin_lock(&conn->gnc_list_lock);
1991         kgnilnd_tx_add_state_locked(tx, conn->gnc_peer, conn, GNILND_TX_LIVE_RDMAQ, 1);
1992         tx->tx_qtime = jiffies;
1993         spin_unlock(&conn->gnc_list_lock);
1994
1995         mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
1996
1997         /* XXX Nic: is this a place we should handle more errors for
1998          * robustness sake */
1999         LASSERT(rrc == GNI_RC_SUCCESS);
2000
2001 }
2002
2003 kgn_rx_t *
2004 kgnilnd_alloc_rx(void)
2005 {
2006         kgn_rx_t        *rx;
2007
2008         rx = kmem_cache_alloc(kgnilnd_data.kgn_rx_cache, GFP_ATOMIC);
2009         if (rx == NULL) {
2010                 CERROR("failed to allocate rx\n");
2011                 return NULL;
2012         }
2013         CDEBUG(D_MALLOC, "slab-alloced 'rx': %lu at %p.\n",
2014                sizeof(*rx), rx);
2015
2016         /* no memset to zero, we'll always fill all members */
2017         return rx;
2018 }
2019
2020 /* release is to just free connection resources
2021  * we use this for the eager path after copying */
2022 void
2023 kgnilnd_release_msg(kgn_conn_t *conn)
2024 {
2025         gni_return_t    rrc;
2026         unsigned long   timestamp;
2027
2028         CDEBUG(D_NET, "consuming %p\n", conn);
2029
2030         timestamp = jiffies;
2031         mutex_lock(&conn->gnc_device->gnd_cq_mutex);
2032         /* delay in jiffies - we are really concerned only with things that
2033          * result in a schedule() or really holding this off for long times .
2034          * NB - mutex_lock could spin for 2 jiffies before going to sleep to wait */
2035         conn->gnc_device->gnd_mutex_delay += (long) jiffies - timestamp;
2036
2037         rrc = kgnilnd_smsg_release(conn->gnc_ephandle);
2038         mutex_unlock(&conn->gnc_device->gnd_cq_mutex);
2039
2040         LASSERTF(rrc == GNI_RC_SUCCESS, "bad rrc %d\n", rrc);
2041         GNIDBG_SMSG_CREDS(D_NET, conn);
2042
2043         return;
2044 }
2045
2046 void
2047 kgnilnd_consume_rx(kgn_rx_t *rx)
2048 {
2049         kgn_conn_t      *conn = rx->grx_conn;
2050         kgn_msg_t       *rxmsg = rx->grx_msg;
2051
2052         /* if we are eager, free the cache alloc'd msg */
2053         if (unlikely(rx->grx_eager)) {
2054                 LIBCFS_FREE(rxmsg, sizeof(*rxmsg) + *kgnilnd_tunables.kgn_max_immediate);
2055                 atomic_dec(&kgnilnd_data.kgn_neager_allocs);
2056
2057                 /* release ref from eager_recv */
2058                 kgnilnd_conn_decref(conn);
2059         } else {
2060                 GNIDBG_MSG(D_NET, rxmsg, "rx %p processed", rx);
2061                 kgnilnd_release_msg(conn);
2062         }
2063
2064         kmem_cache_free(kgnilnd_data.kgn_rx_cache, rx);
2065         CDEBUG(D_MALLOC, "slab-freed 'rx': %lu at %p.\n",
2066                sizeof(*rx), rx);
2067
2068         return;
2069 }
2070
2071 int
2072 kgnilnd_send(lnet_ni_t *ni, void *private, lnet_msg_t *lntmsg)
2073 {
2074         lnet_hdr_t       *hdr = &lntmsg->msg_hdr;
2075         int               type = lntmsg->msg_type;
2076         lnet_process_id_t target = lntmsg->msg_target;
2077         int               target_is_router = lntmsg->msg_target_is_router;
2078         int               routing = lntmsg->msg_routing;
2079         unsigned int      niov = lntmsg->msg_niov;
2080         struct iovec     *iov = lntmsg->msg_iov;
2081         lnet_kiov_t      *kiov = lntmsg->msg_kiov;
2082         unsigned int      offset = lntmsg->msg_offset;
2083         unsigned int      nob = lntmsg->msg_len;
2084         unsigned int      msg_vmflush = lntmsg->msg_vmflush;
2085         kgn_net_t        *net = ni->ni_data;
2086         kgn_tx_t         *tx;
2087         int               rc = 0;
2088         int               mpflag = 0;
2089         int               reverse_rdma_flag = *kgnilnd_tunables.kgn_reverse_rdma;
2090
2091         /* NB 'private' is different depending on what we're sending.... */
2092         LASSERT(!in_interrupt());
2093
2094         CDEBUG(D_NET, "sending msg type %d with %d bytes in %d frags to %s\n",
2095                type, nob, niov, libcfs_id2str(target));
2096
2097         LASSERTF(nob == 0 || niov > 0,
2098                 "lntmsg %p nob %d niov %d\n", lntmsg, nob, niov);
2099         LASSERTF(niov <= LNET_MAX_IOV,
2100                 "lntmsg %p niov %d\n", lntmsg, niov);
2101
2102         /* payload is either all vaddrs or all pages */
2103         LASSERTF(!(kiov != NULL && iov != NULL),
2104                 "lntmsg %p kiov %p iov %p\n", lntmsg, kiov, iov);
2105
2106         if (msg_vmflush)
2107                 mpflag = cfs_memory_pressure_get_and_set();
2108
2109         switch (type) {
2110         default:
2111                 CERROR("lntmsg %p with unexpected type %d\n",
2112                         lntmsg, type);
2113                 LBUG();
2114
2115         case LNET_MSG_ACK:
2116                 LASSERTF(nob == 0, "lntmsg %p nob %d\n",
2117                         lntmsg, nob);
2118                 break;
2119
2120         case LNET_MSG_GET:
2121                 LASSERT(niov == 0);
2122                 LASSERT(nob == 0);
2123
2124                 if (routing || target_is_router)
2125                         break;                  /* send IMMEDIATE */
2126
2127                 /* it is safe to do direct GET with out mapping buffer for RDMA as we
2128                  * check the eventual sink buffer here - if small enough, remote
2129                  * end is perfectly capable of returning data in short message -
2130                  * The magic is that we call lnet_parse in kgnilnd_recv with rdma_req=0
2131                  * for IMMEDIATE messages which will have it send a real reply instead
2132                  * of doing kgnilnd_recv to have the RDMA continued */
2133                 if (lntmsg->msg_md->md_length <= *kgnilnd_tunables.kgn_max_immediate)
2134                        break;
2135
2136                 if ((reverse_rdma_flag & GNILND_REVERSE_GET) == 0)
2137                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_REQ, ni->ni_nid);
2138                 else
2139                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_REQ_REV, ni->ni_nid);
2140
2141                 if (tx == NULL) {
2142                         rc = -ENOMEM;
2143                         goto out;
2144                 }
2145                 /* slightly different options as we might actually have a GET with a
2146                  * MD_KIOV set but a non-NULL md_iov.iov */
2147                 if ((lntmsg->msg_md->md_options & LNET_MD_KIOV) == 0)
2148                         rc = kgnilnd_setup_rdma_buffer(tx, lntmsg->msg_md->md_niov,
2149                                                       lntmsg->msg_md->md_iov.iov, NULL,
2150                                                       0, lntmsg->msg_md->md_length);
2151                 else
2152                         rc = kgnilnd_setup_rdma_buffer(tx, lntmsg->msg_md->md_niov,
2153                                                       NULL, lntmsg->msg_md->md_iov.kiov,
2154                                                       0, lntmsg->msg_md->md_length);
2155                 if (rc != 0) {
2156                         CERROR("unable to setup buffer: %d\n", rc);
2157                         kgnilnd_tx_done(tx, rc);
2158                         rc = -EIO;
2159                         goto out;
2160                 }
2161
2162                 tx->tx_lntmsg[1] = lnet_create_reply_msg(ni, lntmsg);
2163                 if (tx->tx_lntmsg[1] == NULL) {
2164                         CERROR("Can't create reply for GET to %s\n",
2165                                libcfs_nid2str(target.nid));
2166                         kgnilnd_tx_done(tx, rc);
2167                         rc = -EIO;
2168                         goto out;
2169                 }
2170
2171                 tx->tx_lntmsg[0] = lntmsg;
2172                 if ((reverse_rdma_flag & GNILND_REVERSE_GET) == 0)
2173                         tx->tx_msg.gnm_u.get.gngm_hdr = *hdr;
2174                 else
2175                         tx->tx_msg.gnm_u.putreq.gnprm_hdr = *hdr;
2176
2177                 /* rest of tx_msg is setup just before it is sent */
2178                 kgnilnd_launch_tx(tx, net, &target);
2179                 goto out;
2180         case LNET_MSG_REPLY:
2181         case LNET_MSG_PUT:
2182                 /* to save on MDDs, we'll handle short kiov by vmap'ing
2183                  * and sending via SMSG */
2184                 if (nob <= *kgnilnd_tunables.kgn_max_immediate)
2185                        break;
2186
2187                 if ((reverse_rdma_flag & GNILND_REVERSE_PUT) == 0)
2188                         tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_REQ, ni->ni_nid);
2189                 else
2190                         tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_REQ_REV, ni->ni_nid);
2191
2192                 if (tx == NULL) {
2193                         rc = -ENOMEM;
2194                         goto out;
2195                 }
2196
2197                 rc = kgnilnd_setup_rdma_buffer(tx, niov, iov, kiov, offset, nob);
2198                 if (rc != 0) {
2199                         kgnilnd_tx_done(tx, rc);
2200                         rc = -EIO;
2201                         goto out;
2202                 }
2203
2204                 tx->tx_lntmsg[0] = lntmsg;
2205                 if ((reverse_rdma_flag & GNILND_REVERSE_PUT) == 0)
2206                         tx->tx_msg.gnm_u.putreq.gnprm_hdr = *hdr;
2207                 else
2208                         tx->tx_msg.gnm_u.get.gngm_hdr = *hdr;
2209
2210                 /* rest of tx_msg is setup just before it is sent */
2211                 kgnilnd_launch_tx(tx, net, &target);
2212                 goto out;
2213         }
2214
2215         /* send IMMEDIATE */
2216
2217         LASSERTF(nob <= *kgnilnd_tunables.kgn_max_immediate,
2218                 "lntmsg 0x%p too large %d\n", lntmsg, nob);
2219
2220         tx = kgnilnd_new_tx_msg(GNILND_MSG_IMMEDIATE, ni->ni_nid);
2221         if (tx == NULL) {
2222                 rc = -ENOMEM;
2223                 goto out;
2224         }
2225
2226         rc = kgnilnd_setup_immediate_buffer(tx, niov, iov, kiov, offset, nob);
2227         if (rc != 0) {
2228                 kgnilnd_tx_done(tx, rc);
2229                 goto out;
2230         }
2231
2232         tx->tx_msg.gnm_u.immediate.gnim_hdr = *hdr;
2233         tx->tx_lntmsg[0] = lntmsg;
2234         kgnilnd_launch_tx(tx, net, &target);
2235
2236 out:
2237         /* use stored value as we could have already finalized lntmsg here from a failed launch */
2238         if (msg_vmflush)
2239                 cfs_memory_pressure_restore(mpflag);
2240         return rc;
2241 }
2242
2243 void
2244 kgnilnd_setup_rdma(lnet_ni_t *ni, kgn_rx_t *rx, lnet_msg_t *lntmsg, int mlen)
2245 {
2246         kgn_conn_t    *conn = rx->grx_conn;
2247         kgn_msg_t     *rxmsg = rx->grx_msg;
2248         unsigned int   niov = lntmsg->msg_niov;
2249         struct iovec  *iov = lntmsg->msg_iov;
2250         lnet_kiov_t   *kiov = lntmsg->msg_kiov;
2251         unsigned int   offset = lntmsg->msg_offset;
2252         unsigned int   nob = lntmsg->msg_len;
2253         int            done_type;
2254         kgn_tx_t      *tx;
2255         int            rc = 0;
2256
2257         switch (rxmsg->gnm_type) {
2258         case GNILND_MSG_PUT_REQ_REV:
2259                 done_type = GNILND_MSG_PUT_DONE_REV;
2260                 nob = mlen;
2261                 break;
2262         case GNILND_MSG_GET_REQ:
2263                 done_type = GNILND_MSG_GET_DONE;
2264                 break;
2265         default:
2266                 CERROR("invalid msg type %s (%d)\n",
2267                         kgnilnd_msgtype2str(rxmsg->gnm_type),
2268                         rxmsg->gnm_type);
2269                 LBUG();
2270         }
2271
2272         tx = kgnilnd_new_tx_msg(done_type, ni->ni_nid);
2273         if (tx == NULL)
2274                 goto failed_0;
2275
2276         rc = kgnilnd_set_tx_id(tx, conn);
2277         if (rc != 0)
2278                 goto failed_1;
2279
2280         rc = kgnilnd_setup_rdma_buffer(tx, niov, iov, kiov, offset, nob);
2281         if (rc != 0)
2282                 goto failed_1;
2283
2284         tx->tx_lntmsg[0] = lntmsg;
2285         tx->tx_getinfo = rxmsg->gnm_u.get;
2286
2287         /* we only queue from kgnilnd_recv - we might get called from other contexts
2288          * and we don't want to block the mutex in those cases */
2289
2290         spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2291         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2292         spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2293         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2294
2295         return;
2296
2297  failed_1:
2298         kgnilnd_tx_done(tx, rc);
2299         kgnilnd_nak_rdma(conn, done_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2300  failed_0:
2301         lnet_finalize(ni, lntmsg, rc);
2302 }
2303
2304 int
2305 kgnilnd_eager_recv(lnet_ni_t *ni, void *private, lnet_msg_t *lntmsg,
2306                    void **new_private)
2307 {
2308         kgn_rx_t        *rx = private;
2309         kgn_conn_t      *conn = rx->grx_conn;
2310         kgn_msg_t       *rxmsg = rx->grx_msg;
2311         kgn_msg_t       *eagermsg = NULL;
2312         kgn_peer_t      *peer = NULL;
2313         kgn_conn_t      *found_conn = NULL;
2314
2315         GNIDBG_MSG(D_NET, rxmsg, "eager recv for conn %p, rxmsg %p, lntmsg %p",
2316                 conn, rxmsg, lntmsg);
2317
2318         if (rxmsg->gnm_payload_len > *kgnilnd_tunables.kgn_max_immediate) {
2319                 GNIDBG_MSG(D_ERROR, rxmsg, "payload too large %d",
2320                         rxmsg->gnm_payload_len);
2321                 return -EPROTO;
2322         }
2323         /* Grab a read lock so the connection doesnt disappear on us
2324          * while we look it up
2325          */
2326         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
2327
2328         peer = kgnilnd_find_peer_locked(rxmsg->gnm_srcnid);
2329         if (peer != NULL)
2330                 found_conn = kgnilnd_find_conn_locked(peer);
2331
2332
2333         /* Verify the connection found is the same one that the message
2334          * is supposed to be using, if it is not output an error message
2335          * and return.
2336          */
2337         if (!peer || !found_conn
2338             || found_conn->gnc_peer_connstamp != rxmsg->gnm_connstamp) {
2339                 read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2340                 CERROR("Couldnt find matching peer %p or conn %p / %p\n",
2341                         peer, conn, found_conn);
2342                 if (found_conn) {
2343                         CERROR("Unexpected connstamp "LPX64"("LPX64" expected)"
2344                                 " from %s", rxmsg->gnm_connstamp,
2345                                 found_conn->gnc_peer_connstamp,
2346                                 libcfs_nid2str(peer->gnp_nid));
2347                 }
2348                 return -ENOTCONN;
2349         }
2350
2351         /* add conn ref to ensure it doesn't go away until all eager
2352          * messages processed */
2353         kgnilnd_conn_addref(conn);
2354
2355         /* Now that we have verified the connection is valid and added a
2356          * reference we can remove the read_lock on the peer_conn_lock */
2357         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2358
2359         /* we have no credits or buffers for this message, so copy it
2360          * somewhere for a later kgnilnd_recv */
2361         if (atomic_read(&kgnilnd_data.kgn_neager_allocs) >=
2362                         *kgnilnd_tunables.kgn_eager_credits) {
2363                 CERROR("Out of eager credits to %s\n",
2364                         libcfs_nid2str(conn->gnc_peer->gnp_nid));
2365                 return -ENOMEM;
2366         }
2367
2368         atomic_inc(&kgnilnd_data.kgn_neager_allocs);
2369
2370         LIBCFS_ALLOC(eagermsg, sizeof(*eagermsg) + *kgnilnd_tunables.kgn_max_immediate);
2371         if (eagermsg == NULL) {
2372                 kgnilnd_conn_decref(conn);
2373                 CERROR("couldn't allocate eager rx message for conn %p to %s\n",
2374                         conn, libcfs_nid2str(conn->gnc_peer->gnp_nid));
2375                 return -ENOMEM;
2376         }
2377
2378         /* copy msg and payload */
2379         memcpy(eagermsg, rxmsg, sizeof(*rxmsg) + rxmsg->gnm_payload_len);
2380         rx->grx_msg = eagermsg;
2381         rx->grx_eager = 1;
2382
2383         /* stash this for lnet_finalize on cancel-on-conn-close */
2384         rx->grx_lntmsg = lntmsg;
2385
2386         /* keep the same rx_t, it just has a new grx_msg now */
2387         *new_private = private;
2388
2389         /* release SMSG buffer */
2390         kgnilnd_release_msg(conn);
2391
2392         return 0;
2393 }
2394
2395 int
2396 kgnilnd_recv(lnet_ni_t *ni, void *private, lnet_msg_t *lntmsg,
2397              int delayed, unsigned int niov,
2398              struct iovec *iov, lnet_kiov_t *kiov,
2399              unsigned int offset, unsigned int mlen, unsigned int rlen)
2400 {
2401         kgn_rx_t    *rx = private;
2402         kgn_conn_t  *conn = rx->grx_conn;
2403         kgn_msg_t   *rxmsg = rx->grx_msg;
2404         kgn_tx_t    *tx;
2405         int          rc = 0;
2406         __u32        pload_cksum;
2407         ENTRY;
2408
2409         LASSERT(!in_interrupt());
2410         LASSERTF(mlen <= rlen, "%d <= %d\n", mlen, rlen);
2411         /* Either all pages or all vaddrs */
2412         LASSERTF(!(kiov != NULL && iov != NULL), "kiov %p iov %p\n",
2413                 kiov, iov);
2414
2415         GNIDBG_MSG(D_NET, rxmsg, "conn %p, rxmsg %p, lntmsg %p"
2416                 " niov=%d kiov=%p iov=%p offset=%d mlen=%d rlen=%d",
2417                 conn, rxmsg, lntmsg,
2418                 niov, kiov, iov, offset, mlen, rlen);
2419
2420         /* we need to lock here as recv can be called from any context */
2421         read_lock(&kgnilnd_data.kgn_peer_conn_lock);
2422         if (rx->grx_eager && conn->gnc_state != GNILND_CONN_ESTABLISHED) {
2423                 read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2424
2425                 /* someone closed the conn after we copied this out, nuke it */
2426                 kgnilnd_consume_rx(rx);
2427                 lnet_finalize(ni, lntmsg, conn->gnc_error);
2428                 RETURN(0);
2429         }
2430         read_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2431
2432         switch (rxmsg->gnm_type) {
2433         default:
2434                 GNIDBG_MSG(D_NETERROR, rxmsg, "conn %p, rx %p, rxmsg %p, lntmsg %p"
2435                 " niov=%d kiov=%p iov=%p offset=%d mlen=%d rlen=%d",
2436                 conn, rx, rxmsg, lntmsg, niov, kiov, iov, offset, mlen, rlen);
2437                 LBUG();
2438
2439         case GNILND_MSG_IMMEDIATE:
2440                 if (mlen > rxmsg->gnm_payload_len) {
2441                         GNIDBG_MSG(D_ERROR, rxmsg,
2442                                 "Immediate message from %s too big: %d > %d",
2443                                 libcfs_nid2str(conn->gnc_peer->gnp_nid), mlen,
2444                                 rxmsg->gnm_payload_len);
2445                         rc = -EINVAL;
2446                         kgnilnd_consume_rx(rx);
2447                         RETURN(rc);
2448                 }
2449
2450                 /* rxmsg[1] is a pointer to the payload, sitting in the buffer
2451                  * right after the kgn_msg_t header - so just 'cute' way of saying
2452                  * rxmsg + sizeof(kgn_msg_t) */
2453
2454                 /* check payload checksum if sent */
2455
2456                 if (*kgnilnd_tunables.kgn_checksum >= 2 &&
2457                         !rxmsg->gnm_payload_cksum &&
2458                         rxmsg->gnm_payload_len != 0)
2459                         GNIDBG_MSG(D_WARNING, rxmsg, "no msg payload checksum when enabled");
2460
2461                 if (rxmsg->gnm_payload_cksum != 0) {
2462                         /* gnm_payload_len set in kgnilnd_sendmsg from tx->tx_nob,
2463                          * which is what is used to calculate the cksum on the TX side */
2464                         pload_cksum = kgnilnd_cksum(&rxmsg[1], rxmsg->gnm_payload_len);
2465
2466                         if (rxmsg->gnm_payload_cksum != pload_cksum) {
2467                                 GNIDBG_MSG(D_NETERROR, rxmsg,
2468                                            "Bad payload checksum (%x expected %x)",
2469                                             pload_cksum, rxmsg->gnm_payload_cksum);
2470                                 switch (*kgnilnd_tunables.kgn_checksum_dump) {
2471                                 case 2:
2472                                         kgnilnd_dump_blob(D_BUFFS, "bad payload checksum",
2473                                                           &rxmsg[1], rxmsg->gnm_payload_len);
2474                                         /* fall through to dump */
2475                                 case 1:
2476                                         libcfs_debug_dumplog();
2477                                         break;
2478                                 default:
2479                                         break;
2480                                 }
2481                                 rc = -ENOKEY;
2482                                 /* checksum problems are fatal, kill the conn */
2483                                 kgnilnd_consume_rx(rx);
2484                                 kgnilnd_close_conn(conn, rc);
2485                                 RETURN(rc);
2486                         }
2487                 }
2488
2489                 if (kiov != NULL)
2490                         lnet_copy_flat2kiov(
2491                                 niov, kiov, offset,
2492                                 *kgnilnd_tunables.kgn_max_immediate,
2493                                 &rxmsg[1], 0, mlen);
2494                 else
2495                         lnet_copy_flat2iov(
2496                                 niov, iov, offset,
2497                                 *kgnilnd_tunables.kgn_max_immediate,
2498                                 &rxmsg[1], 0, mlen);
2499
2500                 kgnilnd_consume_rx(rx);
2501                 lnet_finalize(ni, lntmsg, 0);
2502                 RETURN(0);
2503
2504         case GNILND_MSG_PUT_REQ:
2505                 /* LNET wants to truncate or drop transaction, sending NAK */
2506                 if (mlen == 0) {
2507                         kgnilnd_consume_rx(rx);
2508                         lnet_finalize(ni, lntmsg, 0);
2509
2510                         /* only error if lntmsg == NULL, otherwise we are just
2511                          * short circuiting the rdma process of 0 bytes */
2512                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2513                                         lntmsg == NULL ? -ENOENT : 0,
2514                                         rxmsg->gnm_u.get.gngm_cookie,
2515                                         ni->ni_nid);
2516                         RETURN(0);
2517                 }
2518                 /* sending ACK with sink buff. info */
2519                 tx = kgnilnd_new_tx_msg(GNILND_MSG_PUT_ACK, ni->ni_nid);
2520                 if (tx == NULL) {
2521                         kgnilnd_consume_rx(rx);
2522                         RETURN(-ENOMEM);
2523                 }
2524
2525                 rc = kgnilnd_set_tx_id(tx, conn);
2526                 if (rc != 0) {
2527                         GOTO(nak_put_req, rc);
2528                 }
2529
2530                 rc = kgnilnd_setup_rdma_buffer(tx, niov, iov, kiov, offset, mlen);
2531                 if (rc != 0) {
2532                         GOTO(nak_put_req, rc);
2533                 }
2534
2535                 tx->tx_msg.gnm_u.putack.gnpam_src_cookie =
2536                         rxmsg->gnm_u.putreq.gnprm_cookie;
2537                 tx->tx_msg.gnm_u.putack.gnpam_dst_cookie = tx->tx_id.txe_cookie;
2538                 tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_addr =
2539                         (__u64)((unsigned long)tx->tx_buffer);
2540                 tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_nob = mlen;
2541
2542                 tx->tx_lntmsg[0] = lntmsg; /* finalize this on RDMA_DONE */
2543                 tx->tx_qtime = jiffies;
2544                 /* we only queue from kgnilnd_recv - we might get called from other contexts
2545                  * and we don't want to block the mutex in those cases */
2546
2547                 spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2548                 kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2549                 spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2550                 kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2551
2552                 kgnilnd_consume_rx(rx);
2553                 RETURN(0);
2554
2555 nak_put_req:
2556                 /* make sure we send an error back when the PUT fails */
2557                 kgnilnd_nak_rdma(conn, rxmsg->gnm_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2558                 kgnilnd_tx_done(tx, rc);
2559                 kgnilnd_consume_rx(rx);
2560
2561                 /* return magic LNet network error */
2562                 RETURN(-EIO);
2563         case GNILND_MSG_GET_REQ_REV:
2564                 /* LNET wants to truncate or drop transaction, sending NAK */
2565                 if (mlen == 0) {
2566                         kgnilnd_consume_rx(rx);
2567                         lnet_finalize(ni, lntmsg, 0);
2568
2569                         /* only error if lntmsg == NULL, otherwise we are just
2570                          * short circuiting the rdma process of 0 bytes */
2571                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2572                                         lntmsg == NULL ? -ENOENT : 0,
2573                                         rxmsg->gnm_u.get.gngm_cookie,
2574                                         ni->ni_nid);
2575                         RETURN(0);
2576                 }
2577                 /* lntmsg can be null when parsing a LNET_GET */
2578                 if (lntmsg != NULL) {
2579                         /* sending ACK with sink buff. info */
2580                         tx = kgnilnd_new_tx_msg(GNILND_MSG_GET_ACK_REV, ni->ni_nid);
2581                         if (tx == NULL) {
2582                                 kgnilnd_consume_rx(rx);
2583                                 RETURN(-ENOMEM);
2584                         }
2585
2586                         rc = kgnilnd_set_tx_id(tx, conn);
2587                         if (rc != 0)
2588                                 GOTO(nak_get_req_rev, rc);
2589
2590
2591                         rc = kgnilnd_setup_rdma_buffer(tx, niov, iov, kiov, offset, mlen);
2592                         if (rc != 0)
2593                                 GOTO(nak_get_req_rev, rc);
2594
2595
2596                         tx->tx_msg.gnm_u.putack.gnpam_src_cookie =
2597                                 rxmsg->gnm_u.putreq.gnprm_cookie;
2598                         tx->tx_msg.gnm_u.putack.gnpam_dst_cookie = tx->tx_id.txe_cookie;
2599                         tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_addr =
2600                                 (__u64)((unsigned long)tx->tx_buffer);
2601                         tx->tx_msg.gnm_u.putack.gnpam_desc.gnrd_nob = mlen;
2602
2603                         tx->tx_lntmsg[0] = lntmsg; /* finalize this on RDMA_DONE */
2604
2605                         /* we only queue from kgnilnd_recv - we might get called from other contexts
2606                          * and we don't want to block the mutex in those cases */
2607
2608                         spin_lock(&tx->tx_conn->gnc_device->gnd_lock);
2609                         kgnilnd_tx_add_state_locked(tx, NULL, tx->tx_conn, GNILND_TX_MAPQ, 1);
2610                         spin_unlock(&tx->tx_conn->gnc_device->gnd_lock);
2611                         kgnilnd_schedule_device(tx->tx_conn->gnc_device);
2612                 } else {
2613                         /* No match */
2614                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2615                                         -ENOENT,
2616                                         rxmsg->gnm_u.get.gngm_cookie,
2617                                         ni->ni_nid);
2618                 }
2619
2620                 kgnilnd_consume_rx(rx);
2621                 RETURN(0);
2622
2623 nak_get_req_rev:
2624                 /* make sure we send an error back when the GET fails */
2625                 kgnilnd_nak_rdma(conn, rxmsg->gnm_type, rc, rxmsg->gnm_u.get.gngm_cookie, ni->ni_nid);
2626                 kgnilnd_tx_done(tx, rc);
2627                 kgnilnd_consume_rx(rx);
2628
2629                 /* return magic LNet network error */
2630                 RETURN(-EIO);
2631
2632
2633         case GNILND_MSG_PUT_REQ_REV:
2634                 /* LNET wants to truncate or drop transaction, sending NAK */
2635                 if (mlen == 0) {
2636                         kgnilnd_consume_rx(rx);
2637                         lnet_finalize(ni, lntmsg, 0);
2638
2639                         /* only error if lntmsg == NULL, otherwise we are just
2640                          * short circuiting the rdma process of 0 bytes */
2641                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2642                                         lntmsg == NULL ? -ENOENT : 0,
2643                                         rxmsg->gnm_u.get.gngm_cookie,
2644                                         ni->ni_nid);
2645                         RETURN(0);
2646                 }
2647
2648                 if (lntmsg != NULL) {
2649                         /* Matched! */
2650                         kgnilnd_setup_rdma(ni, rx, lntmsg, mlen);
2651                 } else {
2652                         /* No match */
2653                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2654                                         -ENOENT,
2655                                         rxmsg->gnm_u.get.gngm_cookie,
2656                                         ni->ni_nid);
2657                 }
2658                 kgnilnd_consume_rx(rx);
2659                 RETURN(0);
2660         case GNILND_MSG_GET_REQ:
2661                 if (lntmsg != NULL) {
2662                         /* Matched! */
2663                         kgnilnd_setup_rdma(ni, rx, lntmsg, mlen);
2664                 } else {
2665                         /* No match */
2666                         kgnilnd_nak_rdma(conn, rxmsg->gnm_type,
2667                                         -ENOENT,
2668                                         rxmsg->gnm_u.get.gngm_cookie,
2669                                         ni->ni_nid);
2670                 }
2671                 kgnilnd_consume_rx(rx);
2672                 RETURN(0);
2673         }
2674         RETURN(0);
2675 }
2676
2677 /* needs write_lock on kgn_peer_conn_lock held */
2678 int
2679 kgnilnd_check_conn_timeouts_locked(kgn_conn_t *conn)
2680 {
2681         unsigned long      timeout, keepalive;
2682         unsigned long      now = jiffies;
2683         unsigned long      newest_last_rx;
2684         kgn_tx_t          *tx;
2685
2686         /* given that we found this conn hanging off a peer, it better damned
2687          * well be connected */
2688         LASSERTF(conn->gnc_state == GNILND_CONN_ESTABLISHED,
2689                  "conn 0x%p->%s with bad state%s\n", conn,
2690                  conn->gnc_peer ? libcfs_nid2str(conn->gnc_peer->gnp_nid)
2691                                : "<?>",
2692                  kgnilnd_conn_state2str(conn));
2693
2694         CDEBUG(D_NET, "checking conn %p->%s timeout %d keepalive %d "
2695                       "rx_diff %lu tx_diff %lu\n",
2696                 conn, libcfs_nid2str(conn->gnc_peer->gnp_nid),
2697                 conn->gnc_timeout, GNILND_TO2KA(conn->gnc_timeout),
2698                 cfs_duration_sec(now - conn->gnc_last_rx_cq),
2699                 cfs_duration_sec(now - conn->gnc_last_tx));
2700
2701         timeout = cfs_time_seconds(conn->gnc_timeout);
2702         keepalive = cfs_time_seconds(GNILND_TO2KA(conn->gnc_timeout));
2703
2704         /* just in case our lack of RX msg processing is gumming up the works - give the
2705          * remove an extra chance */
2706
2707         newest_last_rx = GNILND_LASTRX(conn);
2708
2709         if (time_after_eq(now, newest_last_rx + timeout)) {
2710                 uint32_t level = D_CONSOLE|D_NETERROR;
2711
2712                 if (conn->gnc_peer->gnp_down == GNILND_RCA_NODE_DOWN) {
2713                         level = D_NET;
2714                 }
2715                         GNIDBG_CONN(level, conn,
2716                         "No gnilnd traffic received from %s for %lu "
2717                         "seconds, terminating connection. Is node down? ",
2718                         libcfs_nid2str(conn->gnc_peer->gnp_nid),
2719                         cfs_duration_sec(now - newest_last_rx));
2720                 return -ETIMEDOUT;
2721         }
2722
2723         /* we don't timeout on last_tx stalls - we are going to trust the
2724          * underlying network to let us know when sends are failing.
2725          * At worst, the peer will timeout our RX stamp and drop the connection
2726          * at that point. We'll then see his CLOSE or at worst his RX
2727          * stamp stop and drop the connection on our end */
2728
2729         if (time_after_eq(now, conn->gnc_last_tx + keepalive)) {
2730                 CDEBUG(D_NET, "sending NOOP -> %s (%p idle %lu(%lu)) "
2731                        "last %lu/%lu/%lu %lus/%lus/%lus\n",
2732                        libcfs_nid2str(conn->gnc_peer->gnp_nid), conn,
2733                        cfs_duration_sec(jiffies - conn->gnc_last_tx),
2734                        keepalive,
2735                        conn->gnc_last_noop_want, conn->gnc_last_noop_sent,
2736                        conn->gnc_last_noop_cq,
2737                        cfs_duration_sec(jiffies - conn->gnc_last_noop_want),
2738                        cfs_duration_sec(jiffies - conn->gnc_last_noop_sent),
2739                        cfs_duration_sec(jiffies - conn->gnc_last_noop_cq));
2740                 set_mb(conn->gnc_last_noop_want, jiffies);
2741                 atomic_inc(&conn->gnc_reaper_noop);
2742                 if (CFS_FAIL_CHECK(CFS_FAIL_GNI_NOOP_SEND))
2743                         return 0;
2744
2745                 tx = kgnilnd_new_tx_msg(GNILND_MSG_NOOP, conn->gnc_peer->gnp_net->gnn_ni->ni_nid);
2746                 if (tx == NULL)
2747                         return 0;
2748                 kgnilnd_queue_tx(conn, tx);
2749         }
2750
2751         return 0;
2752 }
2753
2754 /* needs write_lock on kgn_peer_conn_lock held */
2755 void
2756 kgnilnd_check_peer_timeouts_locked(kgn_peer_t *peer, struct list_head *todie,
2757                                     struct list_head *souls)
2758 {
2759         unsigned long           timeout;
2760         kgn_conn_t             *conn, *connN = NULL;
2761         kgn_tx_t               *tx, *txN;
2762         int                     rc = 0;
2763         int                     count = 0;
2764         int                     reconnect;
2765         short                   releaseconn = 0;
2766         unsigned long           first_rx = 0;
2767
2768         CDEBUG(D_NET, "checking peer 0x%p->%s for timeouts; interval %lus\n",
2769                 peer, libcfs_nid2str(peer->gnp_nid),
2770                 peer->gnp_reconnect_interval);
2771
2772         timeout = cfs_time_seconds(MAX(*kgnilnd_tunables.kgn_timeout,
2773                                        GNILND_MIN_TIMEOUT));
2774
2775         conn = kgnilnd_find_conn_locked(peer);
2776         if (conn) {
2777                 /* if there is a valid conn, check the queues for timeouts */
2778                 rc = kgnilnd_check_conn_timeouts_locked(conn);
2779                 if (rc) {
2780                         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_RX_CLOSE_CLOSING)) {
2781                                 /* simulate a RX CLOSE after the timeout but before
2782                                  * the scheduler thread gets it */
2783                                 conn->gnc_close_recvd = GNILND_CLOSE_INJECT1;
2784                                 conn->gnc_peer_error = -ETIMEDOUT;
2785                         }
2786                         /* Once we mark closed, any of the scheduler threads could
2787                          * get it and move through before we hit the fail loc code */
2788                         kgnilnd_close_conn_locked(conn, rc);
2789                 } else {
2790                         /* first_rx is used to decide when to release a conn from purgatory.
2791                          */
2792                         first_rx = conn->gnc_first_rx;
2793                 }
2794         }
2795
2796         /* now regardless of starting new conn, find tx on peer queue that
2797          * are old and smell bad - do this first so we don't trigger
2798          * reconnect on empty queue if we timeout all */
2799         list_for_each_entry_safe(tx, txN, &peer->gnp_tx_queue, tx_list) {
2800                 if (time_after_eq(jiffies, tx->tx_qtime + timeout)) {
2801                         if (count == 0) {
2802                                 LCONSOLE_INFO("could not send to %s due to connection"
2803                                        " setup failure after %lu seconds\n",
2804                                        libcfs_nid2str(peer->gnp_nid),
2805                                        cfs_duration_sec(jiffies - tx->tx_qtime));
2806                         }
2807                         kgnilnd_tx_del_state_locked(tx, peer, NULL,
2808                                                    GNILND_TX_ALLOCD);
2809                         list_add_tail(&tx->tx_list, todie);
2810                         count++;
2811                 }
2812         }
2813
2814         if (count || peer->gnp_connecting == GNILND_PEER_KILL) {
2815                 CDEBUG(D_NET, "canceling %d tx for peer 0x%p->%s\n",
2816                         count, peer, libcfs_nid2str(peer->gnp_nid));
2817                 /* if we nuked all the TX, stop peer connection attempt (if there is one..) */
2818                 if (list_empty(&peer->gnp_tx_queue) ||
2819                         peer->gnp_connecting == GNILND_PEER_KILL) {
2820                         /* we pass down todie to use a common function - but we know there are
2821                          * no TX to add */
2822                         kgnilnd_cancel_peer_connect_locked(peer, todie);
2823                 }
2824         }
2825
2826         /* Don't reconnect if we are still trying to clear out old conns.
2827          * This prevents us sending traffic on the new mbox before ensuring we are done
2828          * with the old one */
2829         reconnect = (peer->gnp_down == GNILND_RCA_NODE_UP) &&
2830                     (atomic_read(&peer->gnp_dirty_eps) == 0);
2831
2832         /* if we are not connected and there are tx on the gnp_tx_queue waiting
2833          * to be sent, we'll check the reconnect interval and fire up a new
2834          * connection request */
2835
2836         if ((peer->gnp_connecting == GNILND_PEER_IDLE) &&
2837             (time_after_eq(jiffies, peer->gnp_reconnect_time)) &&
2838              !list_empty(&peer->gnp_tx_queue) && reconnect) {
2839
2840                 CDEBUG(D_NET, "starting connect to %s\n",
2841                         libcfs_nid2str(peer->gnp_nid));
2842                 LASSERTF(peer->gnp_connecting == GNILND_PEER_IDLE, "Peer was idle and we"
2843                         "have a write_lock, state issue %d\n", peer->gnp_connecting);
2844
2845                 peer->gnp_connecting = GNILND_PEER_CONNECT;
2846                 kgnilnd_peer_addref(peer); /* extra ref for connd */
2847
2848                 spin_lock(&peer->gnp_net->gnn_dev->gnd_connd_lock);
2849                 list_add_tail(&peer->gnp_connd_list,
2850                               &peer->gnp_net->gnn_dev->gnd_connd_peers);
2851                 spin_unlock(&peer->gnp_net->gnn_dev->gnd_connd_lock);
2852
2853                 kgnilnd_schedule_dgram(peer->gnp_net->gnn_dev);
2854         }
2855
2856         /* fail_loc to allow us to delay release of purgatory */
2857         if (CFS_FAIL_CHECK(CFS_FAIL_GNI_PURG_REL_DELAY))
2858                 return;
2859
2860         /* This check allows us to verify that the new conn is actually being used. This allows us to
2861          * pull the old conns out of purgatory if they have actually seen traffic.
2862          * We only release a conn from purgatory during stack reset, admin command, or when a peer reconnects
2863          */
2864         if (first_rx &&
2865                 time_after(jiffies, first_rx + cfs_time_seconds(*kgnilnd_tunables.kgn_hardware_timeout))) {
2866                 CDEBUG(D_INFO, "We can release peer %s conn's from purgatory %lu\n",
2867                         libcfs_nid2str(peer->gnp_nid), first_rx + cfs_time_seconds(*kgnilnd_tunables.kgn_hardware_timeout));
2868                 releaseconn = 1;
2869         }
2870
2871         list_for_each_entry_safe (conn, connN, &peer->gnp_conns, gnc_list) {
2872         /* check for purgatory timeouts */
2873                 if (conn->gnc_in_purgatory) {
2874                         /* We cannot detach this conn from purgatory if it has not been closed so we reschedule it
2875                          * that way the next time we check it we can detach it from purgatory
2876                          */
2877
2878                         if (conn->gnc_state != GNILND_CONN_DONE) {
2879                                 /* Skip over conns that are currently not DONE. If they arent already scheduled
2880                                  * for completion something in the state machine is broken.
2881                                  */
2882                                 continue;
2883                         }
2884
2885                         /* We only detach a conn that is in purgatory if we have received a close message,
2886                          * we have a new valid connection that has successfully received data, or an admin
2887                          * command tells us we need to detach.
2888                          */
2889
2890                         if (conn->gnc_close_recvd || releaseconn || conn->gnc_needs_detach) {
2891                                 unsigned long   waiting;
2892
2893                                 waiting = (long) jiffies - conn->gnc_last_rx_cq;
2894
2895                                 /* C.E: The remote peer is expected to close the
2896                                  * connection (see kgnilnd_check_conn_timeouts)
2897                                  * via the reaper thread and nuke out the MDD and
2898                                  * FMA resources after conn->gnc_timeout has expired
2899                                  * without an FMA RX */
2900                                 CDEBUG(D_NET, "Reconnected to %s in %lds or admin forced detach, dropping "
2901                                         " held resources\n",
2902                                         libcfs_nid2str(conn->gnc_peer->gnp_nid),
2903                                         cfs_duration_sec(waiting));
2904
2905                                 kgnilnd_detach_purgatory_locked(conn, souls);
2906                         }
2907                 }
2908         }
2909
2910         return;
2911 }
2912
2913 void
2914 kgnilnd_reaper_check(int idx)
2915 {
2916         struct list_head  *peers = &kgnilnd_data.kgn_peers[idx];
2917         struct list_head  *ctmp, *ctmpN;
2918         struct list_head   geriatrics;
2919         struct list_head   souls;
2920
2921         INIT_LIST_HEAD(&geriatrics);
2922         INIT_LIST_HEAD(&souls);
2923
2924         write_lock(&kgnilnd_data.kgn_peer_conn_lock);
2925
2926         list_for_each_safe(ctmp, ctmpN, peers) {
2927                 kgn_peer_t        *peer = NULL;
2928
2929                 /* don't timeout stuff if the network is mucked or shutting down */
2930                 if (kgnilnd_check_hw_quiesce()) {
2931                         break;
2932                 }
2933                 peer = list_entry(ctmp, kgn_peer_t, gnp_list);
2934
2935                 kgnilnd_check_peer_timeouts_locked(peer, &geriatrics, &souls);
2936         }
2937
2938         write_unlock(&kgnilnd_data.kgn_peer_conn_lock);
2939
2940         kgnilnd_txlist_done(&geriatrics, -EHOSTUNREACH);
2941         kgnilnd_release_purgatory_list(&souls);
2942 }
2943
2944 void
2945 kgnilnd_update_reaper_timeout(long timeout)
2946 {
2947         LASSERT(timeout > 0);
2948
2949         spin_lock(&kgnilnd_data.kgn_reaper_lock);
2950
2951         if (timeout < kgnilnd_data.kgn_new_min_timeout)
2952                 kgnilnd_data.kgn_new_min_timeout = timeout;
2953
2954         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
2955 }
2956
2957 static void
2958 kgnilnd_reaper_poke_with_stick(unsigned long arg)
2959 {
2960         wake_up(&kgnilnd_data.kgn_reaper_waitq);
2961 }
2962
2963 int
2964 kgnilnd_reaper(void *arg)
2965 {
2966         long               timeout;
2967         int                i;
2968         int                hash_index = 0;
2969         unsigned long      next_check_time = jiffies;
2970         long               current_min_timeout = MAX_SCHEDULE_TIMEOUT;
2971         struct timer_list  timer;
2972         DEFINE_WAIT(wait);
2973
2974         cfs_block_allsigs();
2975
2976         /* all gnilnd threads need to run fairly urgently */
2977         set_user_nice(current, *kgnilnd_tunables.kgn_nice);
2978         spin_lock(&kgnilnd_data.kgn_reaper_lock);
2979
2980         while (!kgnilnd_data.kgn_shutdown) {
2981                 /* I wake up every 'p' seconds to check for timeouts on some
2982                  * more peers.  I try to check every connection 'n' times
2983                  * within the global minimum of all keepalive and timeout
2984                  * intervals, to ensure I attend to every connection within
2985                  * (n+1)/n times its timeout intervals. */
2986                 const int     p = GNILND_REAPER_THREAD_WAKE;
2987                 const int     n = GNILND_REAPER_NCHECKS;
2988                 int           chunk;
2989                 /* to quiesce or to not quiesce, that is the question */
2990                 if (unlikely(kgnilnd_data.kgn_quiesce_trigger)) {
2991                         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
2992                         KGNILND_SPIN_QUIESCE;
2993                         spin_lock(&kgnilnd_data.kgn_reaper_lock);
2994                 }
2995
2996                 /* careful with the jiffy wrap... */
2997                 timeout = (long)(next_check_time - jiffies);
2998
2999                 if (timeout > 0) {
3000                         prepare_to_wait(&kgnilnd_data.kgn_reaper_waitq, &wait,
3001                                         TASK_INTERRUPTIBLE);
3002                         spin_unlock(&kgnilnd_data.kgn_reaper_lock);
3003                         setup_timer(&timer, kgnilnd_reaper_poke_with_stick,
3004                                     next_check_time);
3005                         mod_timer(&timer, (long) jiffies + timeout);
3006
3007                         /* check flag variables before comitting */
3008                         if (!kgnilnd_data.kgn_shutdown &&
3009                             !kgnilnd_data.kgn_quiesce_trigger) {
3010                                 CDEBUG(D_INFO, "schedule timeout %ld (%lu sec)\n",
3011                                        timeout, cfs_duration_sec(timeout));
3012                                 schedule();
3013                                 CDEBUG(D_INFO, "awake after schedule\n");
3014                         }
3015
3016                         del_singleshot_timer_sync(&timer);
3017                         spin_lock(&kgnilnd_data.kgn_reaper_lock);
3018                         finish_wait(&kgnilnd_data.kgn_reaper_waitq, &wait);
3019                         continue;
3020                 }
3021
3022                 /* new_min_timeout is set from the conn timeouts and keepalive
3023                  * this should end up with a min timeout of
3024                  * GNILND_TIMEOUT2KEEPALIVE(t) or roughly LND_TIMEOUT/2 */
3025                 if (kgnilnd_data.kgn_new_min_timeout < current_min_timeout) {
3026                         current_min_timeout = kgnilnd_data.kgn_new_min_timeout;
3027                         CDEBUG(D_NET, "Set new min timeout %ld\n",
3028                                current_min_timeout);
3029                 }
3030
3031                 spin_unlock(&kgnilnd_data.kgn_reaper_lock);
3032
3033                 /* Compute how many table entries to check now so I get round
3034                  * the whole table fast enough given that I do this at fixed
3035                  * intervals of 'p' seconds) */
3036                 chunk = *kgnilnd_tunables.kgn_peer_hash_size;
3037                 if (kgnilnd_data.kgn_new_min_timeout > n * p)