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