Whamcloud - gitweb
LU-9817 lnet: safe access to msg
[fs/lustre-release.git] / lnet / lnet / lib-move.c
1 /*
2  * GPL HEADER START
3  *
4  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
5  *
6  * This program is free software; you can redistribute it and/or modify
7  * it under the terms of the GNU General Public License version 2 only,
8  * as published by the Free Software Foundation.
9  *
10  * This program is distributed in the hope that it will be useful, but
11  * WITHOUT ANY WARRANTY; without even the implied warranty of
12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
13  * General Public License version 2 for more details (a copy is included
14  * in the LICENSE file that accompanied this code).
15  *
16  * You should have received a copy of the GNU General Public License
17  * version 2 along with this program; If not, see
18  * http://www.gnu.org/licenses/gpl-2.0.html
19  *
20  * GPL HEADER END
21  */
22 /*
23  * Copyright (c) 2003, 2010, Oracle and/or its affiliates. All rights reserved.
24  * Use is subject to license terms.
25  *
26  * Copyright (c) 2012, 2016, Intel Corporation.
27  */
28 /*
29  * This file is part of Lustre, http://www.lustre.org/
30  * Lustre is a trademark of Sun Microsystems, Inc.
31  *
32  * lnet/lnet/lib-move.c
33  *
34  * Data movement routines
35  */
36
37 #define DEBUG_SUBSYSTEM S_LNET
38
39 #include <lnet/lib-lnet.h>
40 #include <linux/nsproxy.h>
41 #include <net/net_namespace.h>
42
43 static int local_nid_dist_zero = 1;
44 module_param(local_nid_dist_zero, int, 0444);
45 MODULE_PARM_DESC(local_nid_dist_zero, "Reserved");
46
47 int
48 lnet_fail_nid(lnet_nid_t nid, unsigned int threshold)
49 {
50         struct lnet_test_peer *tp;
51         struct list_head *el;
52         struct list_head *next;
53         struct list_head  cull;
54
55         /* NB: use lnet_net_lock(0) to serialize operations on test peers */
56         if (threshold != 0) {
57                 /* Adding a new entry */
58                 LIBCFS_ALLOC(tp, sizeof(*tp));
59                 if (tp == NULL)
60                         return -ENOMEM;
61
62                 tp->tp_nid = nid;
63                 tp->tp_threshold = threshold;
64
65                 lnet_net_lock(0);
66                 list_add_tail(&tp->tp_list, &the_lnet.ln_test_peers);
67                 lnet_net_unlock(0);
68                 return 0;
69         }
70
71         /* removing entries */
72         INIT_LIST_HEAD(&cull);
73
74         lnet_net_lock(0);
75
76         list_for_each_safe(el, next, &the_lnet.ln_test_peers) {
77                 tp = list_entry(el, struct lnet_test_peer, tp_list);
78
79                 if (tp->tp_threshold == 0 ||    /* needs culling anyway */
80                     nid == LNET_NID_ANY ||      /* removing all entries */
81                     tp->tp_nid == nid) {        /* matched this one */
82                         list_del(&tp->tp_list);
83                         list_add(&tp->tp_list, &cull);
84                 }
85         }
86
87         lnet_net_unlock(0);
88
89         while (!list_empty(&cull)) {
90                 tp = list_entry(cull.next, struct lnet_test_peer, tp_list);
91
92                 list_del(&tp->tp_list);
93                 LIBCFS_FREE(tp, sizeof(*tp));
94         }
95         return 0;
96 }
97
98 static int
99 fail_peer (lnet_nid_t nid, int outgoing)
100 {
101         struct lnet_test_peer *tp;
102         struct list_head *el;
103         struct list_head *next;
104         struct list_head  cull;
105         int               fail = 0;
106
107         INIT_LIST_HEAD(&cull);
108
109         /* NB: use lnet_net_lock(0) to serialize operations on test peers */
110         lnet_net_lock(0);
111
112         list_for_each_safe(el, next, &the_lnet.ln_test_peers) {
113                 tp = list_entry(el, struct lnet_test_peer, tp_list);
114
115                 if (tp->tp_threshold == 0) {
116                         /* zombie entry */
117                         if (outgoing) {
118                                 /* only cull zombies on outgoing tests,
119                                  * since we may be at interrupt priority on
120                                  * incoming messages. */
121                                 list_del(&tp->tp_list);
122                                 list_add(&tp->tp_list, &cull);
123                         }
124                         continue;
125                 }
126
127                 if (tp->tp_nid == LNET_NID_ANY ||       /* fail every peer */
128                     nid == tp->tp_nid) {                /* fail this peer */
129                         fail = 1;
130
131                         if (tp->tp_threshold != LNET_MD_THRESH_INF) {
132                                 tp->tp_threshold--;
133                                 if (outgoing &&
134                                     tp->tp_threshold == 0) {
135                                         /* see above */
136                                         list_del(&tp->tp_list);
137                                         list_add(&tp->tp_list, &cull);
138                                 }
139                         }
140                         break;
141                 }
142         }
143
144         lnet_net_unlock(0);
145
146         while (!list_empty(&cull)) {
147                 tp = list_entry(cull.next, struct lnet_test_peer, tp_list);
148                 list_del(&tp->tp_list);
149
150                 LIBCFS_FREE(tp, sizeof(*tp));
151         }
152
153         return fail;
154 }
155
156 unsigned int
157 lnet_iov_nob(unsigned int niov, struct kvec *iov)
158 {
159         unsigned int nob = 0;
160
161         LASSERT(niov == 0 || iov != NULL);
162         while (niov-- > 0)
163                 nob += (iov++)->iov_len;
164
165         return (nob);
166 }
167 EXPORT_SYMBOL(lnet_iov_nob);
168
169 void
170 lnet_copy_iov2iov(unsigned int ndiov, struct kvec *diov, unsigned int doffset,
171                   unsigned int nsiov, struct kvec *siov, unsigned int soffset,
172                   unsigned int nob)
173 {
174         /* NB diov, siov are READ-ONLY */
175         unsigned int  this_nob;
176
177         if (nob == 0)
178                 return;
179
180         /* skip complete frags before 'doffset' */
181         LASSERT(ndiov > 0);
182         while (doffset >= diov->iov_len) {
183                 doffset -= diov->iov_len;
184                 diov++;
185                 ndiov--;
186                 LASSERT(ndiov > 0);
187         }
188
189         /* skip complete frags before 'soffset' */
190         LASSERT(nsiov > 0);
191         while (soffset >= siov->iov_len) {
192                 soffset -= siov->iov_len;
193                 siov++;
194                 nsiov--;
195                 LASSERT(nsiov > 0);
196         }
197
198         do {
199                 LASSERT(ndiov > 0);
200                 LASSERT(nsiov > 0);
201                 this_nob = MIN(diov->iov_len - doffset,
202                                siov->iov_len - soffset);
203                 this_nob = MIN(this_nob, nob);
204
205                 memcpy((char *)diov->iov_base + doffset,
206                        (char *)siov->iov_base + soffset, this_nob);
207                 nob -= this_nob;
208
209                 if (diov->iov_len > doffset + this_nob) {
210                         doffset += this_nob;
211                 } else {
212                         diov++;
213                         ndiov--;
214                         doffset = 0;
215                 }
216
217                 if (siov->iov_len > soffset + this_nob) {
218                         soffset += this_nob;
219                 } else {
220                         siov++;
221                         nsiov--;
222                         soffset = 0;
223                 }
224         } while (nob > 0);
225 }
226 EXPORT_SYMBOL(lnet_copy_iov2iov);
227
228 int
229 lnet_extract_iov(int dst_niov, struct kvec *dst,
230                  int src_niov, struct kvec *src,
231                  unsigned int offset, unsigned int len)
232 {
233         /* Initialise 'dst' to the subset of 'src' starting at 'offset',
234          * for exactly 'len' bytes, and return the number of entries.
235          * NB not destructive to 'src' */
236         unsigned int    frag_len;
237         unsigned int    niov;
238
239         if (len == 0)                           /* no data => */
240                 return (0);                     /* no frags */
241
242         LASSERT(src_niov > 0);
243         while (offset >= src->iov_len) {      /* skip initial frags */
244                 offset -= src->iov_len;
245                 src_niov--;
246                 src++;
247                 LASSERT(src_niov > 0);
248         }
249
250         niov = 1;
251         for (;;) {
252                 LASSERT(src_niov > 0);
253                 LASSERT((int)niov <= dst_niov);
254
255                 frag_len = src->iov_len - offset;
256                 dst->iov_base = ((char *)src->iov_base) + offset;
257
258                 if (len <= frag_len) {
259                         dst->iov_len = len;
260                         return (niov);
261                 }
262
263                 dst->iov_len = frag_len;
264
265                 len -= frag_len;
266                 dst++;
267                 src++;
268                 niov++;
269                 src_niov--;
270                 offset = 0;
271         }
272 }
273 EXPORT_SYMBOL(lnet_extract_iov);
274
275
276 unsigned int
277 lnet_kiov_nob(unsigned int niov, lnet_kiov_t *kiov)
278 {
279         unsigned int  nob = 0;
280
281         LASSERT(niov == 0 || kiov != NULL);
282         while (niov-- > 0)
283                 nob += (kiov++)->kiov_len;
284
285         return (nob);
286 }
287 EXPORT_SYMBOL(lnet_kiov_nob);
288
289 void
290 lnet_copy_kiov2kiov(unsigned int ndiov, lnet_kiov_t *diov, unsigned int doffset,
291                     unsigned int nsiov, lnet_kiov_t *siov, unsigned int soffset,
292                     unsigned int nob)
293 {
294         /* NB diov, siov are READ-ONLY */
295         unsigned int    this_nob;
296         char           *daddr = NULL;
297         char           *saddr = NULL;
298
299         if (nob == 0)
300                 return;
301
302         LASSERT (!in_interrupt ());
303
304         LASSERT (ndiov > 0);
305         while (doffset >= diov->kiov_len) {
306                 doffset -= diov->kiov_len;
307                 diov++;
308                 ndiov--;
309                 LASSERT(ndiov > 0);
310         }
311
312         LASSERT(nsiov > 0);
313         while (soffset >= siov->kiov_len) {
314                 soffset -= siov->kiov_len;
315                 siov++;
316                 nsiov--;
317                 LASSERT(nsiov > 0);
318         }
319
320         do {
321                 LASSERT(ndiov > 0);
322                 LASSERT(nsiov > 0);
323                 this_nob = MIN(diov->kiov_len - doffset,
324                                siov->kiov_len - soffset);
325                 this_nob = MIN(this_nob, nob);
326
327                 if (daddr == NULL)
328                         daddr = ((char *)kmap(diov->kiov_page)) +
329                                 diov->kiov_offset + doffset;
330                 if (saddr == NULL)
331                         saddr = ((char *)kmap(siov->kiov_page)) +
332                                 siov->kiov_offset + soffset;
333
334                 /* Vanishing risk of kmap deadlock when mapping 2 pages.
335                  * However in practice at least one of the kiovs will be mapped
336                  * kernel pages and the map/unmap will be NOOPs */
337
338                 memcpy (daddr, saddr, this_nob);
339                 nob -= this_nob;
340
341                 if (diov->kiov_len > doffset + this_nob) {
342                         daddr += this_nob;
343                         doffset += this_nob;
344                 } else {
345                         kunmap(diov->kiov_page);
346                         daddr = NULL;
347                         diov++;
348                         ndiov--;
349                         doffset = 0;
350                 }
351
352                 if (siov->kiov_len > soffset + this_nob) {
353                         saddr += this_nob;
354                         soffset += this_nob;
355                 } else {
356                         kunmap(siov->kiov_page);
357                         saddr = NULL;
358                         siov++;
359                         nsiov--;
360                         soffset = 0;
361                 }
362         } while (nob > 0);
363
364         if (daddr != NULL)
365                 kunmap(diov->kiov_page);
366         if (saddr != NULL)
367                 kunmap(siov->kiov_page);
368 }
369 EXPORT_SYMBOL(lnet_copy_kiov2kiov);
370
371 void
372 lnet_copy_kiov2iov (unsigned int niov, struct kvec *iov, unsigned int iovoffset,
373                     unsigned int nkiov, lnet_kiov_t *kiov, unsigned int kiovoffset,
374                     unsigned int nob)
375 {
376         /* NB iov, kiov are READ-ONLY */
377         unsigned int    this_nob;
378         char           *addr = NULL;
379
380         if (nob == 0)
381                 return;
382
383         LASSERT (!in_interrupt ());
384
385         LASSERT (niov > 0);
386         while (iovoffset >= iov->iov_len) {
387                 iovoffset -= iov->iov_len;
388                 iov++;
389                 niov--;
390                 LASSERT(niov > 0);
391         }
392
393         LASSERT(nkiov > 0);
394         while (kiovoffset >= kiov->kiov_len) {
395                 kiovoffset -= kiov->kiov_len;
396                 kiov++;
397                 nkiov--;
398                 LASSERT(nkiov > 0);
399         }
400
401         do {
402                 LASSERT(niov > 0);
403                 LASSERT(nkiov > 0);
404                 this_nob = MIN(iov->iov_len - iovoffset,
405                                kiov->kiov_len - kiovoffset);
406                 this_nob = MIN(this_nob, nob);
407
408                 if (addr == NULL)
409                         addr = ((char *)kmap(kiov->kiov_page)) +
410                                 kiov->kiov_offset + kiovoffset;
411
412                 memcpy((char *)iov->iov_base + iovoffset, addr, this_nob);
413                 nob -= this_nob;
414
415                 if (iov->iov_len > iovoffset + this_nob) {
416                         iovoffset += this_nob;
417                 } else {
418                         iov++;
419                         niov--;
420                         iovoffset = 0;
421                 }
422
423                 if (kiov->kiov_len > kiovoffset + this_nob) {
424                         addr += this_nob;
425                         kiovoffset += this_nob;
426                 } else {
427                         kunmap(kiov->kiov_page);
428                         addr = NULL;
429                         kiov++;
430                         nkiov--;
431                         kiovoffset = 0;
432                 }
433
434         } while (nob > 0);
435
436         if (addr != NULL)
437                 kunmap(kiov->kiov_page);
438 }
439 EXPORT_SYMBOL(lnet_copy_kiov2iov);
440
441 void
442 lnet_copy_iov2kiov(unsigned int nkiov, lnet_kiov_t *kiov, unsigned int kiovoffset,
443                    unsigned int niov, struct kvec *iov, unsigned int iovoffset,
444                    unsigned int nob)
445 {
446         /* NB kiov, iov are READ-ONLY */
447         unsigned int    this_nob;
448         char           *addr = NULL;
449
450         if (nob == 0)
451                 return;
452
453         LASSERT (!in_interrupt ());
454
455         LASSERT (nkiov > 0);
456         while (kiovoffset >= kiov->kiov_len) {
457                 kiovoffset -= kiov->kiov_len;
458                 kiov++;
459                 nkiov--;
460                 LASSERT(nkiov > 0);
461         }
462
463         LASSERT(niov > 0);
464         while (iovoffset >= iov->iov_len) {
465                 iovoffset -= iov->iov_len;
466                 iov++;
467                 niov--;
468                 LASSERT(niov > 0);
469         }
470
471         do {
472                 LASSERT(nkiov > 0);
473                 LASSERT(niov > 0);
474                 this_nob = MIN(kiov->kiov_len - kiovoffset,
475                                iov->iov_len - iovoffset);
476                 this_nob = MIN(this_nob, nob);
477
478                 if (addr == NULL)
479                         addr = ((char *)kmap(kiov->kiov_page)) +
480                                 kiov->kiov_offset + kiovoffset;
481
482                 memcpy (addr, (char *)iov->iov_base + iovoffset, this_nob);
483                 nob -= this_nob;
484
485                 if (kiov->kiov_len > kiovoffset + this_nob) {
486                         addr += this_nob;
487                         kiovoffset += this_nob;
488                 } else {
489                         kunmap(kiov->kiov_page);
490                         addr = NULL;
491                         kiov++;
492                         nkiov--;
493                         kiovoffset = 0;
494                 }
495
496                 if (iov->iov_len > iovoffset + this_nob) {
497                         iovoffset += this_nob;
498                 } else {
499                         iov++;
500                         niov--;
501                         iovoffset = 0;
502                 }
503         } while (nob > 0);
504
505         if (addr != NULL)
506                 kunmap(kiov->kiov_page);
507 }
508 EXPORT_SYMBOL(lnet_copy_iov2kiov);
509
510 int
511 lnet_extract_kiov(int dst_niov, lnet_kiov_t *dst,
512                   int src_niov, lnet_kiov_t *src,
513                   unsigned int offset, unsigned int len)
514 {
515         /* Initialise 'dst' to the subset of 'src' starting at 'offset',
516          * for exactly 'len' bytes, and return the number of entries.
517          * NB not destructive to 'src' */
518         unsigned int    frag_len;
519         unsigned int    niov;
520
521         if (len == 0)                           /* no data => */
522                 return (0);                     /* no frags */
523
524         LASSERT(src_niov > 0);
525         while (offset >= src->kiov_len) {      /* skip initial frags */
526                 offset -= src->kiov_len;
527                 src_niov--;
528                 src++;
529                 LASSERT(src_niov > 0);
530         }
531
532         niov = 1;
533         for (;;) {
534                 LASSERT(src_niov > 0);
535                 LASSERT((int)niov <= dst_niov);
536
537                 frag_len = src->kiov_len - offset;
538                 dst->kiov_page = src->kiov_page;
539                 dst->kiov_offset = src->kiov_offset + offset;
540
541                 if (len <= frag_len) {
542                         dst->kiov_len = len;
543                         LASSERT(dst->kiov_offset + dst->kiov_len <= PAGE_SIZE);
544                         return niov;
545                 }
546
547                 dst->kiov_len = frag_len;
548                 LASSERT(dst->kiov_offset + dst->kiov_len <= PAGE_SIZE);
549
550                 len -= frag_len;
551                 dst++;
552                 src++;
553                 niov++;
554                 src_niov--;
555                 offset = 0;
556         }
557 }
558 EXPORT_SYMBOL(lnet_extract_kiov);
559
560 void
561 lnet_ni_recv(struct lnet_ni *ni, void *private, struct lnet_msg *msg,
562              int delayed, unsigned int offset, unsigned int mlen,
563              unsigned int rlen)
564 {
565         unsigned int  niov = 0;
566         struct kvec *iov = NULL;
567         lnet_kiov_t  *kiov = NULL;
568         int           rc;
569
570         LASSERT (!in_interrupt ());
571         LASSERT (mlen == 0 || msg != NULL);
572
573         if (msg != NULL) {
574                 LASSERT(msg->msg_receiving);
575                 LASSERT(!msg->msg_sending);
576                 LASSERT(rlen == msg->msg_len);
577                 LASSERT(mlen <= msg->msg_len);
578                 LASSERT(msg->msg_offset == offset);
579                 LASSERT(msg->msg_wanted == mlen);
580
581                 msg->msg_receiving = 0;
582
583                 if (mlen != 0) {
584                         niov = msg->msg_niov;
585                         iov  = msg->msg_iov;
586                         kiov = msg->msg_kiov;
587
588                         LASSERT (niov > 0);
589                         LASSERT ((iov == NULL) != (kiov == NULL));
590                 }
591         }
592
593         rc = (ni->ni_net->net_lnd->lnd_recv)(ni, private, msg, delayed,
594                                              niov, iov, kiov, offset, mlen,
595                                              rlen);
596         if (rc < 0)
597                 lnet_finalize(msg, rc);
598 }
599
600 static void
601 lnet_setpayloadbuffer(struct lnet_msg *msg)
602 {
603         struct lnet_libmd *md = msg->msg_md;
604
605         LASSERT(msg->msg_len > 0);
606         LASSERT(!msg->msg_routing);
607         LASSERT(md != NULL);
608         LASSERT(msg->msg_niov == 0);
609         LASSERT(msg->msg_iov == NULL);
610         LASSERT(msg->msg_kiov == NULL);
611
612         msg->msg_niov = md->md_niov;
613         if ((md->md_options & LNET_MD_KIOV) != 0)
614                 msg->msg_kiov = md->md_iov.kiov;
615         else
616                 msg->msg_iov = md->md_iov.iov;
617 }
618
619 void
620 lnet_prep_send(struct lnet_msg *msg, int type, struct lnet_process_id target,
621                unsigned int offset, unsigned int len)
622 {
623         msg->msg_type = type;
624         msg->msg_target = target;
625         msg->msg_len = len;
626         msg->msg_offset = offset;
627
628         if (len != 0)
629                 lnet_setpayloadbuffer(msg);
630
631         memset (&msg->msg_hdr, 0, sizeof (msg->msg_hdr));
632         msg->msg_hdr.type           = cpu_to_le32(type);
633         msg->msg_hdr.dest_pid       = cpu_to_le32(target.pid);
634         /* src_nid will be set later */
635         msg->msg_hdr.src_pid        = cpu_to_le32(the_lnet.ln_pid);
636         msg->msg_hdr.payload_length = cpu_to_le32(len);
637 }
638
639 static void
640 lnet_ni_send(struct lnet_ni *ni, struct lnet_msg *msg)
641 {
642         void   *priv = msg->msg_private;
643         int     rc;
644
645         LASSERT (!in_interrupt ());
646         LASSERT (LNET_NETTYP(LNET_NIDNET(ni->ni_nid)) == LOLND ||
647                  (msg->msg_txcredit && msg->msg_peertxcredit));
648
649         rc = (ni->ni_net->net_lnd->lnd_send)(ni, priv, msg);
650         if (rc < 0)
651                 lnet_finalize(msg, rc);
652 }
653
654 static int
655 lnet_ni_eager_recv(struct lnet_ni *ni, struct lnet_msg *msg)
656 {
657         int     rc;
658
659         LASSERT(!msg->msg_sending);
660         LASSERT(msg->msg_receiving);
661         LASSERT(!msg->msg_rx_ready_delay);
662         LASSERT(ni->ni_net->net_lnd->lnd_eager_recv != NULL);
663
664         msg->msg_rx_ready_delay = 1;
665         rc = (ni->ni_net->net_lnd->lnd_eager_recv)(ni, msg->msg_private, msg,
666                                                   &msg->msg_private);
667         if (rc != 0) {
668                 CERROR("recv from %s / send to %s aborted: "
669                        "eager_recv failed %d\n",
670                        libcfs_nid2str(msg->msg_rxpeer->lpni_nid),
671                        libcfs_id2str(msg->msg_target), rc);
672                 LASSERT(rc < 0); /* required by my callers */
673         }
674
675         return rc;
676 }
677
678 /*
679  * This function can be called from two paths:
680  *      1. when sending a message
681  *      2. when decommiting a message (lnet_msg_decommit_tx())
682  * In both these cases the peer_ni should have it's reference count
683  * acquired by the caller and therefore it is safe to drop the spin
684  * lock before calling lnd_query()
685  */
686 static void
687 lnet_ni_query_locked(struct lnet_ni *ni, struct lnet_peer_ni *lp)
688 {
689         cfs_time_t last_alive = 0;
690         int cpt = lnet_cpt_of_nid_locked(lp->lpni_nid, ni);
691
692         LASSERT(lnet_peer_aliveness_enabled(lp));
693         LASSERT(ni->ni_net->net_lnd->lnd_query != NULL);
694
695         lnet_net_unlock(cpt);
696         (ni->ni_net->net_lnd->lnd_query)(ni, lp->lpni_nid, &last_alive);
697         lnet_net_lock(cpt);
698
699         lp->lpni_last_query = cfs_time_current();
700
701         if (last_alive != 0) /* NI has updated timestamp */
702                 lp->lpni_last_alive = last_alive;
703 }
704
705 /* NB: always called with lnet_net_lock held */
706 static inline int
707 lnet_peer_is_alive (struct lnet_peer_ni *lp, cfs_time_t now)
708 {
709         int        alive;
710         cfs_time_t deadline;
711
712         LASSERT (lnet_peer_aliveness_enabled(lp));
713
714         /*
715          * Trust lnet_notify() if it has more recent aliveness news, but
716          * ignore the initial assumed death (see lnet_peers_start_down()).
717          */
718         spin_lock(&lp->lpni_lock);
719         if (!lp->lpni_alive && lp->lpni_alive_count > 0 &&
720             cfs_time_aftereq(lp->lpni_timestamp, lp->lpni_last_alive)) {
721                 spin_unlock(&lp->lpni_lock);
722                 return 0;
723         }
724
725         deadline =
726           cfs_time_add(lp->lpni_last_alive,
727                        cfs_time_seconds(lp->lpni_net->net_tunables.
728                                         lct_peer_timeout));
729         alive = cfs_time_after(deadline, now);
730
731         /*
732          * Update obsolete lp_alive except for routers assumed to be dead
733          * initially, because router checker would update aliveness in this
734          * case, and moreover lpni_last_alive at peer creation is assumed.
735          */
736         if (alive && !lp->lpni_alive &&
737             !(lnet_isrouter(lp) && lp->lpni_alive_count == 0)) {
738                 spin_unlock(&lp->lpni_lock);
739                 lnet_notify_locked(lp, 0, 1, lp->lpni_last_alive);
740         } else {
741                 spin_unlock(&lp->lpni_lock);
742         }
743
744         return alive;
745 }
746
747
748 /* NB: returns 1 when alive, 0 when dead, negative when error;
749  *     may drop the lnet_net_lock */
750 static int
751 lnet_peer_alive_locked (struct lnet_ni *ni, struct lnet_peer_ni *lp)
752 {
753         cfs_time_t now = cfs_time_current();
754
755         if (!lnet_peer_aliveness_enabled(lp))
756                 return -ENODEV;
757
758         if (lnet_peer_is_alive(lp, now))
759                 return 1;
760
761         /*
762          * Peer appears dead, but we should avoid frequent NI queries (at
763          * most once per lnet_queryinterval seconds).
764          */
765         if (lp->lpni_last_query != 0) {
766                 static const int lnet_queryinterval = 1;
767
768                 cfs_time_t next_query =
769                            cfs_time_add(lp->lpni_last_query,
770                                         cfs_time_seconds(lnet_queryinterval));
771
772                 if (cfs_time_before(now, next_query)) {
773                         if (lp->lpni_alive)
774                                 CWARN("Unexpected aliveness of peer %s: "
775                                       "%d < %d (%d/%d)\n",
776                                       libcfs_nid2str(lp->lpni_nid),
777                                       (int)now, (int)next_query,
778                                       lnet_queryinterval,
779                                       lp->lpni_net->net_tunables.lct_peer_timeout);
780                         return 0;
781                 }
782         }
783
784         /* query NI for latest aliveness news */
785         lnet_ni_query_locked(ni, lp);
786
787         if (lnet_peer_is_alive(lp, now))
788                 return 1;
789
790         lnet_notify_locked(lp, 0, 0, lp->lpni_last_alive);
791         return 0;
792 }
793
794 /**
795  * \param msg The message to be sent.
796  * \param do_send True if lnet_ni_send() should be called in this function.
797  *        lnet_send() is going to lnet_net_unlock immediately after this, so
798  *        it sets do_send FALSE and I don't do the unlock/send/lock bit.
799  *
800  * \retval LNET_CREDIT_OK If \a msg sent or OK to send.
801  * \retval LNET_CREDIT_WAIT If \a msg blocked for credit.
802  * \retval -EHOSTUNREACH If the next hop of the message appears dead.
803  * \retval -ECANCELED If the MD of the message has been unlinked.
804  */
805 static int
806 lnet_post_send_locked(struct lnet_msg *msg, int do_send)
807 {
808         struct lnet_peer_ni     *lp = msg->msg_txpeer;
809         struct lnet_ni          *ni = msg->msg_txni;
810         int                     cpt = msg->msg_tx_cpt;
811         struct lnet_tx_queue    *tq = ni->ni_tx_queues[cpt];
812
813         /* non-lnet_send() callers have checked before */
814         LASSERT(!do_send || msg->msg_tx_delayed);
815         LASSERT(!msg->msg_receiving);
816         LASSERT(msg->msg_tx_committed);
817
818         /* NB 'lp' is always the next hop */
819         if ((msg->msg_target.pid & LNET_PID_USERFLAG) == 0 &&
820             lnet_peer_alive_locked(ni, lp) == 0) {
821                 the_lnet.ln_counters[cpt]->drop_count++;
822                 the_lnet.ln_counters[cpt]->drop_length += msg->msg_len;
823                 lnet_net_unlock(cpt);
824                 if (msg->msg_txpeer)
825                         atomic_inc(&msg->msg_txpeer->lpni_stats.drop_count);
826                 if (msg->msg_txni)
827                         atomic_inc(&msg->msg_txni->ni_stats.drop_count);
828
829                 CNETERR("Dropping message for %s: peer not alive\n",
830                         libcfs_id2str(msg->msg_target));
831                 if (do_send)
832                         lnet_finalize(msg, -EHOSTUNREACH);
833
834                 lnet_net_lock(cpt);
835                 return -EHOSTUNREACH;
836         }
837
838         if (msg->msg_md != NULL &&
839             (msg->msg_md->md_flags & LNET_MD_FLAG_ABORTED) != 0) {
840                 lnet_net_unlock(cpt);
841
842                 CNETERR("Aborting message for %s: LNetM[DE]Unlink() already "
843                         "called on the MD/ME.\n",
844                         libcfs_id2str(msg->msg_target));
845                 if (do_send)
846                         lnet_finalize(msg, -ECANCELED);
847
848                 lnet_net_lock(cpt);
849                 return -ECANCELED;
850         }
851
852         if (!msg->msg_peertxcredit) {
853                 spin_lock(&lp->lpni_lock);
854                 LASSERT((lp->lpni_txcredits < 0) ==
855                         !list_empty(&lp->lpni_txq));
856
857                 msg->msg_peertxcredit = 1;
858                 lp->lpni_txqnob += msg->msg_len + sizeof(struct lnet_hdr);
859                 lp->lpni_txcredits--;
860
861                 if (lp->lpni_txcredits < lp->lpni_mintxcredits)
862                         lp->lpni_mintxcredits = lp->lpni_txcredits;
863
864                 if (lp->lpni_txcredits < 0) {
865                         msg->msg_tx_delayed = 1;
866                         list_add_tail(&msg->msg_list, &lp->lpni_txq);
867                         spin_unlock(&lp->lpni_lock);
868                         return LNET_CREDIT_WAIT;
869                 }
870                 spin_unlock(&lp->lpni_lock);
871         }
872
873         if (!msg->msg_txcredit) {
874                 LASSERT((tq->tq_credits < 0) ==
875                         !list_empty(&tq->tq_delayed));
876
877                 msg->msg_txcredit = 1;
878                 tq->tq_credits--;
879                 atomic_dec(&ni->ni_tx_credits);
880
881                 if (tq->tq_credits < tq->tq_credits_min)
882                         tq->tq_credits_min = tq->tq_credits;
883
884                 if (tq->tq_credits < 0) {
885                         msg->msg_tx_delayed = 1;
886                         list_add_tail(&msg->msg_list, &tq->tq_delayed);
887                         return LNET_CREDIT_WAIT;
888                 }
889         }
890
891         if (do_send) {
892                 lnet_net_unlock(cpt);
893                 lnet_ni_send(ni, msg);
894                 lnet_net_lock(cpt);
895         }
896         return LNET_CREDIT_OK;
897 }
898
899
900 static struct lnet_rtrbufpool *
901 lnet_msg2bufpool(struct lnet_msg *msg)
902 {
903         struct lnet_rtrbufpool  *rbp;
904         int                     cpt;
905
906         LASSERT(msg->msg_rx_committed);
907
908         cpt = msg->msg_rx_cpt;
909         rbp = &the_lnet.ln_rtrpools[cpt][0];
910
911         LASSERT(msg->msg_len <= LNET_MTU);
912         while (msg->msg_len > (unsigned int)rbp->rbp_npages * PAGE_SIZE) {
913                 rbp++;
914                 LASSERT(rbp < &the_lnet.ln_rtrpools[cpt][LNET_NRBPOOLS]);
915         }
916
917         return rbp;
918 }
919
920 static int
921 lnet_post_routed_recv_locked(struct lnet_msg *msg, int do_recv)
922 {
923         /* lnet_parse is going to lnet_net_unlock immediately after this, so it
924          * sets do_recv FALSE and I don't do the unlock/send/lock bit.
925          * I return LNET_CREDIT_WAIT if msg blocked and LNET_CREDIT_OK if
926          * received or OK to receive */
927         struct lnet_peer_ni *lp = msg->msg_rxpeer;
928         struct lnet_rtrbufpool *rbp;
929         struct lnet_rtrbuf *rb;
930
931         LASSERT (msg->msg_iov == NULL);
932         LASSERT (msg->msg_kiov == NULL);
933         LASSERT (msg->msg_niov == 0);
934         LASSERT (msg->msg_routing);
935         LASSERT (msg->msg_receiving);
936         LASSERT (!msg->msg_sending);
937
938         /* non-lnet_parse callers only receive delayed messages */
939         LASSERT(!do_recv || msg->msg_rx_delayed);
940
941         if (!msg->msg_peerrtrcredit) {
942                 spin_lock(&lp->lpni_lock);
943                 LASSERT((lp->lpni_rtrcredits < 0) ==
944                         !list_empty(&lp->lpni_rtrq));
945
946                 msg->msg_peerrtrcredit = 1;
947                 lp->lpni_rtrcredits--;
948                 if (lp->lpni_rtrcredits < lp->lpni_minrtrcredits)
949                         lp->lpni_minrtrcredits = lp->lpni_rtrcredits;
950
951                 if (lp->lpni_rtrcredits < 0) {
952                         /* must have checked eager_recv before here */
953                         LASSERT(msg->msg_rx_ready_delay);
954                         msg->msg_rx_delayed = 1;
955                         list_add_tail(&msg->msg_list, &lp->lpni_rtrq);
956                         spin_unlock(&lp->lpni_lock);
957                         return LNET_CREDIT_WAIT;
958                 }
959                 spin_unlock(&lp->lpni_lock);
960         }
961
962         rbp = lnet_msg2bufpool(msg);
963
964         if (!msg->msg_rtrcredit) {
965                 msg->msg_rtrcredit = 1;
966                 rbp->rbp_credits--;
967                 if (rbp->rbp_credits < rbp->rbp_mincredits)
968                         rbp->rbp_mincredits = rbp->rbp_credits;
969
970                 if (rbp->rbp_credits < 0) {
971                         /* must have checked eager_recv before here */
972                         LASSERT(msg->msg_rx_ready_delay);
973                         msg->msg_rx_delayed = 1;
974                         list_add_tail(&msg->msg_list, &rbp->rbp_msgs);
975                         return LNET_CREDIT_WAIT;
976                 }
977         }
978
979         LASSERT(!list_empty(&rbp->rbp_bufs));
980         rb = list_entry(rbp->rbp_bufs.next, struct lnet_rtrbuf, rb_list);
981         list_del(&rb->rb_list);
982
983         msg->msg_niov = rbp->rbp_npages;
984         msg->msg_kiov = &rb->rb_kiov[0];
985
986         if (do_recv) {
987                 int cpt = msg->msg_rx_cpt;
988
989                 lnet_net_unlock(cpt);
990                 lnet_ni_recv(msg->msg_rxni, msg->msg_private, msg, 1,
991                              0, msg->msg_len, msg->msg_len);
992                 lnet_net_lock(cpt);
993         }
994         return LNET_CREDIT_OK;
995 }
996
997 void
998 lnet_return_tx_credits_locked(struct lnet_msg *msg)
999 {
1000         struct lnet_peer_ni     *txpeer = msg->msg_txpeer;
1001         struct lnet_ni          *txni = msg->msg_txni;
1002         struct lnet_msg         *msg2;
1003
1004         if (msg->msg_txcredit) {
1005                 struct lnet_ni       *ni = msg->msg_txni;
1006                 struct lnet_tx_queue *tq = ni->ni_tx_queues[msg->msg_tx_cpt];
1007
1008                 /* give back NI txcredits */
1009                 msg->msg_txcredit = 0;
1010
1011                 LASSERT((tq->tq_credits < 0) ==
1012                         !list_empty(&tq->tq_delayed));
1013
1014                 tq->tq_credits++;
1015                 atomic_inc(&ni->ni_tx_credits);
1016                 if (tq->tq_credits <= 0) {
1017                         msg2 = list_entry(tq->tq_delayed.next,
1018                                           struct lnet_msg, msg_list);
1019                         list_del(&msg2->msg_list);
1020
1021                         LASSERT(msg2->msg_txni == ni);
1022                         LASSERT(msg2->msg_tx_delayed);
1023                         LASSERT(msg2->msg_tx_cpt == msg->msg_tx_cpt);
1024
1025                         (void) lnet_post_send_locked(msg2, 1);
1026                 }
1027         }
1028
1029         if (msg->msg_peertxcredit) {
1030                 /* give back peer txcredits */
1031                 msg->msg_peertxcredit = 0;
1032
1033                 spin_lock(&txpeer->lpni_lock);
1034                 LASSERT((txpeer->lpni_txcredits < 0) ==
1035                         !list_empty(&txpeer->lpni_txq));
1036
1037                 txpeer->lpni_txqnob -= msg->msg_len + sizeof(struct lnet_hdr);
1038                 LASSERT(txpeer->lpni_txqnob >= 0);
1039
1040                 txpeer->lpni_txcredits++;
1041                 if (txpeer->lpni_txcredits <= 0) {
1042                         int msg2_cpt;
1043
1044                         msg2 = list_entry(txpeer->lpni_txq.next,
1045                                               struct lnet_msg, msg_list);
1046                         list_del(&msg2->msg_list);
1047                         spin_unlock(&txpeer->lpni_lock);
1048
1049                         LASSERT(msg2->msg_txpeer == txpeer);
1050                         LASSERT(msg2->msg_tx_delayed);
1051
1052                         msg2_cpt = msg2->msg_tx_cpt;
1053
1054                         /*
1055                          * The msg_cpt can be different from the msg2_cpt
1056                          * so we need to make sure we lock the correct cpt
1057                          * for msg2.
1058                          * Once we call lnet_post_send_locked() it is no
1059                          * longer safe to access msg2, since it could've
1060                          * been freed by lnet_finalize(), but we still
1061                          * need to relock the correct cpt, so we cache the
1062                          * msg2_cpt for the purpose of the check that
1063                          * follows the call to lnet_pose_send_locked().
1064                          */
1065                         if (msg2_cpt != msg->msg_tx_cpt) {
1066                                 lnet_net_unlock(msg->msg_tx_cpt);
1067                                 lnet_net_lock(msg2_cpt);
1068                         }
1069                         (void) lnet_post_send_locked(msg2, 1);
1070                         if (msg2_cpt != msg->msg_tx_cpt) {
1071                                 lnet_net_unlock(msg2_cpt);
1072                                 lnet_net_lock(msg->msg_tx_cpt);
1073                         }
1074                 } else {
1075                         spin_unlock(&txpeer->lpni_lock);
1076                 }
1077         }
1078
1079         if (txni != NULL) {
1080                 msg->msg_txni = NULL;
1081                 lnet_ni_decref_locked(txni, msg->msg_tx_cpt);
1082         }
1083
1084         if (txpeer != NULL) {
1085                 /*
1086                  * TODO:
1087                  * Once the patch for the health comes in we need to set
1088                  * the health of the peer ni to bad when we fail to send
1089                  * a message.
1090                  * int status = msg->msg_ev.status;
1091                  * if (status != 0)
1092                  *      lnet_set_peer_ni_health_locked(txpeer, false)
1093                  */
1094                 msg->msg_txpeer = NULL;
1095                 lnet_peer_ni_decref_locked(txpeer);
1096         }
1097 }
1098
1099 void
1100 lnet_schedule_blocked_locked(struct lnet_rtrbufpool *rbp)
1101 {
1102         struct lnet_msg *msg;
1103
1104         if (list_empty(&rbp->rbp_msgs))
1105                 return;
1106         msg = list_entry(rbp->rbp_msgs.next,
1107                          struct lnet_msg, msg_list);
1108         list_del(&msg->msg_list);
1109
1110         (void)lnet_post_routed_recv_locked(msg, 1);
1111 }
1112
1113 void
1114 lnet_drop_routed_msgs_locked(struct list_head *list, int cpt)
1115 {
1116         struct lnet_msg *msg;
1117         struct lnet_msg *tmp;
1118
1119         lnet_net_unlock(cpt);
1120
1121         list_for_each_entry_safe(msg, tmp, list, msg_list) {
1122                 lnet_ni_recv(msg->msg_rxni, msg->msg_private, NULL,
1123                              0, 0, 0, msg->msg_hdr.payload_length);
1124                 list_del_init(&msg->msg_list);
1125                 lnet_finalize(msg, -ECANCELED);
1126         }
1127
1128         lnet_net_lock(cpt);
1129 }
1130
1131 void
1132 lnet_return_rx_credits_locked(struct lnet_msg *msg)
1133 {
1134         struct lnet_peer_ni *rxpeer = msg->msg_rxpeer;
1135         struct lnet_ni *rxni = msg->msg_rxni;
1136         struct lnet_msg *msg2;
1137
1138         if (msg->msg_rtrcredit) {
1139                 /* give back global router credits */
1140                 struct lnet_rtrbuf *rb;
1141                 struct lnet_rtrbufpool *rbp;
1142
1143                 /* NB If a msg ever blocks for a buffer in rbp_msgs, it stays
1144                  * there until it gets one allocated, or aborts the wait
1145                  * itself */
1146                 LASSERT(msg->msg_kiov != NULL);
1147
1148                 rb = list_entry(msg->msg_kiov, struct lnet_rtrbuf, rb_kiov[0]);
1149                 rbp = rb->rb_pool;
1150
1151                 msg->msg_kiov = NULL;
1152                 msg->msg_rtrcredit = 0;
1153
1154                 LASSERT(rbp == lnet_msg2bufpool(msg));
1155
1156                 LASSERT((rbp->rbp_credits > 0) ==
1157                         !list_empty(&rbp->rbp_bufs));
1158
1159                 /* If routing is now turned off, we just drop this buffer and
1160                  * don't bother trying to return credits.  */
1161                 if (!the_lnet.ln_routing) {
1162                         lnet_destroy_rtrbuf(rb, rbp->rbp_npages);
1163                         goto routing_off;
1164                 }
1165
1166                 /* It is possible that a user has lowered the desired number of
1167                  * buffers in this pool.  Make sure we never put back
1168                  * more buffers than the stated number. */
1169                 if (unlikely(rbp->rbp_credits >= rbp->rbp_req_nbuffers)) {
1170                         /* Discard this buffer so we don't have too
1171                          * many. */
1172                         lnet_destroy_rtrbuf(rb, rbp->rbp_npages);
1173                         rbp->rbp_nbuffers--;
1174                 } else {
1175                         list_add(&rb->rb_list, &rbp->rbp_bufs);
1176                         rbp->rbp_credits++;
1177                         if (rbp->rbp_credits <= 0)
1178                                 lnet_schedule_blocked_locked(rbp);
1179                 }
1180         }
1181
1182 routing_off:
1183         if (msg->msg_peerrtrcredit) {
1184                 /* give back peer router credits */
1185                 msg->msg_peerrtrcredit = 0;
1186
1187                 spin_lock(&rxpeer->lpni_lock);
1188                 LASSERT((rxpeer->lpni_rtrcredits < 0) ==
1189                         !list_empty(&rxpeer->lpni_rtrq));
1190
1191                 rxpeer->lpni_rtrcredits++;
1192
1193                 /* drop all messages which are queued to be routed on that
1194                  * peer. */
1195                 if (!the_lnet.ln_routing) {
1196                         struct list_head drop;
1197                         INIT_LIST_HEAD(&drop);
1198                         list_splice_init(&rxpeer->lpni_rtrq, &drop);
1199                         spin_unlock(&rxpeer->lpni_lock);
1200                         lnet_drop_routed_msgs_locked(&drop, msg->msg_rx_cpt);
1201                 } else if (rxpeer->lpni_rtrcredits <= 0) {
1202                         msg2 = list_entry(rxpeer->lpni_rtrq.next,
1203                                           struct lnet_msg, msg_list);
1204                         list_del(&msg2->msg_list);
1205                         spin_unlock(&rxpeer->lpni_lock);
1206                         (void) lnet_post_routed_recv_locked(msg2, 1);
1207                 } else {
1208                         spin_unlock(&rxpeer->lpni_lock);
1209                 }
1210         }
1211         if (rxni != NULL) {
1212                 msg->msg_rxni = NULL;
1213                 lnet_ni_decref_locked(rxni, msg->msg_rx_cpt);
1214         }
1215         if (rxpeer != NULL) {
1216                 msg->msg_rxpeer = NULL;
1217                 lnet_peer_ni_decref_locked(rxpeer);
1218         }
1219 }
1220
1221 static int
1222 lnet_compare_peers(struct lnet_peer_ni *p1, struct lnet_peer_ni *p2)
1223 {
1224         if (p1->lpni_txqnob < p2->lpni_txqnob)
1225                 return 1;
1226
1227         if (p1->lpni_txqnob > p2->lpni_txqnob)
1228                 return -1;
1229
1230         if (p1->lpni_txcredits > p2->lpni_txcredits)
1231                 return 1;
1232
1233         if (p1->lpni_txcredits < p2->lpni_txcredits)
1234                 return -1;
1235
1236         return 0;
1237 }
1238
1239 static int
1240 lnet_compare_routes(struct lnet_route *r1, struct lnet_route *r2)
1241 {
1242         struct lnet_peer_ni *p1 = r1->lr_gateway;
1243         struct lnet_peer_ni *p2 = r2->lr_gateway;
1244         int r1_hops = (r1->lr_hops == LNET_UNDEFINED_HOPS) ? 1 : r1->lr_hops;
1245         int r2_hops = (r2->lr_hops == LNET_UNDEFINED_HOPS) ? 1 : r2->lr_hops;
1246         int rc;
1247
1248         if (r1->lr_priority < r2->lr_priority)
1249                 return 1;
1250
1251         if (r1->lr_priority > r2->lr_priority)
1252                 return -1;
1253
1254         if (r1_hops < r2_hops)
1255                 return 1;
1256
1257         if (r1_hops > r2_hops)
1258                 return -1;
1259
1260         rc = lnet_compare_peers(p1, p2);
1261         if (rc)
1262                 return rc;
1263
1264         if (r1->lr_seq - r2->lr_seq <= 0)
1265                 return 1;
1266
1267         return -1;
1268 }
1269
1270 static struct lnet_peer_ni *
1271 lnet_find_route_locked(struct lnet_net *net, lnet_nid_t target,
1272                        lnet_nid_t rtr_nid)
1273 {
1274         struct lnet_remotenet   *rnet;
1275         struct lnet_route               *route;
1276         struct lnet_route               *best_route;
1277         struct lnet_route               *last_route;
1278         struct lnet_peer_ni     *lpni_best;
1279         struct lnet_peer_ni     *lp;
1280         int                     rc;
1281
1282         /* If @rtr_nid is not LNET_NID_ANY, return the gateway with
1283          * rtr_nid nid, otherwise find the best gateway I can use */
1284
1285         rnet = lnet_find_rnet_locked(LNET_NIDNET(target));
1286         if (rnet == NULL)
1287                 return NULL;
1288
1289         lpni_best = NULL;
1290         best_route = last_route = NULL;
1291         list_for_each_entry(route, &rnet->lrn_routes, lr_list) {
1292                 lp = route->lr_gateway;
1293
1294                 if (!lnet_is_route_alive(route))
1295                         continue;
1296
1297                 if (net != NULL && lp->lpni_net != net)
1298                         continue;
1299
1300                 if (lp->lpni_nid == rtr_nid) /* it's pre-determined router */
1301                         return lp;
1302
1303                 if (lpni_best == NULL) {
1304                         best_route = last_route = route;
1305                         lpni_best = lp;
1306                         continue;
1307                 }
1308
1309                 /* no protection on below fields, but it's harmless */
1310                 if (last_route->lr_seq - route->lr_seq < 0)
1311                         last_route = route;
1312
1313                 rc = lnet_compare_routes(route, best_route);
1314                 if (rc < 0)
1315                         continue;
1316
1317                 best_route = route;
1318                 lpni_best = lp;
1319         }
1320
1321         /* set sequence number on the best router to the latest sequence + 1
1322          * so we can round-robin all routers, it's race and inaccurate but
1323          * harmless and functional  */
1324         if (best_route != NULL)
1325                 best_route->lr_seq = last_route->lr_seq + 1;
1326         return lpni_best;
1327 }
1328
1329 static struct lnet_ni *
1330 lnet_get_best_ni(struct lnet_net *local_net, struct lnet_ni *cur_ni,
1331                  int md_cpt)
1332 {
1333         struct lnet_ni *ni = NULL, *best_ni = cur_ni;
1334         unsigned int shortest_distance;
1335         int best_credits;
1336
1337         if (best_ni == NULL) {
1338                 shortest_distance = UINT_MAX;
1339                 best_credits = INT_MIN;
1340         } else {
1341                 shortest_distance = cfs_cpt_distance(lnet_cpt_table(), md_cpt,
1342                                                      best_ni->ni_dev_cpt);
1343                 best_credits = atomic_read(&best_ni->ni_tx_credits);
1344         }
1345
1346         while ((ni = lnet_get_next_ni_locked(local_net, ni))) {
1347                 unsigned int distance;
1348                 int ni_credits;
1349
1350                 if (!lnet_is_ni_healthy_locked(ni))
1351                         continue;
1352
1353                 ni_credits = atomic_read(&ni->ni_tx_credits);
1354
1355                 /*
1356                  * calculate the distance from the CPT on which
1357                  * the message memory is allocated to the CPT of
1358                  * the NI's physical device
1359                  */
1360                 distance = cfs_cpt_distance(lnet_cpt_table(),
1361                                             md_cpt,
1362                                             ni->ni_dev_cpt);
1363
1364                 /*
1365                  * All distances smaller than the NUMA range
1366                  * are treated equally.
1367                  */
1368                 if (distance < lnet_numa_range)
1369                         distance = lnet_numa_range;
1370
1371                 /*
1372                  * Select on shorter distance, then available
1373                  * credits, then round-robin.
1374                  */
1375                 if (distance > shortest_distance) {
1376                         continue;
1377                 } else if (distance < shortest_distance) {
1378                         shortest_distance = distance;
1379                 } else if (ni_credits < best_credits) {
1380                         continue;
1381                 } else if (ni_credits == best_credits) {
1382                         if (best_ni && (best_ni)->ni_seq <= ni->ni_seq)
1383                                 continue;
1384                 }
1385                 best_ni = ni;
1386                 best_credits = ni_credits;
1387         }
1388
1389         return best_ni;
1390 }
1391
1392 static int
1393 lnet_select_pathway(lnet_nid_t src_nid, lnet_nid_t dst_nid,
1394                     struct lnet_msg *msg, lnet_nid_t rtr_nid)
1395 {
1396         struct lnet_ni          *best_ni;
1397         struct lnet_peer_ni     *best_lpni;
1398         struct lnet_peer_ni     *best_gw;
1399         struct lnet_peer_ni     *lpni;
1400         struct lnet_peer_ni     *final_dst;
1401         struct lnet_peer        *peer;
1402         struct lnet_peer_net    *peer_net;
1403         struct lnet_net         *local_net;
1404         __u32                   seq;
1405         int                     cpt, cpt2, rc;
1406         bool                    routing;
1407         bool                    routing2;
1408         bool                    ni_is_pref;
1409         bool                    preferred;
1410         bool                    local_found;
1411         int                     best_lpni_credits;
1412         int                     md_cpt;
1413
1414         /*
1415          * get an initial CPT to use for locking. The idea here is not to
1416          * serialize the calls to select_pathway, so that as many
1417          * operations can run concurrently as possible. To do that we use
1418          * the CPT where this call is being executed. Later on when we
1419          * determine the CPT to use in lnet_message_commit, we switch the
1420          * lock and check if there was any configuration change.  If none,
1421          * then we proceed, if there is, then we restart the operation.
1422          */
1423         cpt = lnet_net_lock_current();
1424
1425         md_cpt = lnet_cpt_of_md(msg->msg_md, msg->msg_offset);
1426         if (md_cpt == CFS_CPT_ANY)
1427                 md_cpt = cpt;
1428
1429 again:
1430         best_ni = NULL;
1431         best_lpni = NULL;
1432         best_gw = NULL;
1433         final_dst = NULL;
1434         local_net = NULL;
1435         routing = false;
1436         routing2 = false;
1437         local_found = false;
1438
1439         seq = lnet_get_dlc_seq_locked();
1440
1441         if (the_lnet.ln_state != LNET_STATE_RUNNING) {
1442                 lnet_net_unlock(cpt);
1443                 return -ESHUTDOWN;
1444         }
1445
1446         peer = lnet_find_or_create_peer_locked(dst_nid, cpt);
1447         if (IS_ERR(peer)) {
1448                 lnet_net_unlock(cpt);
1449                 return PTR_ERR(peer);
1450         }
1451
1452         /* If peer is not healthy then can not send anything to it */
1453         if (!lnet_is_peer_healthy_locked(peer)) {
1454                 lnet_net_unlock(cpt);
1455                 return -EHOSTUNREACH;
1456         }
1457
1458         if (!peer->lp_multi_rail && lnet_get_num_peer_nis(peer) > 1) {
1459                 lnet_net_unlock(cpt);
1460                 CERROR("peer %s is declared to be non MR capable, "
1461                        "yet configured with more than one NID\n",
1462                        libcfs_nid2str(dst_nid));
1463                 return -EINVAL;
1464         }
1465
1466         /*
1467          * STEP 1: first jab at determining best_ni
1468          * if src_nid is explicitly specified, then best_ni is already
1469          * pre-determiend for us. Otherwise we need to select the best
1470          * one to use later on
1471          */
1472         if (src_nid != LNET_NID_ANY) {
1473                 best_ni = lnet_nid2ni_locked(src_nid, cpt);
1474                 if (!best_ni) {
1475                         lnet_net_unlock(cpt);
1476                         LCONSOLE_WARN("Can't send to %s: src %s is not a "
1477                                       "local nid\n", libcfs_nid2str(dst_nid),
1478                                       libcfs_nid2str(src_nid));
1479                         return -EINVAL;
1480                 }
1481         }
1482
1483         if (msg->msg_type == LNET_MSG_REPLY ||
1484             msg->msg_type == LNET_MSG_ACK ||
1485             !peer->lp_multi_rail ||
1486             best_ni) {
1487                 /*
1488                  * for replies we want to respond on the same peer_ni we
1489                  * received the message on if possible. If not, then pick
1490                  * a peer_ni to send to
1491                  *
1492                  * if the peer is non-multi-rail then you want to send to
1493                  * the dst_nid provided as well.
1494                  *
1495                  * If the best_ni has already been determined, IE the
1496                  * src_nid has been specified, then use the
1497                  * destination_nid provided as well, since we're
1498                  * continuing a series of related messages for the same
1499                  * RPC.
1500                  *
1501                  * It is expected to find the lpni using dst_nid, since we
1502                  * created it earlier.
1503                  */
1504                 best_lpni = lnet_find_peer_ni_locked(dst_nid);
1505                 if (best_lpni)
1506                         lnet_peer_ni_decref_locked(best_lpni);
1507
1508                 if (best_lpni && !lnet_get_net_locked(LNET_NIDNET(dst_nid))) {
1509                         /*
1510                          * this lpni is not on a local network so we need
1511                          * to route this reply.
1512                          */
1513                         best_gw = lnet_find_route_locked(NULL,
1514                                                          best_lpni->lpni_nid,
1515                                                          rtr_nid);
1516                         if (best_gw) {
1517                                 /*
1518                                 * RULE: Each node considers only the next-hop
1519                                 *
1520                                 * We're going to route the message, so change the peer to
1521                                 * the router.
1522                                 */
1523                                 LASSERT(best_gw->lpni_peer_net);
1524                                 LASSERT(best_gw->lpni_peer_net->lpn_peer);
1525                                 peer = best_gw->lpni_peer_net->lpn_peer;
1526
1527                                 /*
1528                                 * if the router is not multi-rail then use the best_gw
1529                                 * found to send the message to
1530                                 */
1531                                 if (!peer->lp_multi_rail)
1532                                         best_lpni = best_gw;
1533                                 else
1534                                         best_lpni = NULL;
1535
1536                                 routing = true;
1537                         } else {
1538                                 best_lpni = NULL;
1539                         }
1540                 } else if (!best_lpni) {
1541                         lnet_net_unlock(cpt);
1542                         CERROR("unable to send msg_type %d to "
1543                               "originating %s. Destination NID not in DB\n",
1544                               msg->msg_type, libcfs_nid2str(dst_nid));
1545                         return -EINVAL;
1546                 }
1547         }
1548
1549         /*
1550          * if the peer is not MR capable, then we should always send to it
1551          * using the first NI in the NET we determined.
1552          */
1553         if (!peer->lp_multi_rail) {
1554                 if (!best_lpni) {
1555                         lnet_net_unlock(cpt);
1556                         CERROR("no route to %s\n",
1557                                libcfs_nid2str(dst_nid));
1558                         return -EHOSTUNREACH;
1559                 }
1560
1561                 /* best ni could be set because src_nid was provided */
1562                 if (!best_ni) {
1563                         best_ni = lnet_net2ni_locked(best_lpni->lpni_net->net_id, cpt);
1564                         if (!best_ni) {
1565                                 lnet_net_unlock(cpt);
1566                                 CERROR("no path to %s from net %s\n",
1567                                 libcfs_nid2str(best_lpni->lpni_nid),
1568                                 libcfs_net2str(best_lpni->lpni_net->net_id));
1569                                 return -EHOSTUNREACH;
1570                         }
1571                 }
1572         }
1573
1574         /*
1575          * if we already found a best_ni because src_nid is specified and
1576          * best_lpni because we are replying to a message then just send
1577          * the message
1578          */
1579         if (best_ni && best_lpni)
1580                 goto send;
1581
1582         /*
1583          * If we already found a best_ni because src_nid is specified then
1584          * pick the peer then send the message
1585          */
1586         if (best_ni)
1587                 goto pick_peer;
1588
1589         /*
1590          * pick the best_ni by going through all the possible networks of
1591          * that peer and see which local NI is best suited to talk to that
1592          * peer.
1593          *
1594          * Locally connected networks will always be preferred over
1595          * a routed network. If there are only routed paths to the peer,
1596          * then the best route is chosen. If all routes are equal then
1597          * they are used in round robin.
1598          */
1599         list_for_each_entry(peer_net, &peer->lp_peer_nets, lpn_on_peer_list) {
1600                 if (!lnet_is_peer_net_healthy_locked(peer_net))
1601                         continue;
1602
1603                 local_net = lnet_get_net_locked(peer_net->lpn_net_id);
1604                 if (!local_net && !routing && !local_found) {
1605                         struct lnet_peer_ni *net_gw;
1606
1607                         lpni = list_entry(peer_net->lpn_peer_nis.next,
1608                                           struct lnet_peer_ni,
1609                                           lpni_on_peer_net_list);
1610
1611                         net_gw = lnet_find_route_locked(NULL,
1612                                                         lpni->lpni_nid,
1613                                                         rtr_nid);
1614                         if (!net_gw)
1615                                 continue;
1616
1617                         if (best_gw) {
1618                                 /*
1619                                  * lnet_find_route_locked() call
1620                                  * will return the best_Gw on the
1621                                  * lpni->lpni_nid network.
1622                                  * However, best_gw and net_gw can
1623                                  * be on different networks.
1624                                  * Therefore need to compare them
1625                                  * to pick the better of either.
1626                                  */
1627                                 if (lnet_compare_peers(best_gw, net_gw) > 0)
1628                                         continue;
1629                                 if (best_gw->lpni_gw_seq <= net_gw->lpni_gw_seq)
1630                                         continue;
1631                         }
1632                         best_gw = net_gw;
1633                         final_dst = lpni;
1634
1635                         routing2 = true;
1636                 } else {
1637                         best_gw = NULL;
1638                         final_dst = NULL;
1639                         routing2 = false;
1640                         local_found = true;
1641                 }
1642
1643                 /*
1644                  * a gw on this network is found, but there could be
1645                  * other better gateways on other networks. So don't pick
1646                  * the best_ni until we determine the best_gw.
1647                  */
1648                 if (best_gw)
1649                         continue;
1650
1651                 /* if no local_net found continue */
1652                 if (!local_net)
1653                         continue;
1654
1655                 /*
1656                  * Iterate through the NIs in this local Net and select
1657                  * the NI to send from. The selection is determined by
1658                  * these 3 criterion in the following priority:
1659                  *      1. NUMA
1660                  *      2. NI available credits
1661                  *      3. Round Robin
1662                  */
1663                 best_ni = lnet_get_best_ni(local_net, best_ni, md_cpt);
1664         }
1665
1666         if (!best_ni && !best_gw) {
1667                 lnet_net_unlock(cpt);
1668                 LCONSOLE_WARN("No local ni found to send from to %s\n",
1669                         libcfs_nid2str(dst_nid));
1670                 return -EINVAL;
1671         }
1672
1673         if (!best_ni) {
1674                 best_ni = lnet_get_best_ni(best_gw->lpni_net, best_ni, md_cpt);
1675                 LASSERT(best_gw && best_ni);
1676
1677                 /*
1678                  * We're going to route the message, so change the peer to
1679                  * the router.
1680                  */
1681                 LASSERT(best_gw->lpni_peer_net);
1682                 LASSERT(best_gw->lpni_peer_net->lpn_peer);
1683                 best_gw->lpni_gw_seq++;
1684                 peer = best_gw->lpni_peer_net->lpn_peer;
1685         }
1686
1687         /*
1688          * Now that we selected the NI to use increment its sequence
1689          * number so the Round Robin algorithm will detect that it has
1690          * been used and pick the next NI.
1691          */
1692         best_ni->ni_seq++;
1693
1694 pick_peer:
1695         /*
1696          * At this point the best_ni is on a local network on which
1697          * the peer has a peer_ni as well
1698          */
1699         peer_net = lnet_peer_get_net_locked(peer,
1700                                             best_ni->ni_net->net_id);
1701         /*
1702          * peer_net is not available or the src_nid is explicitly defined
1703          * and the peer_net for that src_nid is unhealthy. find a route to
1704          * the destination nid.
1705          */
1706         if (!peer_net ||
1707             (src_nid != LNET_NID_ANY &&
1708              !lnet_is_peer_net_healthy_locked(peer_net))) {
1709                 best_gw = lnet_find_route_locked(best_ni->ni_net,
1710                                                  dst_nid,
1711                                                  rtr_nid);
1712                 /*
1713                  * if no route is found for that network then
1714                  * move onto the next peer_ni in the peer
1715                  */
1716                 if (!best_gw) {
1717                         lnet_net_unlock(cpt);
1718                         LCONSOLE_WARN("No route to peer from %s\n",
1719                                 libcfs_nid2str(best_ni->ni_nid));
1720                         return -EHOSTUNREACH;
1721                 }
1722
1723                 CDEBUG(D_NET, "Best route to %s via %s for %s %d\n",
1724                         libcfs_nid2str(dst_nid),
1725                         libcfs_nid2str(best_gw->lpni_nid),
1726                         lnet_msgtyp2str(msg->msg_type), msg->msg_len);
1727
1728                 routing2 = true;
1729                 /*
1730                  * RULE: Each node considers only the next-hop
1731                  *
1732                  * We're going to route the message, so change the peer to
1733                  * the router.
1734                  */
1735                 LASSERT(best_gw->lpni_peer_net);
1736                 LASSERT(best_gw->lpni_peer_net->lpn_peer);
1737                 peer = best_gw->lpni_peer_net->lpn_peer;
1738         } else if (!lnet_is_peer_net_healthy_locked(peer_net)) {
1739                 /*
1740                  * this peer_net is unhealthy but we still have an opportunity
1741                  * to find another peer_net that we can use
1742                  */
1743                 __u32 net_id = peer_net->lpn_net_id;
1744                 LCONSOLE_WARN("peer net %s unhealthy\n",
1745                               libcfs_net2str(net_id));
1746                 goto again;
1747         }
1748
1749         /*
1750          * Look at the peer NIs for the destination peer that connect
1751          * to the chosen net. If a peer_ni is preferred when using the
1752          * best_ni to communicate, we use that one. If there is no
1753          * preferred peer_ni, or there are multiple preferred peer_ni,
1754          * the available transmit credits are used. If the transmit
1755          * credits are equal, we round-robin over the peer_ni.
1756          */
1757         lpni = NULL;
1758         best_lpni_credits = INT_MIN;
1759         preferred = false;
1760         best_lpni = NULL;
1761         while ((lpni = lnet_get_next_peer_ni_locked(peer, peer_net, lpni))) {
1762                 /*
1763                  * if this peer ni is not healthy just skip it, no point in
1764                  * examining it further
1765                  */
1766                 if (!lnet_is_peer_ni_healthy_locked(lpni))
1767                         continue;
1768                 ni_is_pref = lnet_peer_is_ni_pref_locked(lpni, best_ni);
1769
1770                 /* if this is a preferred peer use it */
1771                 if (!preferred && ni_is_pref) {
1772                         preferred = true;
1773                 } else if (preferred && !ni_is_pref) {
1774                         /*
1775                          * this is not the preferred peer so let's ignore
1776                          * it.
1777                          */
1778                         continue;
1779                 } else if (lpni->lpni_txcredits < best_lpni_credits) {
1780                         /*
1781                          * We already have a peer that has more credits
1782                          * available than this one. No need to consider
1783                          * this peer further.
1784                          */
1785                         continue;
1786                 } else if (lpni->lpni_txcredits == best_lpni_credits) {
1787                         /*
1788                          * The best peer found so far and the current peer
1789                          * have the same number of available credits let's
1790                          * make sure to select between them using Round
1791                          * Robin
1792                          */
1793                         if (best_lpni) {
1794                                 if (best_lpni->lpni_seq <= lpni->lpni_seq)
1795                                         continue;
1796                         }
1797                 }
1798
1799                 best_lpni = lpni;
1800                 best_lpni_credits = lpni->lpni_txcredits;
1801         }
1802
1803         /* if we still can't find a peer ni then we can't reach it */
1804         if (!best_lpni) {
1805                 __u32 net_id = (peer_net) ? peer_net->lpn_net_id :
1806                         LNET_NIDNET(dst_nid);
1807                 lnet_net_unlock(cpt);
1808                 LCONSOLE_WARN("no peer_ni found on peer net %s\n",
1809                                 libcfs_net2str(net_id));
1810                 return -EHOSTUNREACH;
1811         }
1812
1813
1814 send:
1815         /* Shortcut for loopback. */
1816         if (best_ni == the_lnet.ln_loni) {
1817                 /* No send credit hassles with LOLND */
1818                 lnet_ni_addref_locked(best_ni, cpt);
1819                 msg->msg_hdr.dest_nid = cpu_to_le64(best_ni->ni_nid);
1820                 if (!msg->msg_routing)
1821                         msg->msg_hdr.src_nid = cpu_to_le64(best_ni->ni_nid);
1822                 msg->msg_target.nid = best_ni->ni_nid;
1823                 lnet_msg_commit(msg, cpt);
1824                 msg->msg_txni = best_ni;
1825                 lnet_net_unlock(cpt);
1826
1827                 return LNET_CREDIT_OK;
1828         }
1829
1830         routing = routing || routing2;
1831
1832         /*
1833          * Increment sequence number of the peer selected so that we
1834          * pick the next one in Round Robin.
1835          */
1836         best_lpni->lpni_seq++;
1837
1838         /*
1839          * grab a reference on the peer_ni so it sticks around even if
1840          * we need to drop and relock the lnet_net_lock below.
1841          */
1842         lnet_peer_ni_addref_locked(best_lpni);
1843
1844         /*
1845          * Use lnet_cpt_of_nid() to determine the CPT used to commit the
1846          * message. This ensures that we get a CPT that is correct for
1847          * the NI when the NI has been restricted to a subset of all CPTs.
1848          * If the selected CPT differs from the one currently locked, we
1849          * must unlock and relock the lnet_net_lock(), and then check whether
1850          * the configuration has changed. We don't have a hold on the best_ni
1851          * yet, and it may have vanished.
1852          */
1853         cpt2 = lnet_cpt_of_nid_locked(best_lpni->lpni_nid, best_ni);
1854         if (cpt != cpt2) {
1855                 lnet_net_unlock(cpt);
1856                 cpt = cpt2;
1857                 lnet_net_lock(cpt);
1858                 if (seq != lnet_get_dlc_seq_locked()) {
1859                         lnet_peer_ni_decref_locked(best_lpni);
1860                         goto again;
1861                 }
1862         }
1863
1864         /*
1865          * store the best_lpni in the message right away to avoid having
1866          * to do the same operation under different conditions
1867          */
1868         msg->msg_txpeer = best_lpni;
1869         msg->msg_txni = best_ni;
1870
1871         /*
1872          * grab a reference for the best_ni since now it's in use in this
1873          * send. the reference will need to be dropped when the message is
1874          * finished in lnet_finalize()
1875          */
1876         lnet_ni_addref_locked(msg->msg_txni, cpt);
1877
1878         /*
1879          * Always set the target.nid to the best peer picked. Either the
1880          * nid will be one of the preconfigured NIDs, or the same NID as
1881          * what was originally set in the target or it will be the NID of
1882          * a router if this message should be routed
1883          */
1884         msg->msg_target.nid = msg->msg_txpeer->lpni_nid;
1885
1886         /*
1887          * lnet_msg_commit assigns the correct cpt to the message, which
1888          * is used to decrement the correct refcount on the ni when it's
1889          * time to return the credits
1890          */
1891         lnet_msg_commit(msg, cpt);
1892
1893         /*
1894          * If we are routing the message then we don't need to overwrite
1895          * the src_nid since it would've been set at the origin. Otherwise
1896          * we are the originator so we need to set it.
1897          */
1898         if (!msg->msg_routing)
1899                 msg->msg_hdr.src_nid = cpu_to_le64(msg->msg_txni->ni_nid);
1900
1901         if (routing) {
1902                 msg->msg_target_is_router = 1;
1903                 msg->msg_target.pid = LNET_PID_LUSTRE;
1904                 /*
1905                  * since we're routing we want to ensure that the
1906                  * msg_hdr.dest_nid is set to the final destination. When
1907                  * the router receives this message it knows how to route
1908                  * it.
1909                  */
1910                 msg->msg_hdr.dest_nid =
1911                         cpu_to_le64(final_dst ? final_dst->lpni_nid : dst_nid);
1912         } else {
1913                 /*
1914                  * if we're not routing set the dest_nid to the best peer
1915                  * ni that we picked earlier in the algorithm.
1916                  */
1917                 msg->msg_hdr.dest_nid = cpu_to_le64(msg->msg_txpeer->lpni_nid);
1918         }
1919
1920         rc = lnet_post_send_locked(msg, 0);
1921
1922         if (!rc)
1923                 CDEBUG(D_NET, "TRACE: %s(%s:%s) -> %s(%s:%s) : %s\n",
1924                        libcfs_nid2str(msg->msg_hdr.src_nid),
1925                        libcfs_nid2str(msg->msg_txni->ni_nid),
1926                        libcfs_nid2str(src_nid),
1927                        libcfs_nid2str(msg->msg_hdr.dest_nid),
1928                        libcfs_nid2str(dst_nid),
1929                        libcfs_nid2str(msg->msg_txpeer->lpni_nid),
1930                        lnet_msgtyp2str(msg->msg_type));
1931
1932         lnet_net_unlock(cpt);
1933
1934         return rc;
1935 }
1936
1937 int
1938 lnet_send(lnet_nid_t src_nid, struct lnet_msg *msg, lnet_nid_t rtr_nid)
1939 {
1940         lnet_nid_t              dst_nid = msg->msg_target.nid;
1941         int                     rc;
1942
1943         /*
1944          * NB: rtr_nid is set to LNET_NID_ANY for all current use-cases,
1945          * but we might want to use pre-determined router for ACK/REPLY
1946          * in the future
1947          */
1948         /* NB: ni != NULL == interface pre-determined (ACK/REPLY) */
1949         LASSERT (msg->msg_txpeer == NULL);
1950         LASSERT (!msg->msg_sending);
1951         LASSERT (!msg->msg_target_is_router);
1952         LASSERT (!msg->msg_receiving);
1953
1954         msg->msg_sending = 1;
1955
1956         LASSERT(!msg->msg_tx_committed);
1957
1958         rc = lnet_select_pathway(src_nid, dst_nid, msg, rtr_nid);
1959         if (rc < 0)
1960                 return rc;
1961
1962         if (rc == LNET_CREDIT_OK)
1963                 lnet_ni_send(msg->msg_txni, msg);
1964
1965         /* rc == LNET_CREDIT_OK or LNET_CREDIT_WAIT */
1966         return 0;
1967 }
1968
1969 void
1970 lnet_drop_message(struct lnet_ni *ni, int cpt, void *private, unsigned int nob)
1971 {
1972         lnet_net_lock(cpt);
1973         the_lnet.ln_counters[cpt]->drop_count++;
1974         the_lnet.ln_counters[cpt]->drop_length += nob;
1975         lnet_net_unlock(cpt);
1976
1977         lnet_ni_recv(ni, private, NULL, 0, 0, 0, nob);
1978 }
1979
1980 static void
1981 lnet_recv_put(struct lnet_ni *ni, struct lnet_msg *msg)
1982 {
1983         struct lnet_hdr *hdr = &msg->msg_hdr;
1984
1985         if (msg->msg_wanted != 0)
1986                 lnet_setpayloadbuffer(msg);
1987
1988         lnet_build_msg_event(msg, LNET_EVENT_PUT);
1989
1990         /* Must I ACK?  If so I'll grab the ack_wmd out of the header and put
1991          * it back into the ACK during lnet_finalize() */
1992         msg->msg_ack = (!lnet_is_wire_handle_none(&hdr->msg.put.ack_wmd) &&
1993                         (msg->msg_md->md_options & LNET_MD_ACK_DISABLE) == 0);
1994
1995         lnet_ni_recv(ni, msg->msg_private, msg, msg->msg_rx_delayed,
1996                      msg->msg_offset, msg->msg_wanted, hdr->payload_length);
1997 }
1998
1999 static int
2000 lnet_parse_put(struct lnet_ni *ni, struct lnet_msg *msg)
2001 {
2002         struct lnet_hdr         *hdr = &msg->msg_hdr;
2003         struct lnet_match_info  info;
2004         int                     rc;
2005         bool                    ready_delay;
2006
2007         /* Convert put fields to host byte order */
2008         hdr->msg.put.match_bits = le64_to_cpu(hdr->msg.put.match_bits);
2009         hdr->msg.put.ptl_index  = le32_to_cpu(hdr->msg.put.ptl_index);
2010         hdr->msg.put.offset     = le32_to_cpu(hdr->msg.put.offset);
2011
2012         /* Primary peer NID. */
2013         info.mi_id.nid  = msg->msg_initiator;
2014         info.mi_id.pid  = hdr->src_pid;
2015         info.mi_opc     = LNET_MD_OP_PUT;
2016         info.mi_portal  = hdr->msg.put.ptl_index;
2017         info.mi_rlength = hdr->payload_length;
2018         info.mi_roffset = hdr->msg.put.offset;
2019         info.mi_mbits   = hdr->msg.put.match_bits;
2020         info.mi_cpt     = lnet_cpt_of_nid(msg->msg_rxpeer->lpni_nid, ni);
2021
2022         msg->msg_rx_ready_delay = ni->ni_net->net_lnd->lnd_eager_recv == NULL;
2023         ready_delay = msg->msg_rx_ready_delay;
2024
2025  again:
2026         rc = lnet_ptl_match_md(&info, msg);
2027         switch (rc) {
2028         default:
2029                 LBUG();
2030
2031         case LNET_MATCHMD_OK:
2032                 lnet_recv_put(ni, msg);
2033                 return 0;
2034
2035         case LNET_MATCHMD_NONE:
2036                 if (ready_delay)
2037                         /* no eager_recv or has already called it, should
2038                          * have been attached on delayed list */
2039                         return 0;
2040
2041                 rc = lnet_ni_eager_recv(ni, msg);
2042                 if (rc == 0) {
2043                         ready_delay = true;
2044                         goto again;
2045                 }
2046                 /* fall through */
2047
2048         case LNET_MATCHMD_DROP:
2049                 CNETERR("Dropping PUT from %s portal %d match %llu"
2050                         " offset %d length %d: %d\n",
2051                         libcfs_id2str(info.mi_id), info.mi_portal,
2052                         info.mi_mbits, info.mi_roffset, info.mi_rlength, rc);
2053
2054                 return -ENOENT; /* -ve: OK but no match */
2055         }
2056 }
2057
2058 static int
2059 lnet_parse_get(struct lnet_ni *ni, struct lnet_msg *msg, int rdma_get)
2060 {
2061         struct lnet_match_info info;
2062         struct lnet_hdr *hdr = &msg->msg_hdr;
2063         struct lnet_process_id source_id;
2064         struct lnet_handle_wire reply_wmd;
2065         int rc;
2066
2067         /* Convert get fields to host byte order */
2068         hdr->msg.get.match_bits   = le64_to_cpu(hdr->msg.get.match_bits);
2069         hdr->msg.get.ptl_index    = le32_to_cpu(hdr->msg.get.ptl_index);
2070         hdr->msg.get.sink_length  = le32_to_cpu(hdr->msg.get.sink_length);
2071         hdr->msg.get.src_offset   = le32_to_cpu(hdr->msg.get.src_offset);
2072
2073         source_id.nid = hdr->src_nid;
2074         source_id.pid = hdr->src_pid;
2075         /* Primary peer NID */
2076         info.mi_id.nid  = msg->msg_initiator;
2077         info.mi_id.pid  = hdr->src_pid;
2078         info.mi_opc     = LNET_MD_OP_GET;
2079         info.mi_portal  = hdr->msg.get.ptl_index;
2080         info.mi_rlength = hdr->msg.get.sink_length;
2081         info.mi_roffset = hdr->msg.get.src_offset;
2082         info.mi_mbits   = hdr->msg.get.match_bits;
2083         info.mi_cpt     = lnet_cpt_of_nid(msg->msg_rxpeer->lpni_nid, ni);
2084
2085         rc = lnet_ptl_match_md(&info, msg);
2086         if (rc == LNET_MATCHMD_DROP) {
2087                 CNETERR("Dropping GET from %s portal %d match %llu"
2088                         " offset %d length %d\n",
2089                         libcfs_id2str(info.mi_id), info.mi_portal,
2090                         info.mi_mbits, info.mi_roffset, info.mi_rlength);
2091                 return -ENOENT; /* -ve: OK but no match */
2092         }
2093
2094         LASSERT(rc == LNET_MATCHMD_OK);
2095
2096         lnet_build_msg_event(msg, LNET_EVENT_GET);
2097
2098         reply_wmd = hdr->msg.get.return_wmd;
2099
2100         lnet_prep_send(msg, LNET_MSG_REPLY, source_id,
2101                        msg->msg_offset, msg->msg_wanted);
2102
2103         msg->msg_hdr.msg.reply.dst_wmd = reply_wmd;
2104
2105         if (rdma_get) {
2106                 /* The LND completes the REPLY from her recv procedure */
2107                 lnet_ni_recv(ni, msg->msg_private, msg, 0,
2108                              msg->msg_offset, msg->msg_len, msg->msg_len);
2109                 return 0;
2110         }
2111
2112         lnet_ni_recv(ni, msg->msg_private, NULL, 0, 0, 0, 0);
2113         msg->msg_receiving = 0;
2114
2115         rc = lnet_send(ni->ni_nid, msg, LNET_NID_ANY);
2116         if (rc < 0) {
2117                 /* didn't get as far as lnet_ni_send() */
2118                 CERROR("%s: Unable to send REPLY for GET from %s: %d\n",
2119                        libcfs_nid2str(ni->ni_nid),
2120                        libcfs_id2str(info.mi_id), rc);
2121
2122                 lnet_finalize(msg, rc);
2123         }
2124
2125         return 0;
2126 }
2127
2128 static int
2129 lnet_parse_reply(struct lnet_ni *ni, struct lnet_msg *msg)
2130 {
2131         void             *private = msg->msg_private;
2132         struct lnet_hdr  *hdr = &msg->msg_hdr;
2133         struct lnet_process_id src = {0};
2134         struct lnet_libmd        *md;
2135         int               rlength;
2136         int               mlength;
2137         int                     cpt;
2138
2139         cpt = lnet_cpt_of_cookie(hdr->msg.reply.dst_wmd.wh_object_cookie);
2140         lnet_res_lock(cpt);
2141
2142         src.nid = hdr->src_nid;
2143         src.pid = hdr->src_pid;
2144
2145         /* NB handles only looked up by creator (no flips) */
2146         md = lnet_wire_handle2md(&hdr->msg.reply.dst_wmd);
2147         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2148                 CNETERR("%s: Dropping REPLY from %s for %s "
2149                         "MD %#llx.%#llx\n",
2150                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2151                         (md == NULL) ? "invalid" : "inactive",
2152                         hdr->msg.reply.dst_wmd.wh_interface_cookie,
2153                         hdr->msg.reply.dst_wmd.wh_object_cookie);
2154                 if (md != NULL && md->md_me != NULL)
2155                         CERROR("REPLY MD also attached to portal %d\n",
2156                                md->md_me->me_portal);
2157
2158                 lnet_res_unlock(cpt);
2159                 return -ENOENT; /* -ve: OK but no match */
2160         }
2161
2162         LASSERT(md->md_offset == 0);
2163
2164         rlength = hdr->payload_length;
2165         mlength = MIN(rlength, (int)md->md_length);
2166
2167         if (mlength < rlength &&
2168             (md->md_options & LNET_MD_TRUNCATE) == 0) {
2169                 CNETERR("%s: Dropping REPLY from %s length %d "
2170                         "for MD %#llx would overflow (%d)\n",
2171                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2172                         rlength, hdr->msg.reply.dst_wmd.wh_object_cookie,
2173                         mlength);
2174                 lnet_res_unlock(cpt);
2175                 return -ENOENT; /* -ve: OK but no match */
2176         }
2177
2178         CDEBUG(D_NET, "%s: Reply from %s of length %d/%d into md %#llx\n",
2179                libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2180                mlength, rlength, hdr->msg.reply.dst_wmd.wh_object_cookie);
2181
2182         lnet_msg_attach_md(msg, md, 0, mlength);
2183
2184         if (mlength != 0)
2185                 lnet_setpayloadbuffer(msg);
2186
2187         lnet_res_unlock(cpt);
2188
2189         lnet_build_msg_event(msg, LNET_EVENT_REPLY);
2190
2191         lnet_ni_recv(ni, private, msg, 0, 0, mlength, rlength);
2192         return 0;
2193 }
2194
2195 static int
2196 lnet_parse_ack(struct lnet_ni *ni, struct lnet_msg *msg)
2197 {
2198         struct lnet_hdr  *hdr = &msg->msg_hdr;
2199         struct lnet_process_id src = {0};
2200         struct lnet_libmd        *md;
2201         int                     cpt;
2202
2203         src.nid = hdr->src_nid;
2204         src.pid = hdr->src_pid;
2205
2206         /* Convert ack fields to host byte order */
2207         hdr->msg.ack.match_bits = le64_to_cpu(hdr->msg.ack.match_bits);
2208         hdr->msg.ack.mlength = le32_to_cpu(hdr->msg.ack.mlength);
2209
2210         cpt = lnet_cpt_of_cookie(hdr->msg.ack.dst_wmd.wh_object_cookie);
2211         lnet_res_lock(cpt);
2212
2213         /* NB handles only looked up by creator (no flips) */
2214         md = lnet_wire_handle2md(&hdr->msg.ack.dst_wmd);
2215         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2216                 /* Don't moan; this is expected */
2217                 CDEBUG(D_NET,
2218                        "%s: Dropping ACK from %s to %s MD %#llx.%#llx\n",
2219                        libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2220                        (md == NULL) ? "invalid" : "inactive",
2221                        hdr->msg.ack.dst_wmd.wh_interface_cookie,
2222                        hdr->msg.ack.dst_wmd.wh_object_cookie);
2223                 if (md != NULL && md->md_me != NULL)
2224                         CERROR("Source MD also attached to portal %d\n",
2225                                md->md_me->me_portal);
2226
2227                 lnet_res_unlock(cpt);
2228                 return -ENOENT;                  /* -ve! */
2229         }
2230
2231         CDEBUG(D_NET, "%s: ACK from %s into md %#llx\n",
2232                libcfs_nid2str(ni->ni_nid), libcfs_id2str(src),
2233                hdr->msg.ack.dst_wmd.wh_object_cookie);
2234
2235         lnet_msg_attach_md(msg, md, 0, 0);
2236
2237         lnet_res_unlock(cpt);
2238
2239         lnet_build_msg_event(msg, LNET_EVENT_ACK);
2240
2241         lnet_ni_recv(ni, msg->msg_private, msg, 0, 0, 0, msg->msg_len);
2242         return 0;
2243 }
2244
2245 /**
2246  * \retval LNET_CREDIT_OK       If \a msg is forwarded
2247  * \retval LNET_CREDIT_WAIT     If \a msg is blocked because w/o buffer
2248  * \retval -ve                  error code
2249  */
2250 int
2251 lnet_parse_forward_locked(struct lnet_ni *ni, struct lnet_msg *msg)
2252 {
2253         int     rc = 0;
2254
2255         if (!the_lnet.ln_routing)
2256                 return -ECANCELED;
2257
2258         if (msg->msg_rxpeer->lpni_rtrcredits <= 0 ||
2259             lnet_msg2bufpool(msg)->rbp_credits <= 0) {
2260                 if (ni->ni_net->net_lnd->lnd_eager_recv == NULL) {
2261                         msg->msg_rx_ready_delay = 1;
2262                 } else {
2263                         lnet_net_unlock(msg->msg_rx_cpt);
2264                         rc = lnet_ni_eager_recv(ni, msg);
2265                         lnet_net_lock(msg->msg_rx_cpt);
2266                 }
2267         }
2268
2269         if (rc == 0)
2270                 rc = lnet_post_routed_recv_locked(msg, 0);
2271         return rc;
2272 }
2273
2274 int
2275 lnet_parse_local(struct lnet_ni *ni, struct lnet_msg *msg)
2276 {
2277         int     rc;
2278
2279         switch (msg->msg_type) {
2280         case LNET_MSG_ACK:
2281                 rc = lnet_parse_ack(ni, msg);
2282                 break;
2283         case LNET_MSG_PUT:
2284                 rc = lnet_parse_put(ni, msg);
2285                 break;
2286         case LNET_MSG_GET:
2287                 rc = lnet_parse_get(ni, msg, msg->msg_rdma_get);
2288                 break;
2289         case LNET_MSG_REPLY:
2290                 rc = lnet_parse_reply(ni, msg);
2291                 break;
2292         default: /* prevent an unused label if !kernel */
2293                 LASSERT(0);
2294                 return -EPROTO;
2295         }
2296
2297         LASSERT(rc == 0 || rc == -ENOENT);
2298         return rc;
2299 }
2300
2301 char *
2302 lnet_msgtyp2str (int type)
2303 {
2304         switch (type) {
2305         case LNET_MSG_ACK:
2306                 return ("ACK");
2307         case LNET_MSG_PUT:
2308                 return ("PUT");
2309         case LNET_MSG_GET:
2310                 return ("GET");
2311         case LNET_MSG_REPLY:
2312                 return ("REPLY");
2313         case LNET_MSG_HELLO:
2314                 return ("HELLO");
2315         default:
2316                 return ("<UNKNOWN>");
2317         }
2318 }
2319
2320 void
2321 lnet_print_hdr(struct lnet_hdr *hdr)
2322 {
2323         struct lnet_process_id src = {
2324                 .nid = hdr->src_nid,
2325                 .pid = hdr->src_pid,
2326         };
2327         struct lnet_process_id dst = {
2328                 .nid = hdr->dest_nid,
2329                 .pid = hdr->dest_pid,
2330         };
2331         char *type_str = lnet_msgtyp2str(hdr->type);
2332
2333         CWARN("P3 Header at %p of type %s\n", hdr, type_str);
2334         CWARN("    From %s\n", libcfs_id2str(src));
2335         CWARN("    To   %s\n", libcfs_id2str(dst));
2336
2337         switch (hdr->type) {
2338         default:
2339                 break;
2340
2341         case LNET_MSG_PUT:
2342                 CWARN("    Ptl index %d, ack md %#llx.%#llx, "
2343                       "match bits %llu\n",
2344                       hdr->msg.put.ptl_index,
2345                       hdr->msg.put.ack_wmd.wh_interface_cookie,
2346                       hdr->msg.put.ack_wmd.wh_object_cookie,
2347                       hdr->msg.put.match_bits);
2348                 CWARN("    Length %d, offset %d, hdr data %#llx\n",
2349                       hdr->payload_length, hdr->msg.put.offset,
2350                       hdr->msg.put.hdr_data);
2351                 break;
2352
2353         case LNET_MSG_GET:
2354                 CWARN("    Ptl index %d, return md %#llx.%#llx, "
2355                       "match bits %llu\n", hdr->msg.get.ptl_index,
2356                       hdr->msg.get.return_wmd.wh_interface_cookie,
2357                       hdr->msg.get.return_wmd.wh_object_cookie,
2358                       hdr->msg.get.match_bits);
2359                 CWARN("    Length %d, src offset %d\n",
2360                       hdr->msg.get.sink_length,
2361                       hdr->msg.get.src_offset);
2362                 break;
2363
2364         case LNET_MSG_ACK:
2365                 CWARN("    dst md %#llx.%#llx, "
2366                       "manipulated length %d\n",
2367                       hdr->msg.ack.dst_wmd.wh_interface_cookie,
2368                       hdr->msg.ack.dst_wmd.wh_object_cookie,
2369                       hdr->msg.ack.mlength);
2370                 break;
2371
2372         case LNET_MSG_REPLY:
2373                 CWARN("    dst md %#llx.%#llx, "
2374                       "length %d\n",
2375                       hdr->msg.reply.dst_wmd.wh_interface_cookie,
2376                       hdr->msg.reply.dst_wmd.wh_object_cookie,
2377                       hdr->payload_length);
2378         }
2379
2380 }
2381
2382 int
2383 lnet_parse(struct lnet_ni *ni, struct lnet_hdr *hdr, lnet_nid_t from_nid,
2384            void *private, int rdma_req)
2385 {
2386         int             rc = 0;
2387         int             cpt;
2388         int             for_me;
2389         struct lnet_msg *msg;
2390         lnet_pid_t     dest_pid;
2391         lnet_nid_t     dest_nid;
2392         lnet_nid_t     src_nid;
2393         struct lnet_peer_ni *lpni;
2394         __u32          payload_length;
2395         __u32          type;
2396
2397         LASSERT (!in_interrupt ());
2398
2399         type = le32_to_cpu(hdr->type);
2400         src_nid = le64_to_cpu(hdr->src_nid);
2401         dest_nid = le64_to_cpu(hdr->dest_nid);
2402         dest_pid = le32_to_cpu(hdr->dest_pid);
2403         payload_length = le32_to_cpu(hdr->payload_length);
2404
2405         for_me = (ni->ni_nid == dest_nid);
2406         cpt = lnet_cpt_of_nid(from_nid, ni);
2407
2408         CDEBUG(D_NET, "TRACE: %s(%s) <- %s : %s\n",
2409                 libcfs_nid2str(dest_nid),
2410                 libcfs_nid2str(ni->ni_nid),
2411                 libcfs_nid2str(src_nid),
2412                 lnet_msgtyp2str(type));
2413
2414         switch (type) {
2415         case LNET_MSG_ACK:
2416         case LNET_MSG_GET:
2417                 if (payload_length > 0) {
2418                         CERROR("%s, src %s: bad %s payload %d (0 expected)\n",
2419                                libcfs_nid2str(from_nid),
2420                                libcfs_nid2str(src_nid),
2421                                lnet_msgtyp2str(type), payload_length);
2422                         return -EPROTO;
2423                 }
2424                 break;
2425
2426         case LNET_MSG_PUT:
2427         case LNET_MSG_REPLY:
2428                 if (payload_length >
2429                     (__u32)(for_me ? LNET_MAX_PAYLOAD : LNET_MTU)) {
2430                         CERROR("%s, src %s: bad %s payload %d "
2431                                "(%d max expected)\n",
2432                                libcfs_nid2str(from_nid),
2433                                libcfs_nid2str(src_nid),
2434                                lnet_msgtyp2str(type),
2435                                payload_length,
2436                                for_me ? LNET_MAX_PAYLOAD : LNET_MTU);
2437                         return -EPROTO;
2438                 }
2439                 break;
2440
2441         default:
2442                 CERROR("%s, src %s: Bad message type 0x%x\n",
2443                        libcfs_nid2str(from_nid),
2444                        libcfs_nid2str(src_nid), type);
2445                 return -EPROTO;
2446         }
2447
2448         if (the_lnet.ln_routing &&
2449             ni->ni_last_alive != ktime_get_real_seconds()) {
2450                 /* NB: so far here is the only place to set NI status to "up */
2451                 lnet_ni_lock(ni);
2452                 ni->ni_last_alive = ktime_get_real_seconds();
2453                 if (ni->ni_status != NULL &&
2454                     ni->ni_status->ns_status == LNET_NI_STATUS_DOWN)
2455                         ni->ni_status->ns_status = LNET_NI_STATUS_UP;
2456                 lnet_ni_unlock(ni);
2457         }
2458
2459         /* Regard a bad destination NID as a protocol error.  Senders should
2460          * know what they're doing; if they don't they're misconfigured, buggy
2461          * or malicious so we chop them off at the knees :) */
2462
2463         if (!for_me) {
2464                 if (LNET_NIDNET(dest_nid) == LNET_NIDNET(ni->ni_nid)) {
2465                         /* should have gone direct */
2466                         CERROR("%s, src %s: Bad dest nid %s "
2467                                "(should have been sent direct)\n",
2468                                 libcfs_nid2str(from_nid),
2469                                 libcfs_nid2str(src_nid),
2470                                 libcfs_nid2str(dest_nid));
2471                         return -EPROTO;
2472                 }
2473
2474                 if (lnet_islocalnid(dest_nid)) {
2475                         /* dest is another local NI; sender should have used
2476                          * this node's NID on its own network */
2477                         CERROR("%s, src %s: Bad dest nid %s "
2478                                "(it's my nid but on a different network)\n",
2479                                 libcfs_nid2str(from_nid),
2480                                 libcfs_nid2str(src_nid),
2481                                 libcfs_nid2str(dest_nid));
2482                         return -EPROTO;
2483                 }
2484
2485                 if (rdma_req && type == LNET_MSG_GET) {
2486                         CERROR("%s, src %s: Bad optimized GET for %s "
2487                                "(final destination must be me)\n",
2488                                 libcfs_nid2str(from_nid),
2489                                 libcfs_nid2str(src_nid),
2490                                 libcfs_nid2str(dest_nid));
2491                         return -EPROTO;
2492                 }
2493
2494                 if (!the_lnet.ln_routing) {
2495                         CERROR("%s, src %s: Dropping message for %s "
2496                                "(routing not enabled)\n",
2497                                 libcfs_nid2str(from_nid),
2498                                 libcfs_nid2str(src_nid),
2499                                 libcfs_nid2str(dest_nid));
2500                         goto drop;
2501                 }
2502         }
2503
2504         /* Message looks OK; we're not going to return an error, so we MUST
2505          * call back lnd_recv() come what may... */
2506
2507         if (!list_empty(&the_lnet.ln_test_peers) &&     /* normally we don't */
2508             fail_peer(src_nid, 0)) {                    /* shall we now? */
2509                 CERROR("%s, src %s: Dropping %s to simulate failure\n",
2510                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2511                        lnet_msgtyp2str(type));
2512                 goto drop;
2513         }
2514
2515         if (!list_empty(&the_lnet.ln_drop_rules) &&
2516             lnet_drop_rule_match(hdr)) {
2517                 CDEBUG(D_NET, "%s, src %s, dst %s: Dropping %s to simulate"
2518                               "silent message loss\n",
2519                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2520                        libcfs_nid2str(dest_nid), lnet_msgtyp2str(type));
2521                 goto drop;
2522         }
2523
2524
2525         msg = lnet_msg_alloc();
2526         if (msg == NULL) {
2527                 CERROR("%s, src %s: Dropping %s (out of memory)\n",
2528                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2529                        lnet_msgtyp2str(type));
2530                 goto drop;
2531         }
2532
2533         /* msg zeroed in lnet_msg_alloc; i.e. flags all clear,
2534          * pointers NULL etc */
2535
2536         msg->msg_type = type;
2537         msg->msg_private = private;
2538         msg->msg_receiving = 1;
2539         msg->msg_rdma_get = rdma_req;
2540         msg->msg_len = msg->msg_wanted = payload_length;
2541         msg->msg_offset = 0;
2542         msg->msg_hdr = *hdr;
2543         /* for building message event */
2544         msg->msg_from = from_nid;
2545         if (!for_me) {
2546                 msg->msg_target.pid     = dest_pid;
2547                 msg->msg_target.nid     = dest_nid;
2548                 msg->msg_routing        = 1;
2549
2550         } else {
2551                 /* convert common msg->hdr fields to host byteorder */
2552                 msg->msg_hdr.type       = type;
2553                 msg->msg_hdr.src_nid    = src_nid;
2554                 msg->msg_hdr.src_pid    = le32_to_cpu(msg->msg_hdr.src_pid);
2555                 msg->msg_hdr.dest_nid   = dest_nid;
2556                 msg->msg_hdr.dest_pid   = dest_pid;
2557                 msg->msg_hdr.payload_length = payload_length;
2558         }
2559
2560         lnet_net_lock(cpt);
2561         lpni = lnet_nid2peerni_locked(from_nid, cpt);
2562         if (IS_ERR(lpni)) {
2563                 lnet_net_unlock(cpt);
2564                 CERROR("%s, src %s: Dropping %s "
2565                        "(error %ld looking up sender)\n",
2566                        libcfs_nid2str(from_nid), libcfs_nid2str(src_nid),
2567                        lnet_msgtyp2str(type), PTR_ERR(lpni));
2568                 lnet_msg_free(msg);
2569                 if (rc == -ESHUTDOWN)
2570                         /* We are shutting down.  Don't do anything more */
2571                         return 0;
2572                 goto drop;
2573         }
2574         msg->msg_rxpeer = lpni;
2575         msg->msg_rxni = ni;
2576         lnet_ni_addref_locked(ni, cpt);
2577         /* Multi-Rail: Primary NID of source. */
2578         msg->msg_initiator = lnet_peer_primary_nid_locked(src_nid);
2579
2580         if (lnet_isrouter(msg->msg_rxpeer)) {
2581                 lnet_peer_set_alive(msg->msg_rxpeer);
2582                 if (avoid_asym_router_failure &&
2583                     LNET_NIDNET(src_nid) != LNET_NIDNET(from_nid)) {
2584                         /* received a remote message from router, update
2585                          * remote NI status on this router.
2586                          * NB: multi-hop routed message will be ignored.
2587                          */
2588                         lnet_router_ni_update_locked(msg->msg_rxpeer,
2589                                                      LNET_NIDNET(src_nid));
2590                 }
2591         }
2592
2593         lnet_msg_commit(msg, cpt);
2594
2595         /* message delay simulation */
2596         if (unlikely(!list_empty(&the_lnet.ln_delay_rules) &&
2597                      lnet_delay_rule_match_locked(hdr, msg))) {
2598                 lnet_net_unlock(cpt);
2599                 return 0;
2600         }
2601
2602         if (!for_me) {
2603                 rc = lnet_parse_forward_locked(ni, msg);
2604                 lnet_net_unlock(cpt);
2605
2606                 if (rc < 0)
2607                         goto free_drop;
2608
2609                 if (rc == LNET_CREDIT_OK) {
2610                         lnet_ni_recv(ni, msg->msg_private, msg, 0,
2611                                      0, payload_length, payload_length);
2612                 }
2613                 return 0;
2614         }
2615
2616         lnet_net_unlock(cpt);
2617
2618         rc = lnet_parse_local(ni, msg);
2619         if (rc != 0)
2620                 goto free_drop;
2621         return 0;
2622
2623  free_drop:
2624         LASSERT(msg->msg_md == NULL);
2625         lnet_finalize(msg, rc);
2626
2627  drop:
2628         lnet_drop_message(ni, cpt, private, payload_length);
2629         return 0;
2630 }
2631 EXPORT_SYMBOL(lnet_parse);
2632
2633 void
2634 lnet_drop_delayed_msg_list(struct list_head *head, char *reason)
2635 {
2636         while (!list_empty(head)) {
2637                 struct lnet_process_id id = {0};
2638                 struct lnet_msg *msg;
2639
2640                 msg = list_entry(head->next, struct lnet_msg, msg_list);
2641                 list_del(&msg->msg_list);
2642
2643                 id.nid = msg->msg_hdr.src_nid;
2644                 id.pid = msg->msg_hdr.src_pid;
2645
2646                 LASSERT(msg->msg_md == NULL);
2647                 LASSERT(msg->msg_rx_delayed);
2648                 LASSERT(msg->msg_rxpeer != NULL);
2649                 LASSERT(msg->msg_hdr.type == LNET_MSG_PUT);
2650
2651                 CWARN("Dropping delayed PUT from %s portal %d match %llu"
2652                       " offset %d length %d: %s\n",
2653                       libcfs_id2str(id),
2654                       msg->msg_hdr.msg.put.ptl_index,
2655                       msg->msg_hdr.msg.put.match_bits,
2656                       msg->msg_hdr.msg.put.offset,
2657                       msg->msg_hdr.payload_length, reason);
2658
2659                 /* NB I can't drop msg's ref on msg_rxpeer until after I've
2660                  * called lnet_drop_message(), so I just hang onto msg as well
2661                  * until that's done */
2662
2663                 lnet_drop_message(msg->msg_rxni, msg->msg_rx_cpt,
2664                                   msg->msg_private, msg->msg_len);
2665                 /*
2666                  * NB: message will not generate event because w/o attached MD,
2667                  * but we still should give error code so lnet_msg_decommit()
2668                  * can skip counters operations and other checks.
2669                  */
2670                 lnet_finalize(msg, -ENOENT);
2671         }
2672 }
2673
2674 void
2675 lnet_recv_delayed_msg_list(struct list_head *head)
2676 {
2677         while (!list_empty(head)) {
2678                 struct lnet_msg *msg;
2679                 struct lnet_process_id id;
2680
2681                 msg = list_entry(head->next, struct lnet_msg, msg_list);
2682                 list_del(&msg->msg_list);
2683
2684                 /* md won't disappear under me, since each msg
2685                  * holds a ref on it */
2686
2687                 id.nid = msg->msg_hdr.src_nid;
2688                 id.pid = msg->msg_hdr.src_pid;
2689
2690                 LASSERT(msg->msg_rx_delayed);
2691                 LASSERT(msg->msg_md != NULL);
2692                 LASSERT(msg->msg_rxpeer != NULL);
2693                 LASSERT(msg->msg_rxni != NULL);
2694                 LASSERT(msg->msg_hdr.type == LNET_MSG_PUT);
2695
2696                 CDEBUG(D_NET, "Resuming delayed PUT from %s portal %d "
2697                        "match %llu offset %d length %d.\n",
2698                         libcfs_id2str(id), msg->msg_hdr.msg.put.ptl_index,
2699                         msg->msg_hdr.msg.put.match_bits,
2700                         msg->msg_hdr.msg.put.offset,
2701                         msg->msg_hdr.payload_length);
2702
2703                 lnet_recv_put(msg->msg_rxni, msg);
2704         }
2705 }
2706
2707 /**
2708  * Initiate an asynchronous PUT operation.
2709  *
2710  * There are several events associated with a PUT: completion of the send on
2711  * the initiator node (LNET_EVENT_SEND), and when the send completes
2712  * successfully, the receipt of an acknowledgment (LNET_EVENT_ACK) indicating
2713  * that the operation was accepted by the target. The event LNET_EVENT_PUT is
2714  * used at the target node to indicate the completion of incoming data
2715  * delivery.
2716  *
2717  * The local events will be logged in the EQ associated with the MD pointed to
2718  * by \a mdh handle. Using a MD without an associated EQ results in these
2719  * events being discarded. In this case, the caller must have another
2720  * mechanism (e.g., a higher level protocol) for determining when it is safe
2721  * to modify the memory region associated with the MD.
2722  *
2723  * Note that LNet does not guarantee the order of LNET_EVENT_SEND and
2724  * LNET_EVENT_ACK, though intuitively ACK should happen after SEND.
2725  *
2726  * \param self Indicates the NID of a local interface through which to send
2727  * the PUT request. Use LNET_NID_ANY to let LNet choose one by itself.
2728  * \param mdh A handle for the MD that describes the memory to be sent. The MD
2729  * must be "free floating" (See LNetMDBind()).
2730  * \param ack Controls whether an acknowledgment is requested.
2731  * Acknowledgments are only sent when they are requested by the initiating
2732  * process and the target MD enables them.
2733  * \param target A process identifier for the target process.
2734  * \param portal The index in the \a target's portal table.
2735  * \param match_bits The match bits to use for MD selection at the target
2736  * process.
2737  * \param offset The offset into the target MD (only used when the target
2738  * MD has the LNET_MD_MANAGE_REMOTE option set).
2739  * \param hdr_data 64 bits of user data that can be included in the message
2740  * header. This data is written to an event queue entry at the target if an
2741  * EQ is present on the matching MD.
2742  *
2743  * \retval  0      Success, and only in this case events will be generated
2744  * and logged to EQ (if it exists).
2745  * \retval -EIO    Simulated failure.
2746  * \retval -ENOMEM Memory allocation failure.
2747  * \retval -ENOENT Invalid MD object.
2748  *
2749  * \see struct lnet_event::hdr_data and lnet_event_kind_t.
2750  */
2751 int
2752 LNetPut(lnet_nid_t self, struct lnet_handle_md mdh, enum lnet_ack_req ack,
2753         struct lnet_process_id target, unsigned int portal,
2754         __u64 match_bits, unsigned int offset,
2755         __u64 hdr_data)
2756 {
2757         struct lnet_msg         *msg;
2758         struct lnet_libmd       *md;
2759         int                     cpt;
2760         int                     rc;
2761
2762         LASSERT(the_lnet.ln_refcount > 0);
2763
2764         if (!list_empty(&the_lnet.ln_test_peers) &&     /* normally we don't */
2765             fail_peer(target.nid, 1)) {                 /* shall we now? */
2766                 CERROR("Dropping PUT to %s: simulated failure\n",
2767                        libcfs_id2str(target));
2768                 return -EIO;
2769         }
2770
2771         msg = lnet_msg_alloc();
2772         if (msg == NULL) {
2773                 CERROR("Dropping PUT to %s: ENOMEM on struct lnet_msg\n",
2774                        libcfs_id2str(target));
2775                 return -ENOMEM;
2776         }
2777         msg->msg_vmflush = !!memory_pressure_get();
2778
2779         cpt = lnet_cpt_of_cookie(mdh.cookie);
2780         lnet_res_lock(cpt);
2781
2782         md = lnet_handle2md(&mdh);
2783         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2784                 CERROR("Dropping PUT (%llu:%d:%s): MD (%d) invalid\n",
2785                        match_bits, portal, libcfs_id2str(target),
2786                        md == NULL ? -1 : md->md_threshold);
2787                 if (md != NULL && md->md_me != NULL)
2788                         CERROR("Source MD also attached to portal %d\n",
2789                                md->md_me->me_portal);
2790                 lnet_res_unlock(cpt);
2791
2792                 lnet_msg_free(msg);
2793                 return -ENOENT;
2794         }
2795
2796         CDEBUG(D_NET, "LNetPut -> %s\n", libcfs_id2str(target));
2797
2798         lnet_msg_attach_md(msg, md, 0, 0);
2799
2800         lnet_prep_send(msg, LNET_MSG_PUT, target, 0, md->md_length);
2801
2802         msg->msg_hdr.msg.put.match_bits = cpu_to_le64(match_bits);
2803         msg->msg_hdr.msg.put.ptl_index = cpu_to_le32(portal);
2804         msg->msg_hdr.msg.put.offset = cpu_to_le32(offset);
2805         msg->msg_hdr.msg.put.hdr_data = hdr_data;
2806
2807         /* NB handles only looked up by creator (no flips) */
2808         if (ack == LNET_ACK_REQ) {
2809                 msg->msg_hdr.msg.put.ack_wmd.wh_interface_cookie =
2810                         the_lnet.ln_interface_cookie;
2811                 msg->msg_hdr.msg.put.ack_wmd.wh_object_cookie =
2812                         md->md_lh.lh_cookie;
2813         } else {
2814                 msg->msg_hdr.msg.put.ack_wmd.wh_interface_cookie =
2815                         LNET_WIRE_HANDLE_COOKIE_NONE;
2816                 msg->msg_hdr.msg.put.ack_wmd.wh_object_cookie =
2817                         LNET_WIRE_HANDLE_COOKIE_NONE;
2818         }
2819
2820         lnet_res_unlock(cpt);
2821
2822         lnet_build_msg_event(msg, LNET_EVENT_SEND);
2823
2824         rc = lnet_send(self, msg, LNET_NID_ANY);
2825         if (rc != 0) {
2826                 CNETERR("Error sending PUT to %s: %d\n",
2827                         libcfs_id2str(target), rc);
2828                 lnet_finalize(msg, rc);
2829         }
2830
2831         /* completion will be signalled by an event */
2832         return 0;
2833 }
2834 EXPORT_SYMBOL(LNetPut);
2835
2836 /*
2837  * The LND can DMA direct to the GET md (i.e. no REPLY msg).  This
2838  * returns a msg for the LND to pass to lnet_finalize() when the sink
2839  * data has been received.
2840  *
2841  * CAVEAT EMPTOR: 'getmsg' is the original GET, which is freed when
2842  * lnet_finalize() is called on it, so the LND must call this first
2843  */
2844 struct lnet_msg *
2845 lnet_create_reply_msg(struct lnet_ni *ni, struct lnet_msg *getmsg)
2846 {
2847         struct lnet_msg *msg = lnet_msg_alloc();
2848         struct lnet_libmd *getmd = getmsg->msg_md;
2849         struct lnet_process_id peer_id = getmsg->msg_target;
2850         int cpt;
2851
2852         LASSERT(!getmsg->msg_target_is_router);
2853         LASSERT(!getmsg->msg_routing);
2854
2855         if (msg == NULL) {
2856                 CERROR("%s: Dropping REPLY from %s: can't allocate msg\n",
2857                        libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id));
2858                 goto drop;
2859         }
2860
2861         cpt = lnet_cpt_of_cookie(getmd->md_lh.lh_cookie);
2862         lnet_res_lock(cpt);
2863
2864         LASSERT(getmd->md_refcount > 0);
2865
2866         if (getmd->md_threshold == 0) {
2867                 CERROR("%s: Dropping REPLY from %s for inactive MD %p\n",
2868                         libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id),
2869                         getmd);
2870                 lnet_res_unlock(cpt);
2871                 goto drop;
2872         }
2873
2874         LASSERT(getmd->md_offset == 0);
2875
2876         CDEBUG(D_NET, "%s: Reply from %s md %p\n",
2877                libcfs_nid2str(ni->ni_nid), libcfs_id2str(peer_id), getmd);
2878
2879         /* setup information for lnet_build_msg_event */
2880         msg->msg_initiator = getmsg->msg_txpeer->lpni_peer_net->lpn_peer->lp_primary_nid;
2881         msg->msg_from = peer_id.nid;
2882         msg->msg_type = LNET_MSG_GET; /* flag this msg as an "optimized" GET */
2883         msg->msg_hdr.src_nid = peer_id.nid;
2884         msg->msg_hdr.payload_length = getmd->md_length;
2885         msg->msg_receiving = 1; /* required by lnet_msg_attach_md */
2886
2887         lnet_msg_attach_md(msg, getmd, getmd->md_offset, getmd->md_length);
2888         lnet_res_unlock(cpt);
2889
2890         cpt = lnet_cpt_of_nid(peer_id.nid, ni);
2891
2892         lnet_net_lock(cpt);
2893         lnet_msg_commit(msg, cpt);
2894         lnet_net_unlock(cpt);
2895
2896         lnet_build_msg_event(msg, LNET_EVENT_REPLY);
2897
2898         return msg;
2899
2900  drop:
2901         cpt = lnet_cpt_of_nid(peer_id.nid, ni);
2902
2903         lnet_net_lock(cpt);
2904         the_lnet.ln_counters[cpt]->drop_count++;
2905         the_lnet.ln_counters[cpt]->drop_length += getmd->md_length;
2906         lnet_net_unlock(cpt);
2907
2908         if (msg != NULL)
2909                 lnet_msg_free(msg);
2910
2911         return NULL;
2912 }
2913 EXPORT_SYMBOL(lnet_create_reply_msg);
2914
2915 void
2916 lnet_set_reply_msg_len(struct lnet_ni *ni, struct lnet_msg *reply,
2917                        unsigned int len)
2918 {
2919         /* Set the REPLY length, now the RDMA that elides the REPLY message has
2920          * completed and I know it. */
2921         LASSERT(reply != NULL);
2922         LASSERT(reply->msg_type == LNET_MSG_GET);
2923         LASSERT(reply->msg_ev.type == LNET_EVENT_REPLY);
2924
2925         /* NB I trusted my peer to RDMA.  If she tells me she's written beyond
2926          * the end of my buffer, I might as well be dead. */
2927         LASSERT(len <= reply->msg_ev.mlength);
2928
2929         reply->msg_ev.mlength = len;
2930 }
2931 EXPORT_SYMBOL(lnet_set_reply_msg_len);
2932
2933 /**
2934  * Initiate an asynchronous GET operation.
2935  *
2936  * On the initiator node, an LNET_EVENT_SEND is logged when the GET request
2937  * is sent, and an LNET_EVENT_REPLY is logged when the data returned from
2938  * the target node in the REPLY has been written to local MD.
2939  *
2940  * On the target node, an LNET_EVENT_GET is logged when the GET request
2941  * arrives and is accepted into a MD.
2942  *
2943  * \param self,target,portal,match_bits,offset See the discussion in LNetPut().
2944  * \param mdh A handle for the MD that describes the memory into which the
2945  * requested data will be received. The MD must be "free floating" (See LNetMDBind()).
2946  *
2947  * \retval  0      Success, and only in this case events will be generated
2948  * and logged to EQ (if it exists) of the MD.
2949  * \retval -EIO    Simulated failure.
2950  * \retval -ENOMEM Memory allocation failure.
2951  * \retval -ENOENT Invalid MD object.
2952  */
2953 int
2954 LNetGet(lnet_nid_t self, struct lnet_handle_md mdh,
2955         struct lnet_process_id target, unsigned int portal,
2956         __u64 match_bits, unsigned int offset)
2957 {
2958         struct lnet_msg         *msg;
2959         struct lnet_libmd       *md;
2960         int                     cpt;
2961         int                     rc;
2962
2963         LASSERT(the_lnet.ln_refcount > 0);
2964
2965         if (!list_empty(&the_lnet.ln_test_peers) &&     /* normally we don't */
2966             fail_peer(target.nid, 1))                   /* shall we now? */
2967         {
2968                 CERROR("Dropping GET to %s: simulated failure\n",
2969                        libcfs_id2str(target));
2970                 return -EIO;
2971         }
2972
2973         msg = lnet_msg_alloc();
2974         if (msg == NULL) {
2975                 CERROR("Dropping GET to %s: ENOMEM on struct lnet_msg\n",
2976                        libcfs_id2str(target));
2977                 return -ENOMEM;
2978         }
2979
2980         cpt = lnet_cpt_of_cookie(mdh.cookie);
2981         lnet_res_lock(cpt);
2982
2983         md = lnet_handle2md(&mdh);
2984         if (md == NULL || md->md_threshold == 0 || md->md_me != NULL) {
2985                 CERROR("Dropping GET (%llu:%d:%s): MD (%d) invalid\n",
2986                        match_bits, portal, libcfs_id2str(target),
2987                        md == NULL ? -1 : md->md_threshold);
2988                 if (md != NULL && md->md_me != NULL)
2989                         CERROR("REPLY MD also attached to portal %d\n",
2990                                md->md_me->me_portal);
2991
2992                 lnet_res_unlock(cpt);
2993
2994                 lnet_msg_free(msg);
2995                 return -ENOENT;
2996         }
2997
2998         CDEBUG(D_NET, "LNetGet -> %s\n", libcfs_id2str(target));
2999
3000         lnet_msg_attach_md(msg, md, 0, 0);
3001
3002         lnet_prep_send(msg, LNET_MSG_GET, target, 0, 0);
3003
3004         msg->msg_hdr.msg.get.match_bits = cpu_to_le64(match_bits);
3005         msg->msg_hdr.msg.get.ptl_index = cpu_to_le32(portal);
3006         msg->msg_hdr.msg.get.src_offset = cpu_to_le32(offset);
3007         msg->msg_hdr.msg.get.sink_length = cpu_to_le32(md->md_length);
3008
3009         /* NB handles only looked up by creator (no flips) */
3010         msg->msg_hdr.msg.get.return_wmd.wh_interface_cookie =
3011                 the_lnet.ln_interface_cookie;
3012         msg->msg_hdr.msg.get.return_wmd.wh_object_cookie =
3013                 md->md_lh.lh_cookie;
3014
3015         lnet_res_unlock(cpt);
3016
3017         lnet_build_msg_event(msg, LNET_EVENT_SEND);
3018
3019         rc = lnet_send(self, msg, LNET_NID_ANY);
3020         if (rc < 0) {
3021                 CNETERR("Error sending GET to %s: %d\n",
3022                         libcfs_id2str(target), rc);
3023                 lnet_finalize(msg, rc);
3024         }
3025
3026         /* completion will be signalled by an event */
3027         return 0;
3028 }
3029 EXPORT_SYMBOL(LNetGet);
3030
3031 /**
3032  * Calculate distance to node at \a dstnid.
3033  *
3034  * \param dstnid Target NID.
3035  * \param srcnidp If not NULL, NID of the local interface to reach \a dstnid
3036  * is saved here.
3037  * \param orderp If not NULL, order of the route to reach \a dstnid is saved
3038  * here.
3039  *
3040  * \retval 0 If \a dstnid belongs to a local interface, and reserved option
3041  * local_nid_dist_zero is set, which is the default.
3042  * \retval positives Distance to target NID, i.e. number of hops plus one.
3043  * \retval -EHOSTUNREACH If \a dstnid is not reachable.
3044  */
3045 int
3046 LNetDist(lnet_nid_t dstnid, lnet_nid_t *srcnidp, __u32 *orderp)
3047 {
3048         struct list_head        *e;
3049         struct lnet_ni *ni = NULL;
3050         struct lnet_remotenet *rnet;
3051         __u32                   dstnet = LNET_NIDNET(dstnid);
3052         int                     hops;
3053         int                     cpt;
3054         __u32                   order = 2;
3055         struct list_head        *rn_list;
3056
3057         /* if !local_nid_dist_zero, I don't return a distance of 0 ever
3058          * (when lustre sees a distance of 0, it substitutes 0@lo), so I
3059          * keep order 0 free for 0@lo and order 1 free for a local NID
3060          * match */
3061
3062         LASSERT(the_lnet.ln_refcount > 0);
3063
3064         cpt = lnet_net_lock_current();
3065
3066         while ((ni = lnet_get_next_ni_locked(NULL, ni))) {
3067                 if (ni->ni_nid == dstnid) {
3068                         if (srcnidp != NULL)
3069                                 *srcnidp = dstnid;
3070                         if (orderp != NULL) {
3071                                 if (LNET_NETTYP(LNET_NIDNET(dstnid)) == LOLND)
3072                                         *orderp = 0;
3073                                 else
3074                                         *orderp = 1;
3075                         }
3076                         lnet_net_unlock(cpt);
3077
3078                         return local_nid_dist_zero ? 0 : 1;
3079                 }
3080
3081                 if (LNET_NIDNET(ni->ni_nid) == dstnet) {
3082                         /* Check if ni was originally created in
3083                          * current net namespace.
3084                          * If not, assign order above 0xffff0000,
3085                          * to make this ni not a priority. */
3086                         if (!net_eq(ni->ni_net_ns, current->nsproxy->net_ns))
3087                                 order += 0xffff0000;
3088
3089                         if (srcnidp != NULL)
3090                                 *srcnidp = ni->ni_nid;
3091                         if (orderp != NULL)
3092                                 *orderp = order;
3093                         lnet_net_unlock(cpt);
3094                         return 1;
3095                 }
3096
3097                 order++;
3098         }
3099
3100         rn_list = lnet_net2rnethash(dstnet);
3101         list_for_each(e, rn_list) {
3102                 rnet = list_entry(e, struct lnet_remotenet, lrn_list);
3103
3104                 if (rnet->lrn_net == dstnet) {
3105                         struct lnet_route *route;
3106                         struct lnet_route *shortest = NULL;
3107                         __u32 shortest_hops = LNET_UNDEFINED_HOPS;
3108                         __u32 route_hops;
3109
3110                         LASSERT(!list_empty(&rnet->lrn_routes));
3111
3112                         list_for_each_entry(route, &rnet->lrn_routes,
3113                                             lr_list) {
3114                                 route_hops = route->lr_hops;
3115                                 if (route_hops == LNET_UNDEFINED_HOPS)
3116                                         route_hops = 1;
3117                                 if (shortest == NULL ||
3118                                     route_hops < shortest_hops) {
3119                                         shortest = route;
3120                                         shortest_hops = route_hops;
3121                                 }
3122                         }
3123
3124                         LASSERT(shortest != NULL);
3125                         hops = shortest_hops;
3126                         if (srcnidp != NULL) {
3127                                 ni = lnet_get_next_ni_locked(
3128                                         shortest->lr_gateway->lpni_net,
3129                                         NULL);
3130                                 *srcnidp = ni->ni_nid;
3131                         }
3132                         if (orderp != NULL)
3133                                 *orderp = order;
3134                         lnet_net_unlock(cpt);
3135                         return hops + 1;
3136                 }
3137                 order++;
3138         }
3139
3140         lnet_net_unlock(cpt);
3141         return -EHOSTUNREACH;
3142 }
3143 EXPORT_SYMBOL(LNetDist);